opensource/lexical
1
0
Fork 0
mirror of https://github.com/facebook/lexical.git synced 2025-08-06 08:30:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Table insert row w/ span support (#4063)

Browse Source
This commit is contained in:
Gerard Rovira
2023-03-10 22:15:27 +00:00
committed by GitHub
parent d5bad41520
commit 390d8d46c5
7 changed files with 210 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
packages/lexical-table/src/LexicalTableUtils.ts
View File

@ -7,10 +7,19 @@
*/
import type {Grid} from './LexicalTableSelection';
import type {LexicalNode} from 'lexical';
import {$findMatchingParent} from '@lexical/utils';
import {$createParagraphNode, $createTextNode} from 'lexical';
import {
$createParagraphNode,
$createTextNode,
$getSelection,
$isRangeSelection,
DEPRECATED_$computeGridMap,
DEPRECATED_$getNodeTriplet,
DEPRECATED_$isGridRowNode,
DEPRECATED_$isGridSelection,
LexicalNode,
} from 'lexical';
import invariant from 'shared/invariant';
import {InsertTableCommandPayloadHeaders} from '.';
@ -217,6 +226,60 @@ export function $insertTableRow(
return tableNode;
}
export function $insertTableRow__EXPERIMENTAL(insertAfter = true): void {
const selection = $getSelection();
invariant(
$isRangeSelection(selection) || DEPRECATED_$isGridSelection(selection),
'Expected a RangeSelection or GridSelection',
);
const focus = selection.focus.getNode();
const [focusCell, , grid] = DEPRECATED_$getNodeTriplet(focus);
const [gridMap, focusCellMap] = DEPRECATED_$computeGridMap(
grid,
focusCell,
focusCell,
);
const {startRow: focusStartRow} = focusCellMap;
if (insertAfter) {
const focusEndRow = focusStartRow + focusCell.__rowSpan - 1;
const focusEndRowMap = gridMap[focusEndRow];
const newRow = $createTableRowNode();
const columnCount = gridMap[0].length;
for (let i = 0; i < columnCount; i++) {
const {cell, startRow} = focusEndRowMap[i];
if (startRow + cell.__rowSpan - 1 <= focusEndRow) {
newRow.append($createTableCellNode(TableCellHeaderStates.NO_STATUS));
} else {
cell.setRowSpan(cell.__rowSpan + 1);
}
}
const focusEndRowNode = grid.getChildAtIndex(focusEndRow);
invariant(
DEPRECATED_$isGridRowNode(focusEndRowNode),
'focusEndRow is not a GridRowNode',
);
focusEndRowNode.insertAfter(newRow);
} else {
const focusStartRowMap = gridMap[focusStartRow];
const newRow = $createTableRowNode();
const columnCount = gridMap[0].length;
for (let i = 0; i < columnCount; i++) {
const {cell, startRow} = focusStartRowMap[i];
if (startRow === focusStartRow) {
newRow.append($createTableCellNode(TableCellHeaderStates.NO_STATUS));
} else {
cell.setRowSpan(cell.__rowSpan + 1);
}
}
const focusStartRowNode = grid.getChildAtIndex(focusStartRow);
invariant(
DEPRECATED_$isGridRowNode(focusStartRowNode),
'focusEndRow is not a GridRowNode',
);
focusStartRowNode.insertBefore(newRow);
}
}
export function $insertTableColumn(
tableNode: TableNode,
targetIndex: number,