feat: shoehorn in an updater function on useNodeState

ThinAirThings · May 23, 2024 · 44f7ccc · 44f7ccc
1 parent d8690c3
commit 44f7ccc
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -53,7 +53,8 @@
     "neo4j-driver": "^5.19.0",
     "next": "^14.2.0-canary.62",
     "zod": "^3.22.5",
-    "zustand": "^4.5.2"
+    "zustand": "^4.5.2",
+    "react": "^18"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.24.4",

diff --git a/src/layers/ReactCache/defineReactCacheLayer.ts b/src/layers/ReactCache/defineReactCacheLayer.ts
@@ -8,6 +8,7 @@ import { enableMapSet, Draft, produce } from 'immer'
 import { immer } from "zustand/middleware/immer";
 import { useImmer } from "@thinairthings/use-immer";
 import { OmitNodeConstants } from "@/src/base/defineBaseGraph";
+import { useEffect } from "react";
 
 
 
@@ -35,7 +36,8 @@ export const defineReactCacheLayer = <
         Node extends UixNode<T, TypeOf<(N[number] & { nodeType: T })['stateDefinition']>> | undefined = undefined
     >(
         nodeType: T,
-        node?: Node
+        node?: Node,
+        updateAction?: (...args: any[]) => Promise<any>
     ) => ReturnType<typeof useImmer<Node extends UixNode<T, TypeOf<(N[number] & { nodeType: T })['stateDefinition']>>
         ? (Omit<TypeOf<(N[number] & { nodeType: T })['stateDefinition']>, keyof TypeOf<(N[number] & { nodeType: T })['stateDefaults']>>
             & TypeOf<(N[number] & { nodeType: T })['stateDefaults']>)
@@ -76,8 +78,15 @@ export const defineReactCacheLayer = <
     // )
     return {
         ...graph,
-        useNodeState: (nodeType, node) => {
+        useNodeState: (nodeType, node, updater) => {
             const [nodeState, updateNodeState] = useImmer(graph.getNodeDefinition(nodeType).stateDefaults.parse(node ?? {}))
+            useEffect(() => {
+                return () => {
+                    (async () => {
+                        await updater?.(nodeState)
+                    })()
+                }
+            }, [nodeState])
             return [nodeState, updateNodeState] as any
         },
     }