feat: 实现朋友圈获取; 实现聊天页面跳转到指定日期

2026-03-25 07:16:51 +00:00 · 2026-01-23 00:13:55 +08:00
parent 07e7bce6a9
commit a5e1bfe49a
19 changed files with 1742 additions and 62 deletions
--- a/electron/services/chatService.ts
+++ b/electron/services/chatService.ts
@@ -74,7 +74,7 @@ const emojiDownloading: Map<string, Promise<string | null>> = new Map()
 class ChatService {
  private configService: ConfigService
  private connected = false
-  private messageCursors: Map<string, { cursor: number; fetched: number; batchSize: number }> = new Map()
+  private messageCursors: Map<string, { cursor: number; fetched: number; batchSize: number; startTime?: number; endTime?: number; ascending?: boolean }> = new Map()
  private readonly messageBatchDefault = 50
  private avatarCache: Map<string, ContactCacheEntry>
  private readonly avatarCacheTtlMs = 10 * 60 * 1000
@@ -430,9 +430,9 @@ class ChatService {
      }

      if (!headImageDbPath) {
-        console.log('[头像加载] 未找到 head_image.db', { 
-          accountDir, 
-          checkedPaths: headImageDbPaths 
+        console.log('[头像加载] 未找到 head_image.db', {
+          accountDir,
+          checkedPaths: headImageDbPaths
        })
        return result
      }
@@ -444,7 +444,7 @@ class ChatService {

      try {
        const stmt = db.prepare('SELECT username, image_buffer FROM head_image WHERE username = ?')
-        
+
        for (const username of usernames) {
          try {
            const row = stmt.get(username) as any
@@ -454,9 +454,9 @@ class ChatService {
              result[username] = `data:image/jpeg;base64,${base64}`
            } else {
              // 只输出没有找到头像的
-              console.log(`[头像加载] 未找到头像: ${username}`, { 
-                hasRow: !!row, 
-                hasBuffer: row ? !!row.image_buffer : false 
+              console.log(`[头像加载] 未找到头像: ${username}`, {
+                hasRow: !!row,
+                hasBuffer: row ? !!row.image_buffer : false
              })
            }
          } catch (e) {
@@ -501,7 +501,10 @@ class ChatService {
  async getMessages(
    sessionId: string,
    offset: number = 0,
-    limit: number = 50
+    limit: number = 50,
+    startTime: number = 0,
+    endTime: number = 0,
+    ascending: boolean = false
  ): Promise<{ success: boolean; messages?: Message[]; hasMore?: boolean; error?: string }> {
    try {
      const connectResult = await this.ensureConnected()
@@ -516,7 +519,14 @@ class ChatService {
      // 1. 没有游标状态
      // 2. offset 为 0 (重新加载会话)
      // 3. batchSize 改变
-      const needNewCursor = !state || offset === 0 || state.batchSize !== batchSize
+      // 4. startTime 改变
+      // 5. ascending 改变
+      const needNewCursor = !state ||
+        offset === 0 ||
+        state.batchSize !== batchSize ||
+        state.startTime !== startTime ||
+        state.endTime !== endTime ||
+        state.ascending !== ascending

      if (needNewCursor) {
        // 关闭旧游标
@@ -529,13 +539,16 @@ class ChatService {
        }

        // 创建新游标
-        const cursorResult = await wcdbService.openMessageCursor(sessionId, batchSize, false, 0, 0)
+        // 注意：WeFlow 数据库中的 create_time 是以秒为单位的
+        const beginTimestamp = startTime > 10000000000 ? Math.floor(startTime / 1000) : startTime
+        const endTimestamp = endTime > 10000000000 ? Math.floor(endTime / 1000) : endTime
+        const cursorResult = await wcdbService.openMessageCursor(sessionId, batchSize, ascending, beginTimestamp, endTimestamp)
        if (!cursorResult.success || !cursorResult.cursor) {
          console.error('[ChatService] 打开消息游标失败:', cursorResult.error)
          return { success: false, error: cursorResult.error || '打开消息游标失败' }
        }

-        state = { cursor: cursorResult.cursor, fetched: 0, batchSize }
+        state = { cursor: cursorResult.cursor, fetched: 0, batchSize, startTime, endTime, ascending }
        this.messageCursors.set(sessionId, state)

        // 如果需要跳过消息(offset > 0),逐批获取但不返回
--- a/electron/services/snsService.ts
+++ b/electron/services/snsService.ts
@@ -0,0 +1,64 @@
+import { wcdbService } from './wcdbService'
+import { ConfigService } from './config'
+import { ContactCacheService } from './contactCacheService'
+
+export interface SnsPost {
+    id: string
+    username: string
+    nickname: string
+    avatarUrl?: string
+    createTime: number
+    contentDesc: string
+    type?: number
+    media: { url: string; thumb: string }[]
+    likes: string[]
+    comments: { id: string; nickname: string; content: string; refCommentId: string; refNickname?: string }[]
+}
+
+class SnsService {
+    private contactCache: ContactCacheService
+
+    constructor() {
+        const config = new ConfigService()
+        this.contactCache = new ContactCacheService(config.get('cachePath') as string)
+    }
+
+    async getTimeline(limit: number = 20, offset: number = 0, usernames?: string[], keyword?: string, startTime?: number, endTime?: number): Promise<{ success: boolean; timeline?: SnsPost[]; error?: string }> {
+        console.log('[SnsService] getTimeline called with:', { limit, offset, usernames, keyword, startTime, endTime })
+
+        const result = await wcdbService.getSnsTimeline(limit, offset, usernames, keyword, startTime, endTime)
+
+        console.log('[SnsService] getSnsTimeline result:', {
+            success: result.success,
+            timelineCount: result.timeline?.length,
+            error: result.error
+        })
+
+        if (result.success && result.timeline) {
+            const enrichedTimeline = result.timeline.map((post: any) => {
+                const contact = this.contactCache.get(post.username)
+
+                // 修复媒体 URL，如果是 http 则尝试用 https (虽然 qpic 可能不支持强制 https，但通常支持)
+                const fixedMedia = post.media.map((m: any) => ({
+                    url: m.url.replace('http://', 'https://'),
+                    thumb: m.thumb.replace('http://', 'https://')
+                }))
+
+                return {
+                    ...post,
+                    avatarUrl: contact?.avatarUrl,
+                    nickname: post.nickname || contact?.displayName || post.username,
+                    media: fixedMedia
+                }
+            })
+
+            console.log('[SnsService] Returning enriched timeline with', enrichedTimeline.length, 'posts')
+            return { ...result, timeline: enrichedTimeline }
+        }
+
+        console.log('[SnsService] Returning result:', result)
+        return result
+    }
+}
+
+export const snsService = new SnsService()
--- a/electron/services/wcdbCore.ts
+++ b/electron/services/wcdbCore.ts
@@ -55,6 +55,7 @@ export class WcdbCore {
  private wcdbGetEmoticonCdnUrl: any = null
  private wcdbGetDbStatus: any = null
  private wcdbGetVoiceData: any = null
+  private wcdbGetSnsTimeline: any = null
  private avatarUrlCache: Map<string, { url?: string; updatedAt: number }> = new Map()
  private readonly avatarCacheTtlMs = 10 * 60 * 1000
  private logTimer: NodeJS.Timeout | null = null
@@ -116,7 +117,8 @@ export class WcdbCore {
  private writeLog(message: string, force = false): void {
    if (!force && !this.isLogEnabled()) return
    const line = `[${new Date().toISOString()}] ${message}`
-    // 移除控制台日志，只写入文件
+    // 同时输出到控制台和文件
+    console.log('[WCDB]', message)
    try {
      const base = this.userDataPath || process.env.WCDB_LOG_DIR || process.cwd()
      const dir = join(base, 'logs')
@@ -385,6 +387,13 @@ export class WcdbCore {
        this.wcdbGetVoiceData = null
      }

+      // wcdb_status wcdb_get_sns_timeline(wcdb_handle handle, int32_t limit, int32_t offset, const char* username, const char* keyword, int32_t start_time, int32_t end_time, char** out_json)
+      try {
+        this.wcdbGetSnsTimeline = this.lib.func('int32 wcdb_get_sns_timeline(int64 handle, int32 limit, int32 offset, const char* username, const char* keyword, int32 startTime, int32 endTime, _Out_ void** outJson)')
+      } catch {
+        this.wcdbGetSnsTimeline = null
+      }
+
      // 初始化
      const initResult = this.wcdbInit()
      if (initResult !== 0) {
@@ -401,8 +410,8 @@ export class WcdbCore {
      this.writeLog(`WCDB 初始化异常: ${errorMsg}`, true)
      lastDllInitError = errorMsg
      // 检查是否是常见的 VC++ 运行时缺失错误
-      if (errorMsg.includes('126') || errorMsg.includes('找不到指定的模块') || 
-          errorMsg.includes('The specified module could not be found')) {
+      if (errorMsg.includes('126') || errorMsg.includes('找不到指定的模块') ||
+        errorMsg.includes('The specified module could not be found')) {
        lastDllInitError = '可能缺少 Visual C++ 运行时库。请安装 Microsoft Visual C++ Redistributable (x64)。'
      } else if (errorMsg.includes('193') || errorMsg.includes('不是有效的 Win32 应用程序')) {
        lastDllInitError = 'DLL 架构不匹配。请确保使用 64 位版本的应用程序。'
@@ -1388,4 +1397,32 @@ export class WcdbCore {
      return { success: false, error: String(e) }
    }
  }
+
+  async getSnsTimeline(limit: number, offset: number, usernames?: string[], keyword?: string, startTime?: number, endTime?: number): Promise<{ success: boolean; timeline?: any[]; error?: string }> {
+    if (!this.ensureReady()) return { success: false, error: 'WCDB 未连接' }
+    if (!this.wcdbGetSnsTimeline) return { success: false, error: '当前 DLL 版本不支持获取朋友圈' }
+    try {
+      const outPtr = [null as any]
+      const usernamesJson = usernames && usernames.length > 0 ? JSON.stringify(usernames) : ''
+      const result = this.wcdbGetSnsTimeline(
+        this.handle,
+        limit,
+        offset,
+        usernamesJson,
+        keyword || '',
+        startTime || 0,
+        endTime || 0,
+        outPtr
+      )
+      if (result !== 0 || !outPtr[0]) {
+        return { success: false, error: `获取朋友圈失败: ${result}` }
+      }
+      const jsonStr = this.decodeJsonPtr(outPtr[0])
+      if (!jsonStr) return { success: false, error: '解析朋友圈数据失败' }
+      const timeline = JSON.parse(jsonStr)
+      return { success: true, timeline }
+    } catch (e) {
+      return { success: false, error: String(e) }
+    }
+  }
 }
--- a/electron/services/wcdbService.ts
+++ b/electron/services/wcdbService.ts
@@ -362,6 +362,13 @@ export class WcdbService {
    return this.callWorker('getVoiceData', { sessionId, createTime, candidates, localId, svrId })
  }

+  /**
+   * 获取朋友圈
+   */
+  async getSnsTimeline(limit: number, offset: number, usernames?: string[], keyword?: string, startTime?: number, endTime?: number): Promise<{ success: boolean; timeline?: any[]; error?: string }> {
+    return this.callWorker('getSnsTimeline', { limit, offset, usernames, keyword, startTime, endTime })
+  }
+
 }

 export const wcdbService = new WcdbService()