From f184723354830a973a92d4b879c8e7fc20db37e6 Mon Sep 17 00:00:00 2001
From: sawka <mike.sawka@gmail.com>
Date: Tue, 28 May 2024 14:24:36 -0700
Subject: [PATCH] add useBlockCache() -- can migrate from useBlockAtom

---
 frontend/app/store/global.ts | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/frontend/app/store/global.ts b/frontend/app/store/global.ts
index f72ffc53a..515b66da2 100644
--- a/frontend/app/store/global.ts
+++ b/frontend/app/store/global.ts
@@ -90,6 +90,22 @@ Events.On("block:ptydata", (event: any) => {
     subject.next(data);
 });
 
+const blockCache = new Map<string, Map<string, any>>();
+
+function useBlockCache<T>(blockId: string, name: string, makeFn: () => T): T {
+    let blockMap = blockCache.get(blockId);
+    if (blockMap == null) {
+        blockMap = new Map<string, any>();
+        blockCache.set(blockId, blockMap);
+    }
+    let value = blockMap.get(name);
+    if (value == null) {
+        value = makeFn();
+        blockMap.set(name, value);
+    }
+    return value as T;
+}
+
 const blockAtomCache = new Map<string, Map<string, jotai.Atom<any>>>();
 
 function useBlockAtom<T>(blockId: string, name: string, makeFn: () => jotai.Atom<T>): jotai.Atom<T> {
@@ -107,4 +123,4 @@ function useBlockAtom<T>(blockId: string, name: string, makeFn: () => jotai.Atom
     return atom as jotai.Atom<T>;
 }
 
-export { WOS, atoms, getBlockSubject, globalStore, useBlockAtom };
+export { WOS, atoms, getBlockSubject, globalStore, useBlockAtom, useBlockCache };