diff --git a/README.md b/README.md index 94cc53b..82b8e8d 100644 --- a/README.md +++ b/README.md @@ -7,7 +7,7 @@ **一款现代化的微信聊天记录查看与分析工具** [![License](https://img.shields.io/badge/license-CC--BY--NC--SA--4.0-blue.svg)](LICENSE) -[![Version](https://img.shields.io/badge/version-2.1.4-green.svg)](package.json) +[![Version](https://img.shields.io/badge/version-2.1.5-green.svg)](package.json) [![Platform](https://img.shields.io/badge/platform-Windows-0078D6.svg?logo=windows)]() [![Electron](https://img.shields.io/badge/Electron-39-47848F.svg?logo=electron)]() [![React](https://img.shields.io/badge/React-19-61DAFB.svg?logo=react)]() diff --git a/electron/main.ts b/electron/main.ts index 33ea131..0aa3231 100644 --- a/electron/main.ts +++ b/electron/main.ts @@ -3,7 +3,7 @@ import { join } from 'path' import { readFileSync, existsSync } from 'fs' import { autoUpdater } from 'electron-updater' import { DatabaseService } from './services/database' -import { DecryptService } from './services/decrypt' + import { wechatDecryptService } from './services/decryptService' import { ConfigService } from './services/config' import { wxKeyService } from './services/wxKeyService' @@ -65,7 +65,7 @@ function isNewerVersion(version1: string, version2: string): boolean { // 单例服务 let dbService: DatabaseService | null = null -let decryptService: DecryptService | null = null + let configService: ConfigService | null = null let logService: LogService | null = null @@ -125,7 +125,7 @@ function createWindow() { // 初始化服务 configService = new ConfigService() dbService = new DatabaseService() - decryptService = new DecryptService() + logService = new LogService(configService) // 记录应用启动日志 @@ -888,11 +888,22 @@ function registerIpcHandlers() { // 解密相关 ipcMain.handle('decrypt:database', async (_, sourcePath: string, key: string, outputPath: string) => { - return decryptService?.decryptDatabase(sourcePath, key, outputPath) + return wechatDecryptService.decryptDatabase(sourcePath, outputPath, key) }) ipcMain.handle('decrypt:image', async (_, imagePath: string) => { - return decryptService?.decryptImage(imagePath) + return null + }) + + // ... (其他 IPC) + + // 监听增量消息推送 + chatService.on('new-messages', (data) => { + BrowserWindow.getAllWindows().forEach(win => { + if (!win.isDestroyed()) { + win.webContents.send('chat:new-messages', data) + } + }) }) // 文件对话框 @@ -916,6 +927,11 @@ function registerIpcHandlers() { return shell.openExternal(url) }) + ipcMain.handle('shell:showItemInFolder', async (_, fullPath: string) => { + const { shell } = await import('electron') + return shell.showItemInFolder(fullPath) + }) + ipcMain.handle('app:getDownloadsPath', async () => { return app.getPath('downloads') }) @@ -1124,7 +1140,7 @@ function registerIpcHandlers() { // 发送状态：准备启动微信 event.sender.send('wxkey:status', { status: '正在安装 Hook...', level: 1 }) - + // 获取微信路径 const wechatPath = customWechatPath || wxKeyService.getWeChatPath() if (!wechatPath) { @@ -1223,10 +1239,10 @@ function registerIpcHandlers() { ipcMain.handle('dbpath:getBestCachePath', async () => { const { existsSync } = require('fs') const { join } = require('path') - + // 按优先级检查磁盘：D、E、F、C const drives = ['D', 'E', 'F', 'C'] - + for (const drive of drives) { const drivePath = `${drive}:\\` if (existsSync(drivePath)) { @@ -1235,7 +1251,7 @@ function registerIpcHandlers() { return { success: true, path: cachePath, drive } } } - + // 如果都没有，返回用户目录下的默认路径 const { app } = require('electron') const defaultPath = join(app.getPath('userData'), 'cache') @@ -1282,19 +1298,19 @@ function registerIpcHandlers() { // 数据库解密 ipcMain.handle('wcdb:decryptDatabase', async (event, dbPath: string, hexKey: string, wxid: string) => { logService?.info('Decrypt', '开始解密数据库', { dbPath, wxid }) - + try { // 使用已有的 dataManagementService 来解密 const result = await dataManagementService.decryptAll() - + if (result.success) { - logService?.info('Decrypt', '解密完成', { - successCount: result.successCount, - failCount: result.failCount + logService?.info('Decrypt', '解密完成', { + successCount: result.successCount, + failCount: result.failCount }) - - return { - success: true, + + return { + success: true, totalFiles: (result.successCount || 0) + (result.failCount || 0), successCount: result.successCount, failCount: result.failCount @@ -1561,6 +1577,11 @@ function registerIpcHandlers() { return true }) + ipcMain.handle('chat:setCurrentSession', async (_, sessionId: string | null) => { + chatService.setCurrentSession(sessionId) + return true + }) + ipcMain.handle('chat:getSessionDetail', async (_, sessionId: string) => { const result = await chatService.getSessionDetail(sessionId) if (!result.success) { @@ -1672,7 +1693,7 @@ function registerIpcHandlers() { if (welcomeWindow && !welcomeWindow.isDestroyed()) { welcomeWindow.close() } - + // 如果主窗口还不存在，创建它 if (!mainWindow || mainWindow.isDestroyed()) { mainWindow = createWindow() @@ -1681,7 +1702,7 @@ function registerIpcHandlers() { mainWindow.show() mainWindow.focus() } - + return true }) @@ -1978,8 +1999,8 @@ function registerIpcHandlers() { try { const { proxyService } = await import('./services/ai/proxyService') const proxyUrl = await proxyService.getSystemProxy() - return { - success: true, + return { + success: true, hasProxy: !!proxyUrl, proxyUrl: proxyUrl || null } @@ -1993,8 +2014,8 @@ function registerIpcHandlers() { const { proxyService } = await import('./services/ai/proxyService') proxyService.clearCache() const proxyUrl = await proxyService.getSystemProxy() - return { - success: true, + return { + success: true, hasProxy: !!proxyUrl, proxyUrl: proxyUrl || null, message: proxyUrl ? `已刷新代理: ${proxyUrl}` : '未检测到代理，使用直连' @@ -2008,8 +2029,8 @@ function registerIpcHandlers() { try { const { proxyService } = await import('./services/ai/proxyService') const success = await proxyService.testProxy(proxyUrl, testUrl) - return { - success, + return { + success, message: success ? '代理连接正常' : '代理连接失败' } } catch (e) { diff --git a/electron/preload.ts b/electron/preload.ts index ab18a4d..1ab8502 100644 --- a/electron/preload.ts +++ b/electron/preload.ts @@ -33,7 +33,8 @@ contextBridge.exposeInMainWorld('electronAPI', { // Shell shell: { openPath: (path: string) => ipcRenderer.invoke('shell:openPath', path), - openExternal: (url: string) => ipcRenderer.invoke('shell:openExternal', url) + openExternal: (url: string) => ipcRenderer.invoke('shell:openExternal', url), + showItemInFolder: (fullPath: string) => ipcRenderer.invoke('shell:showItemInFolder', fullPath) }, // App @@ -198,12 +199,18 @@ contextBridge.exposeInMainWorld('electronAPI', { downloadEmoji: (cdnUrl: string, md5?: string, productId?: string, createTime?: number) => ipcRenderer.invoke('chat:downloadEmoji', cdnUrl, md5, productId, createTime), close: () => ipcRenderer.invoke('chat:close'), refreshCache: () => ipcRenderer.invoke('chat:refreshCache'), + setCurrentSession: (sessionId: string | null) => ipcRenderer.invoke('chat:setCurrentSession', sessionId), getSessionDetail: (sessionId: string) => ipcRenderer.invoke('chat:getSessionDetail', sessionId), getVoiceData: (sessionId: string, msgId: string, createTime?: number) => ipcRenderer.invoke('chat:getVoiceData', sessionId, msgId, createTime), onSessionsUpdated: (callback: (sessions: any[]) => void) => { const listener = (_: any, sessions: any[]) => callback(sessions) ipcRenderer.on('chat:sessions-updated', listener) return () => ipcRenderer.removeListener('chat:sessions-updated', listener) + }, + onNewMessages: (callback: (data: { sessionId: string; messages: any[] }) => void) => { + const listener = (_: any, data: any) => callback(data) + ipcRenderer.on('chat:new-messages', listener) + return () => ipcRenderer.removeListener('chat:new-messages', listener) } }, diff --git a/electron/services/chatService.ts b/electron/services/chatService.ts index b2cfdc4..470e33a 100644 --- a/electron/services/chatService.ts +++ b/electron/services/chatService.ts @@ -55,6 +55,11 @@ export interface Message { voiceDuration?: number // 语音时长（秒） // 商店表情相关 productId?: string + // 文件消息相关 + fileName?: string // 文件名 + fileSize?: number // 文件大小（字节） + fileExt?: string // 文件扩展名 + fileMd5?: string // 文件 MD5 } export interface Contact { @@ -104,11 +109,24 @@ class ChatService extends EventEmitter { private syncTimer: NodeJS.Timeout | null = null private lastDbCheckTime: number = 0 + // 增量同步相关 + private currentSessionId: string | null = null + // 记录每个会话已读取的最大 sortSeq (用于此后的增量查询) + private sessionCursor: Map = new Map() + constructor() { super() this.configService = new ConfigService() } + /** + * 设置当前聚焦的会话 ID + * 用于增量同步时只推送当前会话的消息 + */ + setCurrentSession(sessionId: string | null): void { + this.currentSessionId = sessionId + } + /** * 清理账号目录名（支持 wxid_ 格式和自定义微信号格式） */ @@ -324,6 +342,69 @@ class ChatService extends EventEmitter { this.dbDir = null } + /** + * 关闭指定的数据库文件（用于增量更新时释放单个文件） + * 这样可以在更新某个数据库时，不影响其他数据库的查询 + */ + closeDatabase(fileName: string): void { + const fileNameLower = fileName.toLowerCase() + + // 检查是否是核心数据库 + if (fileNameLower === 'session.db' && this.sessionDb) { + try { this.sessionDb.close() } catch { } + this.sessionDb = null + return + } + + if (fileNameLower === 'contact.db' && this.contactDb) { + try { this.contactDb.close() } catch { } + this.contactDb = null + this.contactColumnsCache = null + return + } + + if (fileNameLower === 'emoticon.db' && this.emoticonDb) { + try { this.emoticonDb.close() } catch { } + this.emoticonDb = null + return + } + + if (fileNameLower === 'emotion.db' && this.emotionDb) { + try { this.emotionDb.close() } catch { } + this.emotionDb = null + return + } + + if (fileNameLower === 'head_image.db' && this.headImageDb) { + try { this.headImageDb.close() } catch { } + this.headImageDb = null + this.avatarBase64Cache.clear() + return + } + + // 检查是否是消息数据库（在缓存中查找） + const entries = Array.from(this.messageDbCache.entries()) + for (let i = 0; i < entries.length; i++) { + const [dbPath, db] = entries[i] + if (dbPath.toLowerCase().endsWith(fileNameLower)) { + try { db.close() } catch { } + this.messageDbCache.delete(dbPath) + this.knownMessageDbFiles.delete(dbPath) + // 清除相关的预编译语句缓存 + const stmtKeys = Array.from(this.preparedStmtCache.keys()) + for (let j = 0; j < stmtKeys.length; j++) { + if (stmtKeys[j].startsWith(dbPath)) { + this.preparedStmtCache.delete(stmtKeys[j]) + } + } + // 清除会话表缓存（因为可能包含这个数据库的信息） + this.sessionTableCache.clear() + this.sessionTableCacheTime = 0 + return + } + } + } + /** * 获取会话列表 */ @@ -676,6 +757,9 @@ class ChatService extends EventEmitter { } catch { // ignore } + + // 尝试推送增量消息 + this.checkNewMessagesForCurrentSession() } /** @@ -907,8 +991,13 @@ class ChatService extends EventEmitter { limit: number = 50 ): Promise<{ success: boolean; messages?: Message[]; hasMore?: boolean; error?: string }> { try { + // 如果数据库未连接，尝试自动重连 + // 这解决了增量更新期间数据库被关闭后，用户无法查询消息的问题 if (!this.dbDir) { - return { success: false, error: '数据库未连接' } + const connectResult = await this.connect() + if (!connectResult.success) { + return { success: false, error: connectResult.error || '数据库未连接' } + } } // 获取当前用户的 wxid @@ -1028,6 +1117,19 @@ class ChatService extends EventEmitter { quotedSender = quoteInfo.sender } + // 解析文件消息 (localType === 49 且 XML 中 type=6) + let fileName: string | undefined + let fileSize: number | undefined + let fileExt: string | undefined + let fileMd5: string | undefined + if (localType === 49 && content) { + const fileInfo = this.parseFileInfo(content) + fileName = fileInfo.fileName + fileSize = fileInfo.fileSize + fileExt = fileInfo.fileExt + fileMd5 = fileInfo.fileMd5 + } + const parsedContent = this.parseMessageContent(content, localType) allMessages.push({ @@ -1048,7 +1150,11 @@ class ChatService extends EventEmitter { imageMd5, imageDatName, videoMd5, - voiceDuration + voiceDuration, + fileName, + fileSize, + fileExt, + fileMd5 }) } } catch (e: any) { @@ -1083,9 +1189,20 @@ class ChatService extends EventEmitter { const hasMore = allMessages.length > offset + limit const messages = allMessages.slice(offset, offset + limit) + // 反转使最新消息在最后（UI 显示顺序） // 反转使最新消息在最后（UI 显示顺序） messages.reverse() + // 更新增量游标（仅在拉取最新一页时） + if (offset === 0 && messages.length > 0) { + const latestMsg = messages[messages.length - 1] + // 记录已读取的最大 sortSeq + const currentCursor = this.sessionCursor.get(sessionId) || 0 + if (latestMsg.sortSeq > currentCursor) { + this.sessionCursor.set(sessionId, latestMsg.sortSeq) + } + } + return { success: true, messages, hasMore } } catch (e) { console.error('ChatService: 获取消息失败:', e) @@ -1263,6 +1380,37 @@ class ChatService extends EventEmitter { } } + /** + * 解析文件消息信息 + * 从 type=6 的文件消息 XML 中提取文件信息 + */ + private parseFileInfo(content: string): { fileName?: string; fileSize?: number; fileExt?: string; fileMd5?: string } { + if (!content) return {} + + try { + // 检查是否是文件消息 (type=6) + const type = this.extractXmlValue(content, 'type') + if (type !== '6') return {} + + // 提取文件名 (title) + const fileName = this.extractXmlValue(content, 'title') + + // 提取文件大小 (totallen) + const totallenStr = this.extractXmlValue(content, 'totallen') + const fileSize = totallenStr ? parseInt(totallenStr, 10) : undefined + + // 提取文件扩展名 (fileext) + const fileExt = this.extractXmlValue(content, 'fileext') + + // 提取文件 MD5 + const fileMd5 = this.extractXmlValue(content, 'md5')?.toLowerCase() + + return { fileName, fileSize, fileExt, fileMd5 } + } catch { + return {} + } + } + /** * 从数据库行中解析图片 dat 文件名 */ @@ -1621,7 +1769,7 @@ class ChatService extends EventEmitter { private shouldKeepSession(username: string): boolean { if (!username) return false if (username.startsWith('gh_')) return false - + // 过滤折叠对话占位符 if (username === '@placeholder_foldgroup') return false @@ -3254,7 +3402,7 @@ class ChatService extends EventEmitter { if (this.syncTimer) { clearInterval(this.syncTimer) this.syncTimer = null - console.log('[ChatService] 停止自动增量同步') + // console.log('[ChatService] 停止自动增量同步') // 减少日志 } } @@ -3319,6 +3467,187 @@ class ChatService extends EventEmitter { console.error('[ChatService] 检查更新出错:', e) } } + + /** + * 检查当前会话的新消息并推送（增量同步） + * 采用 Push 模式，主动将新解密的消息推送到前端 + */ + private checkNewMessagesForCurrentSession(): void { + if (!this.currentSessionId) return + + // 如果没有游标，说明尚未加载过历史消息，暂不推送（避免数据不连续） + const cursor = this.sessionCursor.get(this.currentSessionId) || 0 + if (cursor === 0) return + + try { + const tables = this.findSessionTables(this.currentSessionId) + if (tables.length === 0) return + + const allNewMessages: Message[] = [] + + // 获取当前用户的 wxid + const myWxid = this.configService.get('myWxid') + const cleanedMyWxid = myWxid ? this.cleanAccountDirName(myWxid) : '' + + for (const { db, tableName, dbPath } of tables) { + // 检查 Name2Id 表 + const hasName2IdTable = this.checkTableExists(db, 'Name2Id') + + // 鲁棒的 myRowId 查找逻辑 (与 getMessages 保持一致) + let myRowId: number | null = null + if (myWxid && hasName2IdTable) { + const cacheKeyOriginal = `${dbPath}:${myWxid}` + const cachedRowIdOriginal = this.myRowIdCache.get(cacheKeyOriginal) + + if (cachedRowIdOriginal !== undefined) { + myRowId = cachedRowIdOriginal + } else { + try { + const row = db.prepare('SELECT rowid FROM Name2Id WHERE user_name = ?').get(myWxid) as any + if (row?.rowid) { + myRowId = row.rowid + this.myRowIdCache.set(cacheKeyOriginal, myRowId) + } else if (cleanedMyWxid && cleanedMyWxid !== myWxid) { + const cacheKeyCleaned = `${dbPath}:${cleanedMyWxid}` + const cachedRowIdCleaned = this.myRowIdCache.get(cacheKeyCleaned) + + if (cachedRowIdCleaned !== undefined) { + myRowId = cachedRowIdCleaned + } else { + const row2 = db.prepare('SELECT rowid FROM Name2Id WHERE user_name = ?').get(cleanedMyWxid) as any + myRowId = row2?.rowid ?? null + this.myRowIdCache.set(cacheKeyCleaned, myRowId) + } + } else { + this.myRowIdCache.set(cacheKeyOriginal, null) + } + } catch { + myRowId = null + } + } + } + + // 构建查询 SQL (查询比 cursor 大的消息) + let sql: string + if (hasName2IdTable && myRowId !== null) { + sql = `SELECT m.*, + CASE WHEN m.real_sender_id = ? THEN 1 ELSE 0 END AS computed_is_send, + n.user_name AS sender_username + FROM ${tableName} m + LEFT JOIN Name2Id n ON m.real_sender_id = n.rowid + WHERE m.sort_seq > ? + ORDER BY m.sort_seq ASC + LIMIT 100` + } else if (hasName2IdTable) { + sql = `SELECT m.*, n.user_name AS sender_username + FROM ${tableName} m + LEFT JOIN Name2Id n ON m.real_sender_id = n.rowid + WHERE m.sort_seq > ? + ORDER BY m.sort_seq ASC + LIMIT 100` + } else { + sql = `SELECT * FROM ${tableName} WHERE sort_seq > ? ORDER BY sort_seq ASC LIMIT 100` + } + + const rows = hasName2IdTable && myRowId !== null + ? db.prepare(sql).all(myRowId, cursor) as any[] + : db.prepare(sql).all(cursor) as any[] + + // 解析消息 + for (const row of rows) { + const content = this.decodeMessageContent(row.message_content, row.compress_content) + const localType = row.local_type || row.type || 1 + const isSend = row.computed_is_send ?? row.is_send ?? null + + let emojiCdnUrl: string | undefined + let emojiMd5: string | undefined + let emojiProductId: string | undefined + let quotedContent: string | undefined + let quotedSender: string | undefined + let imageMd5: string | undefined + let imageDatName: string | undefined + let videoMd5: string | undefined + let voiceDuration: number | undefined + let fileName: string | undefined + let fileSize: number | undefined + let fileExt: string | undefined + let fileMd5: string | undefined + + if (localType === 47 && content) { + const emojiInfo = this.parseEmojiInfo(content) + emojiCdnUrl = emojiInfo.cdnUrl + emojiMd5 = emojiInfo.md5 + emojiProductId = emojiInfo.productId + } else if (localType === 3 && content) { + const imageInfo = this.parseImageInfo(content) + imageMd5 = imageInfo.md5 + imageDatName = this.parseImageDatNameFromRow(row) + } else if (localType === 43 && content) { + videoMd5 = this.parseVideoMd5(content) + } else if (localType === 34 && content) { + voiceDuration = this.parseVoiceDuration(content) + } else if (localType === 49 && content) { + // 解析文件消息 + const fileInfo = this.parseFileInfo(content) + fileName = fileInfo.fileName + fileSize = fileInfo.fileSize + fileExt = fileInfo.fileExt + fileMd5 = fileInfo.fileMd5 + } else if (localType === 244813135921 || (content && content.includes('57'))) { + const quoteInfo = this.parseQuoteMessage(content) + quotedContent = quoteInfo.content + quotedSender = quoteInfo.sender + } + + const parsedContent = this.parseMessageContent(content, localType) + + allNewMessages.push({ + localId: row.local_id || 0, + serverId: row.server_id || 0, + localType, + createTime: row.create_time || 0, + sortSeq: row.sort_seq || 0, + isSend, + senderUsername: row.sender_username || null, + parsedContent, + rawContent: content, + emojiCdnUrl, + emojiMd5, + productId: emojiProductId, + quotedContent, + quotedSender, + imageMd5, + imageDatName, + videoMd5, + voiceDuration, + fileName, + fileSize, + fileExt, + fileMd5 + }) + } + } + + if (allNewMessages.length > 0) { + // 排序 + allNewMessages.sort((a, b) => a.sortSeq - b.sortSeq) + + // 更新游标 + const maxSeq = allNewMessages[allNewMessages.length - 1].sortSeq + this.sessionCursor.set(this.currentSessionId, maxSeq) + + // 推送事件 + this.emit('new-messages', { + sessionId: this.currentSessionId, + messages: allNewMessages + }) + // console.log(`[ChatService] 推送增量消息: ${allNewMessages.length} 条`) + } + + } catch (e) { + // console.error('[ChatService] 增量同步失败:', e) + } + } } export const chatService = new ChatService() diff --git a/electron/services/dataManagementService.ts b/electron/services/dataManagementService.ts index d32c16a..863a8a5 100644 --- a/electron/services/dataManagementService.ts +++ b/electron/services/dataManagementService.ts @@ -345,6 +345,10 @@ class DataManagementService { const time = new Date().toLocaleTimeString() console.error(`[${time}] [数据解密] 解密失败: ${file.fileName}`, result.error) } + + // 关键：强制让出主线程时间片，防止批量处理时 UI 卡死 + // 即使是 Worker 解密，连续的 IPC 通信和主线程调度也会导致卡顿 + await new Promise(resolve => setTimeout(resolve, 10)) } // 完成 @@ -385,13 +389,10 @@ class DataManagementService { return { success: false, error: '请先在设置页面配置解密密钥' } } - // 关闭所有可能占用数据库文件的服务 - chatService.close() + // 不再关闭整个 chatService，而是在更新每个文件前只关闭那个特定的数据库 + // 这样用户可以在增量更新时继续查看其他会话的消息 imageDecryptService.clearHardlinkCache() - // 等待所有数据库连接完全关闭（给一些时间让文件句柄释放） - await new Promise(resolve => setTimeout(resolve, 2000)) - let successCount = 0 let failCount = 0 const totalFiles = filesToUpdate.length @@ -401,7 +402,7 @@ class DataManagementService { // 在处理每个文件前让出时间片，避免阻塞UI const time = new Date().toLocaleTimeString() - console.log(`[${time}] [增量同步] 正在同步数据库: ${file.fileName} (${i + 1}/${totalFiles})`) + // console.log(`[${time}] [增量同步] 正在同步数据库: ${file.fileName} (${i + 1}/${totalFiles})`) // 减少日志 if (i > 0) { await new Promise(resolve => setImmediate(resolve)) } @@ -432,6 +433,12 @@ class DataManagementService { } const backupPath = file.decryptedPath + '.old.' + Date.now() + + // 在备份/覆盖文件前，先关闭该数据库的连接，释放文件锁 + chatService.closeDatabase(file.fileName) + // 等待文件句柄释放 + await new Promise(resolve => setTimeout(resolve, 100)) + if (fs.existsSync(file.decryptedPath!)) { // 尝试备份文件，如果失败则重试几次 let backupSuccess = false @@ -551,8 +558,8 @@ class DataManagementService { if (result.success) { successCount++ - const time = new Date().toLocaleTimeString() - console.log(`[${time}] [增量同步] 同步成功: ${file.fileName}`) + // const time = new Date().toLocaleTimeString() + // console.log(`[${time}] [增量同步] 同步成功: ${file.fileName}`) // 减少日志 if (fs.existsSync(backupPath)) { try { fs.unlinkSync(backupPath) } catch { } } @@ -564,6 +571,9 @@ class DataManagementService { try { fs.renameSync(backupPath, file.decryptedPath!) } catch { } } } + + // 关键：强制让出主线程时间片，防止批量处理时 UI 卡死 + await new Promise(resolve => setTimeout(resolve, 10)) } this.sendProgress({ type: 'complete' }) @@ -1346,7 +1356,7 @@ class DataManagementService { // 检查更新频率限制（最多每5秒更新一次） const now = Date.now() const timeSinceLastUpdate = now - this.lastUpdateTime - const MIN_UPDATE_INTERVAL = 5000 // 最小更新间隔：5秒 + const MIN_UPDATE_INTERVAL = 1000 // 最小更新间隔：1秒 (配合 DLL 极速解密) if (timeSinceLastUpdate < MIN_UPDATE_INTERVAL) { // 如果距离上次更新不足5秒，延迟到满足间隔 @@ -1365,8 +1375,9 @@ class DataManagementService { } // 等待文件写入完成（微信写入数据库可能需要一些时间） - // 延迟2秒，确保文件完全写入完成 - await new Promise(resolve => setTimeout(resolve, 2000)) + // 等待文件写入完成（微信写入数据库可能需要一些时间） + // 延迟1秒，确保文件完全写入完成 + await new Promise(resolve => setTimeout(resolve, 1000)) // 触发更新 this.triggerUpdate() @@ -1391,7 +1402,7 @@ class DataManagementService { // 检查更新频率限制 const now = Date.now() const timeSinceLastUpdate = now - this.lastUpdateTime - const MIN_UPDATE_INTERVAL = 2000 // 最小更新间隔：2秒 (原来是 5 秒) + const MIN_UPDATE_INTERVAL = 1000 // 最小更新间隔：1秒 if (timeSinceLastUpdate < MIN_UPDATE_INTERVAL) { // 延迟到满足间隔 @@ -1430,7 +1441,7 @@ class DataManagementService { // 检查更新频率限制 const now = Date.now() const timeSinceLastUpdate = now - this.lastUpdateTime - const MIN_UPDATE_INTERVAL = 2000 // 最小更新间隔减少到 2 秒 (原来是 5 秒) + const MIN_UPDATE_INTERVAL = 1000 // 最小更新间隔减少到 1 秒 if (timeSinceLastUpdate < MIN_UPDATE_INTERVAL) { const remainingTime = MIN_UPDATE_INTERVAL - timeSinceLastUpdate @@ -1470,8 +1481,8 @@ class DataManagementService { if (result.success) { // 通知监听器更新完成 - const time = new Date().toLocaleTimeString() - console.log(`[${time}] [自动更新] 增量同步完成, 成功更新 ${result.successCount} 个文件`) + // const time = new Date().toLocaleTimeString() + // console.log(`[${time}] [自动更新] 增量同步完成, 成功更新 ${result.successCount} 个文件`) // 减少日志 this.updateListeners.forEach(listener => listener(false)) return { success: true, updated: result.successCount! > 0 } } else { diff --git a/electron/services/decryptService.ts b/electron/services/decryptService.ts index 51fed60..5b81ba9 100644 --- a/electron/services/decryptService.ts +++ b/electron/services/decryptService.ts @@ -1,103 +1,22 @@ -import * as crypto from 'crypto' -import * as fs from 'fs' -import * as path from 'path' - -// 常量定义 -const PAGE_SIZE = 4096 -const KEY_SIZE = 32 -const SALT_SIZE = 16 -const IV_SIZE = 16 -const HMAC_SIZE = 64 // SHA512 -const AES_BLOCK_SIZE = 16 -const ITER_COUNT = 256000 // Windows v4 迭代次数 -const SQLITE_HEADER = 'SQLite format 3\x00' - -// 计算保留字节数 (IV + HMAC, 向上取整到 AES 块大小) -const RESERVE = Math.ceil((IV_SIZE + HMAC_SIZE) / AES_BLOCK_SIZE) * AES_BLOCK_SIZE +import { nativeDecryptService } from './nativeDecryptService' /** * 微信数据库解密服务 (Windows v4) + * 纯原生 DLL 实现封装 */ export class WeChatDecryptService { - /** - * XOR 字节数组 - */ - private xorBytes(data: Buffer, xorValue: number): Buffer { - const result = Buffer.alloc(data.length) - for (let i = 0; i < data.length; i++) { - result[i] = data[i] ^ xorValue - } - return result - } - - /** - * 派生加密密钥和 MAC 密钥 - */ - private deriveKeys(key: Buffer, salt: Buffer): { encKey: Buffer; macKey: Buffer } { - // 生成加密密钥 - const encKey = crypto.pbkdf2Sync(key, salt, ITER_COUNT, KEY_SIZE, 'sha512') - - // 生成 MAC 密钥的盐 (XOR 0x3a) - const macSalt = this.xorBytes(salt, 0x3a) - - // 生成 MAC 密钥 - const macKey = crypto.pbkdf2Sync(encKey, macSalt, 2, KEY_SIZE, 'sha512') - - return { encKey, macKey } - } - /** * 验证密钥是否正确 + * 目前未实现单独的验证逻辑，依赖解密过程中的验证 */ validateKey(dbPath: string, hexKey: string): boolean { - try { - const key = Buffer.from(hexKey, 'hex') - if (key.length !== KEY_SIZE) { - console.error('密钥长度错误:', key.length) - return false - } - - // 读取第一页 - const fd = fs.openSync(dbPath, 'r') - const page1 = Buffer.alloc(PAGE_SIZE) - fs.readSync(fd, page1, 0, PAGE_SIZE, 0) - fs.closeSync(fd) - - // 检查是否已解密 - if (page1.slice(0, 15).toString() === SQLITE_HEADER.slice(0, 15)) { - console.log('数据库已经是解密状态') - return true - } - - // 获取盐值 - const salt = page1.slice(0, SALT_SIZE) - - // 派生密钥 - const { macKey } = this.deriveKeys(key, salt) - - // 计算 HMAC - const dataEnd = PAGE_SIZE - RESERVE + IV_SIZE - const hmac = crypto.createHmac('sha512', macKey) - hmac.update(page1.slice(SALT_SIZE, dataEnd)) - - // 添加页码 (little-endian, 从1开始) - const pageNoBytes = Buffer.alloc(4) - pageNoBytes.writeUInt32LE(1, 0) - hmac.update(pageNoBytes) - - const calculatedMAC = hmac.digest() - const storedMAC = page1.slice(dataEnd, dataEnd + HMAC_SIZE) - - return calculatedMAC.equals(storedMAC) - } catch (e) { - console.error('验证密钥失败:', e) - return false - } + return true } /** * 解密数据库 + * 使用原生 DLL 解密（高性能、异步不卡顿） */ async decryptDatabase( inputPath: string, @@ -105,155 +24,30 @@ export class WeChatDecryptService { hexKey: string, onProgress?: (current: number, total: number) => void ): Promise<{ success: boolean; error?: string }> { + + // 检查服务是否可用 + if (!nativeDecryptService.isAvailable()) { + return { success: false, error: '原生解密服务不可用：DLL 加载失败或 Worker 未启动' } + } + try { - const key = Buffer.from(hexKey, 'hex') - if (key.length !== KEY_SIZE) { - return { success: false, error: '密钥长度错误' } - } + // console.log(`[Decrypt] 开始解密: ${inputPath} -> ${outputPath}`) // 减少日志 - // 获取文件信息 - const stats = fs.statSync(inputPath) - const fileSize = stats.size - const totalPages = Math.ceil(fileSize / PAGE_SIZE) + // 使用异步 DLL 解密 + const result = await nativeDecryptService.decryptDatabaseAsync(inputPath, outputPath, hexKey, onProgress) - // 读取第一页获取盐值 - const fd = fs.openSync(inputPath, 'r') - const page1 = Buffer.alloc(PAGE_SIZE) - fs.readSync(fd, page1, 0, PAGE_SIZE, 0) - - // 检查是否已解密 - if (page1.slice(0, 15).toString() === SQLITE_HEADER.slice(0, 15)) { - fs.closeSync(fd) - // 已解密，直接复制 - fs.copyFileSync(inputPath, outputPath) + if (result.success) { + // console.log('[Decrypt] 解密成功') // 减少日志 return { success: true } + } else { + console.warn(`[Decrypt] 解密失败: ${result.error}`) + return { success: false, error: result.error } } - - const salt = page1.slice(0, SALT_SIZE) - - // 验证密钥 - if (!this.validateKey(inputPath, hexKey)) { - fs.closeSync(fd) - return { success: false, error: '密钥验证失败' } - } - - // 派生密钥 - const { encKey, macKey } = this.deriveKeys(key, salt) - - // 确保输出目录存在 - const outputDir = path.dirname(outputPath) - if (!fs.existsSync(outputDir)) { - fs.mkdirSync(outputDir, { recursive: true }) - } - - // 创建输出文件 - const outFd = fs.openSync(outputPath, 'w') - - // 写入 SQLite 头 - fs.writeSync(outFd, Buffer.from(SQLITE_HEADER)) - - // 处理每一页 - const pageBuf = Buffer.alloc(PAGE_SIZE) - - for (let pageNum = 0; pageNum < totalPages; pageNum++) { - // 读取一页 - const bytesRead = fs.readSync(fd, pageBuf, 0, PAGE_SIZE, pageNum * PAGE_SIZE) - - if (bytesRead === 0) break - - // 检查是否全为零 - let allZeros = true - for (let i = 0; i < bytesRead; i++) { - if (pageBuf[i] !== 0) { - allZeros = false - break - } - } - - if (allZeros) { - // 写入零页面（第一页需要减去盐值大小） - if (pageNum === 0) { - fs.writeSync(outFd, pageBuf.slice(SALT_SIZE, bytesRead)) - } else { - fs.writeSync(outFd, pageBuf.slice(0, bytesRead)) - } - continue - } - - // 解密页面 - const decrypted = this.decryptPage(pageBuf, encKey, macKey, pageNum) - - // 直接写入解密后的页面数据 - fs.writeSync(outFd, decrypted) - - // 进度回调 与 让出时间片 - // 减少到每50页让出一次，提高响应性（对于大文件更重要） - if (pageNum % 50 === 0) { - if (onProgress) { - onProgress(pageNum + 1, totalPages) - } - // 让出事件循环，防止界面卡死 - await new Promise(resolve => setImmediate(resolve)) - } - } - - fs.closeSync(fd) - fs.closeSync(outFd) - - if (onProgress) { - onProgress(totalPages, totalPages) - } - - return { success: true } } catch (e) { - console.error('解密数据库失败:', e) + console.error('[Decrypt] 调用异常:', e) return { success: false, error: String(e) } } } - - /** - * 解密单个页面 - */ - private decryptPage(pageBuf: Buffer, encKey: Buffer, macKey: Buffer, pageNum: number): Buffer { - const offset = pageNum === 0 ? SALT_SIZE : 0 - - // 验证 HMAC - const hmac = crypto.createHmac('sha512', macKey) - hmac.update(pageBuf.slice(offset, PAGE_SIZE - RESERVE + IV_SIZE)) - - const pageNoBytes = Buffer.alloc(4) - pageNoBytes.writeUInt32LE(pageNum + 1, 0) - hmac.update(pageNoBytes) - - const calculatedMAC = hmac.digest() - const hashMacStart = PAGE_SIZE - RESERVE + IV_SIZE - const storedMAC = pageBuf.slice(hashMacStart, hashMacStart + HMAC_SIZE) - - if (!calculatedMAC.equals(storedMAC)) { - console.warn(`页面 ${pageNum} HMAC 验证失败`) - } - - // 获取 IV - const iv = pageBuf.slice(PAGE_SIZE - RESERVE, PAGE_SIZE - RESERVE + IV_SIZE) - - // 解密数据 - const encrypted = pageBuf.slice(offset, PAGE_SIZE - RESERVE) - const decipher = crypto.createDecipheriv('aes-256-cbc', encKey, iv) - decipher.setAutoPadding(false) - - const decrypted = Buffer.concat([ - decipher.update(encrypted), - decipher.final() - ]) - - // 组合解密后的页面：解密数据 + 保留区域 - const result = Buffer.concat([ - decrypted, - pageBuf.slice(PAGE_SIZE - RESERVE) - ]) - - return result - } } export const wechatDecryptService = new WeChatDecryptService() diff --git a/electron/services/nativeDecryptService.ts b/electron/services/nativeDecryptService.ts new file mode 100644 index 0000000..3237527 --- /dev/null +++ b/electron/services/nativeDecryptService.ts @@ -0,0 +1,215 @@ +/** + * 原生 DLL 解密服务 (Worker 多线程版) + * + * 使用独立的 Worker 线程加载 DLL 并执行解密， + * 彻底避免主线程阻塞，并支持实时进度回报。 + */ + +import * as path from 'path' +import * as fs from 'fs' +import { app } from 'electron' +import { Worker } from 'worker_threads' + +// 简单的 ID 生成器 +const generateId = () => Math.random().toString(36).substring(2, 15) + +interface DecryptTask { + resolve: (value: { success: boolean; error?: string }) => void + onProgress?: (current: number, total: number) => void +} + +/** + * 原生解密服务 + */ +export class NativeDecryptService { + private worker: Worker | null = null + private dllPath: string | null = null + private initialized: boolean = false + private initError: string | null = null + private tasks: Map = new Map() + + constructor() { + this.init() + } + + /** + * 初始化服务和 Worker + */ + private init(): void { + if (this.initialized) return + + try { + // 1. 查找 DLL 路径 + this.dllPath = this.findDllPath() + if (!this.dllPath) { + this.initError = '未找到 wcdb_decrypt.dll' + console.warn('[NativeDecrypt] ' + this.initError) + return + } + + // 2. 查找 Worker 脚本路径 + const workerScript = this.findWorkerPath() + if (!workerScript) { + this.initError = '未找到 decryptWorker.js' + console.warn('[NativeDecrypt] ' + this.initError) + return + } + + console.log('[NativeDecrypt] 启动 Worker:', workerScript) + console.log('[NativeDecrypt] DLL 路径:', this.dllPath) + + // 3. 启动 Worker 线程 + this.worker = new Worker(workerScript, { + workerData: { dllPath: this.dllPath } + }) + + // 4. 监听 Worker 消息 + this.worker.on('message', (msg) => this.handleWorkerMessage(msg)) + this.worker.on('error', (err: Error) => { + console.error('[NativeDecrypt] Worker 错误:', err) + this.initError = `Worker error: ${err.message}` + }) + this.worker.on('exit', (code) => { + if (code !== 0) { + console.error(`[NativeDecrypt] Worker 异常退出，代码: ${code}`) + this.worker = null + this.initialized = false + } + }) + + this.initialized = true + + } catch (e) { + this.initError = `初始化失败: ${e}` + console.error('[NativeDecrypt]', this.initError) + } + } + + /** + * 处理 Worker 发来的消息 + */ + private handleWorkerMessage(msg: any): void { + if (msg.type === 'ready') { + console.log('[NativeDecrypt] Worker 已就绪') + return + } + + const task = this.tasks.get(msg.id) + if (!task) return + + switch (msg.type) { + case 'success': + task.resolve({ success: true }) + this.tasks.delete(msg.id) + break + + case 'error': + task.resolve({ success: false, error: msg.error }) + this.tasks.delete(msg.id) + break + + case 'progress': + if (task.onProgress) { + task.onProgress(msg.current, msg.total) + } + break + } + } + + /** + * 查找 DLL 路径 + */ + private findDllPath(): string | null { + const candidates: string[] = [] + if (app.isPackaged) { + candidates.push( + path.join(process.resourcesPath, 'wcdb_decrypt.dll'), + path.join(process.resourcesPath, 'resources', 'wcdb_decrypt.dll'), + path.join(process.resourcesPath, 'app.asar.unpacked', 'resources', 'wcdb_decrypt.dll') + ) + } else { + candidates.push( + path.join(app.getAppPath(), 'resources', 'wcdb_decrypt.dll'), + path.join(app.getAppPath(), 'native-dlls', 'wcdb_decrypt', 'build', 'bin', 'Release', 'wcdb_decrypt.dll') + ) + } + return candidates.find(p => fs.existsSync(p)) || null + } + + /** + * 查找 Worker 脚本路径 + */ + private findWorkerPath(): string | null { + const candidates: string[] = [] + + if (app.isPackaged) { + // 生产模式：Worker 被编译到 dist-electron/workers 目录 + candidates.push( + path.join(process.resourcesPath, 'app.asar.unpacked', 'dist-electron', 'workers', 'decryptWorker.js'), + path.join(process.resourcesPath, 'dist-electron', 'workers', 'decryptWorker.js'), + path.join(__dirname, 'workers', 'decryptWorker.js'), + path.join(__dirname, '..', 'workers', 'decryptWorker.js') + ) + } else { + // 开发模式：Worker 在源码目录 + candidates.push( + path.join(app.getAppPath(), 'electron', 'workers', 'decryptWorker.js'), + path.join(__dirname, '..', 'workers', 'decryptWorker.js') + ) + } + + const found = candidates.find(p => fs.existsSync(p)) + if (found) { + console.log('[NativeDecrypt] 找到 Worker:', found) + } else { + console.error('[NativeDecrypt] 未找到 Worker，尝试的路径:', candidates) + } + return found || null + } + + /** + * 检查服务是否可用 + */ + isAvailable(): boolean { + return this.initialized && this.worker !== null + } + + /** + * 异步解密数据库（通过 Worker） + */ + async decryptDatabaseAsync( + inputPath: string, + outputPath: string, + hexKey: string, + onProgress?: (current: number, total: number) => void + ): Promise<{ success: boolean; error?: string }> { + if (!this.worker) { + // 如果 Worker 挂了，尝试重启 + if (!this.initialized && !this.initError) { + this.init() + } + if (!this.worker) { + return { success: false, error: this.initError || 'Worker 未启动' } + } + } + + return new Promise((resolve) => { + const id = generateId() + + // 注册任务 + this.tasks.set(id, { resolve, onProgress }) + + // 发送消息 + this.worker!.postMessage({ + type: 'decrypt', + id, + inputPath, + outputPath, + hexKey + }) + }) + } +} + +// 导出单例 +export const nativeDecryptService = new NativeDecryptService() diff --git a/electron/workers/decryptWorker.js b/electron/workers/decryptWorker.js new file mode 100644 index 0000000..10e01c5 --- /dev/null +++ b/electron/workers/decryptWorker.js @@ -0,0 +1,88 @@ +const { parentPort, workerData } = require('worker_threads') +const path = require('path') +const fs = require('fs') +const koffi = require('koffi') + +// 从 workerData 获取 DLL 路径 +const { dllPath } = workerData + +if (!dllPath || !fs.existsSync(dllPath)) { + parentPort?.postMessage({ type: 'error', error: 'DLL path not found: ' + dllPath }) + process.exit(1) +} + +try { + // 加载 DLL + const lib = koffi.load(dllPath) + + // 定义回调类型 + const ProgressCallback = koffi.proto('void ProgressCallback(int current, int total)') + + // 绑定函数 (这里使用同步版本，因为 Worker 本身就是独立的线程) + const Wcdb_DecryptDatabaseWithProgress = lib.func('int Wcdb_DecryptDatabaseWithProgress(const char* inputPath, const char* outputPath, const char* hexKey, ProgressCallback* callback)') + const Wcdb_GetLastErrorMsg = lib.func('int Wcdb_GetLastErrorMsg(char* buffer, int size)') + + // 监听主线程消息 + parentPort?.on('message', (message) => { + if (message.type === 'decrypt') { + const { id, inputPath, outputPath, hexKey } = message + + try { + // 确保输出目录存在 + const outputDir = path.dirname(outputPath) + if (!fs.existsSync(outputDir)) { + fs.mkdirSync(outputDir, { recursive: true }) + } + + // 定义进度回调 (带节流，每 100ms 更新一次) + let lastUpdate = 0 + + const onProgress = koffi.register((current, total) => { + const now = Date.now() + if (now - lastUpdate > 100 || current === total || current === 1) { + lastUpdate = now + parentPort?.postMessage({ + type: 'progress', + id, + current, + total + }) + } + }, koffi.pointer(ProgressCallback)) + + // 执行解密 + const result = Wcdb_DecryptDatabaseWithProgress(inputPath, outputPath, hexKey, onProgress) + + // 注销回调以释放资源 + koffi.unregister(onProgress) + + if (result === 0) { + parentPort?.postMessage({ type: 'success', id }) + } else { + // 获取错误信息 + const buffer = Buffer.alloc(512) + Wcdb_GetLastErrorMsg(buffer, 512) + const errorMsg = buffer.toString('utf8').replace(/\0+$/, '') + + parentPort?.postMessage({ + type: 'error', + id, + error: errorMsg || `ErrorCode: ${result}` + }) + } + } catch (err) { + parentPort?.postMessage({ + type: 'error', + id, + error: String(err) + }) + } + } + }) + + // 通知主线程 Worker 已就绪 + parentPort?.postMessage({ type: 'ready' }) + +} catch (err) { + parentPort?.postMessage({ type: 'error', error: String(err) }) +} diff --git a/package.json b/package.json index 0f9cf85..6570487 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "ciphertalk", - "version": "2.1.4", + "version": "2.1.5", "description": "密语 - 微信聊天记录查看工具", "author": "ILoveBingLu", "license": "CC-BY-NC-SA-4.0", @@ -20,6 +20,8 @@ "dependencies": { "@types/dompurify": "^3.0.5", "@types/marked": "^5.0.2", + "@types/react-virtualized-auto-sizer": "^1.0.4", + "@types/react-window": "^1.8.8", "better-sqlite3": "^12.5.0", "dom-to-image-more": "^3.7.2", "dompurify": "^3.3.1", @@ -40,6 +42,8 @@ "react": "^19.2.3", "react-dom": "^19.2.3", "react-router-dom": "^7.1.1", + "react-virtualized-auto-sizer": "^2.0.2", + "react-window": "^2.2.5", "sherpa-onnx-node": "^1.12.23", "silk-wasm": "^3.7.1", "wechat-emojis": "^1.0.2", @@ -134,7 +138,10 @@ "asarUnpack": [ "node_modules/ffmpeg-static/**/*", "node_modules/silk-wasm/**/*", - "node_modules/sherpa-onnx-node/**/*" + "node_modules/sherpa-onnx-node/**/*", + "node_modules/koffi/**/*", + "dist-electron/workers/**/*", + "resources/**/*" ] } } diff --git a/resources/wcdb_decrypt.dll b/resources/wcdb_decrypt.dll new file mode 100644 index 0000000..41f5259 Binary files /dev/null and b/resources/wcdb_decrypt.dll differ diff --git a/src/pages/ChatPage.scss b/src/pages/ChatPage.scss index c3f54f6..323dea5 100644 --- a/src/pages/ChatPage.scss +++ b/src/pages/ChatPage.scss @@ -1644,6 +1644,85 @@ } } +// 文件消息卡片 +.file-message { + width: 240px; + background: var(--card-bg); + border-radius: 8px; + padding: 12px; + display: flex; + gap: 12px; + align-items: center; + border: 1px solid var(--border-color); + cursor: pointer; + transition: all 0.2s ease; + user-select: none; + + &:hover { + background: var(--bg-hover); + border-color: var(--primary); + transform: translateY(-1px); + box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1); + } + + &:active { + transform: translateY(0); + } + + .file-icon { + flex-shrink: 0; + width: 44px; + height: 44px; + background: linear-gradient(135deg, #4a90d9 0%, #357abd 100%); + border-radius: 8px; + display: flex; + align-items: center; + justify-content: center; + color: white; + + svg { + width: 24px; + height: 24px; + } + } + + .file-info { + flex: 1; + min-width: 0; + overflow: hidden; + + .file-name { + font-size: 14px; + font-weight: 500; + color: var(--text-primary); + line-height: 1.3; + white-space: nowrap; + overflow: hidden; + text-overflow: ellipsis; + margin-bottom: 4px; + } + + .file-meta { + font-size: 12px; + color: var(--text-tertiary); + } + } +} + +// 发送的文件消息样式 +.message-bubble.sent .file-message { + background: #fff; + border: 1px solid rgba(0, 0, 0, 0.1); + + .file-name { + color: #333; + } + + .file-meta { + color: #999; + } +} + // 转账消息卡片 .transfer-message { width: 240px; diff --git a/src/pages/ChatPage.tsx b/src/pages/ChatPage.tsx index 984580c..292782f 100644 --- a/src/pages/ChatPage.tsx +++ b/src/pages/ChatPage.tsx @@ -1,14 +1,23 @@ -import { useState, useEffect, useRef, useCallback } from 'react' +import { useState, useEffect, useRef, useCallback, memo } from 'react' import { createPortal } from 'react-dom' -import { Search, MessageSquare, AlertCircle, Loader2, RefreshCw, X, ChevronDown, Info, Calendar, Database, Hash, Image as ImageIcon, Play, Video, Copy, ZoomIn, CheckSquare, Check, Edit, Link, Sparkles } from 'lucide-react' +import { Search, MessageSquare, AlertCircle, Loader2, RefreshCw, X, ChevronDown, Info, Calendar, Database, Hash, Image as ImageIcon, Play, Video, Copy, ZoomIn, CheckSquare, Check, Edit, Link, Sparkles, FileText, FileArchive } from 'lucide-react' import { useChatStore } from '../stores/chatStore' import { useUpdateStatusStore } from '../stores/updateStatusStore' import ChatBackground from '../components/ChatBackground' import MessageContent from '../components/MessageContent' import { getImageXorKey, getImageAesKey, getQuoteStyle } from '../services/config' +import { LRUCache } from '../utils/lruCache' import type { ChatSession, Message } from '../types/models' +import { List, RowComponentProps } from 'react-window' import './ChatPage.scss' +interface SessionRowData { + sessions: ChatSession[] + currentSessionId: string | null + onSelect: (s: ChatSession) => void + formatTime: (t: number) => string +} + interface ChatPageProps { @@ -155,6 +164,48 @@ function SessionAvatar({ session, size = 48 }: { session: ChatSession; size?: nu ) } +// 会话列表行组件（使用 memo 优化性能） +const SessionRow = (props: RowComponentProps) => { + const { index, style, sessions, currentSessionId, onSelect, formatTime } = props + const session = sessions[index] + + return ( +
onSelect(session)} + > + +
+
+ {session.displayName || session.username} + {formatTime(session.lastTimestamp || session.sortTimestamp)} +
+
+ + {(() => { + const summary = session.summary || '暂无消息' + const firstLine = summary.split('\n')[0] + const hasMoreLines = summary.includes('\n') + return ( + <> + + {hasMoreLines && ...} + + ) + })()} + + {session.unreadCount > 0 && ( + + {session.unreadCount > 99 ? '99+' : session.unreadCount} + + )} +
+
+
+ ) +} + function ChatPage(_props: ChatPageProps) { const [quoteStyle, setQuoteStyle] = useState<'default' | 'wechat'>('default') @@ -322,14 +373,14 @@ function ChatPage(_props: ChatPageProps) { // 合并：保留顺序，只更新变化的字段 const merged = result.sessions!.map(newSession => { const oldSession = oldSessionsMap.get(newSession.username) - + // 如果是新会话，直接返回 if (!oldSession) { return newSession } // 检查是否有实质性变化 - const hasChanges = + const hasChanges = oldSession.summary !== newSession.summary || oldSession.lastTimestamp !== newSession.lastTimestamp || oldSession.unreadCount !== newSession.unreadCount || @@ -439,6 +490,40 @@ function ChatPage(_props: ChatPageProps) { } } + // 监听增量消息推送 + useEffect(() => { + // 告知后端当前会话 + window.electronAPI.chat.setCurrentSession(currentSessionId) + + const cleanup = window.electronAPI.chat.onNewMessages((data: { sessionId: string; messages: Message[] }) => { + if (data.sessionId === currentSessionId && data.messages && data.messages.length > 0) { + setMessages((prev: Message[]) => { + // 使用 sortSeq 去重 + const newMsgs = data.messages.filter((nm: Message) => + !prev.some((pm: Message) => pm.sortSeq === nm.sortSeq) + ) + if (newMsgs.length === 0) return prev + + return [...prev, ...newMsgs] + }) + + // 平滑滚动到底部 + requestAnimationFrame(() => scrollToBottom(true)) + } + }) + + return () => { + cleanup() + } + }, [currentSessionId]) + + // 组件卸载时取消当前会话 + useEffect(() => { + return () => { + window.electronAPI.chat.setCurrentSession(null) + } + }, []) + // 选择会话 const handleSelectSession = (session: ChatSession) => { if (session.username === currentSessionId) { @@ -559,11 +644,11 @@ function ChatPage(_props: ChatPageProps) { useEffect(() => { if (!isConnected) return - // 监听会话列表更新 + // 监听会话列表更新 const removeSessionsListener = window.electronAPI.chat.onSessionsUpdated?.(async (newSessions) => { // 更新增量更新时间戳 lastIncrementalUpdateTime = Date.now() - + // 智能合并更新会话列表，避免闪烁 setSessions((prevSessions: ChatSession[]) => { // 如果之前没有会话，直接设置 @@ -579,14 +664,14 @@ function ChatPage(_props: ChatPageProps) { // 合并：保留顺序，只更新变化的字段 const merged = newSessions.map(newSession => { const oldSession = oldSessionsMap.get(newSession.username) - + // 如果是新会话，直接返回 if (!oldSession) { return newSession } // 检查是否有实质性变化 - const hasChanges = + const hasChanges = oldSession.summary !== newSession.summary || oldSession.lastTimestamp !== newSession.lastTimestamp || oldSession.unreadCount !== newSession.unreadCount || @@ -817,43 +902,22 @@ function ChatPage(_props: ChatPageProps) { ))} ) : filteredSessions.length > 0 ? ( -
- {filteredSessions.map(session => ( -
handleSelectSession(session)} - > - -
-
- {session.displayName || session.username} - {formatSessionTime(session.lastTimestamp || session.sortTimestamp)} -
-
- - {(() => { - const summary = session.summary || '暂无消息' - const firstLine = summary.split('\n')[0] - const hasMoreLines = summary.includes('\n') - return ( - <> - - {hasMoreLines && ...} - - ) - })()} - - {session.unreadCount > 0 && ( - - {session.unreadCount > 99 ? '99+' : session.unreadCount} - - )} -
-
-
- ))} +
+ {/* @ts-ignore - 类型定义不匹配但不影响运行 */} +
+ ) : (
@@ -1357,10 +1421,10 @@ function ChatPage(_props: ChatPageProps) { ) } -// 前端表情包缓存 -const emojiDataUrlCache = new Map() -// 前端图片缓存 -const imageDataUrlCache = new Map() +// 前端表情包缓存 (LRU 限制) +const emojiDataUrlCache = new LRUCache(200) +// 前端图片缓存 (LRU 限制) +const imageDataUrlCache = new LRUCache(50) // 图片解密队列管理 const imageDecryptQueue: Array<() => Promise> = [] @@ -1387,7 +1451,7 @@ function enqueueDecrypt(fn: () => Promise) { } // 视频信息缓存（带时间戳） -const videoInfoCache = new Map { if (isGroupChat && !isSent && message.senderUsername) { setIsLoadingSender(true) @@ -1855,7 +1919,7 @@ function MessageBubble({ message, session, showTime, myAvatarUrl, isGroupChat, h setSenderName(result.displayName) } setIsLoadingSender(false) - }).catch(() => { + }).catch(() => { setIsLoadingSender(false) }) } @@ -2436,26 +2500,122 @@ function MessageBubble({ message, session, showTime, myAvatarUrl, isGroupChat, h ) } + // 文件消息 (type=6)：渲染为文件卡片 + if (appMsgType === '6') { + // 优先使用从接口获取的文件信息，否则从 XML 解析 + const fileName = message.fileName || title || '文件' + const fileSize = message.fileSize + const fileExt = message.fileExt || fileName.split('.').pop()?.toLowerCase() || '' + const fileMd5 = message.fileMd5 + + // 格式化文件大小 + const formatFileSize = (bytes: number | undefined): string => { + if (!bytes) return '' + if (bytes < 1024) return `${bytes} B` + if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)} KB` + if (bytes < 1024 * 1024 * 1024) return `${(bytes / 1024 / 1024).toFixed(1)} MB` + return `${(bytes / 1024 / 1024 / 1024).toFixed(1)} GB` + } + + // 根据扩展名选择图标 + const getFileIcon = (ext: string) => { + const archiveExts = ['zip', 'rar', '7z', 'tar', 'gz', 'bz2'] + if (archiveExts.includes(ext)) { + return + } + return + } + + // 点击文件消息，定位到文件所在文件夹并选中文件 + const handleFileClick = async () => { + try { + // 获取用户设置的微信原始存储目录（不是解密缓存目录） + const wechatDir = await window.electronAPI.config.get('dbPath') as string + if (!wechatDir) { + console.error('未设置微信存储目录') + return + } + + // 获取当前用户信息 + const userInfo = await window.electronAPI.chat.getMyUserInfo() + if (!userInfo.success || !userInfo.userInfo) { + console.error('无法获取用户信息') + return + } + + const wxid = userInfo.userInfo.wxid + + // 文件存储在 {微信存储目录}\{账号文件夹}\msg\file\{年-月}\ 目录下 + // 根据消息创建时间计算日期目录 + const msgDate = new Date(message.createTime * 1000) + const year = msgDate.getFullYear() + const month = String(msgDate.getMonth() + 1).padStart(2, '0') + const dateFolder = `${year}-${month}` + + // 构建完整文件路径（包括文件名） + const filePath = `${wechatDir}\\${wxid}\\msg\\file\\${dateFolder}\\${fileName}` + + // 使用 showItemInFolder 在文件管理器中定位并选中文件 + try { + await window.electronAPI.shell.showItemInFolder(filePath) + } catch (err) { + // 如果文件不存在或路径错误，尝试只打开文件夹 + console.warn('无法定位到具体文件，尝试打开文件夹:', err) + const fileDir = `${wechatDir}\\${wxid}\\msg\\file\\${dateFolder}` + const result = await window.electronAPI.shell.openPath(fileDir) + + // 如果还是失败，打开上级目录 + if (result) { + console.warn('无法打开月份文件夹，尝试打开上级目录') + const parentDir = `${wechatDir}\\${wxid}\\msg\\file` + await window.electronAPI.shell.openPath(parentDir) + } + } + } catch (error) { + console.error('打开文件夹失败:', error) + } + } + + return ( +
+
+ {getFileIcon(fileExt)} +
+
+
{fileName}
+
+ {fileSize ? formatFileSize(fileSize) : ''} +
+
+
+ ) + } + // 转账消息 (type=2000)：渲染为转账卡片 if (appMsgType === '2000') { try { const content = message.rawContent || message.parsedContent || '' const parser = new DOMParser() const doc = parser.parseFromString(content, 'text/xml') - + const feedesc = doc.querySelector('feedesc')?.textContent || '' const payMemo = doc.querySelector('pay_memo')?.textContent || '' const paysubtype = doc.querySelector('paysubtype')?.textContent || '1' - + // paysubtype: 1=待收款, 3=已收款 const isReceived = paysubtype === '3' - + return (
- - + +
diff --git a/src/pages/DataManagementPage.tsx b/src/pages/DataManagementPage.tsx index 9d8da43..d64dd6c 100644 --- a/src/pages/DataManagementPage.tsx +++ b/src/pages/DataManagementPage.tsx @@ -111,6 +111,17 @@ function DataManagementPage() { } const handleDecryptAll = async () => { + // 先检查是否配置了解密密钥 + const decryptKey = await window.electronAPI.config.get('decryptKey') + if (!decryptKey) { + showMessage('请先在设置页面配置解密密钥', false) + // 3秒后自动跳转到设置页面 + setTimeout(() => { + window.location.hash = '#/settings' + }, 3000) + return + } + // 检查聊天窗口是否打开 const isChatOpen = await window.electronAPI.window.isChatWindowOpen() if (isChatOpen) { diff --git a/src/stores/chatStore.ts b/src/stores/chatStore.ts index fda8b32..52f12f8 100644 --- a/src/stores/chatStore.ts +++ b/src/stores/chatStore.ts @@ -36,7 +36,7 @@ export interface ChatState { setFilteredSessions: (sessions: ChatSession[]) => void setCurrentSession: (sessionId: string | null) => void setLoadingSessions: (loading: boolean) => void - setMessages: (messages: Message[]) => void + setMessages: (messages: Message[] | ((prev: Message[]) => Message[])) => void appendMessages: (messages: Message[], prepend?: boolean) => void setLoadingMessages: (loading: boolean) => void setLoadingMore: (loading: boolean) => void @@ -82,24 +82,26 @@ export const useChatStore = create((set, get) => ({ setLoadingSessions: (loading) => set({ isLoadingSessions: loading }), - setMessages: (messages) => set({ messages }), + setMessages: (messages) => set((state) => ({ + messages: typeof messages === 'function' ? messages(state.messages) : messages + })), appendMessages: (newMessages, prepend = false) => set((state) => { // 使用与后端一致的多维 Key (serverId + localId + createTime + sortSeq) 进行去重 const existingKeys = new Set( state.messages.map(m => `${m.serverId}-${m.localId}-${m.createTime}-${m.sortSeq}`) ) - + // 过滤掉已存在的消息 const uniqueNewMessages = newMessages.filter( msg => !existingKeys.has(`${msg.serverId}-${msg.localId}-${msg.createTime}-${msg.sortSeq}`) ) - + // 如果没有新消息，直接返回原状态 if (uniqueNewMessages.length === 0) { return state } - + return { messages: prepend ? [...uniqueNewMessages, ...state.messages] diff --git a/src/types/electron.d.ts b/src/types/electron.d.ts index ebeec38..b480009 100644 --- a/src/types/electron.d.ts +++ b/src/types/electron.d.ts @@ -46,6 +46,7 @@ export interface ElectronAPI { shell: { openPath: (path: string) => Promise openExternal: (url: string) => Promise + showItemInFolder: (fullPath: string) => Promise } app: { getDownloadsPath: () => Promise @@ -202,6 +203,8 @@ export interface ElectronAPI { downloadEmoji: (cdnUrl: string, md5?: string, productId?: string, createTime?: number) => Promise<{ success: boolean; localPath?: string; error?: string }> close: () => Promise refreshCache: () => Promise + setCurrentSession: (sessionId: string | null) => Promise + onNewMessages: (callback: (data: { sessionId: string; messages: Message[] }) => void) => () => void getSessionDetail: (sessionId: string) => Promise<{ success: boolean detail?: { @@ -387,7 +390,7 @@ export interface ElectronAPI { successCount?: number error?: string }> - onProgress: (callback: (data: { + onProgress: (callback: (data: { current?: number total?: number currentSession?: string diff --git a/src/types/models.ts b/src/types/models.ts index 6dbd2da..7be2b9e 100644 --- a/src/types/models.ts +++ b/src/types/models.ts @@ -55,6 +55,11 @@ export interface Message { videoMd5?: string rawContent?: string productId?: string + // 文件消息相关 + fileName?: string // 文件名 + fileSize?: number // 文件大小（字节） + fileExt?: string // 文件扩展名 + fileMd5?: string // 文件 MD5 } // 分析数据 diff --git a/src/utils/lruCache.ts b/src/utils/lruCache.ts new file mode 100644 index 0000000..3a473c6 --- /dev/null +++ b/src/utils/lruCache.ts @@ -0,0 +1,50 @@ +/** + * 简单的 LRU (Least Recently Used) 缓存实现 + * 用于限制内存中缓存对象的数量，防止内存泄漏 + */ +export class LRUCache { + private capacity: number; + private cache: Map; + + constructor(capacity: number) { + this.capacity = capacity; + this.cache = new Map(); + } + + get(key: K): V | undefined { + if (!this.cache.has(key)) return undefined; + + // 刷新项目：先删除再添加，使其成为最新的（排在 Map 末尾） + const value = this.cache.get(key)!; + this.cache.delete(key); + this.cache.set(key, value); + return value; + } + + set(key: K, value: V): void { + if (this.cache.has(key)) { + // 如果已存在，删除旧的以便重新添加到末尾 + this.cache.delete(key); + } else if (this.cache.size >= this.capacity) { + // 如果达到容量上限，删除第一个项目（最久未使用的） + // Map.keys().next().value 获取的是插入顺序最早的那个 + const firstKey = this.cache.keys().next().value; + if (firstKey !== undefined) { + this.cache.delete(firstKey); + } + } + this.cache.set(key, value); + } + + has(key: K): boolean { + return this.cache.has(key); + } + + clear(): void { + this.cache.clear(); + } + + get size(): number { + return this.cache.size; + } +} diff --git a/vite.config.ts b/vite.config.ts index fb9cd58..046f1ed 100644 --- a/vite.config.ts +++ b/vite.config.ts @@ -35,6 +35,18 @@ export default defineConfig({ } } }, + { + // 数据库解密 Worker 线程 + entry: 'electron/workers/decryptWorker.js', + vite: { + build: { + outDir: 'dist-electron/workers', + rollupOptions: { + external: ['koffi'] + } + } + } + }, { // 语音转写 Worker 线程 entry: 'electron/transcribeWorker.ts',