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

package/lib/es/tree/treeUtil.js

@@ -7,25 +7,10 @@ import _isObject from "lodash/isObject";
 import _max from "lodash/max";
 import _uniq from "lodash/uniq";
 import _difference from "lodash/difference";
-import _concatInstanceProperty from "@babel/runtime-corejs3/core-js-stable/instance/concat";
-import _mapInstanceProperty from "@babel/runtime-corejs3/core-js-stable/instance/map";
-import _Object$assign from "@babel/runtime-corejs3/core-js-stable/object/assign";
-import _forEachInstanceProperty from "@babel/runtime-corejs3/core-js-stable/instance/for-each";
-import _Object$entries from "@babel/runtime-corejs3/core-js-stable/object/entries";
-import _Array$isArray from "@babel/runtime-corejs3/core-js-stable/array/is-array";
-import _includesInstanceProperty from "@babel/runtime-corejs3/core-js-stable/instance/includes";
-import _Set from "@babel/runtime-corejs3/core-js-stable/set";
-import _everyInstanceProperty from "@babel/runtime-corejs3/core-js-stable/instance/every";
-import _Object$keys from "@babel/runtime-corejs3/core-js-stable/object/keys";
-import _someInstanceProperty from "@babel/runtime-corejs3/core-js-stable/instance/some";
-import _filterInstanceProperty from "@babel/runtime-corejs3/core-js-stable/instance/filter";
-import _Object$values from "@babel/runtime-corejs3/core-js-stable/object/values";
 const DRAG_OFFSET = 0.45;
 
 function getPosition(level, index) {
-  var _context;
-
-  return _concatInstanceProperty(_context = "".concat(level, "-")).call(_context, index);
+  return "".concat(level, "-").concat(index);
 }
 
 function isValid(val) {
@@ -48,18 +33,17 @@ export function flattenTreeData(treeNodeList, expandedKeys) {
 
   function flatten(list) {
     let parent = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : null;
-    return _mapInstanceProperty(list).call(list, (treeNode, index) => {
+    return list.map((treeNode, index) => {
       const pos = getPosition(parent ? parent.pos : '0', index);
       const mergedKey = treeNode.key; // Add FlattenDataNode into list
 
-      const flattenNode = _Object$assign(_Object$assign({}, _pick(treeNode, ['key', 'label', 'value', 'icon', 'disabled', 'isLeaf'])), {
+      const flattenNode = Object.assign(Object.assign({}, _pick(treeNode, ['key', 'label', 'value', 'icon', 'disabled', 'isLeaf'])), {
         parent,
         pos,
         children: null,
         data: treeNode,
         _innerDataTag: true
       });
-
       const isBooleanFilteredShownKeys = typeof filteredShownKeys === 'boolean';
 
       if (!filterSearch || !isBooleanFilteredShownKeys && filteredShownKeys.has(mergedKey)) {
@@ -82,8 +66,6 @@ export function flattenTreeData(treeNodeList, expandedKeys) {
   return flattenList;
 }
 export function convertJsonToData(treeJson) {
-  var _context3;
-
   const treeData = [];
 
   const traverseNode = (key, children, path, res) => {
@@ -96,22 +78,17 @@ export function convertJsonToData(treeJson) {
     };
 
     if (_isObject(children)) {
-      var _context2;
-
       const newChildren = [];
-
-      _forEachInstanceProperty(_context2 = _Object$entries(children)).call(_context2, c => {
+      Object.entries(children).forEach(c => {
         traverseNode(c[0], c[1], currPath, newChildren);
       });
-
       newNode.children = newChildren;
     }
 
     res.push(newNode);
   };
 
-  _forEachInstanceProperty(_context3 = _Object$entries(treeJson)).call(_context3, item => traverseNode(item[0], item[1], [], treeData));
-
+  Object.entries(treeJson).forEach(item => traverseNode(item[0], item[1], [], treeData));
   return treeData;
 }
 /**
@@ -125,7 +102,7 @@ export function traverseDataNodes(treeNodes, callback) {
 
     if (node) {
       const data = {
-        data: _Object$assign({}, node),
+        data: Object.assign({}, node),
         ind,
         pos,
         key: node.key !== null ? node.key : pos,
@@ -137,7 +114,7 @@ export function traverseDataNodes(treeNodes, callback) {
 
 
     if (children) {
-      _forEachInstanceProperty(children).call(children, (subNode, subIndex) => {
+      children.forEach((subNode, subIndex) => {
         processNode(subNode, subIndex, {
           node,
           pos,
@@ -166,8 +143,7 @@ export function convertDataToEntities(dataNodes) {
       key,
       parentPos
     } = data;
-
-    const entity = _Object$assign({}, data);
+    const entity = Object.assign({}, data);
 
     const value = _get(entity, 'data.value', null);
 
@@ -196,9 +172,9 @@ export function findKeysForValues(valueList, valueEntities) {
     return [];
   }
 
-  if (!isMultiple && _Array$isArray(valueList)) {
+  if (!isMultiple && Array.isArray(valueList)) {
     valueList = valueList.length ? [valueList[0]] : [];
-  } else if (!_Array$isArray(valueList)) {
+  } else if (!Array.isArray(valueList)) {
     valueList = [valueList];
   }
 
@@ -207,13 +183,11 @@ export function findKeysForValues(valueList, valueEntities) {
   }
 
   const res = [];
-
-  _forEachInstanceProperty(valueList).call(valueList, val => {
+  valueList.forEach(val => {
     if (val in valueEntities) {
       res.push(valueEntities[val]);
     }
   });
-
   return res;
 }
 export function findDescendantKeys(selectedKeys, options) {
@@ -231,32 +205,29 @@ export function findDescendantKeys(selectedKeys, options) {
     const hasChildren = isValid(children);
 
     if (hasChildren) {
-      _forEachInstanceProperty(children).call(children, child => {
+      children.forEach(child => {
         res.push(child.key);
         findChild(options[child.key]);
       });
     }
   };
 
-  _forEachInstanceProperty(selectedKeys).call(selectedKeys, item => {
+  selectedKeys.forEach(item => {
     if (self) {
       res.push(item);
     }
 
     findChild(options[item]);
   });
-
   return res;
 }
 export function findChildKeys(keys, options) {
   let omitKeys = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : [];
   const res = [];
-  keys && _forEachInstanceProperty(keys).call(keys, key => {
-    var _context4;
-
+  keys && keys.forEach(key => {
     const opts = options[key];
-    opts && opts.children && _forEachInstanceProperty(_context4 = opts.children).call(_context4, child => {
-      if (!omitKeys.length || !_includesInstanceProperty(omitKeys).call(omitKeys, child.key)) {
+    opts && opts.children && opts.children.forEach(child => {
+      if (!omitKeys.length || !omitKeys.includes(child.key)) {
         res.push(child.key);
       }
     });
@@ -281,28 +252,25 @@ export function findLeafKeys(keys, options) {
     if (isLeaf) {
       res.push(item.key);
     } else {
-      _forEachInstanceProperty(children).call(children, child => {
+      children.forEach(child => {
         findChild(options[child.key]);
       });
     }
   };
 
-  _forEachInstanceProperty(keys).call(keys, item => {
+  keys.forEach(item => {
     findChild(options[item]);
   });
-
   return res;
 }
 export function findSiblingKeys(selectedKeys, options) {
   let self = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : true;
   const par = [];
-
-  _forEachInstanceProperty(selectedKeys).call(selectedKeys, item => {
+  selectedKeys.forEach(item => {
     if (options[item] && options[item].parent) {
       par.push(options[item].parent.key);
     }
   });
-
   const res = findChildKeys(_uniq(par), options, self ? [] : selectedKeys);
   return res;
 }
@@ -317,21 +285,19 @@ export function findAncestorKeys(selectedKeys, options) {
     }
   };
 
-  _forEachInstanceProperty(selectedKeys).call(selectedKeys, item => {
+  selectedKeys.forEach(item => {
     options[item] && findPar(options[item]);
 
     if (self) {
       res.push(item);
     }
   });
-
   return res;
 }
 
 function getSortedKeyList(keyList, keyEntities) {
   const levelMap = {};
-
-  _forEachInstanceProperty(keyList).call(keyList, key => {
+  keyList.forEach(key => {
     if (!keyEntities[key]) {
       return;
     }
@@ -346,12 +312,11 @@ function getSortedKeyList(keyList, keyEntities) {
       levelMap[level] = [key];
     }
   });
-
   return levelMap;
 }
 
 export function calcCheckedKeys(values, keyEntities) {
-  const keyList = _Array$isArray(values) ? values : [values];
+  const keyList = Array.isArray(values) ? values : [values];
   const descendantKeys = findDescendantKeys(keyList, keyEntities, true);
   /**
    * Recursively find the parent element. Because the incoming nodes are all checked,
@@ -359,8 +324,8 @@ export function calcCheckedKeys(values, keyEntities) {
    *  disabled+unchecked nodes, their ancestor nodes will definitely not be checked
    */
 
-  const checkedKeys = new _Set([...descendantKeys]);
-  let halfCheckedKeys = new _Set([]);
+  const checkedKeys = new Set([...descendantKeys]);
+  let halfCheckedKeys = new Set([]);
   let visited = [];
   const levelMap = getSortedKeyList(keyList, keyEntities);
 
@@ -371,19 +336,18 @@ export function calcCheckedKeys(values, keyEntities) {
       level
     } = node; // If the node does not have a parent node, or the node has been processed just now, no processing is done
 
-    if (!parent || _includesInstanceProperty(visited).call(visited, key)) {
+    if (!parent || visited.includes(key)) {
       return;
     }
 
     const siblingKeys = findSiblingKeys([key], keyEntities); // visited for caching to avoid double counting
 
     visited = [...visited, ...siblingKeys];
-
-    const allChecked = _everyInstanceProperty(siblingKeys).call(siblingKeys, siblingKey => checkedKeys.has(siblingKey));
+    const allChecked = siblingKeys.every(siblingKey => checkedKeys.has(siblingKey));
 
     if (!allChecked) {
       const ancestorKeys = findAncestorKeys([key], keyEntities, false);
-      halfCheckedKeys = new _Set([...halfCheckedKeys, ...ancestorKeys]);
+      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
 
@@ -397,12 +361,9 @@ export function calcCheckedKeys(values, keyEntities) {
 
 
   while (!_isEmpty(levelMap)) {
-    var _context5, _context6;
-
-    const maxLevel = _max(_mapInstanceProperty(_context5 = _Object$keys(levelMap)).call(_context5, key => Number(key)));
-
-    _forEachInstanceProperty(_context6 = levelMap[maxLevel]).call(_context6, key => calcCurrLevel(keyEntities[key]));
+    const maxLevel = _max(Object.keys(levelMap).map(key => Number(key)));
 
+    levelMap[maxLevel].forEach(key => calcCurrLevel(keyEntities[key]));
     delete levelMap[maxLevel];
   }
 
@@ -418,23 +379,23 @@ export function calcExpandedKeys() {
   let keyEntities = arguments.length > 1 ? arguments[1] : undefined;
   let autoExpandParent = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : true;
 
-  if (!_Array$isArray(keyList)) {
+  if (!Array.isArray(keyList)) {
     keyList = [keyList];
   }
 
   if (autoExpandParent) {
     const ancestorKeys = findAncestorKeys(keyList, keyEntities, true);
-    return new _Set(ancestorKeys);
+    return new Set(ancestorKeys);
   }
 
-  return new _Set(keyList);
+  return new Set(keyList);
 }
 /* Calculate the expanded node by value */
 // eslint-disable-next-line max-len
 
 export function calcExpandedKeysForValues(value, keyEntities, isMultiple, valueEntities) {
   const keys = findKeysForValues(value, valueEntities, isMultiple);
-  return new _Set(findAncestorKeys(keys, keyEntities, false));
+  return new Set(findAncestorKeys(keys, keyEntities, false));
 }
 export function calcMotionKeys(oldKeySet, newKeySet, keyEntities) {
   let motionType = 'show';
@@ -477,10 +438,8 @@ export function filter(sugInput, option, filterTreeNode, filterProps) {
 
   if (typeof filterTreeNode === 'boolean') {
     filterFn = (targetVal, val) => {
-      var _context7;
-
       const input = targetVal.toLowerCase();
-      return _includesInstanceProperty(_context7 = val.toString().toLowerCase()).call(_context7, input);
+      return val.toString().toLowerCase().includes(input);
     };
   }
 
@@ -491,7 +450,7 @@ export function filter(sugInput, option, filterTreeNode, filterProps) {
   return filterFn(sugInput, target);
 }
 export function normalizedArr(val) {
-  if (!_Array$isArray(val)) {
+  if (!Array.isArray(val)) {
     return [val];
   } else {
     return val;
@@ -500,10 +459,10 @@ export function normalizedArr(val) {
 export function normalizeKeyList(keyList, keyEntities) {
   let leafOnly = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
   const res = [];
-  const keyListSet = new _Set(keyList);
+  const keyListSet = new Set(keyList);
 
   if (!leafOnly) {
-    _forEachInstanceProperty(keyList).call(keyList, key => {
+    keyList.forEach(key => {
       if (!keyEntities[key]) {
         return;
       }
@@ -519,7 +478,7 @@ export function normalizeKeyList(keyList, keyEntities) {
       res.push(key);
     });
   } else {
-    _forEachInstanceProperty(keyList).call(keyList, key => {
+    keyList.forEach(key => {
       if (keyEntities[key] && !isValid(keyEntities[key].children)) {
         res.push(key);
       }
@@ -532,9 +491,7 @@ export function getMotionKeys(eventKey, expandedKeys, keyEntities) {
   const res = [];
 
   const getChild = itemKey => {
-    var _context8;
-
-    keyEntities[itemKey].children && _forEachInstanceProperty(_context8 = keyEntities[itemKey].children).call(_context8, item => {
+    keyEntities[itemKey].children && keyEntities[itemKey].children.forEach(item => {
       const {
         key
       } = item;
@@ -553,7 +510,7 @@ export function getMotionKeys(eventKey, expandedKeys, keyEntities) {
 export function calcCheckedKeysForChecked(key, keyEntities, checkedKeys, halfCheckedKeys) {
   const descendantKeys = findDescendantKeys([key], keyEntities, true);
   const nodeItem = keyEntities[key];
-  checkedKeys = new _Set([...checkedKeys, key]);
+  checkedKeys = new Set([...checkedKeys, key]);
 
   const calcCurrLevel = node => {
     if (!node.parent) {
@@ -566,11 +523,11 @@ export function calcCheckedKeysForChecked(key, keyEntities, checkedKeys, halfChe
     } = node;
     const siblingKeys = findSiblingKeys([key], keyEntities); // eslint-disable-next-line @typescript-eslint/no-shadow
 
-    const allChecked = _everyInstanceProperty(siblingKeys).call(siblingKeys, key => checkedKeys.has(key));
+    const allChecked = siblingKeys.every(key => checkedKeys.has(key));
 
     if (!allChecked) {
       const ancestorKeys = findAncestorKeys([key], keyEntities, false);
-      halfCheckedKeys = new _Set([...halfCheckedKeys, ...ancestorKeys]);
+      halfCheckedKeys = new Set([...halfCheckedKeys, ...ancestorKeys]);
     } else {
       const par = node.parent;
       checkedKeys.add(par.key);
@@ -580,7 +537,7 @@ export function calcCheckedKeysForChecked(key, keyEntities, checkedKeys, halfChe
 
   calcCurrLevel(nodeItem);
   return {
-    checkedKeys: new _Set([...checkedKeys, ...descendantKeys]),
+    checkedKeys: new Set([...checkedKeys, ...descendantKeys]),
     halfCheckedKeys
   };
 } // eslint-disable-next-line max-len
@@ -588,8 +545,7 @@ export function calcCheckedKeysForChecked(key, keyEntities, checkedKeys, halfChe
 export function calcCheckedKeysForUnchecked(key, keyEntities, checkedKeys, halfCheckedKeys) {
   const descendantKeys = findDescendantKeys([key], keyEntities, true);
   const nodeItem = keyEntities[key];
-
-  _forEachInstanceProperty(descendantKeys).call(descendantKeys, descendantKey => {
+  descendantKeys.forEach(descendantKey => {
     if (checkedKeys.has(descendantKey)) {
       checkedKeys.delete(descendantKey);
     }
@@ -618,18 +574,16 @@ export function calcCheckedKeysForUnchecked(key, keyEntities, checkedKeys, halfC
     } = node;
     const siblingKeys = findSiblingKeys([key], keyEntities); // eslint-disable-next-line @typescript-eslint/no-shadow
 
-    const anyChecked = _someInstanceProperty(siblingKeys).call(siblingKeys, key => checkedKeys.has(key) || halfCheckedKeys.has(key));
-
+    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
 
     if (anyChecked) {
-      _forEachInstanceProperty(ancestorKeys).call(ancestorKeys, itemKey => {
+      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
-
     } else {
       if (checkedKeys.has(par.key)) {
         checkedKeys.delete(par.key);
@@ -650,8 +604,6 @@ export function calcCheckedKeysForUnchecked(key, keyEntities, checkedKeys, halfC
   };
 }
 export function filterTreeData(info) {
-  var _context9, _context10;
-
   const {
     showFilteredOnly,
     keyEntities,
@@ -662,29 +614,28 @@ export function filterTreeData(info) {
     prevExpandedKeys
   } = info;
   let filteredOptsKeys = [];
-  filteredOptsKeys = _mapInstanceProperty(_context9 = _filterInstanceProperty(_context10 = _Object$values(keyEntities)).call(_context10, item => filter(inputValue, item.data, filterTreeNode, filterProps))).call(_context9, item => item.key);
+  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 = _filterInstanceProperty(prevExpandedKeys).call(prevExpandedKeys, key => Boolean(keyEntities[key]));
-
-    expandedOptsKeys = _concatInstanceProperty(expandedOptsKeys).call(expandedOptsKeys, prevExpandedValidKeys);
+    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);
+  const filteredShownKeys = new Set([...shownChildKeys, ...expandedOptsKeys]);
+  const flattenNodes = flattenTreeData(treeData, new Set(expandedOptsKeys), showFilteredOnly && filteredShownKeys);
   return {
     flattenNodes,
-    filteredKeys: new _Set(filteredOptsKeys),
-    filteredExpandedKeys: new _Set(expandedOptsKeys),
+    filteredKeys: new Set(filteredOptsKeys),
+    filteredExpandedKeys: new Set(expandedOptsKeys),
     filteredShownKeys
   };
 } // return data.value if data.value exist else fall back to key
 
 export function getValueOrKey(data) {
-  if (_Array$isArray(data)) {
-    return _mapInstanceProperty(data).call(data, item => _get(item, 'value', item.key));
+  if (Array.isArray(data)) {
+    return data.map(item => _get(item, 'value', item.key));
   }
 
   return _get(data, 'value', data.key);
@@ -700,13 +651,10 @@ export function normalizeValue(value, withObject) {
 }
 export function updateKeys(keySet, keyEntities) {
   const keyArr = [...keySet];
-  return _filterInstanceProperty(keyArr).call(keyArr, key => key in keyEntities);
+  return keyArr.filter(key => key in keyEntities);
 }
 export function calcDisabledKeys(keyEntities) {
-  var _context11;
-
-  const disabledKeys = _filterInstanceProperty(_context11 = _Object$keys(keyEntities)).call(_context11, key => keyEntities[key].data.disabled);
-
+  const disabledKeys = Object.keys(keyEntities).filter(key => keyEntities[key].data.disabled);
   const {
     checkedKeys
   } = calcCheckedKeys(disabledKeys, keyEntities);