'init'

app/providers/DialogProvider/useDialogsCache.ts

@@ -0,0 +1,69 @@
+import { useMemory } from "../MemoryProvider/useMemory";
+import { Message } from "./DialogProvider";
+
+export interface DialogCache {
+    publicKey: string;
+    messages: Message[];
+}
+
+export function useDialogsCache() {
+    const [dialogsCache, setDialogsCache] = useMemory<DialogCache[]>("dialogs-cache", [], true);
+    
+    const getDialogCache = (publicKey: string) => {
+        const found = dialogsCache.find((cache) => cache.publicKey == publicKey);
+        if(!found){
+            return [];
+        }
+        return found.messages;
+    }
+
+    const addOrUpdateDialogCache = (publicKey: string, messages: Message[]) => {
+        const existingIndex = dialogsCache.findIndex((cache) => cache.publicKey == publicKey);
+        let newCache = [...dialogsCache];
+        if(existingIndex !== -1){
+            newCache[existingIndex].messages = messages;
+        }else{
+            newCache.push({publicKey, messages});
+        }
+        setDialogsCache(newCache);
+    }
+
+    const updateAttachmentInDialogCache = (attachment_id: string, blob: string) => {
+        /**
+         * TODO: Optimize this function to avoid full map if possible
+         */
+        let newCache = dialogsCache.map((cache) => {
+            let newMessages = cache.messages.map((message) => {
+                if(message.attachments){
+                    let newAttachments = message.attachments.map((attachment) => {
+                        if(attachment.id == attachment_id){
+                            return {
+                                ...attachment,
+                                blob: blob
+                            }
+                        }
+                        return attachment;
+                    });
+                    return {
+                        ...message,
+                        attachments: newAttachments
+                    }
+                }
+                return message;
+            });
+            return {
+                ...cache,
+                messages: newMessages
+            }
+        });
+        setDialogsCache(newCache);
+    }
+
+    return {
+        getDialogCache,
+        addOrUpdateDialogCache,
+        dialogsCache,
+        updateAttachmentInDialogCache,
+        setDialogsCache
+    }
+}