fix(tree): fix expandAll dynamic change behavior

Tencent · Jan 23, 2025 · 37951bf · 37951bf
1 parent 57d4b92
commit 37951bf
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/tree/hooks/useTreeStore.ts b/src/tree/hooks/useTreeStore.ts
@@ -217,6 +217,19 @@ export default function useTreeStore(state: TypeTreeState) {
     });
   });
 
+  watch(refProps.expandAll, (nVal) => {
+    if (nVal) {
+      const valueList = store
+        .getNodes()
+        .filter((node) => Array.isArray(node.children) && node.children.length)
+        .map((node) => node.value);
+      store.setExpanded(valueList);
+    } else {
+      store.replaceExpanded(prevExpanded);
+      prevExpanded = null;
+    }
+  });
+
   return {
     store,
     rebuild,