feat: 支持微信 4.0 数据库解密与富媒体消息查看

- 接入微信 4.0 WCDB 数据库解密，兼容微信 3.0 解密方式
- 支持图片解密及图片预览、缩放、旋转和拖动
- 支持语音解密与播放
- 支持表情包、引用、分享、名片、位置及通话消息解析
- 支持联系人和群聊真实头像
- 优化群聊、联系人分类及排序
- 修复复合消息类型和压缩消息内容解析
- 完善原生解密库打包及数据库错误提示

src/main/index.ts

@@ -3,20 +3,53 @@ import { join } from 'path'
 import { electronApp, optimizer, is } from '@electron-toolkit/utils'
 import icon from '../../resources/icon.png?asset'
 import { WechatDb, Contact, WechatMessage } from './wechat-db'
+import { VoiceService } from './voice-service'
+import { StickerService } from './sticker-service'
+import {
+  parseImageDatNameFromRow,
+  parseMessageContent,
+  parseStickerMessageFromRow
+} from './message-parser'
+import { ImageDecryptService } from './image-decrypt-service'
 
 let wechatDb: WechatDb | null = null
+let voiceService: VoiceService | null = null
+let imageDecryptService: ImageDecryptService | null = null
+let stickerService: StickerService | null = null
+const BUILD_MARK = 'wechat4-open-account-continues-after-init-1000'
+
+// WechatExplorer's WCDB native library runs InitProtection before wcdb_init.
+// In dev, matching the host app name avoids failing the native protection gate.
+app.setName('WechatExplorer')
 
 const MSG_TYPE_DICT: Record<number, string> = {
   1: '普通文本',
   3: '图片',
   34: '语音',
+  42: '名片',
   43: '视频',
   47: '表情包',
   48: '位置',
   49: '分享消息',
+  50: '通话',
   10000: '系统消息'
 }
 
