@soimy/dingtalk 3.2.0 → 3.4.0

package/src/message-utils.ts

@@ -1,4 +1,232 @@
-import type { DingTalkInboundMessage, MessageContent, SendMessageOptions } from "./types";
+import type { AtMention, DingTalkInboundMessage, MessageContent, QuotedInfo, SendMessageOptions } from "./types";
+
+
+interface DingTalkDocMeta {
+  spaceId: string;
+  fileId: string;
+}
+
+function parseBizCustomActionUrl(url: string | undefined): DingTalkDocMeta | null {
+  if (!url || typeof url !== "string") {
+    return null;
+  }
+
+  const queryIndex = url.indexOf("?");
+  if (queryIndex < 0 || queryIndex === url.length - 1) {
+    return null;
+  }
+
+  try {
+    const params = new URLSearchParams(url.slice(queryIndex + 1));
+    const route = params.get("route");
+    const type = params.get("type");
+    const spaceId = params.get("spaceId");
+    const fileId = params.get("fileId");
+    if (route !== "previewDentry" || type !== "file" || !spaceId || !fileId) {
+      return null;
+    }
+    return { spaceId, fileId };
+  } catch {
+    return null;
+  }
+}
+
+function extractRichTextQuoteParts(
+  richText: Array<Record<string, unknown>> | undefined,
+): { summary: string; pictureDownloadCode?: string; pictureDownloadCodes?: string[] } | null {
+  if (!Array.isArray(richText) || richText.length === 0) {
+    return null;
+  }
+
+  const textParts: string[] = [];
+  const pictureDownloadCodes: string[] = [];
+
+  for (const part of richText) {
+    const partType =
+      typeof part.msgType === "string"
+        ? part.msgType
+        : typeof part.type === "string"
+          ? part.type
+          : undefined;
+    const textValue =
+      typeof part.content === "string"
+        ? part.content
+        : typeof part.text === "string"
+          ? part.text
+          : undefined;
+
+    if ((partType === "text" || partType === undefined) && textValue) {
+      textParts.push(textValue);
+      continue;
+    }
+    if (partType === "emoji" && textValue) {
+      textParts.push(textValue);
+      continue;
+    }
+    if (partType === "picture") {
+      textParts.push("[图片]");
+      if (typeof part.downloadCode === "string" && part.downloadCode.trim()) {
+        pictureDownloadCodes.push(part.downloadCode.trim());
+      }
+      continue;
+    }
+    if (partType === "at") {
+      const atName =
+        typeof part.atName === "string"
+          ? part.atName
+          : typeof textValue === "string"
+            ? textValue
+            : "某人";
+      textParts.push(`@${atName}`);
+      continue;
+    }
+    if (textValue) {
+      textParts.push(textValue);
+    }
+  }
+
+  const summary = textParts.join("").trim();
+  const uniquePictureDownloadCodes = [...new Set(pictureDownloadCodes)];
+  const pictureDownloadCode = uniquePictureDownloadCodes[0];
+  if (!summary && !pictureDownloadCode) {
+    return null;
+  }
+  return {
+    summary,
+    pictureDownloadCode,
+    pictureDownloadCodes: uniquePictureDownloadCodes.length > 0 ? uniquePictureDownloadCodes : undefined,
+  };
+}
+
+function trimString(value: string | undefined): string | undefined {
+  if (typeof value !== "string") {
+    return undefined;
+  }
+  const trimmed = value.trim();
+  return trimmed.length > 0 ? trimmed : undefined;
+}
+
+function buildQuotedMessageTypePlaceholder(messageType: string | undefined, fileName?: string): string | undefined {
+  switch (messageType) {
+    case "text":
+      return undefined;
+    case "picture":
+      return "<media:image>";
+    case "audio":
+      return "<media:voice>";
+    case "video":
+      return "<media:video>";
+    case "file":
+    case "unknownMsgType":
+      return fileName ? `<media:file> (${fileName})` : "<media:file>";
+    case "interactiveCardFile":
+      return "[钉钉文档]";
+    case "interactiveCard":
+      return "[interactiveCard消息]";
+    case "richText":
+      return "[富文本消息]";
+    default:
+      return messageType ? `[Quoted ${messageType}]` : undefined;
+  }
+}
+
+function buildLegacyQuoteMessagePreview(message: DingTalkInboundMessage["quoteMessage"]): {
+  previewText?: string;
+  previewMessageType?: string;
+  previewSenderId?: string;
+} {
+  const previewMessageType = trimString(message?.msgtype);
+  return {
+    previewText:
+      trimString(message?.text?.content) ||
+      buildQuotedMessageTypePlaceholder(previewMessageType),
+    previewMessageType,
+    previewSenderId: trimString(message?.senderId),
+  };
+}
+
+function buildRepliedMessagePreview(params: {
+  data: DingTalkInboundMessage;
+  repliedMsg: NonNullable<NonNullable<DingTalkInboundMessage["text"]>["repliedMsg"]>;
+}): Partial<QuotedInfo> {
+  const { data, repliedMsg } = params;
+  const repliedMsgType = trimString(repliedMsg.msgType);
+  const content = repliedMsg.content;
+  const fileName = trimString(content?.fileName);
+  const richTextQuote = extractRichTextQuoteParts(content?.richText);
+  const docMeta = parseBizCustomActionUrl(content?.biz_custom_action_url);
+
+  if (repliedMsgType === "picture") {
+    return {
+      mediaDownloadCode: trimString(content?.downloadCode),
+      mediaType: trimString(content?.downloadCode) ? "image" : undefined,
+      previewText: buildQuotedMessageTypePlaceholder("picture"),
+      previewMessageType: "picture",
+      previewSenderId: trimString(repliedMsg.senderId),
+    };
+  }
+
+  if (repliedMsgType === "richText") {
+    return {
+      mediaDownloadCode: richTextQuote?.pictureDownloadCode,
+      mediaType: richTextQuote?.pictureDownloadCode ? "image" : undefined,
+      previewText:
+        trimString(richTextQuote?.summary) || buildQuotedMessageTypePlaceholder("richText"),
+      previewMessageType: "richText",
+      previewSenderId: trimString(repliedMsg.senderId),
+    };
+  }
+
+  if (repliedMsgType === "unknownMsgType") {
+    return {
+      isQuotedFile: true,
+      fileCreatedAt: repliedMsg.createdAt,
+      previewText: buildQuotedMessageTypePlaceholder("unknownMsgType", fileName),
+      previewMessageType: "unknownMsgType",
+      previewFileName: fileName,
+      previewSenderId: trimString(repliedMsg.senderId),
+    };
+  }
+
+  if (repliedMsgType === "interactiveCard") {
+    const isBotCard = repliedMsg.senderId === data.chatbotUserId;
+    if (isBotCard) {
+      return {
+        isQuotedCard: true,
+        cardCreatedAt: repliedMsg.createdAt,
+        processQueryKey: trimString(data.originalProcessQueryKey),
+        previewText:
+          trimString(content?.text) || buildQuotedMessageTypePlaceholder("interactiveCard"),
+        previewMessageType: "interactiveCard",
+        previewSenderId: trimString(repliedMsg.senderId),
+      };
+    }
+
+    return {
+      isQuotedDocCard: true,
+      fileCreatedAt: repliedMsg.createdAt,
+      previewText:
+        trimString(content?.text) ||
+        buildQuotedMessageTypePlaceholder(docMeta ? "interactiveCardFile" : "interactiveCard"),
+      previewMessageType: docMeta ? "interactiveCardFile" : "interactiveCard",
+      previewSenderId: trimString(repliedMsg.senderId),
+    };
+  }
+
+  const textPreview =
+    trimString(content?.text) ||
+    trimString(richTextQuote?.summary) ||
+    buildQuotedMessageTypePlaceholder(repliedMsgType, fileName);
+
+  return {
+    mediaDownloadCode: richTextQuote?.pictureDownloadCode,
+    mediaType: richTextQuote?.pictureDownloadCode ? "image" : undefined,
+    previewText: textPreview,
+    previewMessageType: repliedMsgType,
+    previewFileName: fileName,
+    previewSenderId: trimString(repliedMsg.senderId),
+  };
+}
 
 /**
  * Auto-detect markdown usage and derive message title.
@@ -24,76 +252,209 @@ export function detectMarkdownAndExtractTitle(
   return { useMarkdown, title };
 }
 
+function isMarkdownTableSeparator(line: string): boolean {
+  const normalized = line.trim();
+  if (!normalized.includes("-")) {
+    return false;
+  }
+  const cells = normalized
+    .replace(/^\|/, "")
+    .replace(/\|$/, "")
+    .split("|")
+    .map((cell) => cell.trim());
+  return cells.length > 0 && cells.every((cell) => /^:?-{3,}:?$/.test(cell));
+}
+
+function isMarkdownTableRow(line: string): boolean {
+  const trimmed = line.trim();
+  return trimmed.includes("|") && !trimmed.startsWith("```");
+}
+
+function parseMarkdownTableRow(line: string): string[] {
+  return line
+    .trim()
+    .replace(/^\|/, "")
+    .replace(/\|$/, "")
+    .split("|")
+    .map((cell) => cell.trim());
+}
+
+function renderMarkdownTable(lines: string[]): string {
+  const rows = lines.map(parseMarkdownTableRow).filter((cells) => cells.length > 0);
+  return rows.map((cells) => cells.join(" | ")).join("  \n");
+}
+
+export function convertMarkdownTablesToPlainText(text: string): string {
+  const lines = text.split("\n");
+  const output: string[] = [];
+  let index = 0;
+  let inCodeFence = false;
+
+  while (index < lines.length) {
+    const line = lines[index] || "";
+    if (line.trim().startsWith("```")) {
+      inCodeFence = !inCodeFence;
+      output.push(line);
+      index += 1;
+      continue;
+    }
+
+    if (
+      !inCodeFence &&
+      index + 1 < lines.length &&
+      isMarkdownTableRow(line) &&
+      isMarkdownTableSeparator(lines[index + 1] || "")
+    ) {
+      const tableLines = [line];
+      index += 2;
+      while (index < lines.length && isMarkdownTableRow(lines[index] || "")) {
+        tableLines.push(lines[index] || "");
+        index += 1;
+      }
+      output.push(renderMarkdownTable(tableLines));
+      continue;
+    }
+
+    output.push(line);
+    index += 1;
+  }
+
+  return output.join("\n");
+}
+
 export function extractMessageContent(data: DingTalkInboundMessage): MessageContent {
   const msgtype = data.msgtype || "text";
+  const atMentions: AtMention[] = [];
 
-  // Normalize quote/reply metadata into a readable text prefix so the agent can understand message context.
-  const formatQuotedContent = (): string => {
-    const textField = data.text as any;
+  // 提取通过 @picker 选中的真实用户的 dingtalkId
+  // 这些是真实钉钉用户，不包含 agent 名（如 @frontend）
+  const atUserDingtalkIds = data.atUsers?.map((u) => u.dingtalkId).filter(Boolean);
+
+  const formatQuotedContent = (): QuotedInfo | null => {
+    const textField = data.text;
 
     if (textField?.isReplyMsg && textField?.repliedMsg) {
       const repliedMsg = textField.repliedMsg;
-      const content = repliedMsg?.content;
+      const repliedMsgType = trimString(repliedMsg.msgType);
+      const content = repliedMsg.content;
+      const repliedMsgId = trimString(repliedMsg.msgId) || trimString(data.originalMsgId);
+      const repliedPreview = buildRepliedMessagePreview({ data, repliedMsg });
 
-      if (content?.text) {
-        const quoteText = content.text.trim();
-        if (quoteText) {
-          return `[引用消息: "${quoteText}"]\n\n`;
+      if (repliedMsgType === "picture" && content?.downloadCode) {
+        return {
+          mediaDownloadCode: content.downloadCode,
+          mediaType: "image",
+          ...repliedPreview,
+        };
+      }
+
+      if (repliedMsgType === "richText") {
+        const richTextQuote = extractRichTextQuoteParts(content?.richText);
+        if (richTextQuote) {
+          return {
+            msgId: repliedMsgId,
+            mediaDownloadCode: richTextQuote.pictureDownloadCode,
+            mediaType: richTextQuote.pictureDownloadCode ? "image" : undefined,
+            ...repliedPreview,
+          };
         }
       }
 
-      if (content?.richText && Array.isArray(content.richText)) {
-        const textParts: string[] = [];
-        for (const part of content.richText) {
-          if (part.msgType === "text" && part.content) {
-            textParts.push(part.content);
-          } else if (part.msgType === "emoji" || part.type === "emoji") {
-            textParts.push(part.content || "[表情]");
-          } else if (part.msgType === "picture" || part.type === "picture") {
-            textParts.push("[图片]");
-          } else if (part.msgType === "at" || part.type === "at") {
-            textParts.push(`@${part.content || part.atName || "某人"}`);
-          } else if (part.text) {
-            textParts.push(part.text);
-          }
+      if (repliedMsgType === "unknownMsgType") {
+        return {
+          isQuotedFile: true,
+          fileCreatedAt: repliedMsg.createdAt,
+          msgId: repliedMsgId,
+          ...repliedPreview,
+        };
+      }
+
+      if (repliedMsgType === "interactiveCard") {
+        const isBotCard = repliedMsg.senderId === data.chatbotUserId;
+        if (isBotCard) {
+          return {
+            isQuotedCard: true,
+            cardCreatedAt: repliedMsg.createdAt,
+            processQueryKey: data.originalProcessQueryKey,
+            msgId: repliedMsgId,
+            ...repliedPreview,
+          };
         }
-        const quoteText = textParts.join("").trim();
-        if (quoteText) {
-          return `[引用消息: "${quoteText}"]\n\n`;
+
+        return {
+          isQuotedDocCard: true,
+          fileCreatedAt: repliedMsg.createdAt,
+          msgId: repliedMsgId,
+          ...repliedPreview,
+        };
+      }
+
+      if (repliedMsgType && repliedMsgId) {
+        return { msgId: repliedMsgId, ...repliedPreview };
+      }
+
+      // No msgType — backward compat: extract text or richText from content.
+      if (content?.text?.trim()) {
+        return repliedMsgId ? { msgId: repliedMsgId, ...repliedPreview } : null;
+      }
+
+      if (content?.richText && Array.isArray(content.richText)) {
+        const richTextQuote = extractRichTextQuoteParts(content.richText);
+        if (richTextQuote) {
+          return {
+            msgId: repliedMsgId,
+            mediaDownloadCode: richTextQuote.pictureDownloadCode,
+            mediaType: richTextQuote.pictureDownloadCode ? "image" : undefined,
+            ...repliedPreview,
+          };
         }
       }
     }
 
-    // Some clients only send originalMsgId for rich media reply messages.
     if (textField?.isReplyMsg && !textField?.repliedMsg && data.originalMsgId) {
-      return `[这是一条引用消息，原消息ID: ${data.originalMsgId}]\n\n`;
+      return {
+        msgId: data.originalMsgId,
+      };
     }
 
     if (data.quoteMessage) {
-      const quoteText = data.quoteMessage.text?.content?.trim() || "";
-      if (quoteText) {
-        return `[引用消息: "${quoteText}"]\n\n`;
+      if (data.quoteMessage.msgId) {
+        return {
+          msgId: data.quoteMessage.msgId,
+          ...buildLegacyQuoteMessagePreview(data.quoteMessage),
+        };
       }
     }
 
-    if (data.content?.quoteContent) {
-      return `[引用消息: "${data.content.quoteContent}"]\n\n`;
-    }
-
-    return "";
+    return null;
   };
 
-  const quotedPrefix = formatQuotedContent();
+  const quoted = formatQuotedContent();
 
-  // Unified extraction by DingTalk msgtype for downstream routing/agent processing.
   if (msgtype === "text") {
-    return { text: quotedPrefix + (data.text?.content?.trim() || ""), messageType: "text" };
+    const textContent = data.text?.content?.trim() || "";
+
+    // Strip quoted prefix before extracting @mentions to avoid matching @names inside quotes.
+    const textForAtExtraction = textContent.replace(/^\[引用[^\]]*\]\s*/, "");
+    // Match @name but exclude email-like patterns (user@domain.com) and emoji (@_@).
+    const atMatches = textForAtExtraction.matchAll(/(?<!\w)@([^\s@.]+)(?!\.\w)/g);
+    for (const match of atMatches) {
+      atMentions.push({ name: match[1].trim() });
+    }
+
+    return {
+      text: textContent,
+      messageType: "text",
+      quoted: quoted ?? undefined,
+      atMentions,
+      atUserDingtalkIds,
+    };
   }
 
   if (msgtype === "richText") {
     const richTextParts = data.content?.richText || [];
     let text = "";
-    let pictureDownloadCode: string | undefined;
+    const pictureDownloadCodes: string[] = [];
     // Keep first image downloadCode while preserving readable text and @mention parts.
     for (const part of richTextParts) {
       if (part.text && (part.type === "text" || part.type === undefined)) {
@@ -101,17 +462,28 @@ export function extractMessageContent(data: DingTalkInboundMessage): MessageCont
       }
       if (part.type === "at" && part.atName) {
         text += `@${part.atName} `;
+        // 提取 @ 提及信息，包括 atUserId
+        atMentions.push({
+          name: part.atName.trim(),
+          userId: part.atUserId,
+        });
       }
-      if (part.type === "picture" && part.downloadCode && !pictureDownloadCode) {
-        pictureDownloadCode = part.downloadCode;
+      if (part.type === "picture" && part.downloadCode) {
+        pictureDownloadCodes.push(part.downloadCode);
       }
     }
+    const uniquePictureDownloadCodes = [...new Set(pictureDownloadCodes)];
+    const pictureDownloadCode = uniquePictureDownloadCodes[0];
     return {
-      text:
-        quotedPrefix + (text.trim() || (pictureDownloadCode ? "<media:image>" : "[富文本消息]")),
+      text: text.trim() || (pictureDownloadCode ? "<media:image>" : "[富文本消息]"),
       mediaPath: pictureDownloadCode,
+      mediaPaths: uniquePictureDownloadCodes.length > 0 ? uniquePictureDownloadCodes : undefined,
       mediaType: pictureDownloadCode ? "image" : undefined,
+      mediaTypes: uniquePictureDownloadCodes.length > 0 ? uniquePictureDownloadCodes.map(() => "image") : undefined,
       messageType: "richText",
+      quoted: quoted ?? undefined,
+      atMentions,
+      atUserDingtalkIds,
     };
   }
 
@@ -121,6 +493,8 @@ export function extractMessageContent(data: DingTalkInboundMessage): MessageCont
       mediaPath: data.content?.downloadCode,
       mediaType: "image",
       messageType: "picture",
+      atMentions,
+      atUserDingtalkIds,
     };
   }
 
