@graffy/common/graph/add.js

Common libraries that used by various Graffy modules.

"use strict";

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

var _findInstanceProperty = require("@babel/runtime-corejs3/core-js-stable/instance/find");

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

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

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

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

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

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

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

function add(base, diff) {
  var changed = false;
  var index = 0;

  for (var _iterator = diff, _isArray = (0, _isArray2["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 node = _ref;
    var cmp = compare(node);
    var nodeIsBranch = (0, _node.isBranch)(node);
    index = (0, _findInstanceProperty(_node))(base, cmp, index);
    var item = base[index];
    var itemIsBranch = (0, _node.isBranch)(item);

    if (!item || cmp(item)) {
      // This node doesn't exist in the base, insert it.
      (0, _splice["default"])(base).call(base, index, 0, clone(node));
      changed = true;
      continue;
    }

    if (nodeIsBranch && itemIsBranch) {
      changed = add(item.children, node.children) || changed;
    } else if (nodeIsBranch || itemIsBranch) {
      throw new Error('add.branch_leaf_mismatch');
    } else {
      item.value += node.value;
    } // If this item was decremented to zero, remove it.


    var size = itemIsBranch ? item.children.length : item.value;

    if (!size) {
      (0, _splice["default"])(base).call(base, index, 1);
      changed = true;
    }
  }

  return changed;
}

function compare(node) {
  return function (item) {
    var v = compareValue(item.key, node.key) || compareValue(item.end, node.end) || compareValue(item.count, node.count);
    return v;
  };
}

function compareValue(a, b) {
  if (a === b) return 0;
  return a < b ? -1 : 1;
}
/*
  We clone to ensure that the original query (passed as diff) never gets modified, even after successive adds.
*/


function clone(node) {
  var _context;

  var copy = (0, _extends2["default"])({}, node);
  if (node.children) copy.children = (0, _map["default"])(_context = node.children).call(_context, function (child) {
    return clone(child);
  });
  return copy;
}

module.exports = exports.default;