@douyinfe/semi-foundation 2.36.0-beta.0 → 2.36.0

package/lib/cjs/tree/treeUtil.js

@@ -31,33 +31,25 @@ exports.normalizeValue = normalizeValue;
 exports.normalizedArr = normalizedArr;
 exports.traverseDataNodes = traverseDataNodes;
 exports.updateKeys = updateKeys;
-
 var _get2 = _interopRequireDefault(require("lodash/get"));
-
 var _pick2 = _interopRequireDefault(require("lodash/pick"));
-
 var _isEmpty2 = _interopRequireDefault(require("lodash/isEmpty"));
-
 var _isUndefined2 = _interopRequireDefault(require("lodash/isUndefined"));
-
 var _isNull2 = _interopRequireDefault(require("lodash/isNull"));
-
 var _isObject2 = _interopRequireDefault(require("lodash/isObject"));
-
 var _max2 = _interopRequireDefault(require("lodash/max"));
-
 var _uniq2 = _interopRequireDefault(require("lodash/uniq"));
-
 var _difference2 = _interopRequireDefault(require("lodash/difference"));
-
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+/**
+ * Part of the utils function implementation process reference
+ * https://github.com/react-component/tree/blob/master/src/util.tsx
+ */
 
 const DRAG_OFFSET = 0.45;
-
 function getPosition(level, index) {
   return `${level}-${index}`;
 }
-
 function isValid(val) {
   return !(0, _isNull2.default)(val) && !(0, _isUndefined2.default)(val);
 }
