clean up fullscreen dialogs on unmount

ibastawisi · Aug 7, 2024 · 570a7bc · 570a7bc
1 parent 3c5d1ac
commit 570a7bc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/editor/plugins/ToolbarPlugin/Dialogs/GraphDialog.tsx b/src/editor/plugins/ToolbarPlugin/Dialogs/GraphDialog.tsx
@@ -78,6 +78,12 @@ function GraphDialog({ editor, node }: { editor: LexicalEditor, node: GraphNode
     closeDialog();
   }
 
+  useEffect(() => {
+    return () => {
+      document.body.classList.remove('fullscreen');
+    }
+  }, []);
+
   return <Dialog open fullScreen onClose={handleClose} disableEscapeKeyDown
     TransitionProps={{
       onEntered() { document.body.classList.add('fullscreen'); },

diff --git a/src/editor/plugins/ToolbarPlugin/Dialogs/Sketch/index.tsx b/src/editor/plugins/ToolbarPlugin/Dialogs/Sketch/index.tsx
@@ -287,6 +287,12 @@ function SketchDialog({ editor, node }: { editor: LexicalEditor, node: ImageNode
 
   const loading = !excalidrawAPI;
 
+  useEffect(() => {
+    return () => {
+      document.body.classList.remove('fullscreen');
+    }
+  }, []);
+
   return <Dialog open fullScreen={true} onClose={handleClose} disableEscapeKeyDown
     TransitionProps={{
       onEntered() { document.body.classList.add('fullscreen') },