jotai-state-tree 1.4.0 → 1.4.1

package/dist/index.js

@@ -1883,6 +1883,9 @@ var ModelType = class _ModelType {
           }
           return descriptor.value;
         }
+        if (propStr === "toggle") {
+          console.log("PROXY GET toggle:", propStr, "in allActions:", propStr in allActions, "keys:", Object.keys(allActions), "node alive:", node.$isAlive);
+        }
         if (propStr in allActions) {
           return allActions[propStr];
         }

package/dist/index.mjs

@@ -895,6 +895,9 @@ var ModelType = class _ModelType {
           }
           return descriptor.value;
         }
+        if (propStr === "toggle") {
+          console.log("PROXY GET toggle:", propStr, "in allActions:", propStr in allActions, "keys:", Object.keys(allActions), "node alive:", node.$isAlive);
+        }
         if (propStr in allActions) {
           return allActions[propStr];
         }

package/dist/react.js

@@ -1249,11 +1249,14 @@ function useHydrateStore(target, snapshot, options) {
     (0, import_utils.useHydrateAtoms)(pairs, { store: options?.store ?? getGlobalStore() });
   }
 }
+var undoManagersCache = /* @__PURE__ */ new WeakMap();
+var timeTravelManagersCache = /* @__PURE__ */ new WeakMap();
 function useUndoManager(target, options) {
-  const managerRef = (0, import_react.useRef)(null);
   const [, forceUpdate] = (0, import_react.useState)(0);
-  if (!managerRef.current) {
-    managerRef.current = createUndoManager(target, options);
+  let manager = target && typeof target === "object" ? undoManagersCache.get(target) : void 0;
+  if (!manager && target && typeof target === "object") {
+    manager = createUndoManager(target, options);
+    undoManagersCache.set(target, manager);
   }
   (0, import_react.useEffect)(() => {
     const disposer = onPatch(target, () => {
@@ -1261,19 +1264,16 @@ function useUndoManager(target, options) {
     });
     return () => {
       disposer();
-      if (managerRef.current) {
-        managerRef.current.dispose();
-        managerRef.current = null;
-      }
     };
   }, [target]);
-  return managerRef.current;
+  return manager;
 }
 function useTimeTravelManager(target, options) {
-  const managerRef = (0, import_react.useRef)(null);
   const [, forceUpdate] = (0, import_react.useState)(0);
-  if (!managerRef.current) {
-    managerRef.current = createTimeTravelManager(target, options);
+  let manager = target && typeof target === "object" ? timeTravelManagersCache.get(target) : void 0;
+  if (!manager && target && typeof target === "object") {
+    manager = createTimeTravelManager(target, options);
+    timeTravelManagersCache.set(target, manager);
   }
   (0, import_react.useEffect)(() => {
     const disposer = onPatch(target, () => {
@@ -1281,13 +1281,9 @@ function useTimeTravelManager(target, options) {
     });
     return () => {
       disposer();
-      if (managerRef.current) {
-        managerRef.current.dispose();
-        managerRef.current = null;
-      }
     };
   }, [target]);
-  return managerRef.current;
+  return manager;
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/react.mjs

@@ -389,11 +389,14 @@ function useHydrateStore(target, snapshot, options) {
     useHydrateAtoms(pairs, { store: options?.store ?? getGlobalStore() });
   }
 }
+var undoManagersCache = /* @__PURE__ */ new WeakMap();
+var timeTravelManagersCache = /* @__PURE__ */ new WeakMap();
 function useUndoManager(target, options) {
-  const managerRef = useRef(null);
   const [, forceUpdate] = useState(0);
-  if (!managerRef.current) {
-    managerRef.current = createUndoManager(target, options);
+  let manager = target && typeof target === "object" ? undoManagersCache.get(target) : void 0;
+  if (!manager && target && typeof target === "object") {
+    manager = createUndoManager(target, options);
+    undoManagersCache.set(target, manager);
   }
   useEffect(() => {
     const disposer = onPatch(target, () => {
@@ -401,19 +404,16 @@ function useUndoManager(target, options) {
     });
     return () => {
       disposer();
-      if (managerRef.current) {
-        managerRef.current.dispose();
-        managerRef.current = null;
-      }
     };
   }, [target]);
-  return managerRef.current;
+  return manager;
 }
 function useTimeTravelManager(target, options) {
-  const managerRef = useRef(null);
   const [, forceUpdate] = useState(0);
-  if (!managerRef.current) {
-    managerRef.current = createTimeTravelManager(target, options);
+  let manager = target && typeof target === "object" ? timeTravelManagersCache.get(target) : void 0;
+  if (!manager && target && typeof target === "object") {
+    manager = createTimeTravelManager(target, options);
+    timeTravelManagersCache.set(target, manager);
   }
   useEffect(() => {
     const disposer = onPatch(target, () => {
@@ -421,13 +421,9 @@ function useTimeTravelManager(target, options) {
     });
     return () => {
       disposer();
-      if (managerRef.current) {
-        managerRef.current.dispose();
-        managerRef.current = null;
-      }
     };
   }, [target]);
-  return managerRef.current;
+  return manager;
 }
 export {
   Observer,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jotai-state-tree",
-  "version": "1.4.0",
+  "version": "1.4.1",
   "description": "MobX-State-Tree API compatible library powered by Jotai",
   "main": "dist/index.js",
   "module": "dist/index.mjs",

package/src/model.ts

@@ -359,6 +359,9 @@ class ModelType<
         }
 
         // Check actions
+        if (propStr === "toggle") {
+          console.log("PROXY GET toggle:", propStr, "in allActions:", propStr in allActions, "keys:", Object.keys(allActions), "node alive:", node.$isAlive);
+        }
         if (propStr in allActions) {
           return allActions[propStr];
         }

package/src/react.ts

@@ -702,6 +702,9 @@ export { hasStateTreeNode };
 // Undo/Redo & Time Travel Hooks
 // ============================================================================
 
+const undoManagersCache = new WeakMap<any, IUndoManager>();
+const timeTravelManagersCache = new WeakMap<any, ITimeTravelManager>();
+
 /**
  * Hook to create and use an UndoManager.
  * Automatically handles the subscription lifecycle and triggers re-renders on changes.
@@ -710,11 +713,12 @@ export function useUndoManager(
   target: unknown,
   options?: IUndoManagerOptions,
 ): IUndoManager {
-  const managerRef = useRef<IUndoManager | null>(null);
   const [, forceUpdate] = useState(0);
 
-  if (!managerRef.current) {
-    managerRef.current = createUndoManager(target, options);
+  let manager = (target && typeof target === "object") ? undoManagersCache.get(target) : undefined;
+  if (!manager && target && typeof target === "object") {
+    manager = createUndoManager(target, options);
+    undoManagersCache.set(target, manager);
   }
 
   useEffect(() => {
@@ -723,14 +727,10 @@ export function useUndoManager(
     });
     return () => {
       disposer();
-      if (managerRef.current) {
-        managerRef.current.dispose();
-        managerRef.current = null;
-      }
     };
   }, [target]);
 
-  return managerRef.current;
+  return manager!;
 }
 
 /**
@@ -744,11 +744,12 @@ export function useTimeTravelManager(
     autoRecord?: boolean;
   },
 ): ITimeTravelManager {
-  const managerRef = useRef<ITimeTravelManager | null>(null);
   const [, forceUpdate] = useState(0);
 
-  if (!managerRef.current) {
-    managerRef.current = createTimeTravelManager(target, options);
+  let manager = (target && typeof target === "object") ? timeTravelManagersCache.get(target) : undefined;
+  if (!manager && target && typeof target === "object") {
+    manager = createTimeTravelManager(target, options);
+    timeTravelManagersCache.set(target, manager);
   }
 
   useEffect(() => {
@@ -757,14 +758,10 @@ export function useTimeTravelManager(
     });
     return () => {
       disposer();
-      if (managerRef.current) {
-        managerRef.current.dispose();
-        managerRef.current = null;
-      }
     };
   }, [target]);
 
-  return managerRef.current;
+  return manager!;
 }
 
 // ============================================================================