UNPKG

antd

Version:

An enterprise-class UI design language and React components implementation

103 lines (78 loc) 2.73 kB
function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _nonIterableSpread(); } function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance"); } function _iterableToArray(iter) { if (Symbol.iterator in Object(iter) || Object.prototype.toString.call(iter) === "[object Arguments]") return Array.from(iter); } function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) { for (var i = 0, arr2 = new Array(arr.length); i < arr.length; i++) { arr2[i] = arr[i]; } return arr2; } } import { getNodeChildren, convertTreeToEntities } from "rc-tree/es/util"; var Record; (function (Record) { Record[Record["None"] = 0] = "None"; Record[Record["Start"] = 1] = "Start"; Record[Record["End"] = 2] = "End"; })(Record || (Record = {})); // TODO: Move this logic into `rc-tree` function traverseNodesKey(rootChildren, callback) { var nodeList = getNodeChildren(rootChildren) || []; function processNode(node) { var key = node.key, children = node.props.children; if (callback(key, node) !== false) { traverseNodesKey(children, callback); } } nodeList.forEach(processNode); } export function getFullKeyList(children) { var _convertTreeToEntitie = convertTreeToEntities(children), keyEntities = _convertTreeToEntitie.keyEntities; return Object.keys(keyEntities); } /** 计算选中范围,只考虑expanded情况以优化性能 */ export function calcRangeKeys(rootChildren, expandedKeys, startKey, endKey) { var keys = []; var record = Record.None; if (startKey && startKey === endKey) { return [startKey]; } if (!startKey || !endKey) { return []; } function matchKey(key) { return key === startKey || key === endKey; } traverseNodesKey(rootChildren, function (key) { if (record === Record.End) { return false; } if (matchKey(key)) { // Match test keys.push(key); if (record === Record.None) { record = Record.Start; } else if (record === Record.Start) { record = Record.End; return false; } } else if (record === Record.Start) { // Append selection keys.push(key); } if (expandedKeys.indexOf(key) === -1) { return false; } return true; }); return keys; } export function convertDirectoryKeysToNodes(rootChildren, keys) { var restKeys = _toConsumableArray(keys); var nodes = []; traverseNodesKey(rootChildren, function (key, node) { var index = restKeys.indexOf(key); if (index !== -1) { nodes.push(node); restKeys.splice(index, 1); } return !!restKeys.length; }); return nodes; } //# sourceMappingURL=util.js.map