refactor: registerMutationListener should initialize its existing nodes

umaranis · Oct 7, 2024 · 55743fb · 55743fb
1 parent d5efec2
commit 55743fb
Showing 1 changed file with 22 additions and 19 deletions.
diff --git a/...e-lexical/src/lib/core/plugins/CodeBlock/CodeActionMenuPlugin/CodeActionMenuPlugin.svelte b/...e-lexical/src/lib/core/plugins/CodeBlock/CodeActionMenuPlugin/CodeActionMenuPlugin.svelte
@@ -83,26 +83,29 @@
   );
 
   onMount(() => {
-    return editor.registerMutationListener(CodeNode, (mutations) => {
-      editor.getEditorState().read(() => {
-        for (const [key, type] of mutations) {
-          switch (type) {
-            case 'created':
-              codeSetRef.add(key);
-              shouldListenMouseMove = codeSetRef.size > 0;
-              break;
-
-            case 'destroyed':
-              codeSetRef.delete(key);
-              shouldListenMouseMove = codeSetRef.size > 0;
-              break;
-
-            default:
-              break;
+    return editor.registerMutationListener(
+      CodeNode,
+      (mutations) => {
+        editor.getEditorState().read(() => {
+          for (const [key, type] of mutations) {
+            switch (type) {
+              case 'created':
+                codeSetRef.add(key);
+                break;
+
+              case 'destroyed':
+                codeSetRef.delete(key);
+                break;
+
+              default:
+                break;
+            }
           }
-        }
-      });
-    });
+        });
+        shouldListenMouseMove = codeSetRef.size > 0;
+      },
+      {skipInitialization: false},
+    );
   });
 
   $: if (shouldListenMouseMove) {