amis-ui 6.14.0-beta.2 → 6.14.0-beta.4

package/lib/components/Tree.js

@@ -1,5 +1,5 @@
 /**
- * amis-ui v6.14.0-beta.2
+ * amis-ui v6.14.0-beta.4
  * Copyright 2018-2025 fex
  */
 
@@ -33,7 +33,7 @@ var TreeSelector = /** @class */ (function (_super) {
         // key: child option, value: parent option;
         _this.relations = new WeakMap();
         _this.levels = new WeakMap();
-        _this.indexes = new WeakMap();
+        _this.nodePaths = new WeakMap();
         _this.startPoint = {
             x: 0,
             y: 0
@@ -61,7 +61,7 @@ var TreeSelector = /** @class */ (function (_super) {
             virtualHeight: 0,
             itemHeight: 0
         };
-        _this.syncUnFolded(props, undefined, true);
+        _this.syncUnFolded(props, true);
         return _this;
     }
     TreeSelector.prototype.componentDidMount = function () {
@@ -94,6 +94,9 @@ var TreeSelector = /** @class */ (function (_super) {
                 options: props.options,
                 labelField: props.labelField
             }, props.enableNodePath);
+            // VirtualList 如果属性不变不会重新渲染
+            // 而value 变化会影响 renderItem 的结果，所以这里让它变一下
+            this.renderItem = this.renderItem.bind(this);
             this.setState({
                 value: newValue,
                 valueSet: new Set(newValue)
@@ -102,7 +105,7 @@ var TreeSelector = /** @class */ (function (_super) {
     };
     TreeSelector.prototype.componentWillUnmount = function () {
         // clear data
-        this.relations = this.levels = this.indexes = new WeakMap();
+        this.relations = this.levels = this.nodePaths = new WeakMap();
         this.unfolded = {};
         if (this.unSensor) {
             this.unSensor();
@@ -123,22 +126,17 @@ var TreeSelector = /** @class */ (function (_super) {
         this.virtualListRef = ref;
         ref && this.handleVirtualHeight();
     };
-    TreeSelector.prototype.syncUnFolded = function (props, unfoldedLevel, initial) {
-        // 传入默认展开层级需要重新初始化unfolded
-        var initFoldedLevel = typeof unfoldedLevel !== 'undefined';
-        var expandLevel = Number(initFoldedLevel ? unfoldedLevel : props.unfoldedLevel);
+    TreeSelector.prototype.syncUnFolded = function (props, initial) {
+        var expandLevel = Number(props.unfoldedLevel);
         // 初始化树节点的展开状态
         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["".concat(level, "-").concat(index)] && !initFoldedLevel) {
-                return;
-            }
+        amisCore.eachTree(props.options, function (node, index, level, paths, indexes) {
+            var unfoldedKey = indexes.concat(index).join('-');
             if (node.children && node.children.length) {
-                var ret = unfolded["".concat(level, "-").concat(index)];
+                var ret = unfolded[unfoldedKey];
                 if (node[deferField] &&
                     node.loaded &&
-                    !initFoldedLevel &&
                     unfoldedField &&
                     node[unfoldedField] !== false) {
                     ret = true;
@@ -151,17 +149,34 @@ var TreeSelector = /** @class */ (function (_super) {
                     ret = !node[foldedField];
                 }
                 else if (initial || typeof ret === 'undefined') {
-                    ret = !!props.initiallyOpen && !initFoldedLevel;
+                    ret = !!props.initiallyOpen;
                     if (!ret && level <= expandLevel) {
                         ret = true;
                     }
                 }
-                unfolded["".concat(level, "-").concat(index)] = ret;
+                unfolded[unfoldedKey] = ret;
             }
         });
         this.flattenOptions(props, initial);
         return unfolded;
     };
+    TreeSelector.prototype.setFoldedOrUnfolded = function (targetLevel, isFolded) {
+        if (isFolded === void 0) { isFolded = false; }
+        var unfolded = this.unfolded;
+        var _a = this.props, options = _a.options, deferField = _a.deferField, onDeferLoad = _a.onDeferLoad;
+        amisCore.eachTree(options, function (node, index, level, paths, indexes) {
+            var unfoldedKey = indexes.concat(index).join('-');
+            if (node[deferField] && !node.loaded) {
+                onDeferLoad === null || onDeferLoad === void 0 ? void 0 : onDeferLoad(node);
+                return;
+            }
+            if (node.children && node.children.length) {
+                var ret = isFolded ? level < targetLevel : level <= targetLevel;
+                unfolded[unfoldedKey] = ret;
+            }
+        });
+        this.flattenOptions(this.props);
+    };
     TreeSelector.prototype.toggleUnfolded = function (node) {
         var unfolded = this.unfolded;
         var _a = this.props, deferField = _a.deferField, onDeferLoad = _a.onDeferLoad, unfoldedField = _a.unfoldedField;
@@ -169,25 +184,22 @@ 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);
+        var path = this.nodePaths.get(node);
         // ！ hack: 在node上直接添加属性，options 在更新的时候旧的字段会保留
         if (node[deferField] && node.loaded) {
-            node[unfoldedField] = !unfolded["".concat(level, "-").concat(index)];
+            node[unfoldedField] = !unfolded[path];
         }
-        unfolded["".concat(level, "-").concat(index)] = !unfolded["".concat(level, "-").concat(index)];
+        unfolded[path] = !unfolded[path];
         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);
+        var path = this.nodePaths.get(node);
         if (parent) {
-            return !!unfolded["".concat(level, "-").concat(index)] && this.isUnfolded(parent);
+            return !!unfolded[path] && this.isUnfolded(parent);
         }
-        return !!unfolded["".concat(level, "-").concat(index)];
+        return !!unfolded[path];
     };
     TreeSelector.prototype.clearSelect = function () {
         var _this = this;
@@ -590,11 +602,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) {
-                    var index = _this.indexes.get(node);
-                    var level = _this.levels.get(node);
-                    _this.unfolded["".concat(level, "-").concat(index)] = false;
+                    var path = _this.nodePaths.get(node);
+                    _this.unfolded[path] = false;
                     _this.flattenOptions();
-                    _this.forceUpdate();
                 }
             }
             else {
@@ -638,13 +648,13 @@ var TreeSelector = /** @class */ (function (_super) {
     TreeSelector.prototype.flattenOptions = function (props, initial) {
         var _this = this;
         var flattenedOptions = [];
-        amisCore.eachTree((props === null || props === void 0 ? void 0 : props.options) || this.props.options, function (item, index, level, paths) {
+        amisCore.eachTree((props === null || props === void 0 ? void 0 : props.options) || this.props.options, function (item, index, level, paths, indexes) {
             var parent = paths[paths.length - 1];
             if (!amisCore.isVisible(item)) {
                 return;
             }
             _this.levels.set(item, level);
-            _this.indexes.set(item, index);
+            _this.nodePaths.set(item, indexes.concat(index).join('-'));
             parent && _this.relations.set(item, parent);
             if (paths.length === 0) {
                 // 父节点
@@ -1080,6 +1090,12 @@ var TreeSelector = /** @class */ (function (_super) {
         tslib.__metadata("design:paramtypes", [Object]),
         tslib.__metadata("design:returntype", void 0)
     ], TreeSelector.prototype, "virtualListRefSetter", null);
+    tslib.__decorate([
+        amisCore.autobind,
+        tslib.__metadata("design:type", Function),
+        tslib.__metadata("design:paramtypes", [Number, Object]),
+        tslib.__metadata("design:returntype", void 0)
+    ], TreeSelector.prototype, "setFoldedOrUnfolded", null);
     tslib.__decorate([
         amisCore.autobind,
         tslib.__metadata("design:type", Function),