From 811bf693eb5b927ec5f8120b56b2ae92f5ba2734 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Tue, 27 Apr 2021 11:41:24 +0200
Subject: [PATCH] feat: add setNodeSelection and setTextSelection commands

---
 docs/src/docPages/api/commands.md             |  2 ++
 .../api/commands/set-node-selection.md        |  3 +++
 .../api/commands/set-text-selection.md        |  3 +++
 docs/src/links.yaml                           |  4 ++++
 .../core/src/commands/setNodeSelection.ts     | 23 +++++++++++++++++++
 .../core/src/commands/setTextSelection.ts     | 23 +++++++++++++++++++
 packages/core/src/extensions/commands.ts      |  6 +++++
 7 files changed, 64 insertions(+)
 create mode 100644 docs/src/docPages/api/commands/set-node-selection.md
 create mode 100644 docs/src/docPages/api/commands/set-text-selection.md
 create mode 100644 packages/core/src/commands/setNodeSelection.ts
 create mode 100644 packages/core/src/commands/setTextSelection.ts

diff --git a/docs/src/docPages/api/commands.md b/docs/src/docPages/api/commands.md
index a9355d0e0..b0c61446b 100644
--- a/docs/src/docPages/api/commands.md
+++ b/docs/src/docPages/api/commands.md
@@ -209,6 +209,8 @@ Have a look at all of the core commands listed below. They should give you a goo
 | .selectNodeBackward() | Select a node backward.                 | [More](/api/commands/select-node-backward)  |
 | .selectNodeForward()  | Select a node forward.                  | [More](/api/commands/select-node-forward)  |
 | .selectParentNode()   | Select the parent node.                 | [More](/api/commands/select-parent-node)  |
+| .setNodeSelection()   | Creates a NodeSelection.                | [More](/api/commands/set-node-selection)  |
+| .setTextSelection()   | Creates a TextSelection.                | [More](/api/commands/set-text-selection)  |
 
 <!-- ## Example use cases
 
diff --git a/docs/src/docPages/api/commands/set-node-selection.md b/docs/src/docPages/api/commands/set-node-selection.md
new file mode 100644
index 000000000..f430004d0
--- /dev/null
+++ b/docs/src/docPages/api/commands/set-node-selection.md
@@ -0,0 +1,3 @@
+# setNodeSelection
+
+TODO
diff --git a/docs/src/docPages/api/commands/set-text-selection.md b/docs/src/docPages/api/commands/set-text-selection.md
new file mode 100644
index 000000000..216625455
--- /dev/null
+++ b/docs/src/docPages/api/commands/set-text-selection.md
@@ -0,0 +1,3 @@
+# setTextSelection
+
+TODO
diff --git a/docs/src/links.yaml b/docs/src/links.yaml
index 03b31ead6..8e050fa8e 100644
--- a/docs/src/links.yaml
+++ b/docs/src/links.yaml
@@ -184,6 +184,10 @@
           link: /api/commands/set-mark
         - title: setNode
           link: /api/commands/set-node
+        - title: setNodeSelection
+          link: /api/commands/set-node-selection
+        - title: setTextSelection
+          link: /api/commands/set-text-selection
         - title: sinkListItem
           link: /api/commands/sink-list-item
         - title: splitBlock
diff --git a/packages/core/src/commands/setNodeSelection.ts b/packages/core/src/commands/setNodeSelection.ts
new file mode 100644
index 000000000..6427d27ba
--- /dev/null
+++ b/packages/core/src/commands/setNodeSelection.ts
@@ -0,0 +1,23 @@
+import { NodeSelection } from 'prosemirror-state'
+import { Command, RawCommands } from '../types'
+
+declare module '@tiptap/core' {
+  interface Commands {
+    setNodeSelection: {
+      /**
+       * Creates a NodeSelection.
+       */
+      setNodeSelection: (position: number) => Command,
+    }
+  }
+}
+
+export const setNodeSelection: RawCommands['setNodeSelection'] = position => ({ tr, dispatch }) => {
+  if (dispatch) {
+    const selection = NodeSelection.create(tr.doc, position)
+
+    tr.setSelection(selection)
+  }
+
+  return true
+}
diff --git a/packages/core/src/commands/setTextSelection.ts b/packages/core/src/commands/setTextSelection.ts
new file mode 100644
index 000000000..e25cfd981
--- /dev/null
+++ b/packages/core/src/commands/setTextSelection.ts
@@ -0,0 +1,23 @@
+import { TextSelection } from 'prosemirror-state'
+import { Command, RawCommands, Range } from '../types'
+
+declare module '@tiptap/core' {
+  interface Commands {
+    setTextSelection: {
+      /**
+       * Creates a TextSelection.
+       */
+      setTextSelection: (range: Range) => Command,
+    }
+  }
+}
+
+export const setTextSelection: RawCommands['setTextSelection'] = range => ({ tr, dispatch }) => {
+  if (dispatch) {
+    const selection = TextSelection.create(tr.doc, range.from, range.to)
+
+    tr.setSelection(selection)
+  }
+
+  return true
+}
diff --git a/packages/core/src/extensions/commands.ts b/packages/core/src/extensions/commands.ts
index a028b51bc..be1c57dbc 100644
--- a/packages/core/src/extensions/commands.ts
+++ b/packages/core/src/extensions/commands.ts
@@ -34,6 +34,8 @@ import * as selectParentNode from '../commands/selectParentNode'
 import * as setContent from '../commands/setContent'
 import * as setMark from '../commands/setMark'
 import * as setNode from '../commands/setNode'
+import * as setNodeSelection from '../commands/setNodeSelection'
+import * as setTextSelection from '../commands/setTextSelection'
 import * as sinkListItem from '../commands/sinkListItem'
 import * as splitBlock from '../commands/splitBlock'
 import * as splitListItem from '../commands/splitListItem'
@@ -84,6 +86,8 @@ export { selectParentNode }
 export { setContent }
 export { setMark }
 export { setNode }
+export { setNodeSelection }
+export { setTextSelection }
 export { sinkListItem }
 export { splitBlock }
 export { splitListItem }
@@ -139,6 +143,8 @@ export const Commands = Extension.create({
       ...setContent,
       ...setMark,
       ...setNode,
+      ...setNodeSelection,
+      ...setTextSelection,
       ...sinkListItem,
       ...splitBlock,
       ...splitListItem,