@kdcloudjs/table 1.0.0

package/lib/table/transforms/treeMode.js

@@ -0,0 +1,286 @@
+"use strict";
+
+var _sliceInstanceProperty2 = require("@babel/runtime-corejs3/core-js-stable/instance/slice");
+
+var _Array$from = require("@babel/runtime-corejs3/core-js-stable/array/from");
+
+var _Symbol = require("@babel/runtime-corejs3/core-js-stable/symbol");
+
+var _getIteratorMethod = require("@babel/runtime-corejs3/core-js/get-iterator-method");
+
+var _getIterator = require("@babel/runtime-corejs3/core-js/get-iterator");
+
+var _interopRequireWildcard = require("@babel/runtime-corejs3/helpers/interopRequireWildcard");
+
+var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.makeTreeModeTransform = makeTreeModeTransform;
+exports.useTreeModeTransform = useTreeModeTransform;
+
+var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/slicedToArray"));
+
+var _toArray2 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/toArray"));
+
+var _defineProperty2 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/defineProperty"));
+
+var _extends3 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/extends"));
+
+var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/toConsumableArray"));
+
+var _getOwnPropertySymbols = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/object/get-own-property-symbols"));
+
+var _set = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/set"));
+
+var _filter = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/filter"));
+
+var _concat = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/concat"));
+
+var _slice = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/slice"));
+
+var _classnames = _interopRequireDefault(require("classnames"));
+
+var _react = _interopRequireWildcard(require("react"));
+
+var _commonViews = require("../common-views");
+
+var _internals = require("../internals");
+
+var _features = require("../pipeline/features");
+
+var _utils = require("../utils");
+
+var _warnTransformsDeprecated = require("./warnTransformsDeprecated");
+
+function _createForOfIteratorHelper(o, allowArrayLike) { var it; if (typeof _Symbol === "undefined" || _getIteratorMethod(o) == null) { 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 = _getIterator(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) { var _context3; if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = _sliceInstanceProperty2(_context3 = Object.prototype.toString.call(o)).call(_context3, 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; }
+
+var __rest = void 0 && (void 0).__rest || function (s, e) {
+  var t = {};
+
+  for (var p in s) {
+    if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
+  }
+
+  if (s != null && typeof _getOwnPropertySymbols.default === "function") for (var i = 0, p = (0, _getOwnPropertySymbols.default)(s); i < p.length; i++) {
+    if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
+  }
+  return t;
+};
+
+var ICON_WIDTH = 16;
+/** @deprecated transform 用法已经过时，请使用 pipeline 来对表格进行拓展 */
+
+function makeTreeModeTransform(_ref) {
+  var onChangeOpenKeys = _ref.onChangeOpenKeys,
+      openKeys = _ref.openKeys,
+      primaryKey = _ref.primaryKey,
+      _ref$iconIndent = _ref.iconIndent,
+      iconIndent = _ref$iconIndent === void 0 ? -6 : _ref$iconIndent,
+      _ref$iconGap = _ref.iconGap,
+      iconGap = _ref$iconGap === void 0 ? 0 : _ref$iconGap,
+      _ref$indentSize = _ref.indentSize,
+      indentSize = _ref$indentSize === void 0 ? 16 : _ref$indentSize,
+      _ref$isLeafNode = _ref.isLeafNode,
+      isLeafNode = _ref$isLeafNode === void 0 ? _utils.isLeafNode : _ref$isLeafNode,
+      _ref$clickArea = _ref.clickArea,
+      clickArea = _ref$clickArea === void 0 ? 'cell' : _ref$clickArea,
+      _ref$treeMetaKey = _ref.treeMetaKey,
+      treeMetaKey = _ref$treeMetaKey === void 0 ? _features.treeMetaSymbol : _ref$treeMetaKey,
+      stopClickEventPropagation = _ref.stopClickEventPropagation;
+  (0, _warnTransformsDeprecated.warnTransformsDeprecated)('makeTreeModeTransform');
+  var openKeySet = new _set.default(openKeys);
+
+  var toggle = function toggle(rowKey) {
+    var expanded = openKeySet.has(rowKey);
+
+    if (expanded) {
+      onChangeOpenKeys((0, _filter.default)(openKeys).call(openKeys, function (key) {
+        return key !== rowKey;
+      }), rowKey, 'collapse');
+    } else {
+      var _context;
+
+      onChangeOpenKeys((0, _concat.default)(_context = []).call(_context, (0, _toConsumableArray2.default)(openKeys), [rowKey]), rowKey, 'expand');
+    }
+  };
+
+  return function (_ref2) {
+    var columns = _ref2.columns,
+        dataSource = _ref2.dataSource;
+    return {
+      columns: processColumns(columns),
+      dataSource: processDataSource(dataSource)
+    };
+
+    function processDataSource(input) {
+      var result = [];
+      dfs(input, 0);
+
+      function dfs(nodes, depth) {
+        if (nodes == null) {
+          return;
+        }
+
+        var _iterator = _createForOfIteratorHelper(nodes),
+            _step;
+
+        try {
+          for (_iterator.s(); !(_step = _iterator.n()).done;) {
+            var node = _step.value;
+            var rowKey = node[primaryKey];
+            var expanded = openKeySet.has(rowKey);
+            var isLeaf = isLeafNode(node, {
+              depth: depth,
+              expanded: expanded,
+              rowKey: rowKey
+            });
+            var treeMeta = {
+              depth: depth,
+              isLeaf: isLeaf,
+              expanded: expanded,
+              rowKey: rowKey
+            };
+            result.push((0, _extends3.default)((0, _defineProperty2.default)({}, treeMetaKey, treeMeta), node));
+
+            if (!isLeaf && expanded) {
+              dfs(node.children, depth + 1);
+            }
+          }
+        } catch (err) {
+          _iterator.e(err);
+        } finally {
+          _iterator.f();
+        }
+      }
+
+      return result;
+    }
+
+    function processColumns(columns) {
+      var _context2;
+
+      if (columns.length === 0) {
+        return columns;
+      }
+
+      var _columns = (0, _toArray2.default)(columns),
+          firstCol = _columns[0],
+          others = (0, _slice.default)(_columns).call(_columns, 1);
+
+      var render = function render(value, record, recordIndex) {
+        var content = _internals.internals.safeRender(firstCol, record, recordIndex);
+
+        if (record[treeMetaKey] == null) {
+          // 没有 treeMeta 信息的话，就返回原先的渲染结果
+          return content;
+        }
+
+        var _record$treeMetaKey = record[treeMetaKey],
+            rowKey = _record$treeMetaKey.rowKey,
+            depth = _record$treeMetaKey.depth,
+            isLeaf = _record$treeMetaKey.isLeaf,
+            expanded = _record$treeMetaKey.expanded;
+        var indent = iconIndent + depth * indentSize;
+
+        if (isLeaf) {
+          return /*#__PURE__*/_react.default.createElement(_commonViews.InlineFlexCell, {
+            className: "expansion-cell leaf"
+          }, /*#__PURE__*/_react.default.createElement("span", {
+            style: {
+              marginLeft: indent + ICON_WIDTH + iconGap
+            }
+          }, content));
+        }
+
+        var onClick = function onClick(e) {
+          if (stopClickEventPropagation) {
+            e.stopPropagation();
+          }
+
+          toggle(rowKey);
+        };
+
+        var expandCls = expanded ? 'expanded' : 'collapsed';
+        return /*#__PURE__*/_react.default.createElement(_commonViews.ExpansionCell, {
+          className: (0, _classnames.default)('expansion-cell', expandCls),
+          style: {
+            cursor: clickArea === 'content' ? 'pointer' : undefined
+          },
+          onClick: clickArea === 'content' ? onClick : undefined
+        }, /*#__PURE__*/_react.default.createElement(_commonViews.icons.CaretRight, {
+          className: (0, _classnames.default)('expansion-icon', expandCls),
+          style: {
+            cursor: clickArea === 'icon' ? 'pointer' : undefined,
+            marginLeft: indent,
+            marginRight: iconGap
+          },
+          onClick: clickArea === 'icon' ? onClick : undefined
+        }), content);
+      };
+
+      var getCellProps = function getCellProps(value, record, rowIndex) {
+        var prevProps = _internals.internals.safeGetCellProps(firstCol, record, rowIndex);
+
+        if (record[treeMetaKey] == null) {
+          // 没有 treeMeta 信息的话，就返回原先的 cellProps
+          return prevProps;
+        }
+
+        var _record$treeMetaKey2 = record[treeMetaKey],
+            isLeaf = _record$treeMetaKey2.isLeaf,
+            rowKey = _record$treeMetaKey2.rowKey;
+
+        if (isLeaf) {
+          return prevProps;
+        }
+
+        return (0, _utils.mergeCellProps)(prevProps, {
+          onClick: function onClick(e) {
+            if (stopClickEventPropagation) {
+              e.stopPropagation();
+            }
+
+            toggle(rowKey);
+          },
+          style: {
+            cursor: 'pointer'
+          }
+        });
+      };
+
+      return (0, _concat.default)(_context2 = [(0, _extends3.default)((0, _extends3.default)({}, firstCol), {
+        title: /*#__PURE__*/_react.default.createElement("span", {
+          style: {
+            marginLeft: iconIndent + ICON_WIDTH + iconGap
+          }
+        }, _internals.internals.safeRenderHeader(firstCol)),
+        render: render,
+        getCellProps: clickArea === 'cell' ? getCellProps : firstCol.getCellProps
+      })]).call(_context2, (0, _toConsumableArray2.default)(others));
+    }
+  };
+}
+/** @deprecated transform 用法已经过时，请使用 pipeline 来对表格进行拓展 */
+
+
+function useTreeModeTransform(_a) {
+  var _a$defaultOpenKeys = _a.defaultOpenKeys,
+      defaultOpenKeys = _a$defaultOpenKeys === void 0 ? [] : _a$defaultOpenKeys,
+      others = __rest(_a, ["defaultOpenKeys"]);
+
+  var _useState = (0, _react.useState)(defaultOpenKeys),
+      _useState2 = (0, _slicedToArray2.default)(_useState, 2),
+      openKeys = _useState2[0],
+      onChangeOpenKeys = _useState2[1];
+
+  return makeTreeModeTransform((0, _extends3.default)({
+    openKeys: openKeys,
+    onChangeOpenKeys: onChangeOpenKeys
+  }, others));
+}

package/lib/table/transforms/visible.d.ts

@@ -0,0 +1,5 @@
+/** @deprecated transform 用法已经过时，请使用 pipeline 来对表格进行拓展 */
+export declare function makeVisibleTransform(visibleCodes: string[]): import("..").Transform<{
+    columns: import("..").ArtColumn[];
+    dataSource: any[];
+}>;

package/lib/table/transforms/visible.js

@@ -0,0 +1,31 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.makeVisibleTransform = makeVisibleTransform;
+
+var _extends2 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/extends"));
+
+var _set = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/set"));
+
+var _utils = require("../utils");
+
+var _warnTransformsDeprecated = require("./warnTransformsDeprecated");
+
+/** @deprecated transform 用法已经过时，请使用 pipeline 来对表格进行拓展 */
+function makeVisibleTransform(visibleCodes) {
+  (0, _warnTransformsDeprecated.warnTransformsDeprecated)('makeVisibleTransform');
+  var set = new _set.default(visibleCodes);
+  return (0, _utils.traverseColumn)(function (column) {
+    if (!(0, _utils.isLeafNode)(column)) {
+      return column;
+    }
+
+    return set.has(column.code) ? column : (0, _extends2.default)((0, _extends2.default)({}, column), {
+      hidden: true
+    });
+  });
+}

package/lib/table/transforms/warnTransformsDeprecated.d.ts

@@ -0,0 +1 @@
+export declare function warnTransformsDeprecated(apiName: string): void;

package/lib/table/transforms/warnTransformsDeprecated.js

@@ -0,0 +1,19 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.warnTransformsDeprecated = warnTransformsDeprecated;
+
+var _set = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/set"));
+
+var warnedSet = new _set.default();
+
+function warnTransformsDeprecated(apiName) {
+  if (!warnedSet.has(apiName)) {
+    warnedSet.add(apiName);
+    console.warn("[kd-table] transform \u7528\u6CD5\u5DF2\u7ECF\u8FC7\u65F6\uFF0C\u8BF7\u4F7F\u7528 pipeline \u6765\u5BF9\u8868\u683C\u8FDB\u884C\u62D3\u5C55" + "\n  \u8BF7\u79FB\u9664\u4EE5\u4E0B API \u7684\u8C03\u7528\uFF1A".concat(apiName));
+  }
+}

package/lib/table/use/useResizeObserver.d.ts

@@ -0,0 +1,2 @@
+declare const useResizeObserver: (callbck: Function, [element]: Array<HTMLElement>) => void;
+export default useResizeObserver;

package/lib/table/use/useResizeObserver.js

@@ -0,0 +1,32 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = void 0;
+
+var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/slicedToArray"));
+
+var _react = _interopRequireDefault(require("react"));
+
+var _resizeObserverPolyfill = _interopRequireDefault(require("resize-observer-polyfill"));
+
+var useResizeObserver = function useResizeObserver(callbck, _ref) {
+  var _ref2 = (0, _slicedToArray2.default)(_ref, 1),
+      element = _ref2[0];
+
+  _react.default.useEffect(function () {
+    var resizeObserver = new _resizeObserverPolyfill.default(function () {
+      callbck();
+    });
+    resizeObserver.observe(element);
+    return function () {
+      resizeObserver.disconnect();
+    };
+  }, []);
+};
+
+var _default = useResizeObserver;
+exports.default = _default;

package/lib/table/utils/applyTransforms.d.ts

@@ -0,0 +1,6 @@
+import { Transform } from '../interfaces';
+/**
+ * 以 input 作为输入，按序使用 transform.
+ *
+ * `applyTransform(input, f1, f2, f3)` 等价于 `f3(f2(f1(input)))` */
+export default function applyTransforms<T>(input: T, ...transforms: Transform<T>[]): T;

package/lib/table/utils/applyTransforms.js

@@ -0,0 +1,24 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = applyTransforms;
+
+var _reduce = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/reduce"));
+
+/**
+ * 以 input 作为输入，按序使用 transform.
+ *
+ * `applyTransform(input, f1, f2, f3)` 等价于 `f3(f2(f1(input)))` */
+function applyTransforms(input) {
+  for (var _len = arguments.length, transforms = new Array(_len > 1 ? _len - 1 : 0), _key = 1; _key < _len; _key++) {
+    transforms[_key - 1] = arguments[_key];
+  }
+
+  return (0, _reduce.default)(transforms).call(transforms, function (v, fn) {
+    return fn(v);
+  }, input);
+}

package/lib/table/utils/buildTree.d.ts

@@ -0,0 +1,36 @@
+declare type WithChildren<T> = T & {
+    children?: WithChildren<T>[];
+};
+/**
+ * 根据 idProp 与 parentIdProp 从对象数组中构建对应的树
+ * 当 A[parentIdProp] === B[idProp] 时，对象A会被移动到对象B的children。
+ * 当一个对象的 parentIdProp 不与其他对象的 idProp 字段相等时，该对象被作为树的顶层节点
+ * @example
+ * const array = [
+ *   { id: 'node-1', parent: 'root' },
+ *   { id: 'node-2', parent: 'root' },
+ *   { id: 'node-3', parent: 'node-2' },
+ *   { id: 'node-4', parent: 'node-2' },
+ *   { id: 'node-5', parent: 'node-4' },
+ * ]
+ * const tree = buildTree('id', 'parent', array)
+ * expect(tree).toEqual([
+ *   { id: 'node-1', parent: 'root' },
+ *   {
+ *     id: 'node-2',
+ *     parent: 'root',
+ *     children: [
+ *       { id: 'node-3', parent: 'node-2' },
+ *       {
+ *         id: 'node-4',
+ *         parent: 'node-2',
+ *         children: [{ id: 'node-5', parent: 'node-4' }],
+ *       },
+ *     ],
+ *   },
+ * ])
+ */
+export default function buildTree<ID extends string, PID extends string, T extends {
+    [key in ID | PID]: string;
+}>(idProp: ID, parentIdProp: PID, items: T[]): WithChildren<T>[];
+export {};

package/lib/table/utils/buildTree.js

@@ -0,0 +1,140 @@
+"use strict";
+
+var _flatMapInstanceProperty = require("@babel/runtime-corejs3/core-js-stable/instance/flat-map");
+
+var _sliceInstanceProperty = require("@babel/runtime-corejs3/core-js-stable/instance/slice");
+
+var _Array$from2 = require("@babel/runtime-corejs3/core-js-stable/array/from");
+
+var _Symbol = require("@babel/runtime-corejs3/core-js-stable/symbol");
+
+var _getIteratorMethod = require("@babel/runtime-corejs3/core-js/get-iterator-method");
+
+var _getIterator = require("@babel/runtime-corejs3/core-js/get-iterator");
+
+var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = buildTree;
+
+var _extends2 = _interopRequireDefault(require("@babel/runtime-corejs3/helpers/extends"));
+
+var _map = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/map"));
+
+var _filter = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/filter"));
+
+var _from = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/array/from"));
+
+var _values = _interopRequireDefault(require("@babel/runtime-corejs3/core-js-stable/instance/values"));
+
+var _others = require("./others");
+
+function _createForOfIteratorHelper(o, allowArrayLike) { var it; if (typeof _Symbol === "undefined" || _getIteratorMethod(o) == null) { 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 = _getIterator(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) { var _context2; if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = _sliceInstanceProperty(_context2 = Object.prototype.toString.call(o)).call(_context2, 8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return _Array$from2(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; }
+
+/**
+ * 根据 idProp 与 parentIdProp 从对象数组中构建对应的树
+ * 当 A[parentIdProp] === B[idProp] 时，对象A会被移动到对象B的children。
+ * 当一个对象的 parentIdProp 不与其他对象的 idProp 字段相等时，该对象被作为树的顶层节点
+ * @example
+ * const array = [
+ *   { id: 'node-1', parent: 'root' },
+ *   { id: 'node-2', parent: 'root' },
+ *   { id: 'node-3', parent: 'node-2' },
+ *   { id: 'node-4', parent: 'node-2' },
+ *   { id: 'node-5', parent: 'node-4' },
+ * ]
+ * const tree = buildTree('id', 'parent', array)
+ * expect(tree).toEqual([
+ *   { id: 'node-1', parent: 'root' },
+ *   {
+ *     id: 'node-2',
+ *     parent: 'root',
+ *     children: [
+ *       { id: 'node-3', parent: 'node-2' },
+ *       {
+ *         id: 'node-4',
+ *         parent: 'node-2',
+ *         children: [{ id: 'node-5', parent: 'node-4' }],
+ *       },
+ *     ],
+ *   },
+ * ])
+ */
+function buildTree(idProp, parentIdProp, items) {
+  var _context;
+
+  var wrapperMap = new _map.default();
+
+  var ensure = function ensure(id) {
+    if (wrapperMap.has(id)) {
+      return wrapperMap.get(id);
+    }
+
+    var wrapper = {
+      id: id,
+      parent: null,
+      item: null,
+      children: []
+    };
+    wrapperMap.set(id, wrapper);
+    return wrapper;
+  };
+
+  var _iterator = _createForOfIteratorHelper(items),
+      _step;
+
+  try {
+    for (_iterator.s(); !(_step = _iterator.n()).done;) {
+      var item = _step.value;
+      var parentWrapper = ensure(item[parentIdProp]);
+      var itemWrapper = ensure(item[idProp]);
+      itemWrapper.parent = parentWrapper;
+      parentWrapper.children.push(itemWrapper);
+      itemWrapper.item = item;
+    }
+  } catch (err) {
+    _iterator.e(err);
+  } finally {
+    _iterator.f();
+  }
+
+  var topLevelWrappers = (0, _flatMapInstanceProperty(_others))((0, _filter.default)(_context = (0, _from.default)((0, _values.default)(wrapperMap).call(wrapperMap))).call(_context, function (wrapper) {
+    return wrapper.parent == null;
+  }), function (wrapper) {
+    return wrapper.children;
+  });
+  return unwrapRecursively(topLevelWrappers); // todo 可能存在无线递归的情况
+
+  function unwrapRecursively(wrapperArray) {
+    var result = [];
+
+    var _iterator2 = _createForOfIteratorHelper(wrapperArray),
+        _step2;
+
+    try {
+      for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
+        var wrapper = _step2.value;
+
+        if (wrapper.children.length === 0) {
+          result.push(wrapper.item);
+        } else {
+          result.push((0, _extends2.default)((0, _extends2.default)({}, wrapper.item), {
+            children: unwrapRecursively(wrapper.children)
+          }));
+        }
+      }
+    } catch (err) {
+      _iterator2.e(err);
+    } finally {
+      _iterator2.f();
+    }
+
+    return result;
+  }
+}

package/lib/table/utils/collectNodes.d.ts

@@ -0,0 +1,8 @@
+import { AbstractTreeNode } from '../interfaces';
+/** 遍历所有节点，并将节点收集到一个数组中.
+ * order 参数可用于指定遍历规则：
+ * * `pre` 前序遍历 （默认）
+ * * `post` 后续遍历
+ * * `leaf-only` 忽略内部节点，只收集叶子节点
+ * */
+export default function collectNodes<T extends AbstractTreeNode>(nodes: T[], order?: 'pre' | 'post' | 'leaf-only'): T[];

package/lib/table/utils/collectNodes.js

@@ -0,0 +1,72 @@
+"use strict";
+
+var _sliceInstanceProperty = require("@babel/runtime-corejs3/core-js-stable/instance/slice");
+
+var _Array$from = require("@babel/runtime-corejs3/core-js-stable/array/from");
+
+var _Symbol = require("@babel/runtime-corejs3/core-js-stable/symbol");
+
+var _getIteratorMethod = require("@babel/runtime-corejs3/core-js/get-iterator-method");
+
+var _getIterator = require("@babel/runtime-corejs3/core-js/get-iterator");
+
+var _interopRequireDefault = require("@babel/runtime-corejs3/helpers/interopRequireDefault");
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = collectNodes;
+
+var _isLeafNode = _interopRequireDefault(require("./isLeafNode"));
+
+function _createForOfIteratorHelper(o, allowArrayLike) { var it; if (typeof _Symbol === "undefined" || _getIteratorMethod(o) == null) { 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 = _getIterator(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) { var _context; if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = _sliceInstanceProperty(_context = Object.prototype.toString.call(o)).call(_context, 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; }
+
+/** 遍历所有节点，并将节点收集到一个数组中.
+ * order 参数可用于指定遍历规则：
+ * * `pre` 前序遍历 （默认）
+ * * `post` 后续遍历
+ * * `leaf-only` 忽略内部节点，只收集叶子节点
+ * */
+function collectNodes(nodes) {
+  var order = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 'pre';
+  var result = [];
+  dfs(nodes);
+  return result;
+
+  function dfs(nodes) {
+    if (nodes == null) {
+      return;
+    }
+
+    var _iterator = _createForOfIteratorHelper(nodes),
+        _step;
+
+    try {
+      for (_iterator.s(); !(_step = _iterator.n()).done;) {
+        var node = _step.value;
+
+        if ((0, _isLeafNode.default)(node)) {
+          result.push(node);
+        } else {
+          if (order === 'pre') {
+            result.push(node);
+            dfs(node.children);
+          } else if (order === 'post') {
+            dfs(node.children);
+            result.push(node);
+          } else {
+            dfs(node.children);
+          }
+        }
+      }
+    } catch (err) {
+      _iterator.e(err);
+    } finally {
+      _iterator.f();
+    }
+  }
+}