Table cell resizer.

fdemian · Aug 2, 2024 · 6cbdc42 · 6cbdc42
1 parent 0477909
commit 6cbdc42
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/src/Kalliope/Plugins/TableCellResizer/index.tsx b/src/Kalliope/Plugins/TableCellResizer/index.tsx
@@ -131,15 +131,19 @@ function TableCellResizer({editor}: {editor: LexicalEditor}): JSX.Element {
       }, 0);
     };
 
-    document.addEventListener('mousemove', onMouseMove);
-    document.addEventListener('mousedown', onMouseDown);
-    document.addEventListener('mouseup', onMouseUp);
+    const removeRootListener = editor.registerRootListener((rootElement, prevRootElement) => {
+      rootElement?.addEventListener('mousemove', onMouseMove);
+      rootElement?.addEventListener('mousedown', onMouseDown);
+      rootElement?.addEventListener('mouseup', onMouseUp);
+
+      prevRootElement?.removeEventListener('mousemove', onMouseMove);
+      prevRootElement?.removeEventListener('mousedown', onMouseDown);
+      prevRootElement?.removeEventListener('mouseup', onMouseUp);
+    });
 
     return () => {
-      document.removeEventListener('mousemove', onMouseMove);
-      document.removeEventListener('mousedown', onMouseDown);
-      document.removeEventListener('mouseup', onMouseUp);
-    };
+      removeRootListener();
+    }
   }, [activeCell, draggingDirection, editor, resetState]);
 
   const isHeightChanging = (direction: MouseDraggingDirection) => {