@vtx/components 4.0.0-beta.5 → 4.0.0-beta.7

package/es/vtx-tree/index.js

@@ -0,0 +1,341 @@
+function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
+var _excluded = ["showSearch", "treeData", "searchPlaceholder", "renderMenuList", "fieldNames", "titleClassName", "expandAction"];
+function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _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 _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
+function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
+function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
+function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
+function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
+function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
+function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
+function _objectWithoutProperties(source, excluded) { if (source == null) return {}; var target = _objectWithoutPropertiesLoose(source, excluded); var key, i; if (Object.getOwnPropertySymbols) { var sourceSymbolKeys = Object.getOwnPropertySymbols(source); for (i = 0; i < sourceSymbolKeys.length; i++) { key = sourceSymbolKeys[i]; if (excluded.indexOf(key) >= 0) continue; if (!Object.prototype.propertyIsEnumerable.call(source, key)) continue; target[key] = source[key]; } } return target; }
+function _objectWithoutPropertiesLoose(source, excluded) { if (source == null) return {}; var target = {}; var sourceKeys = Object.keys(source); var key, i; for (i = 0; i < sourceKeys.length; i++) { key = sourceKeys[i]; if (excluded.indexOf(key) >= 0) continue; target[key] = source[key]; } return target; }
+function _createForOfIteratorHelper(o, allowArrayLike) { var it = typeof Symbol !== "undefined" && o[Symbol.iterator] || o["@@iterator"]; if (!it) { if (Array.isArray(o) || (it = _unsupportedIterableToArray(o)) || allowArrayLike && o && typeof o.length === "number") { if (it) o = it; var i = 0; var F = function F() {}; return { s: F, n: function n() { if (i >= o.length) return { done: true }; return { done: false, value: o[i++] }; }, e: function e(_e) { throw _e; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var normalCompletion = true, didErr = false, err; return { s: function s() { it = it.call(o); }, n: function n() { var step = it.next(); normalCompletion = step.done; return step; }, e: function e(_e2) { didErr = true; err = _e2; }, f: function f() { try { if (!normalCompletion && it.return != null) it.return(); } finally { if (didErr) throw err; } } }; }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+import React, { forwardRef, useEffect, useImperativeHandle, useMemo, useRef, useState, useContext } from 'react';
+import PropTypes from 'prop-types';
+import { ConfigProvider, Input, Tooltip, Tree } from 'antd';
+import { DownOutlined } from '@ant-design/icons';
+import { calcRangeKeys, convertDirectoryKeysToNodes } from "./utils";
+import { useDebounceFn, useMemoizedFn } from 'ahooks';
+import { parseTreeData } from '@vtx/utils';
+import classnames from 'classnames';
+import { useStyle } from "./style/index";
+import { VtxProvider, useIntl } from "../vtx-provider";
+import { jsx as _jsx } from "react/jsx-runtime";
+import { jsxs as _jsxs } from "react/jsx-runtime";
+var getParentKey = function getParentKey(key, tree) {
+  var parentKey = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : null;
+  var _iterator = _createForOfIteratorHelper(tree),
+    _step;
+  try {
+    for (_iterator.s(); !(_step = _iterator.n()).done;) {
+      var node = _step.value;
+      if (node.key === key) return parentKey;
+      if (node.children) {
+        var found = getParentKey(key, node.children, node.key);
+        if (found !== undefined) return found;
+      }
+    }
+  } catch (err) {
+    _iterator.e(err);
+  } finally {
+    _iterator.f();
+  }
+  return undefined;
+};
+var VtxTree = /*#__PURE__*/forwardRef(function (propsAll, ref) {
+  var intl = useIntl();
+  var _propsAll$showSearch = propsAll.showSearch,
+    showSearch = _propsAll$showSearch === void 0 ? true : _propsAll$showSearch,
+    _propsAll$treeData = propsAll.treeData,
+    treeData = _propsAll$treeData === void 0 ? [] : _propsAll$treeData,
+    _propsAll$searchPlace = propsAll.searchPlaceholder,
+    searchPlaceholder = _propsAll$searchPlace === void 0 ? intl.getMessage('tree.searchPlaceholder', '请输入要查询的关键字') : _propsAll$searchPlace,
+    renderMenuList = propsAll.renderMenuList,
+    fieldNames = propsAll.fieldNames,
+    titleClassName = propsAll.titleClassName,
+    _propsAll$expandActio = propsAll.expandAction,
+    expandAction = _propsAll$expandActio === void 0 ? 'click' : _propsAll$expandActio,
+    props = _objectWithoutProperties(propsAll, _excluded);
+  var _useContext = useContext(VtxProvider),
+    getPrefixCls = _useContext.getPrefixCls;
+  var prefixCls = getPrefixCls('tree', props.prefixCls);
+  var _useStyle = useStyle(prefixCls),
+    wrapSSR = _useStyle.wrapSSR,
+    hashId = _useStyle.hashId;
+  var treeRef = useRef();
+  var _useState = useState(0),
+    _useState2 = _slicedToArray(_useState, 2),
+    height = _useState2[0],
+    setHeight = _useState2[1];
+  var _useState3 = useState([]),
+    _useState4 = _slicedToArray(_useState3, 2),
+    expandedKeys = _useState4[0],
+    setExpandedKeys = _useState4[1];
+  var _useState5 = useState(props.autoExpandParent || false),
+    _useState6 = _slicedToArray(_useState5, 2),
+    autoExpandParent = _useState6[0],
+    setAutoExpandParent = _useState6[1];
+  var _useState7 = useState(props.selectedKeys || props.defaultSelectedKeys || []),
+    _useState8 = _slicedToArray(_useState7, 2),
+    selectedKeys = _useState8[0],
+    setSelectedKeys = _useState8[1];
+  var inputRef = useRef();
+  var lastSelectedKey = React.useRef();
+  var cachedSelectedKeys = React.useRef();
+  var dataList = useMemo(function () {
+    var dataList = [];
+    var titleKey = (fieldNames === null || fieldNames === void 0 ? void 0 : fieldNames.title) || 'title';
+    var generateList = function generateList(data) {
+      for (var i = 0; i < data.length; i++) {
+        var node = data[i];
+        var key = node.key;
+        dataList.push({
+          key: key,
+          title: node[titleKey]
+        });
+        if (node.children) {
+          generateList(node.children);
+        }
+      }
+    };
+    generateList(treeData);
+    return dataList;
+  }, [treeData]);
+  useEffect(function () {
+    if ('selectedKeys' in props) {
+      setSelectedKeys(props.selectedKeys);
+    }
+  }, [props.selectedKeys]);
+  useEffect(function () {
+    if ('expandedKeys' in props) {
+      setExpandedKeys(props.expandedKeys);
+    }
+  }, [props.expandedKeys]);
+  var onExpand = function onExpand(newExpandedKeys, info) {
+    var _props$onExpand;
+    setAutoExpandParent(false);
+    (_props$onExpand = props.onExpand) === null || _props$onExpand === void 0 || _props$onExpand.call(props, newExpandedKeys, info);
+    if (!('expandedKeys' in props)) {
+      setExpandedKeys(newExpandedKeys);
+    }
+  };
+  var _useState9 = useState(treeData),
+    _useState10 = _slicedToArray(_useState9, 2),
+    newTreeData = _useState10[0],
+    setNewTreeData = _useState10[1];
+  useEffect(function () {
+    if (treeData !== null && treeData !== void 0 && treeData.length) {
+      filterTreeData(inputRef.current ? inputRef.current.input.value : '');
+    }
+  }, [treeData]);
+  var filterTreeData = useMemoizedFn(function () {
+    var searchValue = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : '';
+    var newTreeData = parseTreeData(treeData, {
+      renderNode: function renderNode(item) {
+        var titleKey = (fieldNames === null || fieldNames === void 0 ? void 0 : fieldNames.title) || 'title';
+        var strTitle = item[titleKey] || '';
+        var index = strTitle.indexOf(searchValue);
+        var beforeStr = strTitle.substring(0, index);
+        var afterStr = strTitle.slice(index + searchValue.length);
+        var className = titleClassName ? titleClassName(item) : '';
+        var title = index > -1 ? /*#__PURE__*/_jsxs("span", {
+          className: className,
+          children: [beforeStr, /*#__PURE__*/_jsx("span", {
+            className: classnames("".concat(prefixCls, "-search-value"), hashId),
+            children: searchValue
+          }), afterStr]
+        }) : /*#__PURE__*/_jsx("span", {
+          className: className,
+          children: strTitle
+        });
+        return _objectSpread(_objectSpread({}, item), {}, _defineProperty(_defineProperty(_defineProperty({}, titleKey, title), "key", item.key), "strTitle", strTitle));
+      },
+      filterNode: function filterNode(item) {
+        return item.strTitle.indexOf(searchValue) > -1;
+      }
+    });
+    return setNewTreeData(newTreeData);
+  });
+  var _useDebounceFn = useDebounceFn(function (value) {
+      var _props$onExpand2;
+      var newExpandedKeys = dataList.map(function (item) {
+        if (item.title.indexOf(value) > -1) {
+          return getParentKey(item.key, treeData);
+        }
+        return null;
+      }).filter(function (item, i, self) {
+        return item && self.indexOf(item) === i;
+      });
+      setAutoExpandParent(true);
+      filterTreeData(value);
+      (_props$onExpand2 = props.onExpand) === null || _props$onExpand2 === void 0 || _props$onExpand2.call(props, newExpandedKeys);
+      if (!('expandedKeys' in props)) {
+        setExpandedKeys(newExpandedKeys);
+      }
+    }, {
+      wait: 300
+    }),
+    onSearch = _useDebounceFn.run;
+  useImperativeHandle(ref, function () {
+    return {
+      onSearch: onSearch
+    };
+  });
+  useEffect(function () {
+    var _treeRef$current2;
+    var resizeObserver = new ResizeObserver(function () {
+      var _treeRef$current;
+      setHeight(treeRef === null || treeRef === void 0 || (_treeRef$current = treeRef.current) === null || _treeRef$current === void 0 ? void 0 : _treeRef$current.offsetHeight);
+    });
+    if (treeRef.current) {
+      resizeObserver.observe(treeRef === null || treeRef === void 0 ? void 0 : treeRef.current, {
+        box: 'border-box'
+      });
+    }
+    setHeight(treeRef === null || treeRef === void 0 || (_treeRef$current2 = treeRef.current) === null || _treeRef$current2 === void 0 ? void 0 : _treeRef$current2.offsetHeight);
+    return function () {
+      resizeObserver.disconnect();
+    };
+  }, []);
+  var onSelect = function onSelect(keys, event) {
+    var _props$onSelect;
+    var multiple = props.multiple;
+    var node = event.node,
+      nativeEvent = event.nativeEvent;
+    var _node$key = node.key,
+      key = _node$key === void 0 ? '' : _node$key;
+    // const newState: DirectoryTreeState = {};
+
+    // We need wrap this event since some value is not same
+    var newEvent = _objectSpread(_objectSpread({}, event), {}, {
+      selected: true // Directory selected always true
+    });
+
+    // Windows / Mac single pick
+    var ctrlPick = (nativeEvent === null || nativeEvent === void 0 ? void 0 : nativeEvent.ctrlKey) || (nativeEvent === null || nativeEvent === void 0 ? void 0 : nativeEvent.metaKey);
+    var shiftPick = nativeEvent === null || nativeEvent === void 0 ? void 0 : nativeEvent.shiftKey;
+
+    // Generate new selected keys
+    var newSelectedKeys;
+    if (multiple && ctrlPick) {
+      // Control click
+      newSelectedKeys = keys;
+      lastSelectedKey.current = key;
+      cachedSelectedKeys.current = newSelectedKeys;
+      newEvent.selectedNodes = convertDirectoryKeysToNodes(treeData, newSelectedKeys);
+    } else if (multiple && shiftPick) {
+      // Shift click
+      newSelectedKeys = Array.from(new Set([].concat(_toConsumableArray(cachedSelectedKeys.current || []), _toConsumableArray(calcRangeKeys({
+        treeData: treeData,
+        expandedKeys: expandedKeys,
+        startKey: key,
+        endKey: lastSelectedKey.current
+      })))));
+      newEvent.selectedNodes = convertDirectoryKeysToNodes(treeData, newSelectedKeys);
+    } else {
+      // Single click
+      newSelectedKeys = [key];
+      lastSelectedKey.current = key;
+      cachedSelectedKeys.current = newSelectedKeys;
+      newEvent.selectedNodes = convertDirectoryKeysToNodes(treeData, newSelectedKeys);
+    }
+    (_props$onSelect = props.onSelect) === null || _props$onSelect === void 0 || _props$onSelect.call(props, newSelectedKeys, newEvent);
+    if (!('selectedKeys' in props)) {
+      setSelectedKeys(newSelectedKeys);
+    }
+  };
+  return wrapSSR( /*#__PURE__*/_jsxs("div", {
+    className: classnames("".concat(prefixCls, "-wraaper"), hashId),
+    children: [showSearch && /*#__PURE__*/_jsx(Input, {
+      style: {
+        marginBottom: 8
+      },
+      placeholder: searchPlaceholder,
+      onChange: function onChange(e) {
+        return onSearch(e.target.value);
+      },
+      ref: inputRef
+    }), /*#__PURE__*/_jsx("div", {
+      className: classnames("".concat(prefixCls), hashId),
+      ref: treeRef,
+      children: /*#__PURE__*/_jsx(Tree, _objectSpread(_objectSpread(_objectSpread({}, props), {}, {
+        height: height,
+        switcherIcon: /*#__PURE__*/_jsx(DownOutlined, {}),
+        showLine: {
+          showLeafIcon: false
+        },
+        blockNode: true,
+        treeData: newTreeData,
+        onExpand: onExpand
+      }, expandedKeys.length === 0 && props.defaultExpandAll ? {} : {
+        expandedKeys: expandedKeys
+      }), {}, {
+        autoExpandParent: autoExpandParent,
+        selectedKeys: selectedKeys,
+        onSelect: onSelect,
+        fieldNames: fieldNames,
+        expandAction: expandAction,
+        titleRender: function titleRender(nodeData) {
+          var titleKey = (fieldNames === null || fieldNames === void 0 ? void 0 : fieldNames.title) || 'title';
+          var title = nodeData[titleKey];
+          if (renderMenuList) {
+            var menuList = renderMenuList(nodeData);
+            return /*#__PURE__*/_jsxs("div", {
+              className: classnames("".concat(prefixCls, "-menu-wrap"), hashId),
+              children: [/*#__PURE__*/_jsx("span", {
+                children: title
+              }), /*#__PURE__*/_jsx("div", {
+                className: classnames("".concat(prefixCls, "-menu-list"), hashId),
+                children: menuList.filter(function (item) {
+                  return item.visible;
+                }).map(function (item) {
+                  return /*#__PURE__*/_jsx(Tooltip, {
+                    title: item.label,
+                    children: /*#__PURE__*/_jsx("span", {
+                      className: classnames("".concat(prefixCls, "-menu-item-icon"), hashId),
+                      onClick: function onClick(e) {
+                        e.stopPropagation();
+                        // eslint-disable-next-line @typescript-eslint/no-unused-expressions
+                        item.onClick && item.onClick(nodeData);
+                      },
+                      children: item.icon
+                    })
+                  }, item.label);
+                })
+              })]
+            });
+          }
+          return title;
+        }
+      }))
+    })]
+  }));
+});
+VtxTree.propTypes = {
+  showSearch: PropTypes.bool,
+  prefixCls: PropTypes.string,
+  treeData: PropTypes.array,
+  selectedKeys: PropTypes.array,
+  defaultSelectedKeys: PropTypes.array,
+  expandedKeys: PropTypes.bool,
+  onExpand: PropTypes.func,
+  multiple: PropTypes.bool,
+  autoExpandParent: PropTypes.bool,
+  onSelect: PropTypes.func,
+  searchPlaceholder: PropTypes.string,
+  renderMenuList: PropTypes.func,
+  titleClassName: PropTypes.func,
+  fieldNames: PropTypes.array,
+  defaultExpandAll: PropTypes.bool,
+  expandAction: PropTypes.oneOfType([PropTypes.string, PropTypes.bool])
+};
+export { VtxTree };
+export default VtxTree;

package/es/vtx-tree/style/index.js

@@ -0,0 +1,50 @@
+function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
+function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
+function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
+function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
+function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
+function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
+import { useStyle as useAntdStyle } from "../../vtx-provider";
+var genVtxTreeStyle = function genVtxTreeStyle(token) {
+  var componentCls = token.componentCls;
+  return _defineProperty({}, componentCls, _defineProperty(_defineProperty(_defineProperty(_defineProperty(_defineProperty(_defineProperty(_defineProperty({
+    flex: 1,
+    minHeight: 0,
+    overflowX: 'auto',
+    overflowY: 'hidden'
+  }, '&-search-value', {
+    color: '#f5222d'
+  }), '&-wraaper', {
+    display: 'flex',
+    flexDirection: 'column',
+    boxSizing: 'border-box',
+    height: '100%'
+  }), '&-menu-wrap', {
+    display: 'flex',
+    justifyContent: 'space-between'
+  }), '&-menu-list', {
+    display: 'flex',
+    alignItems: 'center',
+    gap: '4px'
+  }), '&-menu-item-icon', {
+    '&:hover': {
+      color: '#1677ff'
+    }
+  }), '.ant-tree .ant-tree-treenode-disabled .ant-tree-node-content-wrapper', {
+    color: 'rgba(0, 0, 0, 0.65) !important'
+  }), '.ant-tree .ant-tree-node-content-wrapper', _defineProperty({
+    display: 'flex'
+  }, '.ant-tree-title', {
+    flex: 1,
+    whiteSpace: 'nowrap',
+    flexWrap: 'nowrap'
+  })));
+};
+export function useStyle(prefixCls) {
+  return useAntdStyle('VtxTree', function (token) {
+    var vtxTreeToken = _objectSpread(_objectSpread({}, token), {}, {
+      componentCls: ".".concat(prefixCls)
+    });
+    return [genVtxTreeStyle(vtxTreeToken)];
+  });
+}

package/es/vtx-tree/utils.js

@@ -0,0 +1,73 @@
+function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _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(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
+function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+function traverseNodesKey(treeData, callback) {
+  function processNode(dataNode) {
+    var key = dataNode.key,
+      children = dataNode.children;
+    if (callback(key, dataNode) !== false) {
+      traverseNodesKey(children || [], callback);
+    }
+  }
+  treeData.forEach(processNode);
+}
+var Record = {
+  None: 0,
+  Start: 1,
+  End: 2
+};
+
+/** 计算选中范围，只考虑expanded情况以优化性能 */
+export function calcRangeKeys(_ref) {
+  var treeData = _ref.treeData,
+    expandedKeys = _ref.expandedKeys,
+    startKey = _ref.startKey,
+    endKey = _ref.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(treeData, 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);
+    }
+    return expandedKeys.includes(key);
+  });
+  return keys;
+}
+export function convertDirectoryKeysToNodes(treeData, keys) {
+  var restKeys = _toConsumableArray(keys);
+  var nodes = [];
+  traverseNodesKey(treeData, function (key, node) {
+    var index = restKeys.indexOf(key);
+    if (index !== -1) {
+      nodes.push(node);
+      restKeys.splice(index, 1);
+    }
+    return !!restKeys.length;
+  });
+  return nodes;
+}