@c-rex/components 0.1.31 → 0.1.33

package/src/stores/favorites-store.ts

@@ -60,29 +60,29 @@ export const useFavoritesStore = create<FavoritesStore>()(
 
 
 const favoriteTopic = (documents: Record<string, { topics: Favorite[] }>, documentId: string, id: string, label: string, color: string): Record<string, { topics: Favorite[] }> => {
-
-    const documentsCopy = { ...documents };
-    const notFound = documents[documentId] == undefined;
-
-    if (notFound) {
-        documentsCopy[documentId] = { topics: [] };
-    }
-
-    documentsCopy[documentId]!.topics.push({ id, label, color });
-
-    return documentsCopy
+    const currentDocument = documents[documentId];
+    const topics = currentDocument?.topics || [];
+
+    return {
+        ...documents,
+        [documentId]: {
+            ...(currentDocument?.label ? { label: currentDocument.label } : {}),
+            topics: [...topics, { id, label, color }],
+        },
+    };
 };
 
 const unfavoriteTopic = (documents: Record<string, { topics: Favorite[] }>, documentId: string, id: string): Record<string, { topics: Favorite[] }> => {
-
-    const documentsCopy = { ...documents };
-    const notFound = documents[documentId] == undefined;
-
-    if (notFound) {
-        return documentsCopy;
+    const currentDocument = documents[documentId];
+    if (!currentDocument) {
+        return documents;
     }
 
-    documentsCopy[documentId]!.topics = documentsCopy[documentId]!.topics.filter(topic => topic.id !== id);
-
-    return documentsCopy
-}
+    return {
+        ...documents,
+        [documentId]: {
+            ...(currentDocument.label ? { label: currentDocument.label } : {}),
+            topics: currentDocument.topics.filter((topic) => topic.id !== id),
+        },
+    };
+}