+function normalizeMsgType(value: string | number | undefined): number {
+  const raw = String(value ?? '').trim()
+  if (!raw) return 0
+
+  try {
+    const parsed = BigInt(raw)
+    const low32 = Number(parsed & 0xffffffffn)
+    return low32 || Number(parsed)
+  } catch {
+    const parsed = Number(raw)
+    if (!Number.isFinite(parsed)) return 0
+    return parsed > 0xffffffff ? parsed >>> 0 : parsed
+  }
+}
+
 function createWindow(): void {
   // 创建浏览器窗口
   const mainWindow = new BrowserWindow({
@@ -52,6 +85,7 @@ function createWindow(): void {
 // 当 Electron 完成初始化并准备好创建浏览器窗口时，将调用此方法
 // 某些 API 只能在此事件发生后使用
 app.whenReady().then(() => {
+  console.log(`WechatExplorer main build: ${BUILD_MARK}`)
   // 为窗口设置应用程序用户模型 ID
   electronApp.setAppUserModelId('com.electron')
 
@@ -67,11 +101,21 @@ app.whenReady().then(() => {
 
   ipcMain.handle('db:init', (_, key: string) => {
     try {
+      const trimmedKey = String(key || '').trim()
+      console.log(
+        `db:init build=${BUILD_MARK} keyLength=${trimmedKey.length} keyPreview=${trimmedKey.slice(0, 6)}...${trimmedKey.slice(-6)}`
+      )
       wechatDb = new WechatDb(key)
-      return true
+      const wcdb4Client = wechatDb.getWcdb4Client()
+      if (wcdb4Client) {
+        voiceService = new VoiceService(wcdb4Client)
+        stickerService = new StickerService(wcdb4Client)
+      }
+      imageDecryptService = null
+      return { success: true }
     } catch (error) {
       console.error('Failed to init DB:', error)
-      return false
+      return { success: false, error: error instanceof Error ? error.message : String(error) }
     }
   })
 
@@ -86,12 +130,14 @@ app.whenReady().then(() => {
     // 1. 处理普通联系人
     for (const user of userList) {
       const md5 = wechatDb.md5(user.m_nsUsrName)
+      const isGroup = user.m_nsUsrName.endsWith('@chatroom')
       existingMd5s.add(md5)
       contacts.push({
         m_nsUsrName: user.m_nsUsrName,
         m_nsNickName: user.nickname || '未知用户',
         md5: md5,
-        type: 'user'
+        type: isGroup ? 'group' : 'user',
+        avatar: typeof user.avatar === 'string' ? user.avatar : undefined
       })
     }
 
@@ -124,17 +170,31 @@ app.whenReady().then(() => {
 
   ipcMain.handle('db:getMessages', (_, userMd5: string, startTime?: number, endTime?: number) => {
     if (!wechatDb) return []
+    const wcdb4Client = wechatDb.getWcdb4Client()
+    const username = wcdb4Client?.getUsernameByMd5(userMd5)
     const rawMessages = wechatDb.getUserMessages(userMd5, startTime, endTime)
-    const groupMembers = wechatDb.getAllGroupMembers()
+    const groupMembers = wechatDb.getGroupMembersForChat(userMd5)
+    const myAvatar = wechatDb.getMyAvatarUrl()
 
     return rawMessages.map((msg: WechatMessage) => {
-      const msgType = parseInt(msg.messageType)
+      const rawMsgType = parseInt(msg.messageType)
+      const msgType = normalizeMsgType(msg.messageType)
       const createTime = parseInt(msg.msgCreateTime)
       const date = new Date(createTime * 1000)
+      const isMine = msg.mesDes !== 1
+      const localId = parseInt(msg.mesLocalID) || 0
 
       let content = msg.msgContent
       let img = ''
       let name = ''
+      if (isMine && myAvatar) {
+        img = myAvatar
+      } else if (typeof msg.senderAvatar === 'string') {
+        img = msg.senderAvatar
+      }
+      if (typeof msg.senderNickname === 'string') {
+        name = msg.senderNickname
+      }
       // 检查内容是否以 wxid 开头并包含冒号
       // 示例: wxid_xxxx:\nContent 或 wxid_xxxx:Content
       if (content && typeof content === 'string') {
@@ -159,14 +219,72 @@ app.whenReady().then(() => {
         }
       }
 
+      // 解析富媒体消息内容
+      let contentData: ReturnType<typeof parseMessageContent> | undefined = undefined
+      let displayType = MSG_TYPE_DICT[msgType] || msg.messageType
+      const inferredMsgType =
+        typeof content === 'string' &&
+        /<appmsg\b|<refermsg\b|&lt;appmsg\b|&lt;refermsg\b/i.test(content)
+          ? 49
+          : msgType
+      if ([3, 42, 47, 48, 49, 50].includes(inferredMsgType)) {
+        try {
+          const parsed =
+            inferredMsgType === 47
+              ? parseStickerMessageFromRow(msg, content)
+              : parseMessageContent(content, inferredMsgType)
+          if (parsed.type !== 'unknown') {
+            content = ''
+          }
+          if (parsed.type === 'image') {
+            const imageDatName = parseImageDatNameFromRow(msg)
+            contentData = { ...parsed, datName: parsed.datName || imageDatName }
+          } else {
+            if (parsed.type === 'sticker' && !parsed.url && parsed.md5 && wcdb4Client) {
+              parsed.url = wcdb4Client.resolveEmoticonCdnUrl(parsed.md5)
+            }
+            contentData = parsed
+          }
+          if (inferredMsgType !== msgType || rawMsgType !== msgType) {
+            displayType = MSG_TYPE_DICT[inferredMsgType] || displayType
+          }
+        } catch {
+          // ignore parse errors
+        }
+      }
+
+      if (
+        !contentData &&
+        typeof content === 'string' &&
+        /^[0-9a-fA-F]{64,}$/.test(content.trim())
+      ) {
+        const parsed = parseStickerMessageFromRow(msg, content)
+        if (parsed.type === 'sticker') {
+          if (!parsed.url && parsed.md5 && wcdb4Client) {
+            parsed.url = wcdb4Client.resolveEmoticonCdnUrl(parsed.md5)
+          }
+          content = ''
+          contentData = parsed
+          displayType = '表情包'
+        }
+      }
+
+      if (msgType === 34) {
+        content = '[语音消息]'
+      }
+
       return {
         id: msg.mesLocalID || Math.random().toString(),
-        from: msg.mesDes === 1 ? 'user' : 'assistant', // 1 通常是接收到的，0 是发送的？需要验证。Swift 说：[1: "user", 0: "assistant"]
-        type: MSG_TYPE_DICT[msgType] || msg.messageType,
+        from: isMine ? 'assistant' : 'user',
+        type: displayType,
         datetime: date.toLocaleString('zh-CN', { hour12: false }),
         content: content,
         img: img,
-        name: name
+        name: name,
+        sessionId: username,
+        localId: localId,
+        createTime: createTime,
+        contentData: contentData
       }
     })
   })
@@ -225,6 +343,56 @@ app.whenReady().then(() => {
     }
   })
 
+  ipcMain.handle(
+    'db:getVoiceData',
+    async (_, sessionId: string, localId: number, createTime: number, svrId?: string | number) => {
+      if (!voiceService) {
+        return { success: false, error: 'VoiceService 未初始化' }
+      }
+      return voiceService.resolveVoice(sessionId, localId, createTime, svrId)
+    }
+  )
+
+  ipcMain.handle('db:parseMessage', async (_, content: string, messageType: number) => {
+    return parseMessageContent(content, messageType)
+  })
+
+  ipcMain.handle(
+    'db:getImage',
+    async (_, imageMd5?: string, imageDatNameOrThumb?: string | boolean, _sessionId?: string) => {
+      void _sessionId
+      if (!imageDecryptService) {
+        // 从环境变量获取密钥
+        const xorKey = import.meta.env.VITE_IMAGE_XOR_KEY || '0x40'
+        const aesKey = import.meta.env.VITE_IMAGE_AES_KEY || ''
+        if (!aesKey) {
+          return { success: false, error: '未配置图片解密密钥' }
+        }
+        imageDecryptService = new ImageDecryptService(xorKey, aesKey, wechatDb?.getWcdb4Client())
+      }
+
+      const imageDatName = typeof imageDatNameOrThumb === 'string' ? imageDatNameOrThumb : undefined
+      const filePath = imageDecryptService.findImageFile(imageMd5, imageDatName)
+      if (!filePath) {
+        return { success: false, error: '未找到图片文件' }
+      }
+
+      const base64 = imageDecryptService.decryptImageToBase64(filePath)
+      if (!base64) {
+        return { success: false, error: '图片解密失败' }
+      }
+
+      return { success: true, data: base64 }
+    }
+  )
+
+  ipcMain.handle('db:getSticker', async (_, cdnUrl?: string, md5?: string) => {
+    if (!stickerService) {
+      stickerService = new StickerService(wechatDb?.getWcdb4Client())
+    }
+    return stickerService.resolveSticker(cdnUrl, md5)
+  })
+
   createWindow()
 
   app.on('activate', function () {