@@ -130,6 +504,8 @@ export function extractMessageContent(data: DingTalkInboundMessage): MessageCont
       mediaPath: data.content?.downloadCode,
       mediaType: "audio",
       messageType: "audio",
+      atMentions,
+      atUserDingtalkIds,
     };
   }
 
@@ -139,6 +515,8 @@ export function extractMessageContent(data: DingTalkInboundMessage): MessageCont
       mediaPath: data.content?.downloadCode,
       mediaType: "video",
       messageType: "video",
+      atMentions,
+      atUserDingtalkIds,
     };
   }
 
@@ -148,9 +526,72 @@ export function extractMessageContent(data: DingTalkInboundMessage): MessageCont
       mediaPath: data.content?.downloadCode,
       mediaType: "file",
       messageType: "file",
+      atMentions,
+      atUserDingtalkIds,
     };
   }
 
-  // Fallback: preserve unknown msgtype as readable marker.
-  return { text: data.text?.content?.trim() || `[${msgtype}消息]`, messageType: msgtype };
+  if (msgtype === "interactiveCard") {
+    const docMeta = parseBizCustomActionUrl(data.content?.biz_custom_action_url);
+    if (docMeta) {
+      return {
+        text: "[钉钉文档]\n\n",
+        messageType: "interactiveCardFile",
+        docSpaceId: docMeta.spaceId,
+        docFileId: docMeta.fileId,
+        quoted: quoted ?? undefined,
+        atMentions,
+        atUserDingtalkIds,
+      };
+    }
+    return {
+      text: data.text?.content?.trim() || "[interactiveCard消息]",
+      messageType: msgtype,
+      quoted: quoted ?? undefined,
+      atMentions,
+      atUserDingtalkIds,
+    };
+  }
+  if (msgtype === "chatRecord") {
+    const content = data.content as Record<string, unknown> | undefined;
+    const summary = typeof content?.summary === "string" ? content.summary.trim() : "";
+    const rawRecord = content?.chatRecord;
+    if (
+      summary === "[]" ||
+      (typeof rawRecord === "string" && rawRecord.trim() === "[]") ||
+      (Array.isArray(rawRecord) && rawRecord.length === 0)
+    ) {
+      return {
+        text: "[系统提示] 没有读到引用记录（chatRecord 为空）。请改用逐条转发、复制原文，或重新转发非空聊天记录。",
+        messageType: "chatRecord",
+        quoted: quoted ?? undefined,
+        atMentions,
+        atUserDingtalkIds,
+      };
+    }
+    if (summary) {
+      return {
+        text: `[聊天记录摘要] ${summary}`,
+        messageType: "chatRecord",
+        quoted: quoted ?? undefined,
+        atMentions,
+        atUserDingtalkIds,
+      };
+    }
+    return {
+      text: "[chatRecord消息: 无可读内容]",
+      messageType: "chatRecord",
+      quoted: quoted ?? undefined,
+      atMentions,
+      atUserDingtalkIds,
+    };
+  }
+
+  return {
+    text: data.text?.content?.trim() || `[${msgtype}消息]`,
+    messageType: msgtype,
+    quoted: quoted ?? undefined,
+    atMentions,
+    atUserDingtalkIds,
+  };
 }