From d03177082d9cb71068970a3e6aa657e8be8c24e2 Mon Sep 17 00:00:00 2001
From: digua <i@digua.me>
Date: Fri, 9 Jan 2026 00:13:16 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=90=84=E4=B8=AA=E8=A7=A3=E6=9E=90?=
 =?UTF-8?q?=E5=99=A8=E9=87=8D=E6=96=B0=E6=94=AF=E6=8C=81=E5=9B=9E=E5=A4=8D?=
 =?UTF-8?q?=E6=B6=88=E6=81=AF=E5=AF=BC=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 electron/main/parser/formats/chatlab-jsonl.ts           | 2 ++
 electron/main/parser/formats/chatlab.ts                 | 2 ++
 electron/main/parser/formats/shuakami-qq-exporter.ts    | 9 +++++++++
 electron/main/parser/formats/tyrrrz-discord-exporter.ts | 1 +
 electron/main/parser/formats/ycccccccy-echotrace.ts     | 2 ++
 5 files changed, 16 insertions(+)

diff --git a/electron/main/parser/formats/chatlab-jsonl.ts b/electron/main/parser/formats/chatlab-jsonl.ts
index 4e7a912..04234b8 100644
--- a/electron/main/parser/formats/chatlab-jsonl.ts
+++ b/electron/main/parser/formats/chatlab-jsonl.ts
@@ -68,6 +68,7 @@ interface JsonlMember {
 interface JsonlMessage {
   _type: 'message'
   sender: string
+  platformMessageId?: string
   accountName: string
   groupNickname?: string
   timestamp: number
@@ -216,6 +217,7 @@ async function* parseChatLabJsonl(options: ParseOptions): AsyncGenerator<ParseEv
           timestamp: parsed.timestamp,
           type: parsed.type as MessageType,
           content: parsed.content,
+          platformMessageId: parsed.platformMessageId,
           replyToMessageId: parsed.replyToMessageId,
         })
         messagesProcessed++
diff --git a/electron/main/parser/formats/chatlab.ts b/electron/main/parser/formats/chatlab.ts
index e6ad1aa..1990479 100644
--- a/electron/main/parser/formats/chatlab.ts
+++ b/electron/main/parser/formats/chatlab.ts
@@ -59,6 +59,7 @@ export const feature: FormatFeature = {
 // ==================== 消息结构 ====================
 
 interface ChatLabMessage {
+  platformMessageId?: string // 消息的平台原始 ID（用于回复关联查询）
   sender: string // platformId
   accountName: string // 发送时的账号名称
   groupNickname?: string // 发送时的群昵称
@@ -205,6 +206,7 @@ async function* parseChatLab(options: ParseOptions): AsyncGenerator<ParseEvent,
         timestamp: msg.timestamp,
         type: msg.type,
         content: msg.content,
+        platformMessageId: msg.platformMessageId,
         replyToMessageId: msg.replyToMessageId,
       })
 
diff --git a/electron/main/parser/formats/shuakami-qq-exporter.ts b/electron/main/parser/formats/shuakami-qq-exporter.ts
index ade16b4..de7263f 100644
--- a/electron/main/parser/formats/shuakami-qq-exporter.ts
+++ b/electron/main/parser/formats/shuakami-qq-exporter.ts
@@ -60,6 +60,7 @@ interface V4RawMessage {
 }
 
 interface V4Message {
+  messageId?: string // 消息的唯一 ID
   timestamp: string
   sender: {
     uid?: string
@@ -73,6 +74,12 @@ interface V4Message {
     text: string
     resources?: Array<{ type: string }>
     emojis?: Array<{ type: string }>
+    reply?: {
+      messageId?: string
+      referencedMessageId?: string // 被引用消息的 ID
+      senderName?: string
+      content?: string
+    }
   }
   rawMessage?: V4RawMessage
 }
@@ -323,12 +330,14 @@ async function* parseV4(options: ParseOptions): AsyncGenerator<ParseEvent, void,
       if (value.isRecalled) textContent = '[已撤回] ' + textContent
 
       messageBatch.push({
+        platformMessageId: value.messageId, // 消息的平台原始 ID
         senderPlatformId: platformId,
         senderAccountName: accountName,
         senderGroupNickname: groupNickname,
         timestamp,
         type,
         content: textContent || null,
+        replyToMessageId: value.content?.reply?.referencedMessageId, // 被引用消息的 ID
       })
 
       messagesProcessed++
diff --git a/electron/main/parser/formats/tyrrrz-discord-exporter.ts b/electron/main/parser/formats/tyrrrz-discord-exporter.ts
index 33819d9..a0c537e 100644
--- a/electron/main/parser/formats/tyrrrz-discord-exporter.ts
+++ b/electron/main/parser/formats/tyrrrz-discord-exporter.ts
@@ -381,6 +381,7 @@ async function* parseDiscordExporter(options: ParseOptions): AsyncGenerator<Pars
       }
 
       batchCollector.push({
+        platformMessageId: msg.id, // 消息的平台原始 ID
         senderPlatformId: author.id,
         senderAccountName: author.name,
         senderGroupNickname: author.nickname || null,
diff --git a/electron/main/parser/formats/ycccccccy-echotrace.ts b/electron/main/parser/formats/ycccccccy-echotrace.ts
index a5a47ab..f81414f 100644
--- a/electron/main/parser/formats/ycccccccy-echotrace.ts
+++ b/electron/main/parser/formats/ycccccccy-echotrace.ts
@@ -451,6 +451,7 @@ async function* parseEchotrace(options: ParseOptions): AsyncGenerator<ParseEvent
       }
 
       return {
+        platformMessageId: String(msg.localId), // 消息的平台原始 ID（用于回复关联查询）
         senderPlatformId: platformId,
         senderAccountName: accountName,
         // echotrace 格式没有单独的群昵称字段，使用 null 而非 undefined（SQLite 兼容）
@@ -458,6 +459,7 @@ async function* parseEchotrace(options: ParseOptions): AsyncGenerator<ParseEvent
         timestamp: msg.createTime,
         type,
         content,
+        // 注意：echotrace 导出格式不包含被引用消息的 ID，所以 replyToMessageId 为空
       }
     }