Fix scrollable table exportDOM

facebook · Nov 29, 2024 · 60801f9 · 60801f9
1 parent 16e4987
commit 60801f9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/packages/lexical-table/src/LexicalTableNode.ts b/packages/lexical-table/src/LexicalTableNode.ts
@@ -227,9 +227,12 @@ export class TableNode extends ElementNode {
   }
 
   exportDOM(editor: LexicalEditor): DOMExportOutput {
+    const {element, after} = super.exportDOM(editor);
     return {
-      ...super.exportDOM(editor),
       after: (tableElement) => {
+        if (after) {
+          tableElement = after(tableElement);
+        }
         if (
           tableElement &&
           isHTMLElement(tableElement) &&
@@ -249,6 +252,10 @@ export class TableNode extends ElementNode {
         }
         return tableElement;
       },
+      element:
+        element && isHTMLElement(element) && element.nodeName !== 'TABLE'
+          ? element.querySelector('table')
+          : element,
     };
   }