amis-ui 6.13.0 → 6.14.0-beta.1

package/lib/components/Tree.js

@@ -1,5 +1,5 @@
 /**
- * amis-ui v6.13.0
+ * amis-ui v6.14.0-beta.1
  * Copyright 2018-2025 fex
  */
 
@@ -29,10 +29,11 @@ var TreeSelector = /** @class */ (function (_super) {
     function TreeSelector(props) {
         var _this = _super.call(this, props) || this;
         // 展开的节点
-        _this.unfolded = new WeakMap();
+        _this.unfolded = {};
         // key: child option, value: parent option;
         _this.relations = new WeakMap();
         _this.levels = new WeakMap();
+        _this.indexes = new WeakMap();
         _this.startPoint = {
             x: 0,
             y: 0
@@ -61,7 +62,6 @@ var TreeSelector = /** @class */ (function (_super) {
             itemHeight: 0
         };
         _this.syncUnFolded(props, undefined, true);
-        _this.flattenOptions(props, true);
         return _this;
     }
     TreeSelector.prototype.componentDidMount = function () {
@@ -83,7 +83,6 @@ var TreeSelector = /** @class */ (function (_super) {
         this.state;
         if (prevProps.options !== props.options) {
             this.syncUnFolded(props);
-            this.flattenOptions(props);
         }
         if (prevProps.value !== props.value ||
             prevProps.options !== props.options) {
@@ -103,7 +102,8 @@ var TreeSelector = /** @class */ (function (_super) {
     };
     TreeSelector.prototype.componentWillUnmount = function () {
         // clear data
-        this.relations = this.unfolded = this.levels = new WeakMap();
+        this.relations = this.levels = this.indexes = new WeakMap();
+        this.unfolded = {};
         if (this.unSensor) {
             this.unSensor();
             delete this.unSensor;
@@ -131,11 +131,11 @@ var TreeSelector = /** @class */ (function (_super) {
         var unfolded = this.unfolded;
         var _a = this.props, deferField = _a.deferField, foldedField = _a.foldedField, unfoldedField = _a.unfoldedField;
         amisCore.eachTree(props.options, function (node, index, level) {
-            if (unfolded.has(node) && !initFoldedLevel) {
+            if (unfolded["".concat(level, "-").concat(index)] && !initFoldedLevel) {
                 return;
             }
             if (node.children && node.children.length) {
-                var ret = true;
+                var ret = unfolded["".concat(level, "-").concat(index)];
                 if (node[deferField] &&
                     node.loaded &&
                     !initFoldedLevel &&
@@ -150,17 +150,16 @@ var TreeSelector = /** @class */ (function (_super) {
                 else if (foldedField && typeof node[foldedField] !== 'undefined') {
                     ret = !node[foldedField];
                 }
-                else {
+                else if (initial || typeof ret === 'undefined') {
                     ret = !!props.initiallyOpen && !initFoldedLevel;
                     if (!ret && level <= expandLevel) {
                         ret = true;
                     }
                 }
-                unfolded.set(node, ret);
+                unfolded["".concat(level, "-").concat(index)] = ret;
             }
         });
-        initFoldedLevel && this.forceUpdate();
-        this.flattenOptions(undefined, initial);
+        this.flattenOptions(props, initial);
         return unfolded;
     };
     TreeSelector.prototype.toggleUnfolded = function (node) {
@@ -170,21 +169,25 @@ var TreeSelector = /** @class */ (function (_super) {
             onDeferLoad === null || onDeferLoad === void 0 ? void 0 : onDeferLoad(node);
             return;
         }
+        var index = this.indexes.get(node);
+        var level = this.levels.get(node);
         // ！ hack: 在node上直接添加属性，options 在更新的时候旧的字段会保留
         if (node[deferField] && node.loaded) {
-            node[unfoldedField] = !unfolded.get(node);
+            node[unfoldedField] = !unfolded["".concat(level, "-").concat(index)];
         }
-        unfolded.set(node, !unfolded.get(node));
+        unfolded["".concat(level, "-").concat(index)] = !unfolded["".concat(level, "-").concat(index)];
         this.flattenOptions();
         this.forceUpdate();
     };
     TreeSelector.prototype.isUnfolded = function (node) {
         var unfolded = this.unfolded;
         var parent = this.relations.get(node);
+        var index = this.indexes.get(node);
+        var level = this.levels.get(node);
         if (parent) {
-            return !!unfolded.get(node) && this.isUnfolded(parent);
+            return !!unfolded["".concat(level, "-").concat(index)] && this.isUnfolded(parent);
         }
-        return !!unfolded.get(node);
+        return !!unfolded["".concat(level, "-").concat(index)];
     };
     TreeSelector.prototype.clearSelect = function () {
         var _this = this;
@@ -587,7 +590,9 @@ var TreeSelector = /** @class */ (function (_super) {
                     y: e.clientY
                 };
                 if ((_a = node === null || node === void 0 ? void 0 : node.children) === null || _a === void 0 ? void 0 : _a.length) {
-                    _this.unfolded.set(node, false);
+                    var index = _this.indexes.get(node);
+                    var level = _this.levels.get(node);
+                    _this.unfolded["".concat(level, "-").concat(index)] = false;
                     _this.flattenOptions();
                     _this.forceUpdate();
                 }
@@ -639,6 +644,7 @@ var TreeSelector = /** @class */ (function (_super) {
                 return;
             }
             _this.levels.set(item, level);
+            _this.indexes.set(item, index);
             parent && _this.relations.set(item, parent);
             if (paths.length === 0) {
                 // 父节点
@@ -1046,7 +1052,7 @@ var TreeSelector = /** @class */ (function (_super) {
         iconField: 'icon',
         deferField: 'defer',
         unfoldedField: 'unfolded',
-        foldedField: 'foled',
+        foldedField: 'folded',
         disabledField: 'disabled',
         joinValues: true,
         extractValue: false,