fix update folder items

egomobile · Aug 1, 2023 · 9dafd9e · 9dafd9e
1 parent 8fcb431
commit 9dafd9e
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 4 deletions.
diff --git a/.ui-dev/src/components/Chatbar/index.tsx b/.ui-dev/src/components/Chatbar/index.tsx
@@ -127,7 +127,15 @@ const Chatbar: React.FC<IChatbarProps> = ({
     const list = [...items];
 
     if (currentFolder) {
-      currentFolder.conversations.push(newConversation);
+      [...list].forEach((item, itemIndex) => {
+        if ('conversations' in item && item.id === currentFolder.id) {
+          list[itemIndex] = {
+            ...item,
+
+            conversations: [...currentFolder.conversations, newConversation]
+          };
+        }
+      });
     } else {
       list.push(newConversation);
     }
@@ -157,11 +165,24 @@ const Chatbar: React.FC<IChatbarProps> = ({
   }, [handleItemsUpdate, items, onConversationDelete]);
 
   const handleUpdateConversation = useCallback((newData: IChatConversation) => {
-    const folder = folders.find((f) => {
+    let folder = folders.find((f) => {
       return f.id === newData.folderId;
     });
 
     const newList = [...items];
+    if (folder) {
+      folder = {
+        ...folder,
+
+        conversations: [...folder.conversations]
+      };
+
+      [...newList].forEach((item, itemIndex) => {
+        if ('conversations' in item && item.id === folder!.id) {
+          newList[itemIndex] = folder!;
+        }
+      });
+    }
 
     const list = folder ? folder.conversations : newList;
     [...list].forEach((item, itemIndex) => {

diff --git a/.ui-dev/src/components/Promptbar/index.tsx b/.ui-dev/src/components/Promptbar/index.tsx
@@ -94,7 +94,15 @@ const Promptbar: React.FC<IPromptbarProps> = ({
     const list = [...items];
 
     if (currentFolder) {
-      currentFolder.prompts.push(newPrompt);
+      [...list].forEach((item, itemIndex) => {
+        if ('prompts' in item && item.id === currentFolder.id) {
+          list[itemIndex] = {
+            ...item,
+
+            prompts: [...currentFolder.prompts, newPrompt]
+          };
+        }
+      });
     } else {
       list.push(newPrompt);
     }
@@ -108,11 +116,24 @@ const Promptbar: React.FC<IPromptbarProps> = ({
   }, []);
 
   const handleUpdatePrompt = useCallback((newData: IChatPrompt) => {
-    const folder = folders.find((f) => {
+    let folder = folders.find((f) => {
       return f.id === newData.folderId;
     });
 
     const newList = [...items];
+    if (folder) {
+      folder = {
+        ...folder,
+
+        prompts: [...folder.prompts]
+      };
+
+      [...newList].forEach((item, itemIndex) => {
+        if ('prompts' in item && item.id === folder!.id) {
+          newList[itemIndex] = folder!;
+        }
+      });
+    }
 
     const list = folder ? folder.prompts : newList;
     [...list].forEach((item, itemIndex) => {