From 3243fc89ce7745bb9d745ff8a2022d76217917c1 Mon Sep 17 00:00:00 2001
From: iamyunsin <yunsin@vip.qq.com>
Date: Fri, 1 Mar 2024 01:22:53 +0800
Subject: [PATCH] fix(react-arborist/provider): There is a bug in using the
 tree.update method to set height.

re #228
---
 packages/react-arborist/src/components/provider.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/react-arborist/src/components/provider.tsx b/packages/react-arborist/src/components/provider.tsx
index 0326c82a..b23f083e 100644
--- a/packages/react-arborist/src/components/provider.tsx
+++ b/packages/react-arborist/src/components/provider.tsx
@@ -57,7 +57,12 @@ export function TreeProvider<T>({
   useMemo(() => {
     updateCount.current += 1;
     api.update(treeProps);
-  }, [...Object.values(treeProps), state.nodes.open]);
+  }, [...Object.values(treeProps)]);
+
+  /* when the node's open or closed state changes, just perform rendering. */
+  useMemo(() => {
+    api.update(api.props);
+  }, [state.nodes.open]);
 
   /* Expose the tree api */
   useImperativeHandle(imperativeHandle, () => api);