jotai-state-tree 1.2.0 → 1.3.0

package/dist/{chunk-3TQNT4MR.mjs → chunk-OAXK4KFM.mjs}

@@ -77,6 +77,9 @@ var StateTreeNode = class {
     this.patchListeners = /* @__PURE__ */ new Set();
     /** Volatile state (non-serialized) */
     this.volatileState = {};
+    /** Cached snapshot for structural sharing optimization */
+    this.cachedSnapshot = void 0;
+    this.isSnapshotDirty = true;
     this.$id = generateNodeId();
     this.$type = type;
     this.$env = env ?? parent?.$env;
@@ -86,6 +89,15 @@ var StateTreeNode = class {
     nodeRegistry.set(this.$id, { node: new WeakRef(this), instance: null });
     nodeFinalizationRegistry.register(this, this.$id, this);
   }
+  invalidateSnapshot() {
+    if (!this.isSnapshotDirty) {
+      this.isSnapshotDirty = true;
+      this.cachedSnapshot = void 0;
+      if (this.$parent) {
+        this.$parent.invalidateSnapshot();
+      }
+    }
+  }
   /** Set the instance reference */
   setInstance(instance) {
     const entry = nodeRegistry.get(this.$id);
@@ -110,6 +122,9 @@ var StateTreeNode = class {
       );
     }
     const oldValue = this.getValue();
+    if (oldValue === value) {
+      return;
+    }
     globalStore.set(this.valueAtom, value);
     this.notifyPatch(
       { op: "replace", path: this.$path, value },
@@ -127,6 +142,7 @@ var StateTreeNode = class {
     child.$env = child.$env ?? this.$env;
     this.children.set(key, child);
     lifecycleHookHandlers.runAfterAttach?.(child);
+    this.invalidateSnapshot();
   }
   /** Recursively update the path of a node and all its children */
   updatePathRecursively(node, newPath) {
@@ -142,6 +158,7 @@ var StateTreeNode = class {
     if (child) {
       child.destroy();
       this.children.delete(key);
+      this.invalidateSnapshot();
     }
   }
   /** Get a child node */
@@ -204,6 +221,7 @@ var StateTreeNode = class {
   }
   /** Notify snapshot listeners */
   notifySnapshotChange() {
+    this.invalidateSnapshot();
     let current = this;
     while (current) {
       const snapshot = getSnapshotFromNode(current);
@@ -261,6 +279,7 @@ var StateTreeNode = class {
   detach() {
     if (this.$parent) {
       lifecycleHookHandlers.runBeforeDetach?.(this);
+      const parent = this.$parent;
       for (const [key, child] of this.$parent.children) {
         if (child === this) {
           this.$parent.children.delete(key);
@@ -269,6 +288,7 @@ var StateTreeNode = class {
       }
       this.$parent = null;
       this.$path = "";
+      parent.invalidateSnapshot();
     }
   }
 };
@@ -283,38 +303,44 @@ function hasStateTreeNode(instance) {
   return instance !== null && typeof instance === "object" && $treenode in instance;
 }
 function getSnapshotFromNode(node) {
+  if (!node.isSnapshotDirty && node.cachedSnapshot !== void 0) {
+    return node.cachedSnapshot;
+  }
   const type = node.$type;
   const value = node.getValue();
+  let snapshot;
   if (type._kind === "model") {
-    const snapshot = {};
+    const modelSnapshot = {};
     const children = node.getChildren();
     for (const [key, childNode] of children) {
-      snapshot[key] = getSnapshotFromNode(childNode);
+      modelSnapshot[key] = getSnapshotFromNode(childNode);
     }
     if (node.postProcessor) {
-      return node.postProcessor(snapshot);
+      snapshot = node.postProcessor(modelSnapshot);
+    } else {
+      snapshot = modelSnapshot;
     }
-    return snapshot;
-  }
-  if (type._kind === "array") {
+  } else if (type._kind === "array") {
     const arr = value;
-    return arr.map((_, index) => {
+    snapshot = arr.map((_, index) => {
       const childNode = node.getChild(String(index));
       return childNode ? getSnapshotFromNode(childNode) : arr[index];
     });
-  }
-  if (type._kind === "map") {
-    const snapshot = {};
+  } else if (type._kind === "map") {
+    const mapSnapshot = {};
     const children = node.getChildren();
     for (const [key, childNode] of children) {
-      snapshot[key] = getSnapshotFromNode(childNode);
+      mapSnapshot[key] = getSnapshotFromNode(childNode);
     }
-    return snapshot;
-  }
-  if (type._kind === "reference") {
-    return node.identifierValue ?? value;
+    snapshot = mapSnapshot;
+  } else if (type._kind === "reference") {
+    snapshot = node.identifierValue ?? value;
+  } else {
+    snapshot = value;
   }
-  return value;
+  node.cachedSnapshot = snapshot;
+  node.isSnapshotDirty = false;
+  return snapshot;
 }
 function applySnapshotToNode(node, snapshot) {
   if (!node.$isAlive) {
@@ -591,6 +617,9 @@ function recordPatches(target) {
   };
 }
 var currentAction = null;
+function getCurrentAction() {
+  return currentAction;
+}
 var actionListeners = /* @__PURE__ */ new Set();
 var actionRecorderHooks = [];
 function registerActionRecorderHook(hook) {
@@ -856,6 +885,7 @@ export {
   onPatch,
   applyPatch,
   recordPatches,
+  getCurrentAction,
   registerActionRecorderHook,
   isActionRunning,
   trackAction,

package/dist/index.d.mts

@@ -1,5 +1,5 @@
-import { I as ISimpleType, a as IType, b as IIdentifierType, c as IIdentifierNumberType, d as ILiteralType, e as IEnumerationType, f as IFrozenType, M as ModelProperties, g as IModelType, h as MixinConfig, i as IMixin, j as IAnyType, k as IArrayType, l as IMapType, m as IOptionalType, n as IMaybeType, o as IMaybeNullType, p as IUnionType, U as UnionOptions, q as ILateType, r as IAnyModelType, R as ReferenceOptions, s as IReferenceType, t as ISafeReferenceType, u as IRefinementType, v as IDisposer, S as SnapshotIn, w as Instance, x as IReversibleJsonPatch } from './tree-B2tSEN1S.mjs';
-export { L as CustomTypeOptions, D as IAnyComplexType, E as IAnyMixin, G as IJsonPatch, z as IMSTArray, A as IMSTMap, y as IStateTreeNode, H as IValidationContext, K as IValidationError, J as IValidationResult, B as ModelInstance, F as ModelSelf, C as SnapshotOut, T as applyPatch, O as applySnapshot, aw as cleanupStaleEntries, ax as clearAllRegistries, aa as clone, aq as cloneDeep, a8 as destroy, a9 as detach, am as findAll, an as findFirst, as as freeze, a2 as getEnv, ag as getGlobalStore, a4 as getIdentifier, af as getMembers, ar as getOrCreatePath, Y as getParent, $ as getParentOfType, a0 as getPath, a1 as getPathParts, av as getRegistryStats, aj as getRelativePath, X as getRoot, N as getSnapshot, ap as getTreeStats, a3 as getType, _ as hasParent, al as haveSameRoot, a5 as isAlive, ak as isAncestor, at as isFrozen, a6 as isRoot, a7 as isStateTreeNode, ao as isValidReference, W as onAction, ay as onLifecycleChange, Q as onPatch, P as onSnapshot, V as recordPatches, ai as resetGlobalStore, ae as resolveIdentifier, ac as resolvePath, ah as setGlobalStore, Z as tryGetParent, ad as tryResolve, au as unfreeze, ab as walk } from './tree-B2tSEN1S.mjs';
+import { I as ISimpleType, a as IType, b as IIdentifierType, c as IIdentifierNumberType, d as ILiteralType, e as IEnumerationType, f as IFrozenType, M as ModelProperties, g as IModelType, h as MixinConfig, i as IMixin, j as IAnyType, k as IArrayType, l as IMapType, m as IOptionalType, n as IMaybeType, o as IMaybeNullType, p as IUnionType, U as UnionOptions, q as ILateType, r as IAnyModelType, R as ReferenceOptions, s as IReferenceType, t as ISafeReferenceType, u as IRefinementType, v as IDisposer, S as SnapshotIn, w as Instance, x as IReversibleJsonPatch } from './tree-BV2K9utF.mjs';
+export { L as CustomTypeOptions, D as IAnyComplexType, E as IAnyMixin, G as IJsonPatch, z as IMSTArray, A as IMSTMap, y as IStateTreeNode, H as IValidationContext, K as IValidationError, J as IValidationResult, B as ModelInstance, F as ModelSelf, C as SnapshotOut, T as applyPatch, O as applySnapshot, aw as cleanupStaleEntries, ax as clearAllRegistries, aa as clone, aq as cloneDeep, a8 as destroy, a9 as detach, am as findAll, an as findFirst, as as freeze, a2 as getEnv, ag as getGlobalStore, a4 as getIdentifier, af as getMembers, ar as getOrCreatePath, Y as getParent, $ as getParentOfType, a0 as getPath, a1 as getPathParts, av as getRegistryStats, aj as getRelativePath, X as getRoot, N as getSnapshot, ap as getTreeStats, a3 as getType, _ as hasParent, al as haveSameRoot, a5 as isAlive, ak as isAncestor, at as isFrozen, a6 as isRoot, a7 as isStateTreeNode, ao as isValidReference, W as onAction, ay as onLifecycleChange, Q as onPatch, P as onSnapshot, V as recordPatches, ai as resetGlobalStore, ae as resolveIdentifier, ac as resolvePath, ah as setGlobalStore, Z as tryGetParent, ad as tryResolve, au as unfreeze, ab as walk } from './tree-BV2K9utF.mjs';
 import 'jotai/vanilla/internals';
 import 'jotai';
 
@@ -362,9 +362,14 @@ interface IMiddlewareHandler {
     (call: IMiddlewareEvent, next: (call: IMiddlewareEvent, callback?: (value: unknown) => unknown) => unknown, abort: (value: unknown) => unknown): unknown;
 }
 /**
- * Add middleware to the global stack
+ * Add middleware to the stack/node
  */
 declare function addMiddleware(target: unknown, handler: IMiddlewareHandler, includeHooks?: boolean): IDisposer;
+/**
+ * Creates an async action (generator function) that can be yielded.
+ * Compatible with MST's flow().
+ */
+declare function flow<Args extends unknown[], R>(generator: (...args: Args) => Generator<Promise<unknown>, R, unknown>): (...args: Args) => Promise<R>;
 interface ISerializedActionCall {
     name: string;
     path: string;
@@ -646,11 +651,6 @@ declare const types: {
     safeDynamicReference: typeof safeDynamicReference;
 };
 
-/**
- * Creates an async action (generator function) that can be yielded.
- * Compatible with MST's flow().
- */
-declare function flow<Args extends unknown[], R>(generator: (...args: Args) => Generator<Promise<unknown>, R, unknown>): (...args: Args) => Promise<R>;
 /**
  * Cast a value to a different type.
  * Useful for working around TypeScript limitations.

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { I as ISimpleType, a as IType, b as IIdentifierType, c as IIdentifierNumberType, d as ILiteralType, e as IEnumerationType, f as IFrozenType, M as ModelProperties, g as IModelType, h as MixinConfig, i as IMixin, j as IAnyType, k as IArrayType, l as IMapType, m as IOptionalType, n as IMaybeType, o as IMaybeNullType, p as IUnionType, U as UnionOptions, q as ILateType, r as IAnyModelType, R as ReferenceOptions, s as IReferenceType, t as ISafeReferenceType, u as IRefinementType, v as IDisposer, S as SnapshotIn, w as Instance, x as IReversibleJsonPatch } from './tree-B2tSEN1S.js';
-export { L as CustomTypeOptions, D as IAnyComplexType, E as IAnyMixin, G as IJsonPatch, z as IMSTArray, A as IMSTMap, y as IStateTreeNode, H as IValidationContext, K as IValidationError, J as IValidationResult, B as ModelInstance, F as ModelSelf, C as SnapshotOut, T as applyPatch, O as applySnapshot, aw as cleanupStaleEntries, ax as clearAllRegistries, aa as clone, aq as cloneDeep, a8 as destroy, a9 as detach, am as findAll, an as findFirst, as as freeze, a2 as getEnv, ag as getGlobalStore, a4 as getIdentifier, af as getMembers, ar as getOrCreatePath, Y as getParent, $ as getParentOfType, a0 as getPath, a1 as getPathParts, av as getRegistryStats, aj as getRelativePath, X as getRoot, N as getSnapshot, ap as getTreeStats, a3 as getType, _ as hasParent, al as haveSameRoot, a5 as isAlive, ak as isAncestor, at as isFrozen, a6 as isRoot, a7 as isStateTreeNode, ao as isValidReference, W as onAction, ay as onLifecycleChange, Q as onPatch, P as onSnapshot, V as recordPatches, ai as resetGlobalStore, ae as resolveIdentifier, ac as resolvePath, ah as setGlobalStore, Z as tryGetParent, ad as tryResolve, au as unfreeze, ab as walk } from './tree-B2tSEN1S.js';
+import { I as ISimpleType, a as IType, b as IIdentifierType, c as IIdentifierNumberType, d as ILiteralType, e as IEnumerationType, f as IFrozenType, M as ModelProperties, g as IModelType, h as MixinConfig, i as IMixin, j as IAnyType, k as IArrayType, l as IMapType, m as IOptionalType, n as IMaybeType, o as IMaybeNullType, p as IUnionType, U as UnionOptions, q as ILateType, r as IAnyModelType, R as ReferenceOptions, s as IReferenceType, t as ISafeReferenceType, u as IRefinementType, v as IDisposer, S as SnapshotIn, w as Instance, x as IReversibleJsonPatch } from './tree-BV2K9utF.js';
+export { L as CustomTypeOptions, D as IAnyComplexType, E as IAnyMixin, G as IJsonPatch, z as IMSTArray, A as IMSTMap, y as IStateTreeNode, H as IValidationContext, K as IValidationError, J as IValidationResult, B as ModelInstance, F as ModelSelf, C as SnapshotOut, T as applyPatch, O as applySnapshot, aw as cleanupStaleEntries, ax as clearAllRegistries, aa as clone, aq as cloneDeep, a8 as destroy, a9 as detach, am as findAll, an as findFirst, as as freeze, a2 as getEnv, ag as getGlobalStore, a4 as getIdentifier, af as getMembers, ar as getOrCreatePath, Y as getParent, $ as getParentOfType, a0 as getPath, a1 as getPathParts, av as getRegistryStats, aj as getRelativePath, X as getRoot, N as getSnapshot, ap as getTreeStats, a3 as getType, _ as hasParent, al as haveSameRoot, a5 as isAlive, ak as isAncestor, at as isFrozen, a6 as isRoot, a7 as isStateTreeNode, ao as isValidReference, W as onAction, ay as onLifecycleChange, Q as onPatch, P as onSnapshot, V as recordPatches, ai as resetGlobalStore, ae as resolveIdentifier, ac as resolvePath, ah as setGlobalStore, Z as tryGetParent, ad as tryResolve, au as unfreeze, ab as walk } from './tree-BV2K9utF.js';
 import 'jotai/vanilla/internals';
 import 'jotai';
 
@@ -362,9 +362,14 @@ interface IMiddlewareHandler {
     (call: IMiddlewareEvent, next: (call: IMiddlewareEvent, callback?: (value: unknown) => unknown) => unknown, abort: (value: unknown) => unknown): unknown;
 }
 /**
- * Add middleware to the global stack
+ * Add middleware to the stack/node
  */
 declare function addMiddleware(target: unknown, handler: IMiddlewareHandler, includeHooks?: boolean): IDisposer;
+/**
+ * Creates an async action (generator function) that can be yielded.
+ * Compatible with MST's flow().
+ */
+declare function flow<Args extends unknown[], R>(generator: (...args: Args) => Generator<Promise<unknown>, R, unknown>): (...args: Args) => Promise<R>;
 interface ISerializedActionCall {
     name: string;
     path: string;
@@ -646,11 +651,6 @@ declare const types: {
     safeDynamicReference: typeof safeDynamicReference;
 };
 
-/**
- * Creates an async action (generator function) that can be yielded.
- * Compatible with MST's flow().
- */
-declare function flow<Args extends unknown[], R>(generator: (...args: Args) => Generator<Promise<unknown>, R, unknown>): (...args: Args) => Promise<R>;
 /**
  * Cast a value to a different type.
  * Useful for working around TypeScript limitations.