@@ -69,19 +61,16 @@ function isValid(val) {
  * need expanded keys, provides `true` means all expanded
  */
 // eslint-disable-next-line max-len
-
-
 function flattenTreeData(treeNodeList, expandedKeys) {
   let filteredShownKeys = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
   const flattenList = [];
   const filterSearch = Boolean(filteredShownKeys);
-
   function flatten(list) {
     let parent = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : null;
     return list.map((treeNode, index) => {
       const pos = getPosition(parent ? parent.pos : '0', index);
-      const mergedKey = treeNode.key; // Add FlattenDataNode into list
-
+      const mergedKey = treeNode.key;
+      // Add FlattenDataNode into list
       const flattenNode = Object.assign(Object.assign({}, (0, _pick2.default)(treeNode, ['key', 'label', 'value', 'icon', 'disabled', 'isLeaf'])), {
         parent,
         pos,
@@ -90,30 +79,24 @@ function flattenTreeData(treeNodeList, expandedKeys) {
         _innerDataTag: true
       });
       const isBooleanFilteredShownKeys = typeof filteredShownKeys === 'boolean';
-
       if (!filterSearch || !isBooleanFilteredShownKeys && filteredShownKeys.has(mergedKey)) {
         flattenList.push(flattenNode);
-      } // Loop treeNode children
+      }
+      // Loop treeNode children
       // eslint-disable-next-line max-len
-
-
       if (expandedKeys.has(mergedKey) && (!filterSearch || !isBooleanFilteredShownKeys && filteredShownKeys.has(mergedKey))) {
         flattenNode.children = flatten(treeNode.children || [], flattenNode);
       } else {
         flattenNode.children = [];
       }
-
       return flattenNode;
     });
   }
-
   flatten(treeNodeList);
   return flattenList;
 }
-
 function convertJsonToData(treeJson) {
   const treeData = [];
-
   const traverseNode = (key, children, path, res) => {
     const currPath = [...path, key];
     const itemKey = currPath.join('-');
@@ -122,7 +105,6 @@ function convertJsonToData(treeJson) {
       label: key,
       value: children
     };
-
     if ((0, _isObject2.default)(children)) {
       const newChildren = [];
       Object.entries(children).forEach(c => {
@@ -130,23 +112,19 @@ function convertJsonToData(treeJson) {
       });
       newNode.children = newChildren;
     }
-
     res.push(newNode);
   };
-
   Object.entries(treeJson).forEach(item => traverseNode(item[0], item[1], [], treeData));
   return treeData;
 }
 /**
  * Traverse all the data by `treeData`.
  */
-
-
 function traverseDataNodes(treeNodes, callback) {
   const processNode = (node, ind, parent) => {
     const children = node ? node.children : treeNodes;
-    const pos = node ? getPosition(parent.pos, ind) : '0'; // Process node if is not root
-
+    const pos = node ? getPosition(parent.pos, ind) : '0';
+    // Process node if is not root
     if (node) {
       const data = {
         data: Object.assign({}, node),
@@ -157,9 +135,8 @@ function traverseDataNodes(treeNodes, callback) {
         level: Number(parent.level) + 1
       };
       callback(data);
-    } // Process children node
-
-
+    }
+    // Process children node
     if (children) {
       children.forEach((subNode, subIndex) => {
         processNode(subNode, subIndex, {
@@ -170,12 +147,9 @@ function traverseDataNodes(treeNodes, callback) {
       });
     }
   };
-
   processNode(null);
 }
 /* Convert data to entities map */
-
-
 function convertDataToEntities(dataNodes) {
   const posEntities = {};
   const keyEntities = {};
@@ -193,16 +167,13 @@ function convertDataToEntities(dataNodes) {
     } = data;
     const entity = Object.assign({}, data);
     const value = (0, _get2.default)(entity, 'data.value', null);
-
     if (value !== null) {
       valueEntities[value] = key;
     }
-
     posEntities[pos] = entity;
-    keyEntities[key] = entity; // Fill children
-
+    keyEntities[key] = entity;
+    // Fill children
     entity.parent = posEntities[parentPos];
-
     if (entity.parent) {
       entity.parent.children = entity.parent.children || [];
       entity.parent.children.push(entity);
@@ -211,25 +182,19 @@ function convertDataToEntities(dataNodes) {
   return wrapper;
 }
 /* Get key by value */
-
-
 function findKeysForValues(valueList, valueEntities) {
   let isMultiple = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
-
   if (!isValid(valueList)) {
     return [];
   }
-
   if (!isMultiple && Array.isArray(valueList)) {
     valueList = valueList.length ? [valueList[0]] : [];
   } else if (!Array.isArray(valueList)) {
     valueList = [valueList];
   }
-
   if ((0, _isEmpty2.default)(valueEntities)) {
     return valueList;
   }
-
   const res = [];
   valueList.forEach(val => {
     if (val in valueEntities) {
@@ -241,21 +206,17 @@ function findKeysForValues(valueList, valueEntities) {
   });
   return res;
 }
-
 function findDescendantKeys(selectedKeys, options) {
   let self = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : true;
   const res = [];
-
   const findChild = item => {
     if (!item) {
       return;
     }
-
     const {
       children
     } = item;
     const hasChildren = isValid(children);
-
     if (hasChildren) {
       children.forEach(child => {
         res.push(child.key);
@@ -263,17 +224,14 @@ function findDescendantKeys(selectedKeys, options) {
       });
     }
   };
-
   selectedKeys.forEach(item => {
     if (self) {
       res.push(item);
     }
-
     findChild(options[item]);
   });
   return res;
 }
-
 function findChildKeys(keys, options) {
   let omitKeys = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : [];
   const res = [];
@@ -288,21 +246,16 @@ function findChildKeys(keys, options) {
   return res;
 }
 /* istanbul ignore next */
-
-
 function findLeafKeys(keys, options) {
   const res = [];
-
   const findChild = item => {
     if (!item) {
       return;
     }
-
     const {
       children
     } = item;
     const isLeaf = !isValid(children);
-
     if (isLeaf) {
       res.push(item.key);
     } else {
@@ -311,13 +264,11 @@ function findLeafKeys(keys, options) {
       });
     }
   };
-
   keys.forEach(item => {
     findChild(options[item]);
   });
   return res;
 }
-
 function findSiblingKeys(selectedKeys, options) {
   let self = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : true;
   const par = [];
@@ -329,39 +280,33 @@ function findSiblingKeys(selectedKeys, options) {
   const res = findChildKeys((0, _uniq2.default)(par), options, self ? [] : selectedKeys);
   return res;
 }
-
 function findAncestorKeys(selectedKeys, options) {
   let self = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : true;
-  const res = []; // Recursively find the parent element
-
+  const res = [];
+  // Recursively find the parent element
   const findPar = item => {
     if (item.parent) {
       res.push(item.parent.key);
       findPar(item.parent);
     }
   };
-
   selectedKeys.forEach(item => {
     options[item] && findPar(options[item]);
-
     if (self) {
       res.push(item);
     }
   });
   return res;
 }
-
 function getSortedKeyList(keyList, keyEntities) {
   const levelMap = {};
   keyList.forEach(key => {
     if (!keyEntities[key]) {
       return;
     }
-
     const {
       level
     } = keyEntities[key];
-
     if (levelMap[level]) {
       levelMap[level].push(key);
     } else {
@@ -370,7 +315,6 @@ function getSortedKeyList(keyList, keyEntities) {
   });
   return levelMap;
 }
-
 function calcCheckedKeys(values, keyEntities) {
   const keyList = Array.isArray(values) ? values : [values];
   const descendantKeys = findDescendantKeys(keyList, keyEntities, true);
@@ -379,103 +323,85 @@ function calcCheckedKeys(values, keyEntities) {
    * their descendants must be checked. That is to say, if the descendant nodes have
    *  disabled+unchecked nodes, their ancestor nodes will definitely not be checked
    */
-
   const checkedKeys = new Set([...descendantKeys]);
   let halfCheckedKeys = new Set([]);
   let visited = [];
   const levelMap = getSortedKeyList(keyList, keyEntities);
-
   const calcCurrLevel = node => {
     const {
       key,
       parent,
       level
-    } = node; // If the node does not have a parent node, or the node has been processed just now, no processing is done
-
+    } = node;
+    // If the node does not have a parent node, or the node has been processed just now, no processing is done
     if (!parent || visited.includes(key)) {
       return;
     }
-
-    const siblingKeys = findSiblingKeys([key], keyEntities); // visited for caching to avoid double counting
-
+    const siblingKeys = findSiblingKeys([key], keyEntities);
+    // visited for caching to avoid double counting
     visited = [...visited, ...siblingKeys];
     const allChecked = siblingKeys.every(siblingKey => checkedKeys.has(siblingKey));
-
     if (!allChecked) {
       const ancestorKeys = findAncestorKeys([key], keyEntities, false);
       halfCheckedKeys = new Set([...halfCheckedKeys, ...ancestorKeys]);
     } else {
-      checkedKeys.add(parent.key); // IMPORTANT! parent level may not exist in original level map; if add to the end directly may destroy the hierarchical order
-
+      checkedKeys.add(parent.key);
+      // IMPORTANT! parent level may not exist in original level map; if add to the end directly may destroy the hierarchical order
       if (level - 1 in levelMap && level) {
         levelMap[level - 1].push(parent.key);
       } else {
         levelMap[level - 1] = [parent.key];
       }
     }
-  }; // Loop keyList from deepest Level to topLevel, bottom up
-
-
+  };
+  // Loop keyList from deepest Level to topLevel, bottom up
   while (!(0, _isEmpty2.default)(levelMap)) {
     const maxLevel = (0, _max2.default)(Object.keys(levelMap).map(key => Number(key)));
     levelMap[maxLevel].forEach(key => calcCurrLevel(keyEntities[key]));
     delete levelMap[maxLevel];
   }
-
   return {
     checkedKeys,
     halfCheckedKeys
   };
 }
 /* Calculate the expanded node by key */
-
-
 function calcExpandedKeys() {
   let keyList = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
   let keyEntities = arguments.length > 1 ? arguments[1] : undefined;
   let autoExpandParent = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : true;
-
   if (!Array.isArray(keyList)) {
     keyList = [keyList];
   }
-
   if (autoExpandParent) {
     const ancestorKeys = findAncestorKeys(keyList, keyEntities, true);
     return new Set(ancestorKeys);
   }
-
   return new Set(keyList);
 }
 /* Calculate the expanded node by value */
 // eslint-disable-next-line max-len
-
-
 function calcExpandedKeysForValues(value, keyEntities, isMultiple, valueEntities) {
   const keys = findKeysForValues(value, valueEntities, isMultiple);
   return new Set(findAncestorKeys(keys, keyEntities, false));
 }
-
 function calcMotionKeys(oldKeySet, newKeySet, keyEntities) {
   let motionType = 'show';
   const oldKeys = [...oldKeySet];
   const newKeys = [...newKeySet];
-
   if (Math.abs(oldKeys.length - newKeys.length) !== 1) {
     return {
       motionType,
       motionKeys: []
     };
   }
-
   let diffKeys = [];
-
   if (oldKeys.length > newKeys.length) {
     motionType = 'hide';
     diffKeys = (0, _difference2.default)(oldKeys, newKeys);
   } else {
     diffKeys = (0, _difference2.default)(newKeys, oldKeys);
   }
-
   return {
     motionType: diffKeys.length === 1 ? motionType : 'show',
     motionKeys: diffKeys.length === 1 ? findDescendantKeys(diffKeys, keyEntities, false) : []
@@ -486,121 +412,98 @@ function calcMotionKeys(oldKeySet, newKeySet, keyEntities) {
  * When filterTreeNode is a function,returns the result of filterTreeNode which called with (sugInput, target, option).
  * The filteredPath parameter will only be passed in when the Cascader calls the filter function
  */
-
-
 function filter(sugInput, option, filterTreeNode, filterProps, filteredPath) {
   if (!filterTreeNode) {
     return true;
   }
-
   let filterFn = filterTreeNode;
   let target = filteredPath !== null && filteredPath !== void 0 ? filteredPath : option;
-
   if (typeof filterTreeNode === 'boolean') {
     filterFn = (targetVal, val) => {
       const input = targetVal.toLowerCase();
       return val.toString().toLowerCase().includes(input);
     };
   }
-
   if (filterProps) {
     target = option[filterProps];
   }
-
   return filterFn(sugInput, target, option);
 }
-
 function normalizedArr(val) {
   if (!Array.isArray(val)) {
     return [val];
   } else {
     return val;
   }
-} // flag is used to determine whether to return when the key does not belong to the keys in keyEntities
+}
+// flag is used to determine whether to return when the key does not belong to the keys in keyEntities
 // export function normalizeKeyList(keyList: any, keyEntities: KeyEntities, leafOnly = false) {
-
-
 function normalizeKeyList(keyList, keyEntities) {
   let leafOnly = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
   let flag = arguments.length > 3 ? arguments[3] : undefined;
   const res = [];
   const keyListSet = new Set(keyList);
-
   if (!leafOnly) {
     keyList.forEach(key => {
       if (!keyEntities[key]) {
         if (flag) {
           res.push(key);
         }
-
         return;
       }
-
       const {
         parent
       } = keyEntities[key];
-
       if (parent && keyListSet.has(parent.key)) {
         return;
       }
-
       res.push(key);
     });
   } else {
     keyList.forEach(key => {
       if (keyEntities[key] && !isValid(keyEntities[key].children)) {
         res.push(key);
-      } // when key is not in keyEntities, if flag is true, key should be push in res
-
-
+      }
+      // when key is not in keyEntities, if flag is true, key should be push in res
       if (!keyEntities[key] && flag) {
         res.push(key);
       }
     });
   }
-
   return res;
 }
-
 function getMotionKeys(eventKey, expandedKeys, keyEntities) {
   const res = [];
-
   const getChild = itemKey => {
     keyEntities[itemKey].children && keyEntities[itemKey].children.forEach(item => {
       const {
         key
       } = item;
       res.push(key);
-
       if (expandedKeys.has(key)) {
         getChild(key);
       }
     });
   };
-
   getChild(eventKey);
   return res;
-} // eslint-disable-next-line max-len
-
-
+}
+// eslint-disable-next-line max-len
 function calcCheckedKeysForChecked(key, keyEntities, checkedKeys, halfCheckedKeys) {
   const descendantKeys = findDescendantKeys([key], keyEntities, true);
   const nodeItem = keyEntities[key];
   checkedKeys = new Set([...checkedKeys, key]);
-
   const calcCurrLevel = node => {
     if (!node.parent) {
       return;
-    } // eslint-disable-next-line @typescript-eslint/no-shadow
-
-
+    }
+    // eslint-disable-next-line @typescript-eslint/no-shadow
     const {
       key
     } = node;
-    const siblingKeys = findSiblingKeys([key], keyEntities); // eslint-disable-next-line @typescript-eslint/no-shadow
-
+    const siblingKeys = findSiblingKeys([key], keyEntities);
+    // eslint-disable-next-line @typescript-eslint/no-shadow
     const allChecked = siblingKeys.every(key => checkedKeys.has(key));
-
     if (!allChecked) {
       const ancestorKeys = findAncestorKeys([key], keyEntities, false);
       halfCheckedKeys = new Set([...halfCheckedKeys, ...ancestorKeys]);
@@ -610,15 +513,13 @@ function calcCheckedKeysForChecked(key, keyEntities, checkedKeys, halfCheckedKey
       calcCurrLevel(par);
     }
   };
-
   calcCurrLevel(nodeItem);
   return {
     checkedKeys: new Set([...checkedKeys, ...descendantKeys]),
     halfCheckedKeys
   };
-} // eslint-disable-next-line max-len
-
-
+}
+// eslint-disable-next-line max-len
 function calcCheckedKeysForUnchecked(key, keyEntities, checkedKeys, halfCheckedKeys) {
   const descendantKeys = findDescendantKeys([key], keyEntities, true);
   const nodeItem = keyEntities[key];
@@ -626,61 +527,54 @@ function calcCheckedKeysForUnchecked(key, keyEntities, checkedKeys, halfCheckedK
     if (checkedKeys.has(descendantKey)) {
       checkedKeys.delete(descendantKey);
     }
-
     if (halfCheckedKeys.has(descendantKey)) {
       halfCheckedKeys.delete(descendantKey);
     }
   });
-
   const calcCurrLevel = node => {
-    const par = node.parent; // no parent
-
+    const par = node.parent;
+    // no parent
     if (!par) {
       return;
-    } // Has a parent node, and the parent node is not checked or halfChecked
-
-
+    }
+    // Has a parent node, and the parent node is not checked or halfChecked
     if (!checkedKeys.has(par.key) && !halfCheckedKeys.has(par.key)) {
       return;
-    } // Has a parent node, and the parent node is checked or halfChecked
+    }
+    // Has a parent node, and the parent node is checked or halfChecked
     // eslint-disable-next-line @typescript-eslint/no-shadow
-
-
     const {
       key
     } = node;
-    const siblingKeys = findSiblingKeys([key], keyEntities); // eslint-disable-next-line @typescript-eslint/no-shadow
-
+    const siblingKeys = findSiblingKeys([key], keyEntities);
+    // eslint-disable-next-line @typescript-eslint/no-shadow
     const anyChecked = siblingKeys.some(key => checkedKeys.has(key) || halfCheckedKeys.has(key));
-    const ancestorKeys = findAncestorKeys([key], keyEntities, false); // If there is checked or halfChecked in the sibling node, you need to change the parent node to halfChecked
-
+    const ancestorKeys = findAncestorKeys([key], keyEntities, false);
+    // If there is checked or halfChecked in the sibling node, you need to change the parent node to halfChecked
     if (anyChecked) {
       ancestorKeys.forEach(itemKey => {
         if (checkedKeys.has(itemKey)) {
           checkedKeys.delete(itemKey);
           halfCheckedKeys.add(itemKey);
         }
-      }); // If there is no checked or halfChecked in the sibling node, you need to change the parent node to unchecked
+      });
+      // If there is no checked or halfChecked in the sibling node, you need to change the parent node to unchecked
     } else {
       if (checkedKeys.has(par.key)) {
         checkedKeys.delete(par.key);
       }
-
       if (halfCheckedKeys.has(par.key)) {
         halfCheckedKeys.delete(par.key);
       }
-
       calcCurrLevel(par);
     }
   };
-
   nodeItem && calcCurrLevel(nodeItem);
   return {
     checkedKeys,
     halfCheckedKeys
   };
 }
-
 function filterTreeData(info) {
   const {
     showFilteredOnly,
@@ -694,12 +588,10 @@ function filterTreeData(info) {
   let filteredOptsKeys = [];
   filteredOptsKeys = Object.values(keyEntities).filter(item => filter(inputValue, item.data, filterTreeNode, filterProps)).map(item => item.key);
   let expandedOptsKeys = findAncestorKeys(filteredOptsKeys, keyEntities, false);
-
   if (prevExpandedKeys.length) {
     const prevExpandedValidKeys = prevExpandedKeys.filter(key => Boolean(keyEntities[key]));
     expandedOptsKeys = expandedOptsKeys.concat(prevExpandedValidKeys);
   }
-
   const shownChildKeys = findDescendantKeys(filteredOptsKeys, keyEntities, true);
   const filteredShownKeys = new Set([...shownChildKeys, ...expandedOptsKeys]);
   const flattenNodes = flattenTreeData(treeData, new Set(expandedOptsKeys), showFilteredOnly && filteredShownKeys);
@@ -709,19 +601,15 @@ function filterTreeData(info) {
     filteredExpandedKeys: new Set(expandedOptsKeys),
     filteredShownKeys
   };
-} // return data.value if data.value exist else fall back to key
-
-
+}
+// return data.value if data.value exist else fall back to key
 function getValueOrKey(data) {
   if (Array.isArray(data)) {
     return data.map(item => (0, _get2.default)(item, 'value', item.key));
   }
-
   return (0, _get2.default)(data, 'value', data.key);
 }
 /* Convert value to string */
-
-
 function normalizeValue(value, withObject) {
   if (withObject && isValid(value)) {
     return getValueOrKey(value);
@@ -729,12 +617,10 @@ function normalizeValue(value, withObject) {
     return value;
   }
 }
-
 function updateKeys(keySet, keyEntities) {
   const keyArr = [...keySet];
   return keyArr.filter(key => key in keyEntities);
 }
-
 function calcDisabledKeys(keyEntities) {
   const disabledKeys = Object.keys(keyEntities).filter(key => keyEntities[key].data.disabled);
   const {
@@ -742,7 +628,6 @@ function calcDisabledKeys(keyEntities) {
   } = calcCheckedKeys(disabledKeys, keyEntities);
   return checkedKeys;
 }
-
 function calcDropRelativePosition(event, treeNode) {
   const {
     clientY
@@ -751,25 +636,21 @@ function calcDropRelativePosition(event, treeNode) {
     top,
     bottom,
     height
-  } = treeNode.nodeInstance.getBoundingClientRect(); // eslint-disable-next-line @typescript-eslint/restrict-plus-operands
-
+  } = treeNode.nodeInstance.getBoundingClientRect();
+  // eslint-disable-next-line @typescript-eslint/restrict-plus-operands
   if (clientY <= top + height * DRAG_OFFSET) {
     return -1;
   }
-
   if (clientY >= bottom - height * DRAG_OFFSET) {
     return 1;
   }
-
   return 0;
 }
-
 function getDragNodesKeys(key, keyEntities) {
   return findDescendantKeys([key], keyEntities, true);
 }
-
 function calcDropActualPosition(pos, relativeDropPos) {
-  const posArr = pos.split('-'); // eslint-disable-next-line @typescript-eslint/restrict-plus-operands
-
+  const posArr = pos.split('-');
+  // eslint-disable-next-line @typescript-eslint/restrict-plus-operands
   return relativeDropPos + Number(posArr[posArr.length - 1]);
 }