refactor: 移除微信 3.0 解密路径, 统一走 4.0 WCDB

- wechat-db.ts 删除 SQLCipher 回退、connectDb、getUser、tryOpenWechat4 等 3.0 专属逻辑
- image-decrypt-service.ts 删除 decryptDatV3 与 version === 0 分支
- index.ts 去除 wcdb4Client 冗余空值检查
- 移除 better-sqlite3-multiple-ciphers 依赖
- README 版本前置改为 4.0+

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

src/main/wechat-db.ts

@@ -1,12 +1,5 @@
-import DatabaseConstructor from 'better-sqlite3-multiple-ciphers'
-import fs from 'fs-extra'
-import path from 'path'
-import crypto from 'crypto'
-import os from 'os'
 import { Wcdb4Client } from './wcdb4-client'
 
-type Database = import('better-sqlite3-multiple-ciphers').Database
-
 export interface UserContact {
   m_nsUsrName: string
   nickname: string
@@ -38,444 +31,122 @@ export interface GroupMemberInfo {
 }
 
 export class WechatDb {
-  private static WECHAT_DIR = path.join(
-    os.homedir(),
-    'Library/Containers/com.tencent.xinWeChat/Data/Library/Application Support/com.tencent.xinWeChat/2.0b4.0.9'
-  )
-
-  private rawKey: string
-  private correctUserId: string | null = null
-  private chatDb: { name: string; db_number: string }[] | null = null
-  private groupMemberCache = new Map<string, GroupMemberInfo | null>()
-  private wcdb4Client: Wcdb4Client | null = null
+  private wcdb4Client: Wcdb4Client
   private chatMd5ToUsername = new Map<string, string>()
-  private wechat4OpenError: string | null = null
 
   constructor(rawKey: string) {
-    this.rawKey = rawKey
     console.log(`Initializing WechatDb with key length: ${rawKey.trim().length}`)
-
-    if (this.tryOpenWechat4()) {
-      this.chatDb = this.getChatDbNumber()
-      return
-    }
-
-    if (!fs.existsSync(WechatDb.WECHAT_DIR)) {
-      throw new Error(`WeChat directory not found at ${WechatDb.WECHAT_DIR}`)
-    }
-
-    if (this.getUser()) {
-      console.log('User found, getting chat DB number')
-      this.chatDb = this.getChatDbNumber()
-    } else {
-      throw new Error(
-        `No valid user found or invalid key${this.wechat4OpenError ? `; WeChat 4.0 error: ${this.wechat4OpenError}` : ''}`
-      )
-    }
-  }
-
-  private tryOpenWechat4(): boolean {
-    try {
-      const client = new Wcdb4Client(this.rawKey)
-      client.open()
-      this.wcdb4Client = client
-      console.log('Opened WeChat 4.0 database with WechatExplorer WCDB native adapter')
-      return true
-    } catch (error) {
-      this.wechat4OpenError = error instanceof Error ? error.message : String(error)
-      console.warn('WeChat 4.0 open failed, fallback to 3.0 SQLCipher mode:', error)
-      this.wcdb4Client = null
-      return false
-    }
-  }
-
-  private connectDb(dbPath: string): Database | null {
-    const targetPath = path.resolve(dbPath)
-    if (!fs.existsSync(targetPath)) {
-      console.error(`❌ 错误：数据库文件不存在于路径：${targetPath}`)
-      return null
-    }
-
-    try {
-      const db = new DatabaseConstructor(targetPath, {
-        // verbose: console.log
-      })
-
-      // 配置加密方案 (Cipher Scheme)
-      db.pragma("cipher='sqlcipher'")
-      db.pragma('cipher_page_size = 1024')
-      db.pragma('legacy=3')
-      db.pragma('kdf_iter = 64000')
-      db.pragma('cipher_hmac_algorithm = HMAC_SHA1')
-      db.pragma('cipher_kdf_algorithm = PBKDF2_HMAC_SHA1')
-
-      const processedKey = this.processRawKey(this.rawKey)
-      // console.log(`🔑 应用密钥 (前6位): ${processedKey.substring(0, 6)}...`)
-
-      db.pragma(`key = "x'${processedKey}'"`)
-
-      // 验证连接
-      db.prepare('SELECT count(*) as count FROM sqlite_master').get()
-
-      return db
-    } catch (error) {
-      console.error(`❌ 连接失败:`, error)
-
-      const errString = String(error)
-      if (errString.includes('file is not a database')) {
-        console.error(
-          "💡 诊断建议：\n1. 请确认是否已安装 'better-sqlite3-multiple-ciphers'。\n2. 检查密钥是否正确（WeChat密钥与设备强绑定）。\n3. 尝试将页大小改为 4096 (db.pragma('cipher_page_size = 4096'))。"
-        )
-      } else if (errString.includes('HMAC')) {
-        console.error(
-          '💡 诊断建议：密钥可能正确，但HMAC算法或页大小不匹配。尝试调整 cipher_hmac_algorithm。'
-        )
-      }
-
-      return null
-    }
-  }
-
-  private processRawKey(key: string): string {
-    // 移除可能的 '0x' 前缀并修剪空格
-    return key.replace(/^0x/i, '').trim()
-  }
-
-  private getUser(): boolean {
-    const keyValuePath = path.join(WechatDb.WECHAT_DIR, 'KeyValue')
-    if (!fs.existsSync(keyValuePath)) return false
-
-    const contents = fs.readdirSync(keyValuePath)
-    const potentialUsers = contents.filter((name) => !name.startsWith('.'))
-
-    for (const user of potentialUsers) {
-      const dbPath = path.join(keyValuePath, user, 'KeyValue.db')
-      if (fs.existsSync(dbPath)) {
-        const db = this.connectDb(dbPath)
-        if (db) {
-          this.correctUserId = user
-          console.log(`Valid user found: ${user}`)
-          db.close()
-          return true
-        }
+    const client = new Wcdb4Client(rawKey)
+    client.open()
+    this.wcdb4Client = client
+    for (const table of client.getChatTables()) {
+      if (table.name.startsWith('Chat_')) {
+        this.chatMd5ToUsername.set(table.name.substring(5), table.db_number)
       }
     }
-    return false
-  }
-
-  private getChatDbNumber(): { name: string; db_number: string }[] {
-    if (this.wcdb4Client) {
-      const chatDb = this.wcdb4Client.getChatTables()
-      this.chatMd5ToUsername.clear()
-      for (const table of chatDb) {
-        if (table.name.startsWith('Chat_')) {
-          this.chatMd5ToUsername.set(table.name.substring(5), table.db_number)
-        }
-      }
-      return chatDb
-    }
-
-    const chatDb: { name: string; db_number: string }[] = []
-    if (!this.correctUserId) return []
-
-    for (let i = 0; i < 10; i++) {
-      const dbPath = path.join(WechatDb.WECHAT_DIR, this.correctUserId, 'Message', `msg_${i}.db`)
-      if (!fs.existsSync(dbPath)) continue
-
-      const db = this.connectDb(dbPath)
-      if (db) {
-        const tables = db
-          .prepare("SELECT name FROM sqlite_master WHERE type='table' AND name LIKE 'Chat_%'")
-          .all() as { name: string }[]
-        for (const table of tables) {
-          chatDb.push({ name: table.name, db_number: `msg_${i}.db` })
-        }
-        db.close()
-      }
-    }
-    return chatDb
   }
 
   public getUserList(nicknameFilter?: string): UserContact[] {
-    if (this.wcdb4Client) {
-      const keyword = (nicknameFilter || '').trim().toLowerCase()
-      return this.wcdb4Client
-        .getSessions()
-        .map((session) => ({
-          m_nsUsrName: session.username,
-          nickname: session.nickname || session.username,
-          avatar: session.avatar
-        }))
-        .filter((contact) => {
-          if (!keyword) return true
-          return (
-            contact.m_nsUsrName.toLowerCase().includes(keyword) ||
-            contact.nickname.toLowerCase().includes(keyword)
-          )
-        })
-    }
-
-    if (!this.correctUserId) return []
-    const dbPath = path.join(WechatDb.WECHAT_DIR, this.correctUserId, 'Contact/wccontact_new2.db')
-    const db = this.connectDb(dbPath)
-    if (!db) return []
-
-    let query = "SELECT m_nsUsrName, nickname FROM WCContact WHERE m_nsUsrName NOT LIKE 'gh_%'"
-    if (nicknameFilter) {
-      query += ` AND nickname LIKE '%${nicknameFilter}%'`
-    }
-
-    const results = db.prepare(query).all() as unknown as UserContact[]
-    db.close()
-    return results
+    const keyword = (nicknameFilter || '').trim().toLowerCase()
+    return this.wcdb4Client
+      .getSessions()
+      .map((session) => ({
+        m_nsUsrName: session.username,
+        nickname: session.nickname || session.username,
+        avatar: session.avatar
+      }))
+      .filter((contact) => {
+        if (!keyword) return true
+        return (
+          contact.m_nsUsrName.toLowerCase().includes(keyword) ||
+          contact.nickname.toLowerCase().includes(keyword)
+        )
+      })
   }
 
   public getAllGroupContacts(): Record<string, string> {
-    if (this.wcdb4Client) {
-      const groupContacts: Record<string, string> = {}
-      for (const session of this.wcdb4Client.getSessions()) {
-        if (session.username.endsWith('@chatroom')) {
-          groupContacts[this.md5(session.username)] = session.nickname || session.username
-        }
-      }
-      return groupContacts
-    }
-
-    if (!this.correctUserId) return {}
-    const dbPath = path.join(WechatDb.WECHAT_DIR, this.correctUserId, 'Group/group_new.db')
-    const db = this.connectDb(dbPath)
-    if (!db) return {}
-
-    const results = db.prepare('SELECT m_nsUsrName, nickname FROM GroupContact').all() as {
-      m_nsUsrName: string
-      nickname: string
-    }[]
-    db.close()
-
     const groupContacts: Record<string, string> = {}
-    for (const row of results) {
-      if (row.m_nsUsrName && row.nickname) {
-        const md5 = this.md5(row.m_nsUsrName)
-        groupContacts[md5] = row.nickname
+    for (const session of this.wcdb4Client.getSessions()) {
+      if (session.username.endsWith('@chatroom')) {
+        groupContacts[this.md5(session.username)] = session.nickname || session.username
       }
     }
     return groupContacts
   }
 
   public getAllGroupMembers(): Record<string, string> {
-    if (this.wcdb4Client) {
-      const members: Record<string, string> = {}
-      for (const session of this.wcdb4Client.getSessions()) {
-        if (!session.username.endsWith('@chatroom')) continue
-        for (const member of this.wcdb4Client.getGroupMembers(session.username)) {
-          if (member.m_nsUsrName) {
-            members[member.m_nsUsrName] = member.nickname || member.m_nsUsrName
-          }
-        }
-      }
-      return members
-    }
-
-    if (!this.correctUserId) return {}
-    const dbPath = path.join(WechatDb.WECHAT_DIR, this.correctUserId, 'Group/group_new.db')
-    const db = this.connectDb(dbPath)
-    if (!db) return {}
-
-    // 查询 GroupMember 表
-    const results = db.prepare('SELECT m_nsUsrName, nickname FROM GroupMember').all() as {
-      m_nsUsrName: string
-      nickname: string
-    }[]
-    db.close()
-
-    const groupMembers: Record<string, string> = {}
-    for (const row of results) {
-      if (row.m_nsUsrName && row.nickname) {
-        // 将 wxid (m_nsUsrName) 映射到昵称
-        groupMembers[row.m_nsUsrName] = row.nickname
-      }
-    }
-    return groupMembers
-  }
-
-  public getGroupMembersForChat(userMd5: string): Record<string, string> {
-    if (this.wcdb4Client) {
-      const username = this.chatMd5ToUsername.get(userMd5)
-      if (!username || !username.endsWith('@chatroom')) return {}
-
-      const members: Record<string, string> = {}
-      for (const member of this.wcdb4Client.getGroupMembers(username)) {
+    const members: Record<string, string> = {}
+    for (const session of this.wcdb4Client.getSessions()) {
+      if (!session.username.endsWith('@chatroom')) continue
+      for (const member of this.wcdb4Client.getGroupMembers(session.username)) {
         if (member.m_nsUsrName) {
           members[member.m_nsUsrName] = member.nickname || member.m_nsUsrName
         }
       }
-      return members
     }
+    return members
+  }
 
-    return this.getAllGroupMembers()
+  public getGroupMembersForChat(userMd5: string): Record<string, string> {
+    const username = this.chatMd5ToUsername.get(userMd5)
+    if (!username || !username.endsWith('@chatroom')) return {}
+
+    const members: Record<string, string> = {}
+    for (const member of this.wcdb4Client.getGroupMembers(username)) {
+      if (member.m_nsUsrName) {
+        members[member.m_nsUsrName] = member.nickname || member.m_nsUsrName
+      }
+    }
+    return members
   }
 
   public getGroupMember(wxid: string, chatroomId?: string): GroupMemberInfo | null {
-    if (this.wcdb4Client && chatroomId) {
-      return (
-        this.wcdb4Client
-          .getGroupMembers(chatroomId)
-          .find((member) => member.m_nsUsrName === wxid) || null
-      )
-    }
-
-    if (!this.correctUserId) return null
-
-    // 检查缓存
-    if (this.groupMemberCache.has(wxid)) {
-      return this.groupMemberCache.get(wxid) || null
-    }
-
-    const dbPath = path.join(WechatDb.WECHAT_DIR, this.correctUserId, 'Group/group_new.db')
-    const db = this.connectDb(dbPath)
-    if (!db) {
-      this.groupMemberCache.set(wxid, null)
-      return null
-    }
-
-    const result = db
-      .prepare(
-        'SELECT m_nsUsrName, nickname, m_nsHeadImgUrl FROM GroupMember WHERE m_nsUsrName = ?'
-      )
-      .get(wxid) as GroupMemberInfo
-
-    db.close()
-
-    // 写入缓存
-    this.groupMemberCache.set(wxid, result || null)
-
-    return result || null
+    if (!chatroomId) return null
+    return (
+      this.wcdb4Client
+        .getGroupMembers(chatroomId)
+        .find((member) => member.m_nsUsrName === wxid) || null
+    )
   }
 
   public getAllChatTables(): { name: string; db_number: string }[] {
-    return this.chatDb || []
+    return this.wcdb4Client.getChatTables()
   }
 
   public getMyAvatarUrl(): string | undefined {
-    return this.wcdb4Client?.getMyAvatarUrl()
+    return this.wcdb4Client.getMyAvatarUrl()
   }
 
-  public getWcdb4Client(): Wcdb4Client | null {
+  public getWcdb4Client(): Wcdb4Client {
     return this.wcdb4Client
   }
 
   public close(): void {
-    this.wcdb4Client?.close()
-    this.wcdb4Client = null
+    this.wcdb4Client.close()
   }
 
   public getUserMessages(userMd5: string, startTime?: number, endTime?: number): WechatMessage[] {
-    if (this.wcdb4Client) {
-      const username = this.chatMd5ToUsername.get(userMd5)
-      if (!username) return []
-      return this.wcdb4Client.getMessages(username, startTime, endTime).map((message) => ({
-        ...message,
-        ...message.raw
-      }))
-    }
-
-    if (!this.chatDb || !this.correctUserId) return []
-
-    const tableName = `Chat_${userMd5}`
-    const tableInfo = this.chatDb.find((t) => t.name === tableName)
-
-    if (!tableInfo) {
-      console.error(`No table found for ${tableName}`)
-      return []
-    }
-
-    const dbPath = path.join(
-      WechatDb.WECHAT_DIR,
-      this.correctUserId,
-      'Message',
-      tableInfo.db_number
-    )
-    const db = this.connectDb(dbPath)
-    if (!db) return []
-
-    try {
-      let query = `SELECT * FROM ${tableName}`
-      const conditions: string[] = []
-
-      // console.log(startTime, endTime, 'time range')
-
-      if (startTime) {
-        conditions.push(`msgCreateTime >= ${startTime}`)
-      }
-      if (endTime) {
-        conditions.push(`msgCreateTime <= ${endTime}`)
-      }
-
-      if (conditions.length > 0) {
-        query += ` WHERE ${conditions.join(' AND ')}`
-      }
-
-      // 添加排序以确保存储正确的顺序
-      query += ` ORDER BY msgCreateTime ASC`
-
-      const results = db.prepare(query).all() as unknown as WechatMessage[]
-      db.close()
-      return results
-    } catch (e) {
-      console.error(`Error querying messages for ${tableName}:`, e)
-      db.close()
-      return []
-    }
+    const username = this.chatMd5ToUsername.get(userMd5)
+    if (!username) return []
+    return this.wcdb4Client.getMessages(username, startTime, endTime).map((message) => ({
+      ...message,
+      ...message.raw
+    }))
   }
 
   public searchAllMessages(keyword: string): string | null {
-    if (this.wcdb4Client) {
-      const lowerKeyword = keyword.trim().toLowerCase()
-      if (!lowerKeyword) return null
-      for (const session of this.wcdb4Client.getSessions()) {
-        const found = this.wcdb4Client
-          .getMessages(session.username)
-          .some((message) => message.msgContent.toLowerCase().includes(lowerKeyword))
-        if (found) return `Chat_${this.md5(session.username)}`
-      }
-      return null
-    }
-
-    if (!this.correctUserId) return null
-
-    for (let i = 0; i < 10; i++) {
-      const dbName = `msg_${i}.db`
-      const dbPath = path.join(WechatDb.WECHAT_DIR, this.correctUserId, 'Message', dbName)
-      if (!fs.existsSync(dbPath)) continue
-
-      const db = this.connectDb(dbPath)
-      if (!db) continue
-
-      const tables = db
-        .prepare("SELECT name FROM sqlite_master WHERE type='table' AND name LIKE 'Chat_%'")
-        .all() as { name: string }[]
-      for (const table of tables) {
-        try {
-          // 首先检查 msgContent 列是否存在以避免错误？
-          // 或者像 Swift 代码中那样使用 try/catch "抑制错误"
-          const results = db
-            .prepare(`SELECT 1 FROM ${table.name} WHERE msgContent LIKE '%${keyword}%' LIMIT 1`)
-            .get()
-          if (results) {
-            console.log(`Found keyword in ${table.name} (${dbName})`)
-            db.close()
-            return table.name
-          }
-        } catch {
-          // 忽略没有 msgContent 的表
-        }
-      }
-      db.close()
+    const lowerKeyword = keyword.trim().toLowerCase()
+    if (!lowerKeyword) return null
+    for (const session of this.wcdb4Client.getSessions()) {
+      const found = this.wcdb4Client
+        .getMessages(session.username)
+        .some((message) => message.msgContent.toLowerCase().includes(lowerKeyword))
+      if (found) return `Chat_${this.md5(session.username)}`
     }
     return null
   }
 
   public md5(str: string): string {
-    return crypto.createHash('md5').update(str).digest('hex')
+    return this.wcdb4Client.md5(str)
   }
 }