chore: Upgrade all of prosemirror (#5366)

Co-authored-by: Apoorv Mishra <apoorvmishra101092@gmail.com>

shared/editor/commands/table.ts

@@ -0,0 +1,152 @@
+import { Fragment, Node, NodeType } from "prosemirror-model";
+import {
+  Command,
+  EditorState,
+  TextSelection,
+  Transaction,
+} from "prosemirror-state";
+import {
+  CellSelection,
+  addRow,
+  isInTable,
+  selectedRect,
+  tableNodeTypes,
+} from "prosemirror-tables";
+import { getCellsInColumn } from "../queries/table";
+
+export function createTable(
+  state: EditorState,
+  rowsCount: number,
+  colsCount: number,
+  withHeaderRow = true,
+  cellContent?: Node
+) {
+  const types = tableNodeTypes(state.schema);
+  const headerCells: Node[] = [];
+  const cells: Node[] = [];
+  const rows: Node[] = [];
+
+  const createCell = (
+    cellType: NodeType,
+    cellContent: Fragment | Node | readonly Node[] | null | undefined
+  ) =>
+    cellContent
+      ? cellType.createChecked(null, cellContent)
+      : cellType.createAndFill();
+
+  for (let index = 0; index < colsCount; index += 1) {
+    const cell = createCell(types.cell, cellContent);
+
+    if (cell) {
+      cells.push(cell);
+    }
+
+    if (withHeaderRow) {
+      const headerCell = createCell(types.header_cell, cellContent);
+
+      if (headerCell) {
+        headerCells.push(headerCell);
+      }
+    }
+  }
+
+  for (let index = 0; index < rowsCount; index += 1) {
+    rows.push(
+      types.row.createChecked(
+        null,
+        withHeaderRow && index === 0 ? headerCells : cells
+      )
+    );
+  }
+
+  return types.table.createChecked(null, rows);
+}
+
+export function addRowAfterAndMoveSelection({
+  index,
+}: {
+  index?: number;
+} = {}): Command {
+  return (state, dispatch) => {
+    if (!isInTable(state)) {
+      return false;
+    }
+
+    if (dispatch) {
+      const rect = selectedRect(state);
+      const indexAfter = index !== undefined ? index + 1 : rect.bottom;
+      const tr = addRow(state.tr, rect, indexAfter);
+      const cells = getCellsInColumn(0)(state);
+
+      // Special case when adding row to the end of the table as the calculated
+      // rect does not include the row that we just added.
+      if (indexAfter !== rect.map.height) {
+        const pos = cells[Math.min(cells.length - 1, indexAfter)];
+        const $pos = tr.doc.resolve(pos);
+        dispatch(tr.setSelection(TextSelection.near($pos)));
+      } else {
+        const $pos = tr.doc.resolve(rect.tableStart + rect.table.nodeSize);
+        dispatch(tr.setSelection(TextSelection.near($pos)));
+      }
+    }
+
+    return true;
+  };
+}
+
+export function setColumnAttr({
+  index,
+  alignment,
+}: {
+  index: number;
+  alignment: string;
+}): Command {
+  return (state, dispatch) => {
+    if (dispatch) {
+      const cells = getCellsInColumn(index)(state) || [];
+      let transaction = state.tr;
+      cells.forEach((pos) => {
+        transaction = transaction.setNodeMarkup(pos, undefined, {
+          alignment,
+        });
+      });
+      dispatch(transaction);
+    }
+    return true;
+  };
+}
+
+export function selectRow(index: number, expand = false) {
+  return (state: EditorState): Transaction => {
+    const rect = selectedRect(state);
+    const pos = rect.map.positionAt(index, 0, rect.table);
+    const $pos = state.doc.resolve(rect.tableStart + pos);
+    const rowSelection =
+      expand && state.selection instanceof CellSelection
+        ? CellSelection.rowSelection(state.selection.$anchorCell, $pos)
+        : CellSelection.rowSelection($pos);
+    return state.tr.setSelection(rowSelection);
+  };
+}
+
+export function selectColumn(index: number, expand = false) {
+  return (state: EditorState): Transaction => {
+    const rect = selectedRect(state);
+    const pos = rect.map.positionAt(0, index, rect.table);
+    const $pos = state.doc.resolve(rect.tableStart + pos);
+    const colSelection =
+      expand && state.selection instanceof CellSelection
+        ? CellSelection.colSelection(state.selection.$anchorCell, $pos)
+        : CellSelection.colSelection($pos);
+    return state.tr.setSelection(colSelection);
+  };
+}
+
+export function selectTable(state: EditorState): Transaction {
+  const rect = selectedRect(state);
+  const map = rect.map.map;
+  const $anchor = state.doc.resolve(rect.tableStart + map[0]);
+  const $head = state.doc.resolve(rect.tableStart + map[map.length - 1]);
+  const tableSelection = new CellSelection($anchor, $head);
+  return state.tr.setSelection(tableSelection);
+}