Fix bmEditing without refresh

1 files changed, 14 insertions, 6 deletions

diff --git a/extension/src/components/FolderBody.tsx b/extension/src/components/FolderBody.tsx
index 9b7c5f4..0d9ddec 100644
--- a/extension/src/components/FolderBody.tsx
+++ b/extension/src/components/FolderBody.tsx
@@ -27,10 +27,18 @@ function FolderBody (props: {id: string}) {
 
     useEffect(() => {
         updateBookmarks();
-        // getBrowser().bookmarks.onRemoved.addListener((id: string, moveInfo) => {
-        //     if (moveInfo.parentId !== props.id) return;
-        //     updateBookmarks();
-        // })
+        getBrowser().bookmarks.onRemoved.addListener((id: string, moveInfo) => {
+            if (moveInfo.parentId !== props.id) return;
+            updateBookmarks();
+        })
+        getBrowser().bookmarks.onMoved.addListener((id: string, moveInfo) => {
+            if (moveInfo.parentId !== props.id && moveInfo.oldParentId !== props.id ) return;
+            updateBookmarks();
+        })
+        getBrowser().bookmarks.onCreated.addListener((id: string, moveInfo) => {
+            if (moveInfo.parentId !== props.id) return;
+            updateBookmarks();
+        })
     }, []);
 
     useEffect(() => {
@@ -43,8 +51,8 @@ function FolderBody (props: {id: string}) {
         <div className={"folderBody"}>
             {children.map(child =>
                 child.children
-                    ? <FolderButton id={child.id} />
-                    : <Bookmark id={child.id} />
+                    ? <FolderButton id={child.id} key={child.id}/>
+                    : <Bookmark id={child.id} key={child.id}/>
             )}
         </div>
     )