amis-ui 3.0.0 → 3.1.0

package/lib/components/Tree.js

@@ -1,5 +1,5 @@
 /**
- * amis-ui v3.0.0
+ * amis-ui v3.1.0
  * Copyright 2018-2023 fex
  */
 
@@ -46,6 +46,7 @@ var TreeSelector = /** @class */ (function (_super) {
         _this.unfolded = new WeakMap();
         // key: child option, value: parent option;
         _this.relations = new WeakMap();
+        _this.levels = new WeakMap();
         _this.startPoint = {
             x: 0,
             y: 0
@@ -99,6 +100,10 @@ var TreeSelector = /** @class */ (function (_super) {
             });
         }
     };
+    TreeSelector.prototype.componentWillUnmount = function () {
+        // clear data
+        this.relations = this.unfolded = this.levels = new WeakMap();
+    };
     /**
      * 展开懒加载节点的父节点
      */
@@ -221,7 +226,7 @@ var TreeSelector = /** @class */ (function (_super) {
             }
             return;
         }
-        if (onlyLeaf && (Array.isArray(node.children) && node.children.length)) {
+        if (onlyLeaf && Array.isArray(node.children) && node.children.length) {
             return;
         }
         this.setState({
@@ -562,18 +567,18 @@ var TreeSelector = /** @class */ (function (_super) {
         var _this = this;
         var flattenedOptions = [];
         amisCore.eachTree((props === null || props === void 0 ? void 0 : props.options) || this.props.options, function (item, index, level, paths) {
-            var parent = paths[paths.length - 2];
+            var parent = paths[paths.length - 1];
             if (!amisCore.isVisible(item)) {
                 return;
             }
-            if (paths.length === 1) {
+            if (paths.length === 0) {
                 // 父节点
                 flattenedOptions.push(item);
             }
             else if (_this.isUnfolded(parent)) {
                 _this.relations.set(item, parent);
+                _this.levels.set(item, level);
                 // 父节点是展开的状态
-                item.level = level;
                 flattenedOptions.push(item);
             }
         });
@@ -727,11 +732,11 @@ var TreeSelector = /** @class */ (function (_super) {
         var isLeaf = (!item.children || !item.children.length) && !item.placeholder;
         var iconValue = item[iconField] ||
             (enableDefaultIcon !== false
-                ? (Array.isArray(item.children) && item.children.length)
+                ? Array.isArray(item.children) && item.children.length
                     ? 'folder'
                     : 'file'
                 : false);
-        var level = item.level ? item.level - 1 : 0;
+        var level = this.levels.has(item) ? this.levels.get(item) - 1 : 0;
         var body = null;
         if (isEditing && editingItem === item) {
             body = this.renderInput(checkbox);
@@ -756,7 +761,9 @@ var TreeSelector = /** @class */ (function (_super) {
                     React__default["default"].createElement(icons.Icon, { icon: "down-arrow-bold", className: "icon" }))) : (React__default["default"].createElement("span", { className: cx('Tree-itemArrowPlaceholder') })),
                 checkbox,
                 React__default["default"].createElement("div", { className: cx('Tree-itemLabel-item') },
-                    showIcon ? (React__default["default"].createElement("i", { className: cx("Tree-itemIcon ".concat((Array.isArray(item.children) && item.children.length) ? 'Tree-folderIcon' : 'Tree-leafIcon')), onClick: function () {
+                    showIcon ? (React__default["default"].createElement("i", { className: cx("Tree-itemIcon ".concat(Array.isArray(item.children) && item.children.length
+                            ? 'Tree-folderIcon'
+                            : 'Tree-leafIcon')), onClick: function () {
                             return !disabled &&
                                 (multiple
                                     ? _this.handleCheck(item, !checked)
@@ -772,6 +779,7 @@ var TreeSelector = /** @class */ (function (_super) {
                             index: index,
                             multiple: multiple,
                             checked: checked,
+                            labelField: labelField,
                             onChange: function () { return _this.handleCheck(item, !checked); },
                             disabled: disabled || item.disabled
                         })