refactor: process mention users after transform is complete

Author: whale4113

apps/chrome-extension/src/common/utils.ts

@@ -1,6 +1,7 @@
 import { toHast } from 'mdast-util-to-hast'
 import { toHtml } from 'hast-util-to-html'
-import type { InvalidTable } from '@dolphin/lark'
+import { Docx, type InvalidTable, type mdast } from '@dolphin/lark'
+import { Second, waitForFunction } from '@dolphin/common'
 
 interface Ref<T> {
   current: T
@@ -109,3 +110,35 @@ export const transformInvalidTablesToHtml = (
     }
   })
 }
+
+export const transformMentionUsers = async (
+  mentionUsers: mdast.InlineCode[],
+): Promise<void> => {
+  for (const user of mentionUsers) {
+    if (user.data?.parentBlockRecordId && user.data.mentionUserId) {
+      await waitForFunction(
+        () =>
+          Docx.locateBlockWithRecordId(
+            user.data?.parentBlockRecordId ?? '',
+          ).then(
+            isSuccess =>
+              isSuccess &&
+              document.querySelector(
+                `a[data-token="${user.data?.mentionUserId ?? ''}"]`,
+              ) !== null,
+          ),
+        {
+          timeout: 3 * Second,
+        },
+      )
+
+      const el: HTMLElement | null = document.querySelector(
+        `a[data-token="${user.data.mentionUserId}"]`,
+      )
+
+      if (el?.innerText) {
+        user.value = '@' + el.innerText
+      }
+    }
+  }
+}

apps/chrome-extension/src/scripts/copy-lark-docx-as-markdown.ts

@@ -5,7 +5,10 @@ import { isDefined } from '@dolphin/common'
 import { CommonTranslationKey, en, Namespace, zh } from '../common/i18n'
 import { confirm } from '../common/notification'
 import { reportBug } from '../common/issue'
-import { transformInvalidTablesToHtml } from '../common/utils'
+import {
+  transformInvalidTablesToHtml,
+  transformMentionUsers,
+} from '../common/utils'
 import {
   getSettings,
   SettingKey,
@@ -69,10 +72,12 @@ const main = async () => {
     SettingKey.TextHighlight,
   ])
 
-  const { root, images, invalidTables } = docx.intoMarkdownAST({
+  const { root, images, invalidTables, mentionUsers } = docx.intoMarkdownAST({
     highlight: settings[SettingKey.TextHighlight],
   })
 
+  await transformMentionUsers(mentionUsers)
+
   const tokens = images
     .map(image => {
       if (!image.data?.token) return null

apps/chrome-extension/src/scripts/download-lark-docx-as-markdown.ts

@@ -11,6 +11,7 @@ import { legacyFileSave } from '../common/legacy'
 import { reportBug } from '../common/issue'
 import {
   transformInvalidTablesToHtml,
+  transformMentionUsers,
   UniqueFileName,
   withSignal,
 } from '../common/utils'
@@ -501,12 +502,15 @@ const main = async (options: { signal?: AbortSignal } = {}) => {
     SettingKey.TextHighlight,
   ])
 
-  const { root, images, files, invalidTables } = docx.intoMarkdownAST({
-    whiteboard: true,
-    diagram: true,
-    file: true,
-    highlight: settings[SettingKey.TextHighlight],
-  })
+  const { root, images, files, invalidTables, mentionUsers } =
+    docx.intoMarkdownAST({
+      whiteboard: true,
+      diagram: true,
+      file: true,
+      highlight: settings[SettingKey.TextHighlight],
+    })
+
+  await transformMentionUsers(mentionUsers)
 
   const recommendName = docx.pageTitle
     ? normalizeFileName(docx.pageTitle.slice(0, OneHundred))

apps/chrome-extension/src/scripts/view-lark-docx-as-markdown.ts

@@ -5,7 +5,10 @@ import { isDefined } from '@dolphin/common'
 import { CommonTranslationKey, en, Namespace, zh } from '../common/i18n'
 import { confirm } from '../common/notification'
 import { reportBug } from '../common/issue'
-import { transformInvalidTablesToHtml } from '../common/utils'
+import {
+  transformInvalidTablesToHtml,
+  transformMentionUsers,
+} from '../common/utils'
 import {
   getSettings,
   SettingKey,
@@ -72,10 +75,12 @@ const main = async () => {
     SettingKey.TextHighlight,
   ])
 
-  const { root, images, invalidTables } = docx.intoMarkdownAST({
+  const { root, images, invalidTables, mentionUsers } = docx.intoMarkdownAST({
     highlight: settings[SettingKey.TextHighlight],
   })
 
+  await transformMentionUsers(mentionUsers)
+
   const tokens = images
     .map(image => {
       if (!image.data?.token) return null