feat: 添加微信数据库实时监控，聊天记录自动刷新

src/main/index.ts

@@ -109,14 +109,22 @@ app.whenReady().then(() => {
     try {
       const trimmedKey = String(key || '').trim()
       console.log(`db:init build=${BUILD_MARK} keyLength=${trimmedKey.length}`)
-      wechatDb = new WechatDb(key)
-      const wcdb4Client = wechatDb.getWcdb4Client()
+      const nextWechatDb = new WechatDb(key)
+      wechatDb?.close()
+      wechatDb = nextWechatDb
+      const wcdb4Client = nextWechatDb.getWcdb4Client()
+      let monitoring = false
       if (wcdb4Client) {
         voiceService = new VoiceService(wcdb4Client)
         stickerService = new StickerService(wcdb4Client)
+        monitoring = wcdb4Client.startMonitor((type, json) => {
+          for (const window of BrowserWindow.getAllWindows()) {
+            if (!window.isDestroyed()) window.webContents.send('wcdb-change', { type, json })
+          }
+        })
       }
       imageDecryptService = null
-      return { success: true }
+      return { success: true, monitoring }
     } catch (error) {
       console.error('Failed to init DB:', error)
       return { success: false, error: error instanceof Error ? error.message : String(error) }
@@ -453,3 +461,8 @@ app.on('window-all-closed', () => {
     app.quit()
   }
 })
+
+app.on('before-quit', () => {
+  wechatDb?.close()
+  wechatDb = null
+})

src/main/wcdb4-client.ts

@@ -3,6 +3,7 @@ import path from 'path'
 import os from 'os'
 import crypto from 'crypto'
 import { createRequire } from 'module'
+import { createConnection, Socket } from 'net'
 
 export interface Wcdb4Session {
   username: string
@@ -132,6 +133,15 @@ export class Wcdb4Client {
   private wcdbGetEmoticonCdnUrl:
     | ((handle: number, dbPath: string, md5: string, outUrl: WcdbVoidOut) => number)
     | null = null
+  private wcdbStartMonitorPipe: (() => number) | null = null
+  private wcdbStopMonitorPipe: (() => void) | null = null
+  private wcdbGetMonitorPipeName: ((outName: WcdbVoidOut) => number) | null = null
+  private monitorPipeClient: Socket | null = null
+  private monitorCallback: ((type: string, json: string) => void) | null = null
+  private monitorConnectTimer: ReturnType<typeof setTimeout> | null = null
+  private monitorReconnectTimer: ReturnType<typeof setTimeout> | null = null
+  private monitorPipePath = ''
+  private monitorStarted = false
 
   constructor(key: string, accountRoot?: string) {
     this.key = key.replace(/^0x/i, '').trim()
@@ -225,6 +235,7 @@ export class Wcdb4Client {
   }
 
   close(): void {
+    this.stopMonitor()
     if (!this.initialized || !this.wcdbShutdown) return
 
     try {
@@ -241,6 +252,142 @@ export class Wcdb4Client {
     this.groupNicknameCache.clear()
   }
 
+  startMonitor(callback: (type: string, json: string) => void): boolean {
+    if (!this.wcdbStartMonitorPipe || !this.wcdbGetMonitorPipeName || !this.koffi) return false
+
+    this.stopMonitor()
+    this.monitorCallback = callback
+
+    try {
+      const startResult = this.wcdbStartMonitorPipe()
+      if (startResult !== 0) {
+        this.monitorCallback = null
+        console.warn(`[WCDB4] wcdb_start_monitor_pipe 失败，错误码: ${startResult}`)
+        return false
+      }
+      this.monitorStarted = true
+
+      const outName: WcdbVoidOut = [null]
+      const nameResult = this.wcdbGetMonitorPipeName(outName)
+      if (nameResult !== 0 || !outName[0]) {
+        console.warn(`[WCDB4] wcdb_get_monitor_pipe_name 失败，错误码: ${nameResult}`)
+        this.stopMonitor()
+        return false
+      }
+
+      try {
+        this.monitorPipePath = this.koffi.decode(outName[0], 'char', -1).trim()
+      } finally {
+        this.wcdbFreeString?.(outName[0])
+      }
+
+      if (!this.monitorPipePath) {
+        this.stopMonitor()
+        return false
+      }
+
+      this.connectMonitorPipe()
+      return true
+    } catch (error) {
+      console.warn('[WCDB4] 启动数据库监听失败:', error)
+      this.stopMonitor()
+      return false
+    }
+  }
+
+  stopMonitor(): void {
+    this.monitorCallback = null
+
+    if (this.monitorConnectTimer) {
+      clearTimeout(this.monitorConnectTimer)
+      this.monitorConnectTimer = null
+    }
+    if (this.monitorReconnectTimer) {
+      clearTimeout(this.monitorReconnectTimer)
+      this.monitorReconnectTimer = null
+    }
+    if (this.monitorPipeClient) {
+      this.monitorPipeClient.destroy()
+      this.monitorPipeClient = null
+    }
+    if (this.monitorStarted && this.wcdbStopMonitorPipe) {
+      try {
+        this.wcdbStopMonitorPipe()
+      } catch {
+        // Native monitor cleanup is best-effort during reconnect or shutdown.
+      }
+    }
+
+    this.monitorStarted = false
+    this.monitorPipePath = ''
+  }
+
+  private connectMonitorPipe(): void {
+    if (!this.monitorCallback || !this.monitorPipePath || this.monitorConnectTimer) return
+
+    this.monitorConnectTimer = setTimeout(() => {
+      this.monitorConnectTimer = null
+      if (!this.monitorCallback || !this.monitorPipePath || this.monitorPipeClient) return
+
+      const socket = createConnection(this.monitorPipePath)
+      this.monitorPipeClient = socket
+      let buffer = ''
+
+      socket.on('data', (data) => {
+        const normalizedChunk = data
+          .toString('utf8')
+          .split('\0')
+          .join('\n')
+          .replace(/}\s*{/g, '}\n{')
+        buffer += normalizedChunk
+
+        const lines = buffer.split(/\r?\n/)
+        buffer = lines.pop() || ''
+        for (const line of lines) this.emitMonitorPayload(line)
+
+        const tail = buffer.trim()
+        if (tail.startsWith('{') && tail.endsWith('}')) {
+          try {
+            JSON.parse(tail)
+            this.emitMonitorPayload(tail)
+            buffer = ''
+          } catch {
+            // Keep the partial payload until the next socket chunk arrives.
+          }
+        }
+      })
+
+      socket.on('error', (error) => {
+        console.warn('[WCDB4] 数据库监听管道异常:', error.message)
+      })
+
+      socket.on('close', () => {
+        if (this.monitorPipeClient === socket) this.monitorPipeClient = null
+        this.scheduleMonitorReconnect()
+      })
+    }, 100)
+  }
+
+  private emitMonitorPayload(rawPayload: string): void {
+    const payload = rawPayload.trim()
+    if (!payload || !this.monitorCallback) return
+
+    try {
+      const parsed = JSON.parse(payload) as { action?: string }
+      this.monitorCallback(parsed.action || 'update', payload)
+    } catch {
+      this.monitorCallback('update', payload)
+    }
+  }
+
+  private scheduleMonitorReconnect(): void {
+    if (this.monitorReconnectTimer || !this.monitorCallback || !this.monitorPipePath) return
+    this.monitorReconnectTimer = setTimeout(() => {
+      this.monitorReconnectTimer = null
+      this.connectMonitorPipe()
+    }, 3000)
+  }
+
   getSessions(): Wcdb4Session[] {
     if (this.cachedSessions) return this.cachedSessions
     if (!this.wcdbGetSessions) return []
@@ -768,6 +915,19 @@ export class Wcdb4Client {
       console.warn('[WCDB4] wcdb_get_emoticon_cdn_url symbol unavailable')
       this.wcdbGetEmoticonCdnUrl = null
     }
+
+    try {
+      this.wcdbStartMonitorPipe = lib.func('int32 wcdb_start_monitor_pipe()') as () => number
+      this.wcdbStopMonitorPipe = lib.func('void wcdb_stop_monitor_pipe()') as () => void
+      this.wcdbGetMonitorPipeName = lib.func(
+        'int32 wcdb_get_monitor_pipe_name(_Out_ void** outName)'
+      ) as (outName: WcdbVoidOut) => number
+    } catch {
+      console.warn('[WCDB4] monitor pipe symbols unavailable')
+      this.wcdbStartMonitorPipe = null
+      this.wcdbStopMonitorPipe = null
+      this.wcdbGetMonitorPipeName = null
+    }
   }
 
   private initProtection(lib: KoffiLibrary, libDir: string): void {
@@ -974,7 +1134,7 @@ export class Wcdb4Client {
     ])
 
     return {
-      mesLocalID: localId || `${createTime}-${crypto.randomUUID()}`,
+      mesLocalID: localId || `${createTime}-${this.md5(JSON.stringify(row))}`,
       mesDes: isSend ? 0 : 1,
       messageType: messageType || '1',
       msgCreateTime: String(createTime),

src/main/wechat-db.ts

@@ -364,6 +364,11 @@ export class WechatDb {
     return this.wcdb4Client
   }
 
+  public close(): void {
+    this.wcdb4Client?.close()
+    this.wcdb4Client = null
+  }
+
   public getUserMessages(userMd5: string, startTime?: number, endTime?: number): WechatMessage[] {
     if (this.wcdb4Client) {
       const username = this.chatMd5ToUsername.get(userMd5)

src/preload/index.d.ts

@@ -34,7 +34,9 @@ declare global {
   interface Window {
     electron: ElectronAPI
     api: {
-      initDb: (key: string) => Promise<boolean | { success: boolean; error?: string }>
+      initDb: (
+        key: string
+      ) => Promise<boolean | { success: boolean; error?: string; monitoring?: boolean }>
       getContacts: (filter?: string) => Promise<Contact[]>
       getMessages: (userMd5: string, startTime?: number, endTime?: number) => Promise<Message[]>
       search: (keyword: string) => Promise<string | null>
@@ -72,6 +74,7 @@ declare global {
       }>
       pasteAndSaveDbKey: () => Promise<{ success: boolean; key?: string; error?: string }>
       clearSavedDbKey: () => Promise<{ success: boolean; error?: string }>
+      onWcdbChange: (callback: (payload: { type: string; json: string }) => void) => () => void
       onDbKeyStatus: (callback: (payload: { message: string }) => void) => () => void
     }
   }

src/preload/index.ts

@@ -28,6 +28,14 @@ const api = {
   autoGetDbKey: () => ipcRenderer.invoke('key:autoGetDbKey'),
   pasteAndSaveDbKey: () => ipcRenderer.invoke('key:pasteAndSaveDbKey'),
   clearSavedDbKey: () => ipcRenderer.invoke('key:clearSavedDbKey'),
+  onWcdbChange: (callback: (payload: { type: string; json: string }) => void) => {
+    const listener = (
+      _event: Electron.IpcRendererEvent,
+      payload: { type: string; json: string }
+    ): void => callback(payload)
+    ipcRenderer.on('wcdb-change', listener)
+    return () => ipcRenderer.removeListener('wcdb-change', listener)
+  },
   onDbKeyStatus: (callback: (payload: { message: string }) => void) => {
     const listener = (_event: Electron.IpcRendererEvent, payload: { message: string }): void =>
       callback(payload)

src/renderer/src/App.tsx

@@ -4,6 +4,22 @@ import ChatWindow from './components/ChatWindow'
 import { Contact, Message } from '../../shared/types'
 
 const MAC_KEY_FAQ_URL = 'https://github.com/hicccc77/WeFlow/blob/main/docs/MAC-KEY-FAQ.md'
+const MESSAGE_MONITOR_DEBOUNCE_MS = 250
+
+const getMessageIdentity = (message: Message): string => {
+  if (message.localId) return `local:${message.localId}`
+  if (message.id) return `id:${message.id}`
+  return `${message.createTime || 0}:${message.from}:${message.type}:${message.content}`
+}
+
+const areMessagesEquivalent = (left: Message[], right: Message[]): boolean => {
+  if (left === right) return true
+  if (left.length !== right.length) return false
+  for (let index = 0; index < left.length; index += 1) {
+    if (getMessageIdentity(left[index]) !== getMessageIdentity(right[index])) return false
+  }
+  return true
+}
 
 function App(): React.ReactElement {
   const [isAuthenticated, setIsAuthenticated] = useState(false)
@@ -19,6 +35,7 @@ function App(): React.ReactElement {
   const [dbKeyStatusKind, setDbKeyStatusKind] = useState<'normal' | 'success' | 'error'>('normal')
   const [showDbKey, setShowDbKey] = useState(false)
   const [showMacKeyFaq, setShowMacKeyFaq] = useState(false)
+  const [isNativeMonitorActive, setIsNativeMonitorActive] = useState(false)
 
   React.useEffect(() => {
     let active = true
@@ -54,6 +71,7 @@ function App(): React.ReactElement {
       const result = await window.api.initDb(keyToUse)
       const success = typeof result === 'boolean' ? result : result.success
       if (success) {
+        setIsNativeMonitorActive(typeof result !== 'boolean' && result.monitoring === true)
         setIsAuthenticated(true)
         loadContacts()
       } else {
@@ -162,19 +180,52 @@ function App(): React.ReactElement {
 
   const handleDateRangeChange = (range: string): void => {
     setDateRange(range)
-    // 如果选择了联系人，则使用新范围重新加载消息
     if (selectedContact) {
-      // 需要调用 handleSelectContact，但它需要一个联系人对象。
-      // 由于状态更新是异步的，可能需要使用状态中的当前联系人，
-      // 此函数内部 'selectedContact' 可从闭包中获得。
-      // 但是，需要确保 'dateRange' 已更新。
-      // 实际上，就在这里使用新范围手动触发获取。
-
       const { startTime, endTime } = getDateRangeParams(range)
       window.api.getMessages(selectedContact.md5, startTime, endTime).then(setMessages)
     }
   }
 
+  React.useEffect(() => {
+    if (!isAuthenticated || !selectedContact || !isNativeMonitorActive) return
+
+    let disposed = false
+    let refreshTimer: number | null = null
+    const contactMd5 = selectedContact.md5
+
+    const refreshCurrentConversation = async (): Promise<void> => {
+      try {
+        const range = getDateRangeParams(dateRange)
+        const latestMessages = await window.api.getMessages(
+          contactMd5,
+          range.startTime,
+          range.endTime
+        )
+        if (!disposed) {
+          setMessages((current) =>
+            areMessagesEquivalent(current, latestMessages) ? current : latestMessages
+          )
+        }
+      } catch (error) {
+        console.warn('[MessageMonitor] 刷新当前会话失败:', error)
+      }
+    }
+
+    const unsubscribe = window.api.onWcdbChange(() => {
+      if (refreshTimer) window.clearTimeout(refreshTimer)
+      refreshTimer = window.setTimeout(() => {
+        refreshTimer = null
+        void refreshCurrentConversation()
+      }, MESSAGE_MONITOR_DEBOUNCE_MS)
+    })
+
+    return () => {
+      disposed = true
+      if (refreshTimer) window.clearTimeout(refreshTimer)
+      unsubscribe()
+    }
+  }, [dateRange, isAuthenticated, isNativeMonitorActive, selectedContact])
+
   const handleSearchContacts = (keyword: string): void => {
     if (!keyword) {
       setFilteredContacts(contacts)