From f9025775ca7ef6182885bdf4f4a992b82942ce9f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Tue, 17 Nov 2020 21:53:39 +0100
Subject: [PATCH] add code commands

---
 docs/src/demos/Examples/Basic/index.vue          |  2 +-
 .../Examples/CollaborativeEditing/index.vue      |  2 +-
 .../Examples/CollaborativeEditingWs/index.vue    |  2 +-
 docs/src/demos/Marks/Code/index.vue              |  2 +-
 packages/extension-code/src/index.ts             | 16 ++++++++++++++--
 5 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/docs/src/demos/Examples/Basic/index.vue b/docs/src/demos/Examples/Basic/index.vue
index 70b52b13..76450865 100644
--- a/docs/src/demos/Examples/Basic/index.vue
+++ b/docs/src/demos/Examples/Basic/index.vue
@@ -10,7 +10,7 @@
       <button @click="editor.chain().focus().toggleStrike().run()" :class="{ 'is-active': editor.isActive('strike') }">
         strike
       </button>
-      <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+      <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
         code
       </button>
       <button @click="editor.chain().focus().removeMarks().run()">
diff --git a/docs/src/demos/Examples/CollaborativeEditing/index.vue b/docs/src/demos/Examples/CollaborativeEditing/index.vue
index 17d156ab..e14f98e0 100644
--- a/docs/src/demos/Examples/CollaborativeEditing/index.vue
+++ b/docs/src/demos/Examples/CollaborativeEditing/index.vue
@@ -10,7 +10,7 @@
       <button @click="editor.chain().focus().toggleStrike().run()" :class="{ 'is-active': editor.isActive('strike') }">
         strike
       </button>
-      <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+      <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
         code
       </button>
       <button @click="editor.chain().focus().removeMarks().run()">
diff --git a/docs/src/demos/Examples/CollaborativeEditingWs/index.vue b/docs/src/demos/Examples/CollaborativeEditingWs/index.vue
index 38ee356f..2bc267ba 100644
--- a/docs/src/demos/Examples/CollaborativeEditingWs/index.vue
+++ b/docs/src/demos/Examples/CollaborativeEditingWs/index.vue
@@ -10,7 +10,7 @@
       <button @click="editor.chain().focus().toggleStrike().run()" :class="{ 'is-active': editor.isActive('strike') }">
         strike
       </button>
-      <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+      <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
         code
       </button>
       <button @click="editor.chain().focus().removeMarks().run()">
diff --git a/docs/src/demos/Marks/Code/index.vue b/docs/src/demos/Marks/Code/index.vue
index ac874df5..365f8208 100644
--- a/docs/src/demos/Marks/Code/index.vue
+++ b/docs/src/demos/Marks/Code/index.vue
@@ -1,6 +1,6 @@
 <template>
   <div v-if="editor">
-    <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+    <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
       code
     </button>
 
diff --git a/packages/extension-code/src/index.ts b/packages/extension-code/src/index.ts
index 5eb5909e..c68b9ff0 100644
--- a/packages/extension-code/src/index.ts
+++ b/packages/extension-code/src/index.ts
@@ -35,18 +35,30 @@ const Code = Mark.create({
 
   addCommands() {
     return {
+      /**
+       * Set a code mark
+       */
+      setCode: (): Command => ({ commands }) => {
+        return commands.addMark('code')
+      },
       /**
        * Toggle inline code
        */
-      code: (): Command => ({ commands }) => {
+      toggleCode: (): Command => ({ commands }) => {
         return commands.toggleMark('code')
       },
+      /**
+       * Set a code mark
+       */
+      unsetCode: (): Command => ({ commands }) => {
+        return commands.addMark('code')
+      },
     }
   },
 
   addKeyboardShortcuts() {
     return {
-      'Mod-`': () => this.editor.commands.code(),
+      'Mod-`': () => this.editor.commands.toggleCode(),
     }
   },