chore: support set indeterminate

Tencent · Dec 4, 2024 · 0777282 · 0777282
1 parent 12a6bb2
commit 0777282
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/src/tree/Tree.tsx b/src/tree/Tree.tsx
@@ -273,24 +273,16 @@ const Tree = forwardRef<TreeInstanceFunctions<TreeOptionData>, TreeProps>((origi
             delete spec.checked;
           }
           if ('indeterminate' in options) {
-            setTreeIndeterminate([...indeterminate, value]);
+            // @ts-ignore
+            setTreeIndeterminate((prevIndeterminate: TreeNodeValue[]) => [...prevIndeterminate, value]);
             setIndeterminate(node, spec.indeterminate, { trigger: 'setItem' });
             delete spec.indeterminate;
           }
           node.set(spec);
         }
       },
     }),
-    [
-      store,
-      setExpanded,
-      setActived,
-      setTreeIndeterminate,
-      indeterminate,
-      setChecked,
-      setIndeterminate,
-      handleScrollToElement,
-    ],
+    [store, setExpanded, setActived, setTreeIndeterminate, setChecked, setIndeterminate, handleScrollToElement],
   );
 
   /* ======== render ======= */

diff --git a/src/tree/hooks/useStore.ts b/src/tree/hooks/useStore.ts
@@ -228,7 +228,7 @@ export function useStore(
   useUpdateLayoutEffect(() => {
     if (Array.isArray(value)) {
       store.replaceChecked(value);
-      const checkedValue = store.getCheckedNodes().map((v: TreeNode) => v.data.value);
+      const checkedValue = store.getCheckedNodes().map((v: TreeNode) => v.data[keys.value || 'value']);
       const indeterminateConflict = checkedValue.filter((v) => indeterminate.includes(v));
       if (indeterminateConflict.length) {
         setTreeIndeterminate(indeterminate.filter((v: TreeNodeValue) => !indeterminateConflict.includes(v)));