feat: add setCellSelection command

ueberdosis · Apr 27, 2021 · eb7e92f · eb7e92f
1 parent 811bf69
commit eb7e92f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/packages/extension-table/src/table.ts b/packages/extension-table/src/table.ts
@@ -24,6 +24,7 @@ import {
   toggleHeaderCell,
   setCellAttr,
   fixTables,
+  CellSelection,
 } from 'prosemirror-tables'
 import { NodeView } from 'prosemirror-view'
 import { TextSelection } from 'prosemirror-state'
@@ -62,6 +63,7 @@ declare module '@tiptap/core' {
       goToNextCell: () => Command,
       goToPreviousCell: () => Command,
       fixTables: () => Command,
+      setCellSelection: (position: { anchorCell: number, headCell?: number }) => Command,
     }
   }
 
@@ -182,6 +184,16 @@ export const Table = Node.create<TableOptions>({
           fixTables(state)
         }
 
+        return true
+      },
+      setCellSelection: position => ({ tr, dispatch }) => {
+        if (dispatch) {
+          const selection = CellSelection.create(tr.doc, position.anchorCell, position.headCell)
+
+          // @ts-ignore
+          tr.setSelection(selection)
+        }
+
         return true
       },
     }