From 06f6f6a222a6cc88879bd93dd8fb90dc1f8d6d5f Mon Sep 17 00:00:00 2001
From: Alexander Onnikov <Alexander.Onnikov@gmail.com>
Date: Thu, 11 Jan 2024 21:28:52 +0700
Subject: [PATCH] TSK-1673 Fix missing mentions (#4340)

Signed-off-by: Alexander Onnikov <alexander.onnikov@xored.com>
---
 packages/text/src/nodes/reference.ts | 42 ++++++++++++++++++++++------
 1 file changed, 33 insertions(+), 9 deletions(-)

diff --git a/packages/text/src/nodes/reference.ts b/packages/text/src/nodes/reference.ts
index d164650d97..71756022c5 100644
--- a/packages/text/src/nodes/reference.ts
+++ b/packages/text/src/nodes/reference.ts
@@ -49,26 +49,35 @@ export function extractReferences (content: ProseMirrorNode): Array<Reference> {
   return result
 }
 
+export interface ReferenceOptions {
+  renderLabel: (props: { options: ReferenceOptions, node: any }) => string
+  suggestion: { char: string }
+}
+
 /**
  * @public
  */
-export const ReferenceNode = Node.create({
+export const ReferenceNode = Node.create<ReferenceOptions>({
   name: 'reference',
   group: 'inline',
-  content: 'inline*',
   inline: true,
 
   addAttributes () {
     return {
       id: getDataAttribute('id'),
-
       objectclass: getDataAttribute('objectclass'),
-
       label: getDataAttribute('label'),
+      class: { default: null }
+    }
+  },
 
-      class: {
-        default: null
-      }
+  addOptions () {
+    return {
+      renderLabel ({ options, node }) {
+        // eslint-disable-next-line
+        return `${options.suggestion.char}${node.attrs.label ?? node.attrs.id}`
+      },
+      suggestion: { char: '@' }
     }
   },
 
@@ -80,7 +89,22 @@ export const ReferenceNode = Node.create({
     ]
   },
 
-  renderHTML ({ HTMLAttributes }) {
-    return ['span', mergeAttributes({ 'data-type': this.name }, HTMLAttributes), 0]
+  renderHTML ({ node, HTMLAttributes }) {
+    const options = this.options
+    return [
+      'span',
+      mergeAttributes(
+        {
+          'data-type': this.name
+        },
+        HTMLAttributes
+      ),
+      this.options.renderLabel({ options, node })
+    ]
+  },
+
+  renderText ({ node }) {
+    const options = this.options
+    return options.renderLabel({ options, node })
   }
 })