feat: also delay rebuildTree with mounted-checks

Author: lukasbach

packages/core/src/core/create-tree.ts

@@ -163,9 +163,10 @@ export const createTree = <T>(
         // TODO this is never used, remove
       },
       setMounted: ({}, isMounted) => {
-        const ref = treeDataRef as TreeDataRef;
-        treeDataRef.current.isMounted = isMounted;
+        const ref = treeDataRef.current as TreeDataRef;
+        ref.isMounted = isMounted;
         if (isMounted) {
+          console.log("apply waiting", ref.waitingForMount?.length);
           ref.waitingForMount?.forEach((cb) => cb());
           ref.waitingForMount = [];
         }
@@ -185,16 +186,29 @@ export const createTree = <T>(
         };
         const ref = treeDataRef.current as TreeDataRef;
         if (ref.isMounted) {
+          console.log("apply substate now");
           apply();
         } else {
+          console.log("apply substate later");
           ref.waitingForMount ??= [];
           ref.waitingForMount.push(apply);
         }
       },
       // TODO rebuildSubTree: (itemId: string) => void;
       rebuildTree: () => {
-        rebuildItemMeta();
-        config.setState?.(state);
+        const ref = treeDataRef.current as TreeDataRef;
+        if (ref.isMounted) {
+          rebuildItemMeta();
+          config.setState?.(state);
+          console.log("rebuild now");
+        } else {
+          ref.waitingForMount ??= [];
+          ref.waitingForMount.push(() => {
+            rebuildItemMeta();
+            config.setState?.(state);
+          });
+          console.log("rebuild later");
+        }
       },
       scheduleRebuildTree: () => {
         rebuildScheduled = true;