@graffy/common/decorate/decorate.js

Common libraries that used by various Graffy modules.

"use strict";

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

exports.__esModule = true;
exports.descend = descend;
exports["default"] = decorate;

var _defineProperty = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/object/define-property"));

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

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

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

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

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

var _pageInfo = _interopRequireDefault(require("./pageInfo"));

var LINK_PLACEHOLDER = (0, _symbol["default"])();

function descend(tree, path) {
  var node = tree;

  for (var _iterator = path, _isArray = (0, _isArray4["default"])(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator2["default"])(_iterator);;) {
    var _ref;

    if (_isArray) {
      if (_i >= _iterator.length) break;
      _ref = _iterator[_i++];
    } else {
      _i = _iterator.next();
      if (_i.done) break;
      _ref = _i.value;
    }

    var key = _ref;
    if (!node) return;
    if (!(key in node)) return undefined;
    node = node[key];
  }

  return node;
}

function decorate(graph, links) {
  if (links === void 0) {
    links = [];
  }

  var result = graph && decorateChildren(graph, links);
  var link;

  while (link = links.shift()) {
    var _link = link,
        from = _link[0],
        key = _link[1],
        path = _link[2];
    var node = descend(result, path);

    if (node === LINK_PLACEHOLDER) {
      // Try this link again later. This is to resolve multi-hop links.
      // TODO: Cycle detection.
      links.push(link);
    } else {
      // if (typeof node === 'undefined' || node === null) {
      //   console.warn('Decorate: Link', path, 'is', node);
      // }
      from[key] = node;
    }
  }

  return result;
}

function decorateChildren(graph, links) {
  var isPage = (0, _some["default"])(graph).call(graph, function (node) {
    return (0, _node.isRange)(node) && node.key !== node.end;
  });

  if (isPage) {
    return decoratePage(graph, links);
  } else {
    return decorateBranch(graph, links);
  }
}

function decoratePage(graph, links) {
  var result = [];

  for (var _iterator2 = graph, _isArray2 = (0, _isArray4["default"])(_iterator2), _i2 = 0, _iterator2 = _isArray2 ? _iterator2 : (0, _getIterator2["default"])(_iterator2);;) {
    var _ref2;

    if (_isArray2) {
      if (_i2 >= _iterator2.length) break;
      _ref2 = _iterator2[_i2++];
    } else {
      _i2 = _iterator2.next();
      if (_i2.done) break;
      _ref2 = _i2.value;
    }

    var node = _ref2;
    if ((0, _node.isRange)(node)) continue;

    if ((0, _node.isLink)(node)) {
      links.push([result, result.length, node.path]);
      result.push(LINK_PLACEHOLDER); // Placeholder that will read replaced.

      continue;
    }

    if ((0, _node.isBranch)(node)) {
      result.push(decorateChildren(node.children, links));
      continue;
    }

    result.push(node.value);
  }

  (0, _defineProperty["default"])(result, 'pageInfo', {
    value: (0, _pageInfo["default"])(graph)
  }); // result.pageInfo = pageInfo(graph);

  return result;
}

function decorateBranch(graph, links) {
  var result = {};

  for (var _iterator3 = graph, _isArray3 = (0, _isArray4["default"])(_iterator3), _i3 = 0, _iterator3 = _isArray3 ? _iterator3 : (0, _getIterator2["default"])(_iterator3);;) {
    var _ref3;

    if (_isArray3) {
      if (_i3 >= _iterator3.length) break;
      _ref3 = _iterator3[_i3++];
    } else {
      _i3 = _iterator3.next();
      if (_i3.done) break;
      _ref3 = _i3.value;
    }

    var node = _ref3;
    var key = node.key;

    if ((0, _node.isRange)(node)) {
      result[key] = null;
      continue;
    }

    if ((0, _node.isLink)(node)) {
      links.push([result, key, node.path]);
      result[key] = LINK_PLACEHOLDER;
      continue;
    }

    if ((0, _node.isBranch)(node)) {
      result[key] = decorateChildren(node.children, links);
      continue;
    }

    if (node.value !== null) result[key] = node.value;
  }

  return result;
}