feat: add label attribute to mention extension (#1322)

ueberdosis · Jun 14, 2021 · c3afe88 · c3afe88
1 parent a3a847f
commit c3afe88
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/packages/extension-mention/src/mention.ts b/packages/extension-mention/src/mention.ts
@@ -62,6 +62,22 @@ export const Mention = Node.create<MentionOptions>({
           }
         },
       },
+      label: {
+        default: null,
+        parseHTML: element => {
+          return {
+            label: element.getAttribute('data-label'),
+          };
+        },
+        renderHTML: attributes => {
+          if (!attributes.label) {
+            return {};
+          }
+          return {
+            'data-label': attributes.label,
+          };
+        },
+      },
     }
   },
 
@@ -77,12 +93,12 @@ export const Mention = Node.create<MentionOptions>({
     return [
       'span',
       mergeAttributes(this.options.HTMLAttributes, HTMLAttributes),
-      `${this.options.suggestion.char}${node.attrs.id}`,
+      `${this.options.suggestion.char}${node.attrs.label ?? node.attrs.id}`,
     ]
   },
 
   renderText({ node }) {
-    return `${this.options.suggestion.char}${node.attrs.id}`
+    return `${this.options.suggestion.char}${node.attrs.label ?? node.attrs.id}`
   },
 
   addKeyboardShortcuts() {