From 8195cbd91da4baf41ae1421ad577168a9fc90d43 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:52:16 +0100
Subject: [PATCH] add underline commands

---
 docs/src/demos/Marks/Underline/index.vue  |  2 +-
 packages/extension-underline/src/index.ts | 19 +++++++++++++++++--
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/docs/src/demos/Marks/Underline/index.vue b/docs/src/demos/Marks/Underline/index.vue
index a1511948..21f41c91 100644
--- a/docs/src/demos/Marks/Underline/index.vue
+++ b/docs/src/demos/Marks/Underline/index.vue
@@ -1,6 +1,6 @@
 <template>
   <div v-if="editor">
-    <button @click="editor.chain().focus().underline().run()" :class="{ 'is-active': editor.isActive('underline') }">
+    <button @click="editor.chain().focus().toggleUnderline().run()" :class="{ 'is-active': editor.isActive('underline') }">
       underline
     </button>
 
diff --git a/packages/extension-underline/src/index.ts b/packages/extension-underline/src/index.ts
index 0f9cb39c..9ddd6b45 100644
--- a/packages/extension-underline/src/index.ts
+++ b/packages/extension-underline/src/index.ts
@@ -30,15 +30,30 @@ const Underline = Mark.create({
 
   addCommands() {
     return {
-      underline: (): Command => ({ commands }) => {
+      /**
+       * Set a underline mark
+       */
+      setUnderline: (): Command => ({ commands }) => {
+        return commands.addMark('underline')
+      },
+      /**
+       * Toggle a underline mark
+       */
+      toggleUnderline: (): Command => ({ commands }) => {
         return commands.toggleMark('underline')
       },
+      /**
+       * Set a underline mark
+       */
+      unsetUnderline: (): Command => ({ commands }) => {
+        return commands.addMark('underline')
+      },
     }
   },
 
   addKeyboardShortcuts() {
     return {
-      'Mod-u': () => this.editor.commands.underline(),
+      'Mod-u': () => this.editor.commands.toggleUnderline(),
     }
   },
 })