import { join, dirname, basename } from 'path' import { appendFileSync, existsSync, mkdirSync, readdirSync, statSync, readFileSync } from 'fs' // DLL 初始化错误信息,用于帮助用户诊断问题 let lastDllInitError: string | null = null export function getLastDllInitError(): string | null { return lastDllInitError } export class WcdbCore { private resourcesPath: string | null = null private userDataPath: string | null = null private logEnabled = false private lib: any = null private koffi: any = null private initialized = false private handle: number | null = null private currentPath: string | null = null private currentKey: string | null = null private currentWxid: string | null = null // 函数引用 private wcdbInitProtection: any = null private wcdbInit: any = null private wcdbShutdown: any = null private wcdbOpenAccount: any = null private wcdbCloseAccount: any = null private wcdbSetMyWxid: any = null private wcdbFreeString: any = null private wcdbUpdateMessage: any = null private wcdbDeleteMessage: any = null private wcdbGetSessions: any = null private wcdbGetMessages: any = null private wcdbGetMessageCount: any = null private wcdbGetDisplayNames: any = null private wcdbGetAvatarUrls: any = null private wcdbGetGroupMemberCount: any = null private wcdbGetGroupMemberCounts: any = null private wcdbGetGroupMembers: any = null private wcdbGetGroupNicknames: any = null private wcdbGetMessageTables: any = null private wcdbGetMessageMeta: any = null private wcdbGetContact: any = null private wcdbGetMessageTableStats: any = null private wcdbGetAggregateStats: any = null private wcdbGetAvailableYears: any = null private wcdbGetAnnualReportStats: any = null private wcdbGetAnnualReportExtras: any = null private wcdbGetDualReportStats: any = null private wcdbGetGroupStats: any = null private wcdbGetMessageDates: any = null private wcdbOpenMessageCursor: any = null private wcdbOpenMessageCursorLite: any = null private wcdbFetchMessageBatch: any = null private wcdbCloseMessageCursor: any = null private wcdbGetLogs: any = null private wcdbExecQuery: any = null private wcdbListMessageDbs: any = null private wcdbListMediaDbs: any = null private wcdbGetMessageById: any = null private wcdbGetEmoticonCdnUrl: any = null private wcdbGetDbStatus: any = null private wcdbGetVoiceData: any = null private wcdbGetSnsTimeline: any = null private wcdbGetSnsAnnualStats: any = null private wcdbVerifyUser: any = null private wcdbStartMonitorPipe: any = null private wcdbStopMonitorPipe: any = null private wcdbGetMonitorPipeName: any = null private monitorPipeClient: any = null private monitorCallback: ((type: string, json: string) => void) | null = null private monitorReconnectTimer: any = null private monitorPipePath: string = '' private avatarUrlCache: Map = new Map() private readonly avatarCacheTtlMs = 10 * 60 * 1000 private logTimer: NodeJS.Timeout | null = null private lastLogTail: string | null = null setPaths(resourcesPath: string, userDataPath: string): void { this.resourcesPath = resourcesPath this.userDataPath = userDataPath } setLogEnabled(enabled: boolean): void { this.logEnabled = enabled if (this.isLogEnabled() && this.initialized) { this.startLogPolling() } else { this.stopLogPolling() } } // 使用命名管道 IPC startMonitor(callback: (type: string, json: string) => void): boolean { if (!this.wcdbStartMonitorPipe) { return false } this.monitorCallback = callback try { const result = this.wcdbStartMonitorPipe() if (result !== 0) { return false } // 从 DLL 获取动态管道名(含 PID) let pipePath = '\\\\.\\pipe\\weflow_monitor' if (this.wcdbGetMonitorPipeName) { try { const namePtr = [null as any] if (this.wcdbGetMonitorPipeName(namePtr) === 0 && namePtr[0]) { pipePath = this.koffi.decode(namePtr[0], 'char', -1) this.wcdbFreeString(namePtr[0]) } } catch {} } this.connectMonitorPipe(pipePath) return true } catch (e) { console.error('[wcdbCore] startMonitor exception:', e) return false } } // 连接命名管道,支持断开后自动重连 private connectMonitorPipe(pipePath: string) { this.monitorPipePath = pipePath const net = require('net') setTimeout(() => { if (!this.monitorCallback) return this.monitorPipeClient = net.createConnection(this.monitorPipePath, () => { }) let buffer = '' this.monitorPipeClient.on('data', (data: Buffer) => { buffer += data.toString('utf8') const lines = buffer.split('\n') buffer = lines.pop() || '' for (const line of lines) { if (line.trim()) { try { const parsed = JSON.parse(line) this.monitorCallback?.(parsed.action || 'update', line) } catch { this.monitorCallback?.('update', line) } } } }) this.monitorPipeClient.on('error', () => { }) this.monitorPipeClient.on('close', () => { this.monitorPipeClient = null this.scheduleReconnect() }) }, 100) } // 定时重连 private scheduleReconnect() { if (this.monitorReconnectTimer || !this.monitorCallback) return this.monitorReconnectTimer = setTimeout(() => { this.monitorReconnectTimer = null if (this.monitorCallback && !this.monitorPipeClient) { this.connectMonitorPipe(this.monitorPipePath) } }, 3000) } stopMonitor(): void { this.monitorCallback = null if (this.monitorReconnectTimer) { clearTimeout(this.monitorReconnectTimer) this.monitorReconnectTimer = null } if (this.monitorPipeClient) { this.monitorPipeClient.destroy() this.monitorPipeClient = null } if (this.wcdbStopMonitorPipe) { this.wcdbStopMonitorPipe() } } // 保留旧方法签名以兼容 setMonitor(callback: (type: string, json: string) => void): boolean { return this.startMonitor(callback) } /** * 获取 DLL 路径 */ private getDllPath(): string { const envDllPath = process.env.WCDB_DLL_PATH if (envDllPath && envDllPath.length > 0) { return envDllPath } // 基础路径探测 const isPackaged = typeof process['resourcesPath'] !== 'undefined' const resourcesPath = isPackaged ? process.resourcesPath : join(process.cwd(), 'resources') const candidates = [ // 环境变量指定 resource 目录 process.env.WCDB_RESOURCES_PATH ? join(process.env.WCDB_RESOURCES_PATH, 'wcdb_api.dll') : null, // 显式 setPaths 设置的路径 this.resourcesPath ? join(this.resourcesPath, 'wcdb_api.dll') : null, // text/resources/wcdb_api.dll (打包常见结构) join(resourcesPath, 'resources', 'wcdb_api.dll'), // items/resourcesPath/wcdb_api.dll (扁平结构) join(resourcesPath, 'wcdb_api.dll'), // CWD fallback join(process.cwd(), 'resources', 'wcdb_api.dll') ].filter(Boolean) as string[] for (const path of candidates) { if (existsSync(path)) return path } return candidates[0] || 'wcdb_api.dll' } private isLogEnabled(): boolean { // 移除 Worker 线程的日志禁用逻辑,允许在 Worker 中记录日志 if (process.env.WCDB_LOG_ENABLED === '1') return true return this.logEnabled } private writeLog(message: string, force = false): void { if (!force && !this.isLogEnabled()) return const line = `[${new Date().toISOString()}] ${message}` // 同时输出到控制台和文件 try { const base = this.userDataPath || process.env.WCDB_LOG_DIR || process.cwd() const dir = join(base, 'logs') if (!existsSync(dir)) mkdirSync(dir, { recursive: true }) appendFileSync(join(dir, 'wcdb.log'), line + '\n', { encoding: 'utf8' }) } catch { } } /** * 递归查找 session.db 文件 */ private findSessionDb(dir: string, depth = 0): string | null { if (depth > 5) return null try { const entries = readdirSync(dir) for (const entry of entries) { if (entry.toLowerCase() === 'session.db') { const fullPath = join(dir, entry) if (statSync(fullPath).isFile()) { return fullPath } } } for (const entry of entries) { const fullPath = join(dir, entry) try { if (statSync(fullPath).isDirectory()) { const found = this.findSessionDb(fullPath, depth + 1) if (found) return found } } catch { } } } catch (e) { console.error('查找 session.db 失败:', e) } return null } private resolveDbStoragePath(basePath: string, wxid: string): string | null { if (!basePath) return null const normalized = basePath.replace(/[\\\\/]+$/, '') if (normalized.toLowerCase().endsWith('db_storage') && existsSync(normalized)) { return normalized } const direct = join(normalized, 'db_storage') if (existsSync(direct)) { return direct } if (wxid) { const viaWxid = join(normalized, wxid, 'db_storage') if (existsSync(viaWxid)) { return viaWxid } // 兼容目录名包含额外后缀(如 wxid_xxx_1234) try { const entries = readdirSync(normalized) const lowerWxid = wxid.toLowerCase() const candidates = entries.filter((entry) => { const entryPath = join(normalized, entry) try { if (!statSync(entryPath).isDirectory()) return false } catch { return false } const lowerEntry = entry.toLowerCase() return lowerEntry === lowerWxid || lowerEntry.startsWith(`${lowerWxid}_`) }) for (const entry of candidates) { const candidate = join(normalized, entry, 'db_storage') if (existsSync(candidate)) { return candidate } } } catch { } } return null } /** * 初始化 WCDB */ async initialize(): Promise { if (this.initialized) return true try { this.koffi = require('koffi') const dllPath = this.getDllPath() if (!existsSync(dllPath)) { console.error('WCDB DLL 不存在:', dllPath) return false } const dllDir = dirname(dllPath) const wcdbCorePath = join(dllDir, 'WCDB.dll') if (existsSync(wcdbCorePath)) { try { this.koffi.load(wcdbCorePath) this.writeLog('预加载 WCDB.dll 成功') } catch (e) { console.warn('预加载 WCDB.dll 失败(可能不是致命的):', e) this.writeLog(`预加载 WCDB.dll 失败: ${String(e)}`) } } const sdl2Path = join(dllDir, 'SDL2.dll') if (existsSync(sdl2Path)) { try { this.koffi.load(sdl2Path) this.writeLog('预加载 SDL2.dll 成功') } catch (e) { console.warn('预加载 SDL2.dll 失败(可能不是致命的):', e) this.writeLog(`预加载 SDL2.dll 失败: ${String(e)}`) } } this.lib = this.koffi.load(dllPath) // InitProtection (Added for security) try { this.wcdbInitProtection = this.lib.func('bool InitProtection(const char* resourcePath)') // 尝试多个可能的资源路径 const resourcePaths = [ dllDir, // DLL 所在目录 dirname(dllDir), // 上级目录 this.resourcesPath, // 配置的资源路径 join(process.cwd(), 'resources') // 开发环境 ].filter(Boolean) let protectionOk = false for (const resPath of resourcePaths) { try { // protectionOk = this.wcdbInitProtection(resPath) if (protectionOk) { // break } } catch (e) { // console.warn(`[WCDB] InitProtection 失败 (${resPath}):`, e) } } if (!protectionOk) { // console.warn('[WCDB] Core security check failed - 继续运行但可能不稳定') // this.writeLog('InitProtection 失败,继续运行') // 不返回 false,允许继续运行 } } catch (e) { // console.warn('InitProtection symbol not found:', e) } // 定义类型 // wcdb_status wcdb_init() this.wcdbInit = this.lib.func('int32 wcdb_init()') // wcdb_status wcdb_shutdown() this.wcdbShutdown = this.lib.func('int32 wcdb_shutdown()') // wcdb_status wcdb_open_account(const char* session_db_path, const char* hex_key, wcdb_handle* out_handle) // wcdb_handle 是 int64_t this.wcdbOpenAccount = this.lib.func('int32 wcdb_open_account(const char* path, const char* key, _Out_ int64* handle)') // wcdb_status wcdb_close_account(wcdb_handle handle) // C 接口是 int64, koffi 返回 handle 是 number 类型 this.wcdbCloseAccount = this.lib.func('int32 wcdb_close_account(int64 handle)') // wcdb_status wcdb_set_my_wxid(wcdb_handle handle, const char* wxid) try { this.wcdbSetMyWxid = this.lib.func('int32 wcdb_set_my_wxid(int64 handle, const char* wxid)') } catch { this.wcdbSetMyWxid = null } // wcdb_status wcdb_update_message(wcdb_handle handle, const char* session_id, int64_t local_id, int32_t create_time, const char* new_content, char** out_error) try { this.wcdbUpdateMessage = this.lib.func('int32 wcdb_update_message(int64 handle, const char* sessionId, int64 localId, int32 createTime, const char* newContent, _Out_ void** outError)') } catch { this.wcdbUpdateMessage = null } // wcdb_status wcdb_delete_message(wcdb_handle handle, const char* session_id, int64_t local_id, char** out_error) try { this.wcdbDeleteMessage = this.lib.func('int32 wcdb_delete_message(int64 handle, const char* sessionId, int64 localId, int32 createTime, const char* dbPathHint, _Out_ void** outError)') } catch { this.wcdbDeleteMessage = null } // void wcdb_free_string(char* ptr) this.wcdbFreeString = this.lib.func('void wcdb_free_string(void* ptr)') // wcdb_status wcdb_get_sessions(wcdb_handle handle, char** out_json) this.wcdbGetSessions = this.lib.func('int32 wcdb_get_sessions(int64 handle, _Out_ void** outJson)') // wcdb_status wcdb_get_messages(wcdb_handle handle, const char* username, int32_t limit, int32_t offset, char** out_json) this.wcdbGetMessages = this.lib.func('int32 wcdb_get_messages(int64 handle, const char* username, int32 limit, int32 offset, _Out_ void** outJson)') // wcdb_status wcdb_get_message_count(wcdb_handle handle, const char* username, int32_t* out_count) this.wcdbGetMessageCount = this.lib.func('int32 wcdb_get_message_count(int64 handle, const char* username, _Out_ int32* outCount)') // wcdb_status wcdb_get_display_names(wcdb_handle handle, const char* usernames_json, char** out_json) this.wcdbGetDisplayNames = this.lib.func('int32 wcdb_get_display_names(int64 handle, const char* usernamesJson, _Out_ void** outJson)') // wcdb_status wcdb_get_avatar_urls(wcdb_handle handle, const char* usernames_json, char** out_json) this.wcdbGetAvatarUrls = this.lib.func('int32 wcdb_get_avatar_urls(int64 handle, const char* usernamesJson, _Out_ void** outJson)') // wcdb_status wcdb_get_group_member_count(wcdb_handle handle, const char* chatroom_id, int32_t* out_count) this.wcdbGetGroupMemberCount = this.lib.func('int32 wcdb_get_group_member_count(int64 handle, const char* chatroomId, _Out_ int32* outCount)') // wcdb_status wcdb_get_group_member_counts(wcdb_handle handle, const char* chatroom_ids_json, char** out_json) try { this.wcdbGetGroupMemberCounts = this.lib.func('int32 wcdb_get_group_member_counts(int64 handle, const char* chatroomIdsJson, _Out_ void** outJson)') } catch { this.wcdbGetGroupMemberCounts = null } // wcdb_status wcdb_get_group_members(wcdb_handle handle, const char* chatroom_id, char** out_json) this.wcdbGetGroupMembers = this.lib.func('int32 wcdb_get_group_members(int64 handle, const char* chatroomId, _Out_ void** outJson)') // wcdb_status wcdb_get_group_nicknames(wcdb_handle handle, const char* chatroom_id, char** out_json) try { this.wcdbGetGroupNicknames = this.lib.func('int32 wcdb_get_group_nicknames(int64 handle, const char* chatroomId, _Out_ void** outJson)') } catch { this.wcdbGetGroupNicknames = null } // wcdb_status wcdb_get_message_tables(wcdb_handle handle, const char* session_id, char** out_json) this.wcdbGetMessageTables = this.lib.func('int32 wcdb_get_message_tables(int64 handle, const char* sessionId, _Out_ void** outJson)') // wcdb_status wcdb_get_message_meta(wcdb_handle handle, const char* db_path, const char* table_name, int32_t limit, int32_t offset, char** out_json) this.wcdbGetMessageMeta = this.lib.func('int32 wcdb_get_message_meta(int64 handle, const char* dbPath, const char* tableName, int32 limit, int32 offset, _Out_ void** outJson)') // wcdb_status wcdb_get_contact(wcdb_handle handle, const char* username, char** out_json) this.wcdbGetContact = this.lib.func('int32 wcdb_get_contact(int64 handle, const char* username, _Out_ void** outJson)') // wcdb_status wcdb_get_message_table_stats(wcdb_handle handle, const char* session_id, char** out_json) this.wcdbGetMessageTableStats = this.lib.func('int32 wcdb_get_message_table_stats(int64 handle, const char* sessionId, _Out_ void** outJson)') // wcdb_status wcdb_get_aggregate_stats(wcdb_handle handle, const char* session_ids_json, int32_t begin_timestamp, int32_t end_timestamp, char** out_json) this.wcdbGetAggregateStats = this.lib.func('int32 wcdb_get_aggregate_stats(int64 handle, const char* sessionIdsJson, int32 begin, int32 end, _Out_ void** outJson)') // wcdb_status wcdb_get_available_years(wcdb_handle handle, const char* session_ids_json, char** out_json) try { this.wcdbGetAvailableYears = this.lib.func('int32 wcdb_get_available_years(int64 handle, const char* sessionIdsJson, _Out_ void** outJson)') } catch { this.wcdbGetAvailableYears = null } // wcdb_status wcdb_get_annual_report_stats(wcdb_handle handle, const char* session_ids_json, int32_t begin_timestamp, int32_t end_timestamp, char** out_json) try { this.wcdbGetAnnualReportStats = this.lib.func('int32 wcdb_get_annual_report_stats(int64 handle, const char* sessionIdsJson, int32 begin, int32 end, _Out_ void** outJson)') } catch { this.wcdbGetAnnualReportStats = null } // wcdb_status wcdb_get_annual_report_extras(wcdb_handle handle, const char* session_ids_json, int32_t begin_timestamp, int32_t end_timestamp, int32_t peak_day_begin, int32_t peak_day_end, char** out_json) try { this.wcdbGetAnnualReportExtras = this.lib.func('int32 wcdb_get_annual_report_extras(int64 handle, const char* sessionIdsJson, int32 begin, int32 end, int32 peakBegin, int32 peakEnd, _Out_ void** outJson)') } catch { this.wcdbGetAnnualReportExtras = null } // wcdb_status wcdb_get_dual_report_stats(wcdb_handle handle, const char* session_id, int32_t begin_timestamp, int32_t end_timestamp, char** out_json) try { this.wcdbGetDualReportStats = this.lib.func('int32 wcdb_get_dual_report_stats(int64 handle, const char* sessionId, int32 begin, int32 end, _Out_ void** outJson)') } catch { this.wcdbGetDualReportStats = null } // wcdb_status wcdb_get_logs(char** out_json) try { this.wcdbGetLogs = this.lib.func('int32 wcdb_get_logs(_Out_ void** outJson)') } catch { this.wcdbGetLogs = null } // wcdb_status wcdb_get_group_stats(wcdb_handle handle, const char* chatroom_id, int32_t begin_timestamp, int32_t end_timestamp, char** out_json) try { this.wcdbGetGroupStats = this.lib.func('int32 wcdb_get_group_stats(int64 handle, const char* chatroomId, int32 begin, int32 end, _Out_ void** outJson)') } catch { this.wcdbGetGroupStats = null } // wcdb_status wcdb_get_message_dates(wcdb_handle handle, const char* session_id, char** out_json) try { this.wcdbGetMessageDates = this.lib.func('int32 wcdb_get_message_dates(int64 handle, const char* sessionId, _Out_ void** outJson)') } catch { this.wcdbGetMessageDates = null } // wcdb_status wcdb_open_message_cursor(wcdb_handle handle, const char* session_id, int32_t batch_size, int32_t ascending, int32_t begin_timestamp, int32_t end_timestamp, wcdb_cursor* out_cursor) this.wcdbOpenMessageCursor = this.lib.func('int32 wcdb_open_message_cursor(int64 handle, const char* sessionId, int32 batchSize, int32 ascending, int32 beginTimestamp, int32 endTimestamp, _Out_ int64* outCursor)') // wcdb_status wcdb_open_message_cursor_lite(wcdb_handle handle, const char* session_id, int32_t batch_size, int32_t ascending, int32_t begin_timestamp, int32_t end_timestamp, wcdb_cursor* out_cursor) try { this.wcdbOpenMessageCursorLite = this.lib.func('int32 wcdb_open_message_cursor_lite(int64 handle, const char* sessionId, int32 batchSize, int32 ascending, int32 beginTimestamp, int32 endTimestamp, _Out_ int64* outCursor)') } catch { this.wcdbOpenMessageCursorLite = null } // wcdb_status wcdb_fetch_message_batch(wcdb_handle handle, wcdb_cursor cursor, char** out_json, int32_t* out_has_more) this.wcdbFetchMessageBatch = this.lib.func('int32 wcdb_fetch_message_batch(int64 handle, int64 cursor, _Out_ void** outJson, _Out_ int32* outHasMore)') // wcdb_status wcdb_close_message_cursor(wcdb_handle handle, wcdb_cursor cursor) this.wcdbCloseMessageCursor = this.lib.func('int32 wcdb_close_message_cursor(int64 handle, int64 cursor)') // wcdb_status wcdb_get_logs(char** out_json) this.wcdbGetLogs = this.lib.func('int32 wcdb_get_logs(_Out_ void** outJson)') // wcdb_status wcdb_exec_query(wcdb_handle handle, const char* db_kind, const char* db_path, const char* sql, char** out_json) this.wcdbExecQuery = this.lib.func('int32 wcdb_exec_query(int64 handle, const char* kind, const char* path, const char* sql, _Out_ void** outJson)') // wcdb_status wcdb_get_emoticon_cdn_url(wcdb_handle handle, const char* db_path, const char* md5, char** out_url) this.wcdbGetEmoticonCdnUrl = this.lib.func('int32 wcdb_get_emoticon_cdn_url(int64 handle, const char* dbPath, const char* md5, _Out_ void** outUrl)') // wcdb_status wcdb_list_message_dbs(wcdb_handle handle, char** out_json) this.wcdbListMessageDbs = this.lib.func('int32 wcdb_list_message_dbs(int64 handle, _Out_ void** outJson)') // wcdb_status wcdb_list_media_dbs(wcdb_handle handle, char** out_json) this.wcdbListMediaDbs = this.lib.func('int32 wcdb_list_media_dbs(int64 handle, _Out_ void** outJson)') // wcdb_status wcdb_get_message_by_id(wcdb_handle handle, const char* session_id, int32 local_id, char** out_json) this.wcdbGetMessageById = this.lib.func('int32 wcdb_get_message_by_id(int64 handle, const char* sessionId, int32 localId, _Out_ void** outJson)') // wcdb_status wcdb_get_db_status(wcdb_handle handle, char** out_json) try { this.wcdbGetDbStatus = this.lib.func('int32 wcdb_get_db_status(int64 handle, _Out_ void** outJson)') } catch { this.wcdbGetDbStatus = null } // wcdb_status wcdb_get_voice_data(wcdb_handle handle, const char* session_id, int32_t create_time, int32_t local_id, int64_t svr_id, const char* candidates_json, char** out_hex) try { this.wcdbGetVoiceData = this.lib.func('int32 wcdb_get_voice_data(int64 handle, const char* sessionId, int32 createTime, int32 localId, int64 svrId, const char* candidatesJson, _Out_ void** outHex)') } catch { 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 } // wcdb_status wcdb_get_sns_annual_stats(wcdb_handle handle, int32_t begin_timestamp, int32_t end_timestamp, char** out_json) try { this.wcdbGetSnsAnnualStats = this.lib.func('int32 wcdb_get_sns_annual_stats(int64 handle, int32 begin, int32 end, _Out_ void** outJson)') } catch { this.wcdbGetSnsAnnualStats = null } // Named pipe IPC for monitoring (replaces callback) try { this.wcdbStartMonitorPipe = this.lib.func('int32 wcdb_start_monitor_pipe()') this.wcdbStopMonitorPipe = this.lib.func('void wcdb_stop_monitor_pipe()') this.wcdbGetMonitorPipeName = this.lib.func('int32 wcdb_get_monitor_pipe_name(_Out_ void** outName)') this.writeLog('Monitor pipe functions loaded') } catch (e) { console.warn('Failed to load monitor pipe functions:', e) this.wcdbStartMonitorPipe = null this.wcdbStopMonitorPipe = null this.wcdbGetMonitorPipeName = null } // void VerifyUser(int64_t hwnd_ptr, const char* message, char* out_result, int max_len) try { this.wcdbVerifyUser = this.lib.func('void VerifyUser(int64 hwnd, const char* message, _Out_ char* outResult, int maxLen)') } catch { this.wcdbVerifyUser = null } // 初始化 const initResult = this.wcdbInit() if (initResult !== 0) { console.error('WCDB 初始化失败:', initResult) return false } this.initialized = true lastDllInitError = null return true } catch (e) { const errorMsg = e instanceof Error ? e.message : String(e) console.error('WCDB 初始化异常:', errorMsg) this.writeLog(`WCDB 初始化异常: ${errorMsg}`, true) lastDllInitError = errorMsg // 检查是否是常见的 VC++ 运行时缺失错误 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 位版本的应用程序。' } return false } } /** * 测试数据库连接 */ async testConnection(dbPath: string, hexKey: string, wxid: string): Promise<{ success: boolean; error?: string; sessionCount?: number }> { try { // 如果当前已经有相同参数的活动连接,直接返回成功 if (this.handle !== null && this.currentPath === dbPath && this.currentKey === hexKey && this.currentWxid === wxid) { return { success: true, sessionCount: 0 } } // 记录当前活动连接,用于在测试结束后恢复(避免影响聊天页等正在使用的连接) const hadActiveConnection = this.handle !== null const prevPath = this.currentPath const prevKey = this.currentKey const prevWxid = this.currentWxid if (!this.initialized) { const initOk = await this.initialize() if (!initOk) { // 返回更详细的错误信息,帮助用户诊断问题 const detailedError = lastDllInitError || 'WCDB 初始化失败' return { success: false, error: detailedError } } } // 构建 db_storage 目录路径 const dbStoragePath = this.resolveDbStoragePath(dbPath, wxid) this.writeLog(`testConnection dbPath=${dbPath} wxid=${wxid} dbStorage=${dbStoragePath || 'null'}`) if (!dbStoragePath || !existsSync(dbStoragePath)) { return { success: false, error: `数据库目录不存在: ${dbPath}` } } // 递归查找 session.db const sessionDbPath = this.findSessionDb(dbStoragePath) this.writeLog(`testConnection sessionDb=${sessionDbPath || 'null'}`) if (!sessionDbPath) { return { success: false, error: `未找到 session.db 文件` } } // 分配输出参数内存 const handleOut = [0] const result = this.wcdbOpenAccount(sessionDbPath, hexKey, handleOut) if (result !== 0) { await this.printLogs() let errorMsg = '数据库打开失败' if (result === -1) errorMsg = '参数错误' else if (result === -2) errorMsg = '密钥错误' else if (result === -3) errorMsg = '数据库打开失败' this.writeLog(`testConnection openAccount failed code=${result}`) return { success: false, error: `${errorMsg} (错误码: ${result})` } } const tempHandle = handleOut[0] if (tempHandle <= 0) { return { success: false, error: '无效的数据库句柄' } } // 测试成功:使用 shutdown 清理资源(包括测试句柄) // 注意:shutdown 会断开当前活动连接,因此需要在测试后尝试恢复之前的连接 try { this.wcdbShutdown() this.handle = null this.currentPath = null this.currentKey = null this.currentWxid = null this.initialized = false } catch (closeErr) { console.error('关闭测试数据库时出错:', closeErr) } // 恢复测试前的连接(如果之前有活动连接) if (hadActiveConnection && prevPath && prevKey && prevWxid) { try { await this.open(prevPath, prevKey, prevWxid) } catch { // 恢复失败则保持断开,由调用方处理 } } return { success: true, sessionCount: 0 } } catch (e) { console.error('测试连接异常:', e) this.writeLog(`testConnection exception: ${String(e)}`) return { success: false, error: String(e) } } } /** * 打印 DLL 内部日志(仅在出错时调用) */ private async printLogs(force = false): Promise { try { if (!this.wcdbGetLogs) return const outPtr = [null as any] const result = this.wcdbGetLogs(outPtr) if (result === 0 && outPtr[0]) { try { const jsonStr = this.koffi.decode(outPtr[0], 'char', -1) this.writeLog(`wcdb_logs: ${jsonStr}`, force) this.wcdbFreeString(outPtr[0]) } catch (e) { // ignore } } } catch (e) { console.error('获取日志失败:', e) this.writeLog(`wcdb_logs failed: ${String(e)}`, force) } } private startLogPolling(): void { if (this.logTimer || !this.isLogEnabled()) return this.logTimer = setInterval(() => { void this.pollLogs() }, 2000) } private stopLogPolling(): void { if (this.logTimer) { clearInterval(this.logTimer) this.logTimer = null } this.lastLogTail = null } private async pollLogs(): Promise { try { if (!this.wcdbGetLogs || !this.isLogEnabled()) return const outPtr = [null as any] const result = this.wcdbGetLogs(outPtr) if (result !== 0 || !outPtr[0]) return let jsonStr = '' try { jsonStr = this.koffi.decode(outPtr[0], 'char', -1) } finally { try { this.wcdbFreeString(outPtr[0]) } catch { } } const logs = JSON.parse(jsonStr) as string[] if (!Array.isArray(logs) || logs.length === 0) return let startIdx = 0 if (this.lastLogTail) { const idx = logs.lastIndexOf(this.lastLogTail) if (idx >= 0) startIdx = idx + 1 } for (let i = startIdx; i < logs.length; i += 1) { this.writeLog(`wcdb: ${logs[i]}`) } this.lastLogTail = logs[logs.length - 1] } catch (e) { // ignore polling errors } } private decodeJsonPtr(outPtr: any): string | null { if (!outPtr) return null try { const jsonStr = this.koffi.decode(outPtr, 'char', -1) this.wcdbFreeString(outPtr) return jsonStr } catch (e) { try { this.wcdbFreeString(outPtr) } catch { } return null } } private ensureReady(): boolean { return this.initialized && this.handle !== null } private normalizeTimestamp(input: number): number { if (!input || input <= 0) return 0 const asNumber = Number(input) if (!Number.isFinite(asNumber)) return 0 // Treat >1e12 as milliseconds. const seconds = asNumber > 1e12 ? Math.floor(asNumber / 1000) : Math.floor(asNumber) const maxInt32 = 2147483647 return Math.min(Math.max(seconds, 0), maxInt32) } private normalizeRange(beginTimestamp: number, endTimestamp: number): { begin: number; end: number } { const normalizedBegin = this.normalizeTimestamp(beginTimestamp) let normalizedEnd = this.normalizeTimestamp(endTimestamp) if (normalizedEnd <= 0) { normalizedEnd = this.normalizeTimestamp(Date.now()) } if (normalizedBegin > 0 && normalizedEnd < normalizedBegin) { normalizedEnd = normalizedBegin } return { begin: normalizedBegin, end: normalizedEnd } } isReady(): boolean { return this.ensureReady() } /** * 打开数据库 */ async open(dbPath: string, hexKey: string, wxid: string): Promise { try { if (!this.initialized) { const initOk = await this.initialize() if (!initOk) return false } // 检查是否已经是当前连接的参数,如果是则直接返回成功,实现"始终保持链接" if (this.handle !== null && this.currentPath === dbPath && this.currentKey === hexKey && this.currentWxid === wxid) { return true } // 如果参数不同,则先关闭原来的连接 if (this.handle !== null) { this.close() // 重新初始化,因为 close 呼叫了 shutdown const initOk = await this.initialize() if (!initOk) return false } const dbStoragePath = this.resolveDbStoragePath(dbPath, wxid) this.writeLog(`open dbPath=${dbPath} wxid=${wxid} dbStorage=${dbStoragePath || 'null'}`) if (!dbStoragePath || !existsSync(dbStoragePath)) { console.error('数据库目录不存在:', dbPath) this.writeLog(`open failed: dbStorage not found for ${dbPath}`) return false } const sessionDbPath = this.findSessionDb(dbStoragePath) this.writeLog(`open sessionDb=${sessionDbPath || 'null'}`) if (!sessionDbPath) { console.error('未找到 session.db 文件') this.writeLog('open failed: session.db not found') return false } const handleOut = [0] const result = this.wcdbOpenAccount(sessionDbPath, hexKey, handleOut) if (result !== 0) { console.error('打开数据库失败:', result) await this.printLogs() this.writeLog(`open failed: openAccount code=${result}`) return false } const handle = handleOut[0] if (handle <= 0) { return false } this.handle = handle this.currentPath = dbPath this.currentKey = hexKey this.currentWxid = wxid this.initialized = true if (this.wcdbSetMyWxid && wxid) { try { this.wcdbSetMyWxid(this.handle, wxid) } catch (e) { // 静默失败 } } if (this.isLogEnabled()) { this.startLogPolling() } this.writeLog(`open ok handle=${handle}`) return true } catch (e) { console.error('打开数据库异常:', e) this.writeLog(`open exception: ${String(e)}`) return false } } /** * 关闭数据库 * 注意:wcdb_close_account 可能导致崩溃,使用 shutdown 代替 */ close(): void { if (this.handle !== null || this.initialized) { try { // 不调用 closeAccount,直接 shutdown this.wcdbShutdown() } catch (e) { console.error('WCDB shutdown 出错:', e) } this.handle = null this.currentPath = null this.currentKey = null this.currentWxid = null this.initialized = false this.stopLogPolling() } } /** * 关闭服务(与 close 相同) */ shutdown(): void { this.close() } /** * 检查是否已连接 */ isConnected(): boolean { return this.initialized && this.handle !== null } async getSessions(): Promise<{ success: boolean; sessions?: any[]; error?: string }> { if (!this.ensureReady()) { this.writeLog('getSessions skipped: not connected') return { success: false, error: 'WCDB 未连接' } } try { // 使用 setImmediate 让事件循环有机会处理其他任务,避免长时间阻塞 await new Promise(resolve => setImmediate(resolve)) const outPtr = [null as any] const result = this.wcdbGetSessions(this.handle, outPtr) // DLL 调用后再次让出控制权 await new Promise(resolve => setImmediate(resolve)) if (result !== 0 || !outPtr[0]) { this.writeLog(`getSessions failed: code=${result}`) return { success: false, error: `获取会话失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析会话失败' } this.writeLog(`getSessions ok size=${jsonStr.length}`) const sessions = JSON.parse(jsonStr) return { success: true, sessions } } catch (e) { this.writeLog(`getSessions exception: ${String(e)}`) return { success: false, error: String(e) } } } async getMessages(sessionId: string, limit: number, offset: number): Promise<{ success: boolean; messages?: any[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const result = this.wcdbGetMessages(this.handle, sessionId, limit, offset, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取消息失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析消息失败' } const messages = JSON.parse(jsonStr) return { success: true, messages } } catch (e) { return { success: false, error: String(e) } } } /** * 获取指定时间之后的新消息 */ async getNewMessages(sessionId: string, minTime: number, limit: number = 1000): Promise<{ success: boolean; messages?: any[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { // 1. 打开游标 (使用 Ascending=1 从指定时间往后查) const openRes = await this.openMessageCursor(sessionId, limit, true, minTime, 0) if (!openRes.success || !openRes.cursor) { return { success: false, error: openRes.error } } const cursor = openRes.cursor try { // 2. 获取批次 const fetchRes = await this.fetchMessageBatch(cursor) if (!fetchRes.success) { return { success: false, error: fetchRes.error } } return { success: true, messages: fetchRes.rows } } finally { // 3. 关闭游标 await this.closeMessageCursor(cursor) } } catch (e) { return { success: false, error: String(e) } } } async getMessageCount(sessionId: string): Promise<{ success: boolean; count?: number; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outCount = [0] const result = this.wcdbGetMessageCount(this.handle, sessionId, outCount) if (result !== 0) { return { success: false, error: `获取消息总数失败: ${result}` } } return { success: true, count: outCount[0] } } catch (e) { return { success: false, error: String(e) } } } async getDisplayNames(usernames: string[]): Promise<{ success: boolean; map?: Record; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (usernames.length === 0) return { success: true, map: {} } try { // 让出控制权,避免阻塞事件循环 await new Promise(resolve => setImmediate(resolve)) const outPtr = [null as any] const result = this.wcdbGetDisplayNames(this.handle, JSON.stringify(usernames), outPtr) // DLL 调用后再次让出控制权 await new Promise(resolve => setImmediate(resolve)) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取昵称失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析昵称失败' } const map = JSON.parse(jsonStr) return { success: true, map } } catch (e) { return { success: false, error: String(e) } } } async getAvatarUrls(usernames: string[]): Promise<{ success: boolean; map?: Record; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (usernames.length === 0) return { success: true, map: {} } try { const now = Date.now() const resultMap: Record = {} const toFetch: string[] = [] const seen = new Set() for (const username of usernames) { if (!username || seen.has(username)) continue seen.add(username) const cached = this.avatarUrlCache.get(username) // 只使用有效的缓存(URL不为空) if (cached && cached.url && cached.url.trim() && now - cached.updatedAt < this.avatarCacheTtlMs) { resultMap[username] = cached.url continue } toFetch.push(username) } if (toFetch.length === 0) { return { success: true, map: resultMap } } // 让出控制权,避免阻塞事件循环 await new Promise(resolve => setImmediate(resolve)) const outPtr = [null as any] const result = this.wcdbGetAvatarUrls(this.handle, JSON.stringify(toFetch), outPtr) // DLL 调用后再次让出控制权 await new Promise(resolve => setImmediate(resolve)) if (result !== 0 || !outPtr[0]) { if (Object.keys(resultMap).length > 0) { return { success: true, map: resultMap, error: `获取头像失败: ${result}` } } return { success: false, error: `获取头像失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) { return { success: false, error: '解析头像失败' } } const map = JSON.parse(jsonStr) as Record for (const username of toFetch) { const url = map[username] if (url && url.trim()) { resultMap[username] = url // 只缓存有效的URL this.avatarUrlCache.set(username, { url, updatedAt: now }) } // 不缓存空URL,下次可以重新尝试 } return { success: true, map: resultMap } } catch (e) { console.error('[wcdbCore] getAvatarUrls 异常:', e) return { success: false, error: String(e) } } } async getGroupMemberCount(chatroomId: string): Promise<{ success: boolean; count?: number; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outCount = [0] const result = this.wcdbGetGroupMemberCount(this.handle, chatroomId, outCount) if (result !== 0) { return { success: false, error: `获取群成员数量失败: ${result}` } } return { success: true, count: outCount[0] } } catch (e) { return { success: false, error: String(e) } } } async getGroupMemberCounts(chatroomIds: string[]): Promise<{ success: boolean; map?: Record; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (chatroomIds.length === 0) return { success: true, map: {} } if (!this.wcdbGetGroupMemberCounts) { const map: Record = {} for (const chatroomId of chatroomIds) { const result = await this.getGroupMemberCount(chatroomId) if (result.success && typeof result.count === 'number') { map[chatroomId] = result.count } } return { success: true, map } } try { const outPtr = [null as any] const result = this.wcdbGetGroupMemberCounts(this.handle, JSON.stringify(chatroomIds), outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取群成员数量失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析群成员数量失败' } const map = JSON.parse(jsonStr) return { success: true, map } } catch (e) { return { success: false, error: String(e) } } } async getGroupMembers(chatroomId: string): Promise<{ success: boolean; members?: any[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const result = this.wcdbGetGroupMembers(this.handle, chatroomId, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取群成员失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析群成员失败' } const members = JSON.parse(jsonStr) return { success: true, members } } catch (e) { return { success: false, error: String(e) } } } async getGroupNicknames(chatroomId: string): Promise<{ success: boolean; nicknames?: Record; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (!this.wcdbGetGroupNicknames) { return { success: false, error: '当前 DLL 版本不支持获取群昵称接口' } } try { const outPtr = [null as any] const result = this.wcdbGetGroupNicknames(this.handle, chatroomId, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取群昵称失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析群昵称失败' } const nicknames = JSON.parse(jsonStr) return { success: true, nicknames } } catch (e) { return { success: false, error: String(e) } } } async getMessageTables(sessionId: string): Promise<{ success: boolean; tables?: any[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const result = this.wcdbGetMessageTables(this.handle, sessionId, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取消息表失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析消息表失败' } const tables = JSON.parse(jsonStr) return { success: true, tables } } catch (e) { return { success: false, error: String(e) } } } async getMessageDates(sessionId: string): Promise<{ success: boolean; dates?: string[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { if (!this.wcdbGetMessageDates) { return { success: false, error: 'DLL 不支持 getMessageDates' } } const outPtr = [null as any] const result = this.wcdbGetMessageDates(this.handle, sessionId, outPtr) if (result !== 0 || !outPtr[0]) { // 空结果也可能是正常的(无消息) return { success: true, dates: [] } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析日期列表失败' } const dates = JSON.parse(jsonStr) return { success: true, dates } } catch (e) { return { success: false, error: String(e) } } } async getMessageTableStats(sessionId: string): Promise<{ success: boolean; tables?: any[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const result = this.wcdbGetMessageTableStats(this.handle, sessionId, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取表统计失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析表统计失败' } const tables = JSON.parse(jsonStr) return { success: true, tables } } catch (e) { return { success: false, error: String(e) } } } async getMessageMeta(dbPath: string, tableName: string, limit: number, offset: number): Promise<{ success: boolean; rows?: any[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const result = this.wcdbGetMessageMeta(this.handle, dbPath, tableName, limit, offset, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取消息元数据失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析消息元数据失败' } const rows = JSON.parse(jsonStr) return { success: true, rows } } catch (e) { return { success: false, error: String(e) } } } async getContact(username: string): Promise<{ success: boolean; contact?: any; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const result = this.wcdbGetContact(this.handle, username, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取联系人失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析联系人失败' } const contact = JSON.parse(jsonStr) return { success: true, contact } } catch (e) { return { success: false, error: String(e) } } } async getAggregateStats(sessionIds: string[], beginTimestamp: number = 0, endTimestamp: number = 0): Promise<{ success: boolean; data?: any; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const normalizedBegin = this.normalizeTimestamp(beginTimestamp) let normalizedEnd = this.normalizeTimestamp(endTimestamp) if (normalizedEnd <= 0) { normalizedEnd = this.normalizeTimestamp(Date.now()) } if (normalizedBegin > 0 && normalizedEnd < normalizedBegin) { normalizedEnd = normalizedBegin } const callAggregate = (ids: string[]) => { const idsAreNumeric = ids.length > 0 && ids.every((id) => /^\d+$/.test(id)) const payloadIds = idsAreNumeric ? ids.map((id) => Number(id)) : ids const outPtr = [null as any] const result = this.wcdbGetAggregateStats(this.handle, JSON.stringify(payloadIds), normalizedBegin, normalizedEnd, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取聚合统计失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) { return { success: false, error: '解析聚合统计失败' } } const data = JSON.parse(jsonStr) return { success: true, data } } let result = callAggregate(sessionIds) if (result.success && result.data && result.data.total === 0 && result.data.idMap) { const idMap = result.data.idMap as Record const reverseMap: Record = {} for (const [id, name] of Object.entries(idMap)) { if (!name) continue reverseMap[name] = id } const numericIds = sessionIds .map((id) => reverseMap[id]) .filter((id) => typeof id === 'string' && /^\d+$/.test(id)) if (numericIds.length > 0) { const retry = callAggregate(numericIds) if (retry.success && retry.data) { result = retry } } } return result } catch (e) { return { success: false, error: String(e) } } } async getAvailableYears(sessionIds: string[]): Promise<{ success: boolean; data?: number[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (!this.wcdbGetAvailableYears) { return { success: false, error: '未支持获取年度列表' } } if (sessionIds.length === 0) return { success: true, data: [] } try { const outPtr = [null as any] const result = this.wcdbGetAvailableYears(this.handle, JSON.stringify(sessionIds), outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取年度列表失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析年度列表失败' } const data = JSON.parse(jsonStr) return { success: true, data } } catch (e) { return { success: false, error: String(e) } } } async getAnnualReportStats(sessionIds: string[], beginTimestamp: number = 0, endTimestamp: number = 0): Promise<{ success: boolean; data?: any; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (!this.wcdbGetAnnualReportStats) { return this.getAggregateStats(sessionIds, beginTimestamp, endTimestamp) } try { const { begin, end } = this.normalizeRange(beginTimestamp, endTimestamp) const outPtr = [null as any] const result = this.wcdbGetAnnualReportStats(this.handle, JSON.stringify(sessionIds), begin, end, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取年度统计失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析年度统计失败' } const data = JSON.parse(jsonStr) return { success: true, data } } catch (e) { return { success: false, error: String(e) } } } async getAnnualReportExtras( sessionIds: string[], beginTimestamp: number = 0, endTimestamp: number = 0, peakDayBegin: number = 0, peakDayEnd: number = 0 ): Promise<{ success: boolean; data?: any; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (!this.wcdbGetAnnualReportExtras) { return { success: false, error: '未支持年度扩展统计' } } if (sessionIds.length === 0) return { success: true, data: {} } try { const { begin, end } = this.normalizeRange(beginTimestamp, endTimestamp) const outPtr = [null as any] const result = this.wcdbGetAnnualReportExtras( this.handle, JSON.stringify(sessionIds), begin, end, this.normalizeTimestamp(peakDayBegin), this.normalizeTimestamp(peakDayEnd), outPtr ) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取年度扩展统计失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析年度扩展统计失败' } const data = JSON.parse(jsonStr) return { success: true, data } } catch (e) { return { success: false, error: String(e) } } } async getGroupStats(chatroomId: string, beginTimestamp: number = 0, endTimestamp: number = 0): Promise<{ success: boolean; data?: any; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (!this.wcdbGetGroupStats) { return this.getAggregateStats([chatroomId], beginTimestamp, endTimestamp) } try { const { begin, end } = this.normalizeRange(beginTimestamp, endTimestamp) const outPtr = [null as any] const result = this.wcdbGetGroupStats(this.handle, chatroomId, begin, end, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取群聊统计失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析群聊统计失败' } const data = JSON.parse(jsonStr) return { success: true, data } } catch (e) { return { success: false, error: String(e) } } } async openMessageCursor(sessionId: string, batchSize: number, ascending: boolean, beginTimestamp: number, endTimestamp: number): Promise<{ success: boolean; cursor?: number; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outCursor = [0] const result = this.wcdbOpenMessageCursor( this.handle, sessionId, batchSize, ascending ? 1 : 0, beginTimestamp, endTimestamp, outCursor ) if (result !== 0 || outCursor[0] <= 0) { await this.printLogs(true) this.writeLog( `openMessageCursor failed: sessionId=${sessionId} batchSize=${batchSize} ascending=${ascending ? 1 : 0} begin=${beginTimestamp} end=${endTimestamp} result=${result} cursor=${outCursor[0]}`, true ) return { success: false, error: `创建游标失败: ${result},请查看日志` } } return { success: true, cursor: outCursor[0] } } catch (e) { await this.printLogs(true) this.writeLog(`openMessageCursor exception: ${String(e)}`, true) return { success: false, error: '创建游标异常,请查看日志' } } } async openMessageCursorLite(sessionId: string, batchSize: number, ascending: boolean, beginTimestamp: number, endTimestamp: number): Promise<{ success: boolean; cursor?: number; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (!this.wcdbOpenMessageCursorLite) { return this.openMessageCursor(sessionId, batchSize, ascending, beginTimestamp, endTimestamp) } try { const outCursor = [0] const result = this.wcdbOpenMessageCursorLite( this.handle, sessionId, batchSize, ascending ? 1 : 0, beginTimestamp, endTimestamp, outCursor ) if (result !== 0 || outCursor[0] <= 0) { await this.printLogs(true) this.writeLog( `openMessageCursorLite failed: sessionId=${sessionId} batchSize=${batchSize} ascending=${ascending ? 1 : 0} begin=${beginTimestamp} end=${endTimestamp} result=${result} cursor=${outCursor[0]}`, true ) return { success: false, error: `创建游标失败: ${result},请查看日志` } } return { success: true, cursor: outCursor[0] } } catch (e) { await this.printLogs(true) this.writeLog(`openMessageCursorLite exception: ${String(e)}`, true) return { success: false, error: '创建游标异常,请查看日志' } } } async fetchMessageBatch(cursor: number): Promise<{ success: boolean; rows?: any[]; hasMore?: boolean; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const outHasMore = [0] const result = this.wcdbFetchMessageBatch(this.handle, cursor, outPtr, outHasMore) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取批次失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析批次失败' } const rows = JSON.parse(jsonStr) return { success: true, rows, hasMore: outHasMore[0] === 1 } } catch (e) { return { success: false, error: String(e) } } } async closeMessageCursor(cursor: number): Promise<{ success: boolean; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const result = this.wcdbCloseMessageCursor(this.handle, cursor) if (result !== 0) { return { success: false, error: `关闭游标失败: ${result}` } } return { success: true } } catch (e) { return { success: false, error: String(e) } } } async getLogs(): Promise<{ success: boolean; logs?: string[]; error?: string }> { if (!this.lib) return { success: false, error: 'DLL 未加载' } if (!this.wcdbGetLogs) return { success: false, error: '接口未就绪' } try { const outPtr = [null as any] const result = this.wcdbGetLogs(outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取日志失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析日志失败' } return { success: true, logs: JSON.parse(jsonStr) } } catch (e) { return { success: false, error: String(e) } } } async execQuery(kind: string, path: string | null, sql: string, params: any[] = []): Promise<{ success: boolean; rows?: any[]; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { if (!this.wcdbExecQuery) return { success: false, error: '接口未就绪' } // 如果提供了参数,使用参数化查询(需要 C++ 层支持) // 注意:当前 wcdbExecQuery 可能不支持参数化,这是一个占位符实现 // TODO: 需要更新 C++ 层的 wcdb_exec_query 以支持参数绑定 if (params && params.length > 0) { console.warn('[wcdbCore] execQuery: 参数化查询暂未在 C++ 层实现,将使用原始 SQL(可能存在注入风险)') } const outPtr = [null as any] const result = this.wcdbExecQuery(this.handle, kind, path || '', sql, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `执行查询失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析查询结果失败' } const rows = JSON.parse(jsonStr) return { success: true, rows } } catch (e) { return { success: false, error: String(e) } } } async getEmoticonCdnUrl(dbPath: string, md5: string): Promise<{ success: boolean; url?: string; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { const outPtr = [null as any] const result = this.wcdbGetEmoticonCdnUrl(this.handle, dbPath, md5, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取表情 URL 失败: ${result}` } } const urlStr = this.decodeJsonPtr(outPtr[0]) if (urlStr === null) return { success: false, error: '解析表情 URL 失败' } return { success: true, url: urlStr || undefined } } catch (e) { return { success: false, error: String(e) } } } async listMessageDbs(): Promise<{ success: boolean; data?: string[]; error?: string }> { if (!this.ensureReady()) return { success: false, error: 'WCDB 未连接' } try { const outPtr = [null as any] const result = this.wcdbListMessageDbs(this.handle, outPtr) if (result !== 0 || !outPtr[0]) return { success: false, error: `获取消息库列表失败: ${result}` } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析消息库列表失败' } const data = JSON.parse(jsonStr) return { success: true, data } } catch (e) { return { success: false, error: String(e) } } } async listMediaDbs(): Promise<{ success: boolean; data?: string[]; error?: string }> { if (!this.ensureReady()) return { success: false, error: 'WCDB 未连接' } try { const outPtr = [null as any] const result = this.wcdbListMediaDbs(this.handle, outPtr) if (result !== 0 || !outPtr[0]) return { success: false, error: `获取媒体库列表失败: ${result}` } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析媒体库列表失败' } const data = JSON.parse(jsonStr) return { success: true, data } } catch (e) { return { success: false, error: String(e) } } } async getMessageById(sessionId: string, localId: number): Promise<{ success: boolean; message?: any; error?: string }> { if (!this.ensureReady()) return { success: false, error: 'WCDB 未连接' } try { const outPtr = [null as any] const result = this.wcdbGetMessageById(this.handle, sessionId, localId, outPtr) if (result !== 0 || !outPtr[0]) return { success: false, error: `查询消息失败: ${result}` } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析消息失败' } const message = JSON.parse(jsonStr) // 处理 wcdb_get_message_by_id 返回空对象的情况 if (Object.keys(message).length === 0) return { success: false, error: '未找到消息' } return { success: true, message } } catch (e) { return { success: false, error: String(e) } } } async getVoiceData(sessionId: string, createTime: number, candidates: string[], localId: number = 0, svrId: string | number = 0): Promise<{ success: boolean; hex?: string; error?: string }> { if (!this.ensureReady()) return { success: false, error: 'WCDB 未连接' } if (!this.wcdbGetVoiceData) return { success: false, error: '当前 DLL 版本不支持获取语音数据' } try { const outPtr = [null as any] const result = this.wcdbGetVoiceData(this.handle, sessionId, createTime, localId, BigInt(svrId || 0), JSON.stringify(candidates), outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取语音数据失败: ${result}` } } const hex = this.decodeJsonPtr(outPtr[0]) if (hex === null) return { success: false, error: '解析语音数据失败' } return { success: true, hex: hex || undefined } } catch (e) { return { success: false, error: String(e) } } } /** * 验证 Windows Hello */ async verifyUser(message: string, hwnd?: string): Promise<{ success: boolean; error?: string }> { if (!this.initialized) { const initOk = await this.initialize() if (!initOk) return { success: false, error: 'WCDB 初始化失败' } } if (!this.wcdbVerifyUser) { return { success: false, error: 'Binding not found: VerifyUser' } } return new Promise((resolve) => { try { // Allocate buffer for result JSON const maxLen = 1024 const outBuf = Buffer.alloc(maxLen) // Call native function const hwndVal = hwnd ? BigInt(hwnd) : BigInt(0) this.wcdbVerifyUser(hwndVal, message || '', outBuf, maxLen) // Parse result const jsonStr = this.koffi.decode(outBuf, 'char', -1) const result = JSON.parse(jsonStr) resolve(result) } catch (e) { resolve({ 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) } } } async getSnsAnnualStats(beginTimestamp: number, endTimestamp: number): Promise<{ success: boolean; data?: any; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } try { if (!this.wcdbGetSnsAnnualStats) { return { success: false, error: 'wcdbGetSnsAnnualStats 未找到' } } await new Promise(resolve => setImmediate(resolve)) const outPtr = [null as any] const result = this.wcdbGetSnsAnnualStats(this.handle, beginTimestamp, endTimestamp, outPtr) await new Promise(resolve => setImmediate(resolve)) if (result !== 0 || !outPtr[0]) { return { success: false, error: `getSnsAnnualStats failed: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: 'Failed to decode JSON' } return { success: true, data: JSON.parse(jsonStr) } } catch (e) { console.error('getSnsAnnualStats 异常:', e) return { success: false, error: String(e) } } } async getDualReportStats(sessionId: string, beginTimestamp: number = 0, endTimestamp: number = 0): Promise<{ success: boolean; data?: any; error?: string }> { if (!this.ensureReady()) { return { success: false, error: 'WCDB 未连接' } } if (!this.wcdbGetDualReportStats) { return { success: false, error: '未支持双人报告统计' } } try { const { begin, end } = this.normalizeRange(beginTimestamp, endTimestamp) const outPtr = [null as any] const result = this.wcdbGetDualReportStats(this.handle, sessionId, begin, end, outPtr) if (result !== 0 || !outPtr[0]) { return { success: false, error: `获取双人报告统计失败: ${result}` } } const jsonStr = this.decodeJsonPtr(outPtr[0]) if (!jsonStr) return { success: false, error: '解析双人报告统计失败' } const data = JSON.parse(jsonStr) return { success: true, data } } catch (e) { return { success: false, error: String(e) } } } /** * 修改消息内容 */ async updateMessage(sessionId: string, localId: number, createTime: number, newContent: string): Promise<{ success: boolean; error?: string }> { if (!this.initialized || !this.wcdbUpdateMessage) return { success: false, error: 'WCDB Not Initialized or Method Missing' } if (!this.handle) return { success: false, error: 'Not Connected' } return new Promise((resolve) => { try { const outError = [null as any] const result = this.wcdbUpdateMessage(this.handle, sessionId, localId, createTime, newContent, outError) if (result !== 0) { let errorMsg = 'Unknown Error' if (outError[0]) { errorMsg = this.decodeJsonPtr(outError[0]) || 'Unknown Error (Decode Failed)' } resolve({ success: false, error: errorMsg }) return } resolve({ success: true }) } catch (e) { resolve({ success: false, error: String(e) }) } }) } /** * 删除消息 */ async deleteMessage(sessionId: string, localId: number, createTime: number, dbPathHint?: string): Promise<{ success: boolean; error?: string }> { if (!this.initialized || !this.wcdbDeleteMessage) return { success: false, error: 'WCDB Not Initialized or Method Missing' } if (!this.handle) return { success: false, error: 'Not Connected' } return new Promise((resolve) => { try { const outError = [null as any] const result = this.wcdbDeleteMessage(this.handle, sessionId, localId, createTime || 0, dbPathHint || '', outError) if (result !== 0) { let errorMsg = 'Unknown Error' if (outError[0]) { errorMsg = this.decodeJsonPtr(outError[0]) || 'Unknown Error (Decode Failed)' } console.error(`[WcdbCore] deleteMessage fail: code=${result}, error=${errorMsg}`) resolve({ success: false, error: errorMsg }) return } resolve({ success: true }) } catch (e) { console.error(`[WcdbCore] deleteMessage exception:`, e) resolve({ success: false, error: String(e) }) } }) } }