handle new row header property

udecode · Nov 3, 2023 · 8b60732 · 8b60732
1 parent 9d2931b
commit 8b60732
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/packages/table/src/transforms/insertTableRow.ts b/packages/table/src/transforms/insertTableRow.ts
@@ -59,16 +59,18 @@ export const insertTableRow = <V extends Value>(
 
   const getEmptyRowNode = () => ({
     type: getPluginType(editor, ELEMENT_TR),
-    children: (trNode.children as TElement[]).map((_, i) =>
-      getEmptyCellNode(editor, {
-        header:
-          header ??
-          (tableEntry[0].children as TElement[]).every(
-            (n) => n.children[i].type === ELEMENT_TH
-          ),
+    children: (trNode.children as TElement[]).map((_, i) => {
+      const hasSingleRow = tableEntry[0].children.length === 1;
+      const isHeaderColumn =
+        !hasSingleRow &&
+        (tableEntry[0].children as TElement[]).every(
+          (n) => n.children[i].type === ELEMENT_TH
+        );
+      return getEmptyCellNode(editor, {
+        header: header ?? isHeaderColumn,
         ...newCellChildren,
-      })
-    ),
+      });
+    }),
   });
 
   withoutNormalizing(editor, () => {