@graffy/common/decorate/decorateQuery.js

Common libraries that used by various Graffy modules.

"use strict";

var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");

exports.__esModule = true;
exports["default"] = decorateQuery;

var _bind = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/bind"));

var _getIterator2 = _interopRequireDefault(require("@babel/runtime-corejs3/core-js/get-iterator"));

var _isArray = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/array/is-array"));

var _getIteratorMethod2 = _interopRequireDefault(require("@babel/runtime-corejs3/core-js/get-iterator-method"));

var _symbol = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/symbol"));

var _from = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/array/from"));

var _slice = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/slice"));

var _some = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/some"));

var _node = require("../node");

var _encode = require("../encode");

var _graph = require("../graph");

function _createForOfIteratorHelperLoose(o) { var _context2; var i = 0; if (typeof _symbol["default"] === "undefined" || (0, _getIteratorMethod2["default"])(o) == null) { if ((0, _isArray["default"])(o) || (o = _unsupportedIterableToArray(o))) return function () { if (i >= o.length) return { done: true }; return { done: false, value: o[i++] }; }; throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } i = (0, _getIterator2["default"])(o); return (0, _bind["default"])(_context2 = i.next).call(_context2, i); }

function _unsupportedIterableToArray(o, minLen) { var _context; if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = (0, _slice["default"])(_context = Object.prototype.toString.call(o)).call(_context, 8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return (0, _from["default"])(n); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }

function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) { arr2[i] = arr[i]; } return arr2; }

function rangeToPage(key, end, count) {
  var page = {};
  page[count > 0 ? 'first' : 'last'] = count;

  if (key !== '') {
    var _keyStep = (0, _graph.keyStep)(key),
        k = _keyStep.key,
        step = _keyStep.step;

    page.after = (0, _encode.decodeKey)(k);
    if (step === 1) page.excludeAfter = true;
  }

  if (end !== "\uFFFF") {
    var _keyStep2 = (0, _graph.keyStep)(end),
        _k = _keyStep2.key,
        _step = _keyStep2.step;

    page.before = (0, _encode.decodeKey)(_k);
    if (_step === -1) page.excludeBefore = true;
  }

  return page;
}

function decorateQuery(query) {
  var result = decorateChildren(query);
  return result;
}

function decorateChildren(query) {
  var isPage = (0, _some["default"])(query).call(query, function (node) {
    return (0, _node.isRange)(node);
  });

  if (isPage) {
    return decoratePage(query);
  } else {
    return decorateBranch(query);
  }
}

function decoratePage(query) {
  var result = [];

  for (var _iterator = _createForOfIteratorHelperLoose(query), _step2; !(_step2 = _iterator()).done;) {
    var node = _step2.value;
    var child = (0, _node.isBranch)(node) ? decorateChildren(node.children) : true;

    if ((0, _node.isRange)(node)) {
      var key = node.key,
          end = node.end,
          count = node.count;
      result.push(rangeToPage(key, end, count), child);
    } else {
      var _key = node.key;
      result.push(rangeToPage(_key, _key, 1), child);
    }
  }

  return result;
}

function decorateBranch(query) {
  var result = {};

  for (var _iterator2 = _createForOfIteratorHelperLoose(query), _step3; !(_step3 = _iterator2()).done;) {
    var node = _step3.value;
    var child = (0, _node.isBranch)(node) ? decorateChildren(node.children) : true;
    var key = node.key;
    result[key] = child;
  }

  return result;
}

module.exports = exports.default;