Read colgroups when deserializing LexicalTableNode

facebook · Oct 14, 2024 · 1467755 · 1467755
1 parent 2e64bfc
commit 1467755
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/packages/lexical-table/src/LexicalTableNode.ts b/packages/lexical-table/src/LexicalTableNode.ts
@@ -33,6 +33,7 @@ import {$isTableCellNode, TableCellNode} from './LexicalTableCellNode';
 import {TableDOMCell, TableDOMTable} from './LexicalTableObserver';
 import {TableRowNode} from './LexicalTableRowNode';
 import {getTable} from './LexicalTableSelectionHelpers';
+import {PIXEL_VALUE_REG_EXP} from './constants'
 
 export type SerializedTableNode = Spread<
   {
@@ -358,6 +359,19 @@ export function $convertTableElement(
   if (domNode.hasAttribute('data-lexical-row-striping')) {
     tableNode.setRowStriping(true);
   }
+  const colGroup = domNode.querySelector(":scope > colgroup");
+  if (colGroup) {
+    let columns: number[] | undefined = [];
+    for (const col of colGroup.querySelectorAll(":scope > col")) {
+      const width = (col as HTMLElement).style.width;
+      if (!width || !PIXEL_VALUE_REG_EXP.test(width)) {
+        columns = undefined;
+        break;
+      }
+      columns.push(parseFloat(width))
+    }
+    if (columns) tableNode.setColWidths(columns);
+  }
   return {node: tableNode};
 }