@txdfe/at/lib/tree/view/util.js

一个设计体系组件库

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.filterChildKey = filterChildKey;
exports.filterParentKey = filterParentKey;
exports.getAllCheckedKeys = getAllCheckedKeys;
exports.isDescendantOrSelf = isDescendantOrSelf;
exports.isSiblingOrSelf = isSiblingOrSelf;
exports.normalizeToArray = normalizeToArray;
function _toConsumableArray(r) { return _arrayWithoutHoles(r) || _iterableToArray(r) || _unsupportedIterableToArray(r) || _nonIterableSpread(); }
function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
function _iterableToArray(r) { if ("undefined" != typeof Symbol && null != r[Symbol.iterator] || null != r["@@iterator"]) return Array.from(r); }
function _arrayWithoutHoles(r) { if (Array.isArray(r)) return _arrayLikeToArray(r); }
function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
/* eslint-disable valid-jsdoc */
function normalizeToArray(keys) {
  if (keys !== undefined && keys !== null && keys !== '') {
    if (Array.isArray(keys)) {
      return _toConsumableArray(keys);
    }
    return [keys];
  }
  return [];
}
/**
 * 过滤子节点
 * @param {Array} keys
 * @param {Object} _k2n
 */
function filterChildKey(keys, _k2n) {
  var newKeys = _toConsumableArray(keys).filter(function (key) {
    return !!_k2n[key];
  }).sort(function (prev, next) {
    return getDepth(prev, _k2n) - getDepth(next, _k2n);
  });
  for (var i = 0; i < newKeys.length; i++) {
    for (var j = 0; j < newKeys.length; j++) {
      if (i !== j && isDescendantOrSelf(_k2n[newKeys[i]].pos, _k2n[newKeys[j]].pos)) {
        newKeys.splice(j, 1);
        j--;
      }
    }
  }
  return newKeys;
}
function filterParentKey(keys, _k2n) {
  var newKeys = _toConsumableArray(keys).filter(function (key) {
    return !!_k2n[key];
  }).sort(function (prev, next) {
    return getDepth(next, _k2n) - getDepth(prev, _k2n);
  });
  for (var i = 0; i < newKeys.length; i++) {
    for (var j = 0; j < newKeys.length; j++) {
      if (i !== j && isDescendantOrSelf(_k2n[newKeys[j]].pos, _k2n[newKeys[i]].pos)) {
        newKeys.splice(j, 1);
        j--;
      }
    }
  }
  return newKeys;
}
function isDescendantOrSelf(currentPos, targetPos) {
  if (!currentPos || !targetPos) {
    return false;
  }
  var currentNums = currentPos.split('-');
  var targetNums = targetPos.split('-');
  return currentNums.length <= targetNums.length && currentNums.every(function (num, index) {
    return num === targetNums[index];
  });
}
function isSiblingOrSelf(currentPos, targetPos) {
  var currentNums = currentPos.split('-').slice(0, -1);
  var targetNums = targetPos.split('-').slice(0, -1);
  return currentNums.length === targetNums.length && currentNums.every(function (num, index) {
    return num === targetNums[index];
  });
}

// eslint-disable-next-line max-statements
function getAllCheckedKeys(checkedKeys, _k2n, _p2n) {
  checkedKeys = normalizeToArray(checkedKeys);
  var filteredKeys = checkedKeys.filter(function (key) {
    return !!_k2n[key];
  });
  var flatKeys = filterChildKey(filteredKeys, _k2n);
  var childChecked = function childChecked(child) {
    return flatKeys.indexOf(child.key) > -1;
  };
  var removeKey = function removeKey(child) {
    return flatKeys.splice(flatKeys.indexOf(child.key), 1);
  };
  var addParentKey = function addParentKey(i, parent) {
    return flatKeys.splice(i, 0, parent.key);
  };
  var keys = _toConsumableArray(flatKeys);
  for (var i = 0; i < keys.length; i++) {
    var pos = _k2n[keys[i]].pos;
    var nums = pos.split('-');
    if (nums.length === 2) {
      break;
    }
    for (var j = nums.length - 2; j > 0; j--) {
      var parentPos = nums.slice(0, j + 1).join('-');
      var parent = _p2n[parentPos];
      var parentChecked = parent.children.every(childChecked);
      if (parentChecked) {
        parent.children.forEach(removeKey);
        addParentKey(i, parent);
      } else {
        break;
      }
    }
  }
  var newKeys = [];
  if (flatKeys.length) {
    flatKeys = flatKeys.reverse();
    var ps = Object.keys(_p2n);
    for (var _i = 0; _i < flatKeys.length; _i++) {
      var _pos = _k2n[flatKeys[_i]].pos;
      for (var _j = 0; _j < ps.length; _j++) {
        if (isDescendantOrSelf(_pos, ps[_j])) {
          newKeys.push(_p2n[ps[_j]].key);
          ps.splice(_j, 1);
          _j--;
        }
      }
    }
  }
  return newKeys;
}
function getDepth(key, _k2n) {
  return _k2n[key].pos.split('-').length;
}