From e8232dd73799c8f48b921f3d034ea6afa572ce8e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Tue, 9 Feb 2021 10:06:13 +0100
Subject: [PATCH] add basic enter command

---
 docs/src/demos/Examples/Default/index.vue | 15 ++++++++++
 packages/core/src/Editor.ts               | 28 ++++++++++++++++++
 packages/core/src/commands/enter.ts       | 36 +++++++++++++++++++++++
 packages/core/src/extensions/commands.ts  |  2 ++
 4 files changed, 81 insertions(+)
 create mode 100644 packages/core/src/commands/enter.ts

diff --git a/docs/src/demos/Examples/Default/index.vue b/docs/src/demos/Examples/Default/index.vue
index c951a44a..19049ce4 100644
--- a/docs/src/demos/Examples/Default/index.vue
+++ b/docs/src/demos/Examples/Default/index.vue
@@ -1,6 +1,21 @@
 <template>
   <div>
     <div v-if="editor">
+      <button
+        @click="editor
+          .can()
+          .chain()
+          .focus()
+          .enter()
+          .insertText('1')
+          .enter()
+          .insertText('2')
+          .enter()
+          .insertText('3')
+          .run()"
+      >
+        enter
+      </button>
       <button @click="editor.chain().focus().toggleBold().run()" :class="{ 'is-active': editor.isActive('bold') }">
         bold
       </button>
diff --git a/packages/core/src/Editor.ts b/packages/core/src/Editor.ts
index a457e564..c4300950 100644
--- a/packages/core/src/Editor.ts
+++ b/packages/core/src/Editor.ts
@@ -307,12 +307,40 @@ export class Editor extends EventEmitter {
     return this.createDocument('')
   }
 
+  public isCapturingTransaction = false
+
+  private capturedTransaction: Transaction | null = null
+
+  public captureTransaction(fn: Function) {
+    this.isCapturingTransaction = true
+    fn()
+    this.isCapturingTransaction = false
+
+    const tr = this.capturedTransaction
+
+    this.capturedTransaction = null
+
+    return tr
+  }
+
   /**
    * The callback over which to send transactions (state updates) produced by the view.
    *
    * @param transaction An editor state transaction
    */
   private dispatchTransaction(transaction: Transaction): void {
+    if (this.isCapturingTransaction) {
+      if (!this.capturedTransaction) {
+        this.capturedTransaction = transaction
+
+        return
+      }
+
+      transaction.steps.forEach(step => this.capturedTransaction?.step(step))
+
+      return
+    }
+
     const state = this.state.apply(transaction)
     const selectionHasChanged = !this.state.selection.eq(state.selection)
 
diff --git a/packages/core/src/commands/enter.ts b/packages/core/src/commands/enter.ts
new file mode 100644
index 00000000..887f8a6c
--- /dev/null
+++ b/packages/core/src/commands/enter.ts
@@ -0,0 +1,36 @@
+import { Command } from '../types'
+
+/**
+ * Press Enter
+ */
+export const enter = (): Command => ({
+  editor,
+  view,
+  tr,
+  dispatch,
+}) => {
+  const keyCode = 13
+  const key = 'Enter'
+  const event = document.createEvent('Event')
+  event.initEvent('keydown', true, true)
+  // @ts-ignore
+  event.keyCode = keyCode
+  // @ts-ignore
+  event.key = key
+  // @ts-ignore
+  event.code = key
+
+  const capturedTransaction = editor.captureTransaction(() => {
+    view.someProp('handleKeyDown', f => f(view, event))
+  })
+
+  capturedTransaction?.steps.forEach(step => {
+    const newStep = step.map(tr.mapping)
+
+    if (newStep && dispatch) {
+      tr.step(newStep)
+    }
+  })
+
+  return true
+}
diff --git a/packages/core/src/extensions/commands.ts b/packages/core/src/extensions/commands.ts
index 2c267821..adf7bdba 100644
--- a/packages/core/src/extensions/commands.ts
+++ b/packages/core/src/extensions/commands.ts
@@ -6,6 +6,7 @@ import * as command from '../commands/command'
 import * as createParagraphNear from '../commands/createParagraphNear'
 import * as deleteRange from '../commands/deleteRange'
 import * as deleteSelection from '../commands/deleteSelection'
+import * as enter from '../commands/enter'
 import * as exitCode from '../commands/exitCode'
 import * as extendMarkRange from '../commands/extendMarkRange'
 import * as first from '../commands/first'
@@ -55,6 +56,7 @@ export const Commands = Extension.create({
       ...createParagraphNear,
       ...deleteRange,
       ...deleteSelection,
+      ...enter,
       ...exitCode,
       ...extendMarkRange,
       ...first,