diff --git a/electron/main.ts b/electron/main.ts index 72cfb45..cfc436f 100644 --- a/electron/main.ts +++ b/electron/main.ts @@ -30,6 +30,7 @@ import { cloudControlService } from './services/cloudControlService' import { destroyNotificationWindow, registerNotificationHandlers, showNotification, setNotificationNavigateHandler } from './windows/notificationWindow' import { httpService } from './services/httpService' import { messagePushService } from './services/messagePushService' +import { insightService } from './services/insightService' import { bizService } from './services/bizService' // 配置自动更新 @@ -1621,6 +1622,19 @@ function registerIpcHandlers() { return result }) + // AI 见解 + ipcMain.handle('insight:testConnection', async () => { + return insightService.testConnection() + }) + + ipcMain.handle('insight:getTodayStats', async () => { + return insightService.getTodayStats() + }) + + ipcMain.handle('insight:triggerTest', async () => { + return insightService.triggerTest() + }) + ipcMain.handle('config:clear', async () => { if (isLaunchAtStartupSupported() && getSystemLaunchAtStartup()) { const result = setSystemLaunchAtStartup(false) @@ -3485,8 +3499,10 @@ app.whenReady().then(async () => { registerIpcHandlers() chatService.addDbMonitorListener((type, json) => { messagePushService.handleDbMonitorChange(type, json) + insightService.handleDbMonitorChange(type, json) }) messagePushService.start() + insightService.start() await delay(200) // 检查配置状态 @@ -3607,6 +3623,7 @@ app.on('before-quit', async () => { if (tray) { try { tray.destroy() } catch {} tray = null } // 通知窗使用 hide 而非 close，退出时主动销毁，避免残留窗口阻塞进程退出。 destroyNotificationWindow() + insightService.stop() // 兜底：5秒后强制退出，防止某个异步任务卡住导致进程残留 const forceExitTimer = setTimeout(() => { console.warn('[App] Force exit after timeout') diff --git a/electron/preload.ts b/electron/preload.ts index ed918b7..e90a078 100644 --- a/electron/preload.ts +++ b/electron/preload.ts @@ -498,5 +498,12 @@ contextBridge.exposeInMainWorld('electronAPI', { start: (port?: number, host?: string) => ipcRenderer.invoke('http:start', port, host), stop: () => ipcRenderer.invoke('http:stop'), status: () => ipcRenderer.invoke('http:status') + }, + + // AI 见解 + insight: { + testConnection: () => ipcRenderer.invoke('insight:testConnection'), + getTodayStats: () => ipcRenderer.invoke('insight:getTodayStats'), + triggerTest: () => ipcRenderer.invoke('insight:triggerTest') } }) diff --git a/electron/services/config.ts b/electron/services/config.ts index 85da9fc..87029c9 100644 --- a/electron/services/config.ts +++ b/electron/services/config.ts @@ -69,10 +69,34 @@ interface ConfigSchema { quoteLayout: 'quote-top' | 'quote-bottom' wordCloudExcludeWords: string[] exportWriteLayout: 'A' | 'B' | 'C' + + // AI 见解 + aiInsightEnabled: boolean + aiInsightApiBaseUrl: string + aiInsightApiKey: string + aiInsightApiModel: string + aiInsightSilenceDays: number + aiInsightAllowContext: boolean + aiInsightWhitelistEnabled: boolean + aiInsightWhitelist: string[] + /** 活跃分析冷却时间（分钟），0 表示无冷却 */ + aiInsightCooldownMinutes: number + /** 沉默联系人扫描间隔（小时） */ + aiInsightScanIntervalHours: number + /** 发送上下文时的最大消息条数 */ + aiInsightContextCount: number + /** 自定义 system prompt，空字符串表示使用内置默认值 */ + aiInsightSystemPrompt: string + /** 是否启用 Telegram 推送 */ + aiInsightTelegramEnabled: boolean + /** Telegram Bot Token */ + aiInsightTelegramToken: string + /** Telegram 接收 Chat ID，逗号分隔，支持多个 */ + aiInsightTelegramChatIds: string } // 需要 safeStorage 加密的字段（普通模式） -const ENCRYPTED_STRING_KEYS: Set = new Set(['decryptKey', 'imageAesKey', 'authPassword', 'httpApiToken']) +const ENCRYPTED_STRING_KEYS: Set = new Set(['decryptKey', 'imageAesKey', 'authPassword', 'httpApiToken', 'aiInsightApiKey']) const ENCRYPTED_BOOL_KEYS: Set = new Set(['authEnabled', 'authUseHello']) const ENCRYPTED_NUMBER_KEYS: Set = new Set(['imageXorKey']) @@ -142,7 +166,22 @@ export class ConfigService { windowCloseBehavior: 'ask', quoteLayout: 'quote-top', wordCloudExcludeWords: [], - exportWriteLayout: 'A' + exportWriteLayout: 'A', + aiInsightEnabled: false, + aiInsightApiBaseUrl: '', + aiInsightApiKey: '', + aiInsightApiModel: 'gpt-4o-mini', + aiInsightSilenceDays: 3, + aiInsightAllowContext: false, + aiInsightWhitelistEnabled: false, + aiInsightWhitelist: [], + aiInsightCooldownMinutes: 120, + aiInsightScanIntervalHours: 4, + aiInsightContextCount: 40, + aiInsightSystemPrompt: '', + aiInsightTelegramEnabled: false, + aiInsightTelegramToken: '', + aiInsightTelegramChatIds: '' } const storeOptions: any = { @@ -690,7 +729,7 @@ export class ConfigService { // === 工具方法 === /** - * 获取当前 wxid 对应的图片密钥，优先从 wxidConfigs 中取，找不到则回退到全局配置 + * 获取当前 wxid 对应的图片密钥，优先从 wxidConfigs 中取，找不到则回退到全局��置 */ getImageKeysForCurrentWxid(): { xorKey: unknown; aesKey: string } { const wxid = this.get('myWxid') diff --git a/electron/services/insightService.ts b/electron/services/insightService.ts new file mode 100644 index 0000000..305cfd6 --- /dev/null +++ b/electron/services/insightService.ts @@ -0,0 +1,829 @@ +/** + * insightService.ts + * + * AI 见解后台服务： + * 1. 监听 DB 变更事件（debounce 500ms 防抖，避免开机/重连时爆发大量事件阻塞主线程） + * 2. 沉默联系人扫描（独立 setInterval，每 4 小时一次） + * 3. 触发后拉取真实聊天上下文（若用户授权），组装 prompt 调用单一 AI 模型 + * 4. 输出 ≤80 字见解，通过现有 showNotification 弹出右下角通知 + * + * 设计原则： + * - 不引入任何额外 npm 依赖，使用 Node 原生 https 模块调用 OpenAI 兼容 API + * - 所有失败静默处理，不影响主流程 + * - 当日触发记录（sessionId + 时间列表）随 prompt 一起发送，让模型自行判断是否克制 + */ + +import https from 'https' +import http from 'http' +import fs from 'fs' +import path from 'path' +import { URL } from 'url' +import { app, Notification } from 'electron' +import { ConfigService } from './config' +import { chatService, ChatSession, Message } from './chatService' + +// ─── 常量 ──────────────────────────────────────────────────────────────────── + +/** + * DB 变更防抖延迟（毫秒）。 + * 设为 2s：微信写库通常是批量操作，500ms 过短会在开机/重连时产生大量连续触发。 + */ +const DB_CHANGE_DEBOUNCE_MS = 2000 + +/** 首次沉默扫描延迟（毫秒），避免启动期间抢占资源 */ +const SILENCE_SCAN_INITIAL_DELAY_MS = 3 * 60 * 1000 + +/** 单次 API 请求超时（毫秒） */ +const API_TIMEOUT_MS = 45_000 + +/** 沉默天数阈值默认值 */ +const DEFAULT_SILENCE_DAYS = 3 + +// ─── 类型 ──────────────────────────────────────────────────────────────────── + +interface TodayTriggerRecord { + /** 该会话今日触发的时间戳列表（毫秒） */ + timestamps: number[] +} + +// ─── 桌面日志 ───────────────────────────────────────────────────────────────── + +/** + * 将日志同时输出到 console 和桌面上的 weflow-insight.log 文件。 + * 文件名带当天日期，每天自动换一个新文件，旧文件保留。 + */ +function insightLog(level: 'INFO' | 'WARN' | 'ERROR', message: string): void { + const now = new Date() + const dateStr = now.toLocaleDateString('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' }).replace(/\//g, '-') + const timeStr = now.toLocaleTimeString('zh-CN', { hour12: false }) + const line = `[${dateStr} ${timeStr}] [${level}] ${message}\n` + + // 同步到 console + if (level === 'ERROR' || level === 'WARN') { + console.warn(`[InsightService] ${message}`) + } else { + console.log(`[InsightService] ${message}`) + } + + // 异步写入桌面日志文件，避免同步磁盘 I/O 阻塞 Electron 主线程事件循环 + try { + const desktopPath = app.getPath('desktop') + const logFile = path.join(desktopPath, `weflow-insight-${dateStr}.log`) + fs.appendFile(logFile, line, 'utf-8', () => { /* 失败静默处理 */ }) + } catch { + // getPath 失败时静默处理 + } +} + +// ─── 工具函数 ───────────────────────────────────────────────────────────────── + +/** + * 绝对拼接 baseUrl 与路径，避免 Node.js URL 相对路径陷阱。 + * + * 例如： + * baseUrl = "https://api.ohmygpt.com/v1" + * path = "/chat/completions" + * 结果为 "https://api.ohmygpt.com/v1/chat/completions" + * + * 如果 baseUrl 末尾没有斜杠，直接用字符串拼接（而非 new URL(path, base)）， + * 因为 new URL("chat/completions", "https://api.example.com/v1") 会错误地 + * 丢弃 v1，变成 https://api.example.com/chat/completions。 + */ +function buildApiUrl(baseUrl: string, path: string): string { + const base = baseUrl.replace(/\/+$/, '') // 去掉末尾斜杠 + const suffix = path.startsWith('/') ? path : `/${path}` + return `${base}${suffix}` +} + +function getStartOfDay(date: Date = new Date()): number { + const d = new Date(date) + d.setHours(0, 0, 0, 0) + return d.getTime() +} + +function formatTimestamp(ts: number): string { + return new Date(ts).toLocaleTimeString('zh-CN', { hour: '2-digit', minute: '2-digit' }) +} + +/** + * 调用 OpenAI 兼容 API（非流式），返回模型第一条消息内容。 + * 使用 Node 原生 https/http 模块，无需任何第三方 SDK。 + */ +function callApi( + apiBaseUrl: string, + apiKey: string, + model: string, + messages: Array<{ role: string; content: string }>, + timeoutMs: number = API_TIMEOUT_MS +): Promise { + return new Promise((resolve, reject) => { + const endpoint = buildApiUrl(apiBaseUrl, '/chat/completions') + let urlObj: URL + try { + urlObj = new URL(endpoint) + } catch (e) { + reject(new Error(`无效的 API URL: ${endpoint}`)) + return + } + + const body = JSON.stringify({ + model, + messages, + max_tokens: 200, + temperature: 0.7, + stream: false + }) + + const options = { + hostname: urlObj.hostname, + port: urlObj.port || (urlObj.protocol === 'https:' ? 443 : 80), + path: urlObj.pathname + urlObj.search, + method: 'POST' as const, + headers: { + 'Content-Type': 'application/json', + 'Content-Length': Buffer.byteLength(body).toString(), + Authorization: `Bearer ${apiKey}` + } + } + + const isHttps = urlObj.protocol === 'https:' + const requestFn = isHttps ? https.request : http.request + const req = requestFn(options, (res) => { + let data = '' + res.on('data', (chunk) => { data += chunk }) + res.on('end', () => { + try { + const parsed = JSON.parse(data) + const content = parsed?.choices?.[0]?.message?.content + if (typeof content === 'string' && content.trim()) { + resolve(content.trim()) + } else { + reject(new Error(`API 返回格式异常: ${data.slice(0, 200)}`)) + } + } catch (e) { + reject(new Error(`JSON 解析失败: ${data.slice(0, 200)}`)) + } + }) + }) + + req.setTimeout(timeoutMs, () => { + req.destroy() + reject(new Error('API 请求超时')) + }) + + req.on('error', (e) => reject(e)) + req.write(body) + req.end() + }) +} + +// ─── InsightService 主类 ────────────────────────────────────────────────────── + +class InsightService { + private readonly config: ConfigService + + /** DB 变更防抖定时器 */ + private dbDebounceTimer: NodeJS.Timeout | null = null + + /** 沉默扫描定时器 */ + private silenceScanTimer: NodeJS.Timeout | null = null + private silenceInitialDelayTimer: NodeJS.Timeout | null = null + + /** 是否正在处理中（防重入） */ + private processing = false + + /** + * 当日触发记录：sessionId -> TodayTriggerRecord + * 每天 00:00 之后自动重置（通过检查日期实现） + */ + private todayTriggers: Map = new Map() + private todayDate = getStartOfDay() + + /** + * 活跃分析冷却记录：sessionId -> 上次分析时间戳（毫秒） + * 同一会话 2 小时内不重复触发活跃分析，防止 DB 频繁变更时爆量调用 API。 + */ + private lastActivityAnalysis: Map = new Map() + + /** + * 跟踪每个会话上次见到的最新消息时间戳，用于判断是否有真正的新消息。 + * sessionId -> lastMessageTimestamp（秒，与微信 DB 保持一致） + */ + private lastSeenTimestamp: Map = new Map() + + /** + * 本地会话快照缓存，避免 analyzeRecentActivity 在每次 DB 变更时都做全量读取。 + * 首次调用时填充，此后只在沉默扫描里刷新（沉默扫描间隔更长，更合适做全量刷新）。 + */ + private sessionCache: ChatSession[] | null = null + /** sessionCache 最后刷新时间戳（ms），超过 15 分钟强制重新拉取 */ + private sessionCacheAt = 0 + /** 缓存 TTL 设为 15 分钟，大幅减少 connect() + getSessions() 调用频率 */ + private static readonly SESSION_CACHE_TTL_MS = 15 * 60 * 1000 + /** 数据库是否已连接（避免重复调用 chatService.connect()） */ + private dbConnected = false + + private started = false + + constructor() { + this.config = ConfigService.getInstance() + } + + // ── 公开 API ──────────────────────────────────────────────────────────────── + + start(): void { + if (this.started) return + this.started = true + insightLog('INFO', '已启动') + this.scheduleSilenceScan() + } + + stop(): void { + this.started = false + this.dbConnected = false + this.sessionCache = null + this.sessionCacheAt = 0 + if (this.dbDebounceTimer !== null) { + clearTimeout(this.dbDebounceTimer) + this.dbDebounceTimer = null + } + if (this.silenceScanTimer !== null) { + clearTimeout(this.silenceScanTimer) + this.silenceScanTimer = null + } + if (this.silenceInitialDelayTimer !== null) { + clearTimeout(this.silenceInitialDelayTimer) + this.silenceInitialDelayTimer = null + } + insightLog('INFO', '已停止') + } + + /** + * 由 main.ts 在 addDbMonitorListener 回调中调用。 + * 加入 2s 防抖，防止开机/重连时大量事件并发阻塞主线程。 + * 如果当前正在处理中，直接忽略此次事件（不创建新的 timer），避免 timer 堆积。 + */ + handleDbMonitorChange(_type: string, _json: string): void { + if (!this.started) return + if (!this.isEnabled()) return + // 正在处理时忽略新事件，避免 timer 堆积 + if (this.processing) return + + if (this.dbDebounceTimer !== null) { + clearTimeout(this.dbDebounceTimer) + } + this.dbDebounceTimer = setTimeout(() => { + this.dbDebounceTimer = null + void this.analyzeRecentActivity() + }, DB_CHANGE_DEBOUNCE_MS) + } + + /** + * 测��� API 连接，返回 { success, message }。 + * 供设置页"测试连接"按钮调用。 + */ + async testConnection(): Promise<{ success: boolean; message: string }> { + const apiBaseUrl = this.config.get('aiInsightApiBaseUrl') as string + const apiKey = this.config.get('aiInsightApiKey') as string + const model = (this.config.get('aiInsightApiModel') as string) || 'gpt-4o-mini' + + if (!apiBaseUrl || !apiKey) { + return { success: false, message: '请先填写 API 地址和 API Key' } + } + + try { + const result = await callApi( + apiBaseUrl, + apiKey, + model, + [{ role: 'user', content: '请回复"连接成功"四个字。' }], + 15_000 + ) + return { success: true, message: `连接成功，模型回复：${result.slice(0, 50)}` } + } catch (e) { + return { success: false, message: `连接失败：${(e as Error).message}` } + } + } + + /** + * 强制立即对最近一个私聊会话触发一次见解（忽略冷却，用于测试）。 + * 返回触发结果描述，供设置页展示。 + */ + async triggerTest(): Promise<{ success: boolean; message: string }> { + insightLog('INFO', '手动触发测试见解...') + const apiBaseUrl = this.config.get('aiInsightApiBaseUrl') as string + const apiKey = this.config.get('aiInsightApiKey') as string + if (!apiBaseUrl || !apiKey) { + return { success: false, message: '请先填写 API 地址和 Key' } + } + try { + const connectResult = await chatService.connect() + if (!connectResult.success) { + return { success: false, message: '数据库连接失败，请先在"数据库连接"页完成配置' } + } + const sessionsResult = await chatService.getSessions() + if (!sessionsResult.success || !sessionsResult.sessions || sessionsResult.sessions.length === 0) { + return { success: false, message: '未找到任何会话，请确认数据库已正确连接' } + } + // 找第一个允许的私聊 + const session = (sessionsResult.sessions as ChatSession[]).find((s) => { + const id = s.username?.trim() || '' + return id && !id.endsWith('@chatroom') && !id.toLowerCase().includes('placeholder') && this.isSessionAllowed(id) + }) + if (!session) { + return { success: false, message: '未找到任何私聊会话（若已启用白名单，请检查是否有勾选的私聊）' } + } + const sessionId = session.username?.trim() || '' + const displayName = session.displayName || sessionId + insightLog('INFO', `测试目标会话：${displayName} (${sessionId})`) + await this.generateInsightForSession({ + sessionId, + displayName, + triggerReason: 'activity' + }) + return { success: true, message: `已向「${displayName}」发送测试见解，请查看右下角弹窗` } + } catch (e) { + return { success: false, message: `测试失败：${(e as Error).message}` } + } + } + + /** 获取今日触发统计（供设置页展示） */ + getTodayStats(): { sessionId: string; count: number; times: string[] }[] { + this.resetIfNewDay() + const result: { sessionId: string; count: number; times: string[] }[] = [] + for (const [sessionId, record] of this.todayTriggers.entries()) { + result.push({ + sessionId, + count: record.timestamps.length, + times: record.timestamps.map(formatTimestamp) + }) + } + return result + } + + // ── 私有方法 ──────────────────────────────────────────────────────────────── + + private isEnabled(): boolean { + return this.config.get('aiInsightEnabled') === true + } + + /** + * 判断某个会话是否允许触发见解。 + * 若白名单未启用，则所有私聊会话均允许； + * 若白名单已启用，则只有在白名单中的会话才允许。 + */ + private isSessionAllowed(sessionId: string): boolean { + const whitelistEnabled = this.config.get('aiInsightWhitelistEnabled') as boolean + if (!whitelistEnabled) return true + const whitelist = (this.config.get('aiInsightWhitelist') as string[]) || [] + return whitelist.includes(sessionId) + } + + /** + * 获取会话列表，优先使用缓存（15 分钟 TTL）。 + * 缓存命中时完全跳过数据库访问，避免频繁 connect() + getSessions() 消耗 CPU。 + * forceRefresh=true 时强制重新拉取（仅用于沉默扫描等低频场景）。 + */ + private async getSessionsCached(forceRefresh = false): Promise { + const now = Date.now() + // 缓存命中：直接返回，零数据库操作 + if ( + !forceRefresh && + this.sessionCache !== null && + now - this.sessionCacheAt < InsightService.SESSION_CACHE_TTL_MS + ) { + return this.sessionCache + } + // 缓存未命中或强制刷新：连接数据库并拉取 + try { + // 只在首次或强制刷新时调用 connect()，避免重复建立连接 + if (!this.dbConnected || forceRefresh) { + const connectResult = await chatService.connect() + if (!connectResult.success) { + insightLog('WARN', '数据库连接失败，使用旧缓存') + return this.sessionCache ?? [] + } + this.dbConnected = true + } + const result = await chatService.getSessions() + if (result.success && result.sessions) { + this.sessionCache = result.sessions as ChatSession[] + this.sessionCacheAt = now + } + } catch (e) { + insightLog('WARN', `获取会话缓存失败: ${(e as Error).message}`) + // 连接可能已断开，下次强制重连 + this.dbConnected = false + } + return this.sessionCache ?? [] + } + + private resetIfNewDay(): void { + const todayStart = getStartOfDay() + if (todayStart > this.todayDate) { + this.todayDate = todayStart + this.todayTriggers.clear() + } + } + + /** + * 记录触发并返回该会话今日所有触发时间（用于组装 prompt）。 + */ + private recordTrigger(sessionId: string): string[] { + this.resetIfNewDay() + const existing = this.todayTriggers.get(sessionId) ?? { timestamps: [] } + existing.timestamps.push(Date.now()) + this.todayTriggers.set(sessionId, existing) + return existing.timestamps.map(formatTimestamp) + } + + /** + * 获取今日全局已触发次数（所有会话合计），用于 prompt 中告知模���全局上下文。 + */ + private getTodayTotalTriggerCount(): number { + this.resetIfNewDay() + let total = 0 + for (const record of this.todayTriggers.values()) { + total += record.timestamps.length + } + return total + } + + // ── 沉默联系人扫描 ────────────────────────────────────────────────────────── + + private scheduleSilenceScan(): void { + // 等待扫描完成后再安排下一次，避免并发堆积 + const scheduleNext = () => { + if (!this.started) return + const intervalHours = (this.config.get('aiInsightScanIntervalHours') as number) || 4 + const intervalMs = Math.max(0.1, intervalHours) * 60 * 60 * 1000 + insightLog('INFO', `下次沉默扫描将在 ${intervalHours} 小时后执行`) + this.silenceScanTimer = setTimeout(async () => { + this.silenceScanTimer = null + await this.runSilenceScan() + scheduleNext() + }, intervalMs) + } + + this.silenceInitialDelayTimer = setTimeout(async () => { + this.silenceInitialDelayTimer = null + await this.runSilenceScan() + scheduleNext() + }, SILENCE_SCAN_INITIAL_DELAY_MS) + } + + private async runSilenceScan(): Promise { + if (!this.isEnabled()) { + insightLog('INFO', '沉默扫描：AI 见解未启用，跳过') + return + } + if (this.processing) { + insightLog('INFO', '沉默扫描：正在处理中，跳过本次') + return + } + + this.processing = true + insightLog('INFO', '开始沉默联系人扫描...') + try { + const silenceDays = (this.config.get('aiInsightSilenceDays') as number) || DEFAULT_SILENCE_DAYS + const thresholdMs = silenceDays * 24 * 60 * 60 * 1000 + const now = Date.now() + + insightLog('INFO', `沉默阈值：${silenceDays} 天`) + + // 沉默扫描间隔较长，强制刷新缓存以获取最新数据 + const sessions = await this.getSessionsCached(true) + if (sessions.length === 0) { + insightLog('WARN', '获取会话列表失败，跳过沉默扫描') + return + } + + insightLog('INFO', `共 ${sessions.length} 个会话，开始过滤...`) + + let silentCount = 0 + for (const session of sessions) { + const sessionId = session.username?.trim() || '' + if (!sessionId || sessionId.endsWith('@chatroom')) continue + if (sessionId.toLowerCase().includes('placeholder')) continue + if (!this.isSessionAllowed(sessionId)) continue + + const lastTimestamp = (session.lastTimestamp || 0) * 1000 + if (!lastTimestamp || lastTimestamp <= 0) continue + + const silentMs = now - lastTimestamp + if (silentMs < thresholdMs) continue + + silentCount++ + const silentDays = Math.floor(silentMs / (24 * 60 * 60 * 1000)) + insightLog('INFO', `发现沉默联系人：${session.displayName || sessionId}，已沉默 ${silentDays} 天`) + + await this.generateInsightForSession({ + sessionId, + displayName: session.displayName || session.username, + triggerReason: 'silence', + silentDays + }) + } + insightLog('INFO', `沉默扫描完成，共发现 ${silentCount} 个沉默联系人`) + } catch (e) { + insightLog('ERROR', `沉默扫描出错: ${(e as Error).message}`) + } finally { + this.processing = false + } + } + + // ── 活跃会话分析 ──────────────────────────────────────────────────────────── + + /** + * 在 DB 变更防抖后执行，分析最近活跃的会话。 + * + * 触发条件（必须同时满足）： + * 1. 会话有真正的新消息（lastTimestamp 比上次见到的更新） + * 2. 该会话距上次活跃分析已超过冷却期 + * + * 白名单启用时：直接使用白名单里的 sessionId，完全跳过 getSessions()。 + * 白名单未启用时：从缓存拉取全量会话后过滤私聊。 + */ + private async analyzeRecentActivity(): Promise { + if (!this.isEnabled()) return + if (this.processing) return + + this.processing = true + try { + const now = Date.now() + const cooldownMinutes = (this.config.get('aiInsightCooldownMinutes') as number) ?? 120 + const cooldownMs = cooldownMinutes * 60 * 1000 + const whitelistEnabled = this.config.get('aiInsightWhitelistEnabled') as boolean + const whitelist = (this.config.get('aiInsightWhitelist') as string[]) || [] + + // 白名单启用且有勾选项时，直接用白名单 sessionId，无需查数据库全量会话列表。 + // 通过拉取该会话最新 1 条消息时间戳判断是否真正有新消息，开销极低。 + if (whitelistEnabled && whitelist.length > 0) { + // 确保数据库已连接（首次时连接，之后复用） + if (!this.dbConnected) { + const connectResult = await chatService.connect() + if (!connectResult.success) return + this.dbConnected = true + } + + for (const sessionId of whitelist) { + if (!sessionId || sessionId.endsWith('@chatroom')) continue + + // 冷却期检查（先过滤，减少不必要的 DB 查询） + if (cooldownMs > 0) { + const lastAnalysis = this.lastActivityAnalysis.get(sessionId) ?? 0 + if (cooldownMs - (now - lastAnalysis) > 0) continue + } + + // 拉取最新 1 条消息，用时间戳判断是否有新消息，避免全量 getSessions() + try { + const msgsResult = await chatService.getLatestMessages(sessionId, 1) + if (!msgsResult.success || !msgsResult.messages || msgsResult.messages.length === 0) continue + + const latestMsg = msgsResult.messages[0] + const latestTs = Number(latestMsg.createTime) || 0 + const lastSeen = this.lastSeenTimestamp.get(sessionId) ?? 0 + + if (latestTs <= lastSeen) continue // 没有新消息 + this.lastSeenTimestamp.set(sessionId, latestTs) + } catch { + continue + } + + insightLog('INFO', `白名单会话 ${sessionId} 有新消息，准备生成见解...`) + this.lastActivityAnalysis.set(sessionId, now) + + // displayName 使用白名单 sessionId，generateInsightForSession 内部会从上下文里获取真实名称 + await this.generateInsightForSession({ + sessionId, + displayName: sessionId, + triggerReason: 'activity' + }) + break // 每次最多处理 1 个会话 + } + return + } + + // 白名单未启用：需要拉取全量会话列表，从中过滤私聊 + const sessions = await this.getSessionsCached() + if (sessions.length === 0) return + + const privateSessions = sessions.filter((s) => { + const id = s.username?.trim() || '' + return id && !id.endsWith('@chatroom') && !id.toLowerCase().includes('placeholder') + }) + + for (const session of privateSessions.slice(0, 10)) { + const sessionId = session.username?.trim() || '' + if (!sessionId) continue + + const currentTimestamp = session.lastTimestamp || 0 + const lastSeen = this.lastSeenTimestamp.get(sessionId) ?? 0 + if (currentTimestamp <= lastSeen) continue + this.lastSeenTimestamp.set(sessionId, currentTimestamp) + + if (cooldownMs > 0) { + const lastAnalysis = this.lastActivityAnalysis.get(sessionId) ?? 0 + if (cooldownMs - (now - lastAnalysis) > 0) continue + } + + insightLog('INFO', `${session.displayName || sessionId} 有新消息，准备生成见解...`) + this.lastActivityAnalysis.set(sessionId, now) + + await this.generateInsightForSession({ + sessionId, + displayName: session.displayName || session.username, + triggerReason: 'activity' + }) + break + } + } catch (e) { + insightLog('ERROR', `活跃分析出错: ${(e as Error).message}`) + } finally { + this.processing = false + } + } + + // ── 核心见解生成 ──────────────────────────────────────────────────────────── + + private async generateInsightForSession(params: { + sessionId: string + displayName: string + triggerReason: 'activity' | 'silence' + silentDays?: number + }): Promise { + const { sessionId, displayName, triggerReason, silentDays } = params + if (!sessionId) return + + const apiBaseUrl = this.config.get('aiInsightApiBaseUrl') as string + const apiKey = this.config.get('aiInsightApiKey') as string + const model = (this.config.get('aiInsightApiModel') as string) || 'gpt-4o-mini' + const allowContext = this.config.get('aiInsightAllowContext') as boolean + const contextCount = (this.config.get('aiInsightContextCount') as number) || 40 + + insightLog('INFO', `generateInsightForSession: sessionId=${sessionId}, reason=${triggerReason}, contextCount=${contextCount}, api=${apiBaseUrl ? '已配置' : '未配置'}`) + + if (!apiBaseUrl || !apiKey) { + insightLog('WARN', 'API 地址或 Key 未配置，跳过见解生成') + return + } + + // ── 构建 prompt ─────────────���───────────────────────────────���──────────── + + // 今日触发统计（让模型具备时间与克制感） + const sessionTriggerTimes = this.recordTrigger(sessionId) + const totalTodayTriggers = this.getTodayTotalTriggerCount() + + let contextSection = '' + if (allowContext) { + try { + const msgsResult = await chatService.getLatestMessages(sessionId, contextCount) + if (msgsResult.success && msgsResult.messages && msgsResult.messages.length > 0) { + const messages: Message[] = msgsResult.messages + const msgLines = messages.map((m) => { + const sender = m.isSend === 1 ? '我' : (displayName || sessionId) + const content = m.rawContent || m.parsedContent || '[非文字消息]' + const time = new Date(Number(m.createTime) * 1000).toLocaleString('zh-CN') + return `[${time}] ${sender}：${content}` + }) + contextSection = `\n\n近期对话记录（最近 ${msgLines.length} 条）：\n${msgLines.join('\n')}` + insightLog('INFO', `已加载 ${msgLines.length} 条上下文消息`) + } + } catch (e) { + insightLog('WARN', `拉取上下文失败: ${(e as Error).message}`) + } + } + + // ── 默认 system prompt（稳定内容，有利于 provider 端 prompt cache 命中）──── + const DEFAULT_SYSTEM_PROMPT = `你是用户的私人关系观察助手，名叫"见解"。你的任务是主动提供有价值的观察和建议。 + +要求： +1. 必须给出见解。基于聊天记录分析对方情绪、话题趋势、关系动态，或给出回复建议、聊天话题推荐。 +2. 控制在 80 字以内，直接、具体、一针见血。不要废话。 +3. 输出纯文本，不使用 Markdown。 +4. 只有在完全没有任何可说的内容时（比如对话只有一条"嗯"），才回复"SKIP"。绝大多数情况下你应该输出见解。` + + // 优先使用用户自定义 prompt，为空则使用默认值 + const customPrompt = (this.config.get('aiInsightSystemPrompt') as string) || '' + const systemPrompt = customPrompt.trim() || DEFAULT_SYSTEM_PROMPT + + // 可变的上下文统计信息放在 user message 里，保持 system prompt 稳定不变 + // 这样 provider 端（Anthropic/OpenAI）能最大化命中 prompt cache，降低费用 + const triggerDesc = + triggerReason === 'silence' + ? `你已经 ${silentDays} 天没有和「${displayName}」聊天了。` + : `你最近和「${displayName}」有新的聊天动态。` + + const todayStatsDesc = + sessionTriggerTimes.length > 1 + ? `今天你已经针对「${displayName}」收到过 ${sessionTriggerTimes.length - 1} 条见解（时间：${sessionTriggerTimes.slice(0, -1).join('、')}），请适当克制。` + : `今天你还没有针对「${displayName}」发出过见解。` + + const globalStatsDesc = `今天全部联系人合计已触发 ${totalTodayTriggers} 条见解。` + + const userPrompt = `触发原因：${triggerDesc} +时间统计：${todayStatsDesc} ${globalStatsDesc}${contextSection} + +请给出你的见解（≤80字）：` + + const endpoint = buildApiUrl(apiBaseUrl, '/chat/completions') + insightLog('INFO', `准备调用 API: ${endpoint}，模型: ${model}`) + + try { + const result = await callApi( + apiBaseUrl, + apiKey, + model, + [ + { role: 'system', content: systemPrompt }, + { role: 'user', content: userPrompt } + ] + ) + + insightLog('INFO', `API 返回原文: ${result.slice(0, 150)}`) + + // 模型主动选择跳过 + if (result.trim().toUpperCase() === 'SKIP' || result.trim().startsWith('SKIP')) { + insightLog('INFO', `模型选择跳过 ${displayName}`) + return + } + + const insight = result.slice(0, 120) + const notifTitle = `见解 · ${displayName}` + + insightLog('INFO', `推送通知 → ${displayName}: ${insight}`) + + // 渠道一：Electron 原生系统通知 + if (Notification.isSupported()) { + const notif = new Notification({ title: notifTitle, body: insight, silent: false }) + notif.show() + } else { + insightLog('WARN', '当前系统不支持原生通知') + } + + // 渠道二：Telegram Bot 推送（可选） + const telegramEnabled = this.config.get('aiInsightTelegramEnabled') as boolean + if (telegramEnabled) { + const telegramToken = (this.config.get('aiInsightTelegramToken') as string) || '' + const telegramChatIds = (this.config.get('aiInsightTelegramChatIds') as string) || '' + if (telegramToken && telegramChatIds) { + const chatIds = telegramChatIds.split(',').map((s) => s.trim()).filter(Boolean) + const telegramText = `【WeFlow】 ${notifTitle}\n\n${insight}` + for (const chatId of chatIds) { + this.sendTelegram(telegramToken, chatId, telegramText).catch((e) => { + insightLog('WARN', `Telegram 推送失败 (chatId=${chatId}): ${(e as Error).message}`) + }) + } + } else { + insightLog('WARN', 'Telegram 已启用但 Token 或 Chat ID 未填写，跳过') + } + } + + insightLog('INFO', `已为 ${displayName} 推送见解`) + } catch (e) { + insightLog('ERROR', `API 调用失败 (${displayName}): ${(e as Error).message}`) + } + } + + /** + * 通过 Telegram Bot API 发送消息。 + * 使用 Node 原生 https 模块，无需第三方依赖。 + */ + private sendTelegram(token: string, chatId: string, text: string): Promise { + return new Promise((resolve, reject) => { + const body = JSON.stringify({ chat_id: chatId, text, parse_mode: 'HTML' }) + const options = { + hostname: 'api.telegram.org', + port: 443, + path: `/bot${token}/sendMessage`, + method: 'POST' as const, + headers: { + 'Content-Type': 'application/json', + 'Content-Length': Buffer.byteLength(body).toString() + } + } + const req = https.request(options, (res) => { + let data = '' + res.on('data', (chunk) => { data += chunk }) + res.on('end', () => { + try { + const parsed = JSON.parse(data) + if (parsed.ok) { + resolve() + } else { + reject(new Error(parsed.description || '未知错误')) + } + } catch { + reject(new Error(`响应解析失败: ${data.slice(0, 100)}`)) + } + }) + }) + req.setTimeout(15_000, () => { req.destroy(); reject(new Error('Telegram 请求超时')) }) + req.on('error', reject) + req.write(body) + req.end() + }) + } +} + +export const insightService = new InsightService() diff --git a/src/pages/SettingsPage.tsx b/src/pages/SettingsPage.tsx index 316228c..35f989f 100644 --- a/src/pages/SettingsPage.tsx +++ b/src/pages/SettingsPage.tsx @@ -10,12 +10,13 @@ import { Eye, EyeOff, FolderSearch, FolderOpen, Search, Copy, RotateCcw, Trash2, Plug, Check, Sun, Moon, Monitor, Palette, Database, HardDrive, Info, RefreshCw, ChevronDown, Download, Mic, - ShieldCheck, Fingerprint, Lock, KeyRound, Bell, Globe, BarChart2, X, UserRound + ShieldCheck, Fingerprint, Lock, KeyRound, Bell, Globe, BarChart2, X, UserRound, + Sparkles, Loader2, CheckCircle2, XCircle } from 'lucide-react' import { Avatar } from '../components/Avatar' import './SettingsPage.scss' -type SettingsTab = 'appearance' | 'notification' | 'antiRevoke' | 'database' | 'models' | 'cache' | 'api' | 'updates' | 'security' | 'about' | 'analytics' +type SettingsTab = 'appearance' | 'notification' | 'antiRevoke' | 'database' | 'models' | 'cache' | 'api' | 'updates' | 'security' | 'about' | 'analytics' | 'insight' const tabs: { id: SettingsTab; label: string; icon: React.ElementType }[] = [ { id: 'appearance', label: '外观', icon: Palette }, @@ -26,6 +27,7 @@ const tabs: { id: SettingsTab; label: string; icon: React.ElementType }[] = [ { id: 'cache', label: '缓存', icon: HardDrive }, { id: 'api', label: 'API 服务', icon: Globe }, { id: 'analytics', label: '分析', icon: BarChart2 }, + { id: 'insight', label: 'AI 见解', icon: Sparkles }, { id: 'security', label: '安全', icon: ShieldCheck }, { id: 'updates', label: '版本更新', icon: RefreshCw }, { id: 'about', label: '关于', icon: Info } @@ -123,7 +125,7 @@ function SettingsPage({ onClose }: SettingsPageProps = {}) { setHttpApiToken(token) await configService.setHttpApiToken(token) - showMessage('已生成并保存新的 Access Token', true) + showMessage('已生成��保存新的 Access Token', true) } const clearApiToken = async () => { @@ -213,6 +215,29 @@ function SettingsPage({ onClose }: SettingsPageProps = {}) { const isClearingCache = isClearingAnalyticsCache || isClearingImageCache || isClearingAllCache + // AI 见解 state + const [aiInsightEnabled, setAiInsightEnabled] = useState(false) + const [aiInsightApiBaseUrl, setAiInsightApiBaseUrl] = useState('') + const [aiInsightApiKey, setAiInsightApiKey] = useState('') + const [aiInsightApiModel, setAiInsightApiModel] = useState('gpt-4o-mini') + const [aiInsightSilenceDays, setAiInsightSilenceDays] = useState(3) + const [aiInsightAllowContext, setAiInsightAllowContext] = useState(false) + const [isTestingInsight, setIsTestingInsight] = useState(false) + const [insightTestResult, setInsightTestResult] = useState<{ success: boolean; message: string } | null>(null) + const [showInsightApiKey, setShowInsightApiKey] = useState(false) + const [isTriggeringInsightTest, setIsTriggeringInsightTest] = useState(false) + const [insightTriggerResult, setInsightTriggerResult] = useState<{ success: boolean; message: string } | null>(null) + const [aiInsightWhitelistEnabled, setAiInsightWhitelistEnabled] = useState(false) + const [aiInsightWhitelist, setAiInsightWhitelist] = useState>(new Set()) + const [insightWhitelistSearch, setInsightWhitelistSearch] = useState('') + const [aiInsightCooldownMinutes, setAiInsightCooldownMinutes] = useState(120) + const [aiInsightScanIntervalHours, setAiInsightScanIntervalHours] = useState(4) + const [aiInsightContextCount, setAiInsightContextCount] = useState(40) + const [aiInsightSystemPrompt, setAiInsightSystemPrompt] = useState('') + const [aiInsightTelegramEnabled, setAiInsightTelegramEnabled] = useState(false) + const [aiInsightTelegramToken, setAiInsightTelegramToken] = useState('') + const [aiInsightTelegramChatIds, setAiInsightTelegramChatIds] = useState('') + const [isWayland, setIsWayland] = useState(false) useEffect(() => { const checkWaylandStatus = async () => { @@ -438,6 +463,37 @@ function SettingsPage({ onClose }: SettingsPageProps = {}) { if (savedWhisperModelDir) setWhisperModelDir(savedWhisperModelDir) + // 加载 AI 见解配置 + const savedAiInsightEnabled = await configService.getAiInsightEnabled() + const savedAiInsightApiBaseUrl = await configService.getAiInsightApiBaseUrl() + const savedAiInsightApiKey = await configService.getAiInsightApiKey() + const savedAiInsightApiModel = await configService.getAiInsightApiModel() + const savedAiInsightSilenceDays = await configService.getAiInsightSilenceDays() + const savedAiInsightAllowContext = await configService.getAiInsightAllowContext() + const savedAiInsightWhitelistEnabled = await configService.getAiInsightWhitelistEnabled() + const savedAiInsightWhitelist = await configService.getAiInsightWhitelist() + const savedAiInsightCooldownMinutes = await configService.getAiInsightCooldownMinutes() + const savedAiInsightScanIntervalHours = await configService.getAiInsightScanIntervalHours() + const savedAiInsightContextCount = await configService.getAiInsightContextCount() + const savedAiInsightSystemPrompt = await configService.getAiInsightSystemPrompt() + const savedAiInsightTelegramEnabled = await configService.getAiInsightTelegramEnabled() + const savedAiInsightTelegramToken = await configService.getAiInsightTelegramToken() + const savedAiInsightTelegramChatIds = await configService.getAiInsightTelegramChatIds() + setAiInsightEnabled(savedAiInsightEnabled) + setAiInsightApiBaseUrl(savedAiInsightApiBaseUrl) + setAiInsightApiKey(savedAiInsightApiKey) + setAiInsightApiModel(savedAiInsightApiModel) + setAiInsightSilenceDays(savedAiInsightSilenceDays) + setAiInsightAllowContext(savedAiInsightAllowContext) + setAiInsightWhitelistEnabled(savedAiInsightWhitelistEnabled) + setAiInsightWhitelist(new Set(savedAiInsightWhitelist)) + setAiInsightCooldownMinutes(savedAiInsightCooldownMinutes) + setAiInsightScanIntervalHours(savedAiInsightScanIntervalHours) + setAiInsightContextCount(savedAiInsightContextCount) + setAiInsightSystemPrompt(savedAiInsightSystemPrompt) + setAiInsightTelegramEnabled(savedAiInsightTelegramEnabled) + setAiInsightTelegramToken(savedAiInsightTelegramToken) + setAiInsightTelegramChatIds(savedAiInsightTelegramChatIds) } catch (e: any) { console.error('加载配置失败:', e) @@ -579,7 +635,7 @@ function SettingsPage({ onClose }: SettingsPageProps = {}) { showMessage(`已切换到${channelLabel}更新渠道，正在检查更新`, true) await handleCheckUpdate() } catch (e: any) { - showMessage(`切换更新渠道失败: ${e}`, false) + showMessage(`切换更新渠道��败: ${e}`, false) } } @@ -820,16 +876,19 @@ function SettingsPage({ onClose }: SettingsPageProps = {}) { } useEffect(() => { - if (activeTab !== 'antiRevoke') return + if (activeTab !== 'antiRevoke' && activeTab !== 'insight') return let canceled = false ;(async () => { try { + // 两个 Tab 都需要会话列表；antiRevoke 还需要额外检查防撤回状态 const sessionIds = await ensureAntiRevokeSessionsLoaded() if (canceled) return - await handleRefreshAntiRevokeStatus(sessionIds) + if (activeTab === 'antiRevoke') { + await handleRefreshAntiRevokeStatus(sessionIds) + } } catch (e: any) { if (!canceled) { - showMessage(`加载防撤回会话失败: ${e?.message || String(e)}`, false) + showMessage(`加载会话失败: ${e?.message || String(e)}`, false) } } })() @@ -1171,7 +1230,7 @@ function SettingsPage({ onClose }: SettingsPageProps = {}) { if (result.success && result.aesKey) { if (typeof result.xorKey === 'number') setImageXorKey(`0x${result.xorKey.toString(16).toUpperCase().padStart(2, '0')}`) setImageAesKey(result.aesKey) - setImageKeyStatus('已获取图片密钥') + setImageKeyStatus('已获取图片��钥') showMessage('已自动获取图片密钥', true) const newXorKey = typeof result.xorKey === 'number' ? result.xorKey : 0 const newAesKey = result.aesKey @@ -1613,7 +1672,7 @@ function SettingsPage({ onClose }: SettingsPageProps = {}) {
- 开启后，收到新消息时将显示桌面弹窗通知 + 开启后，收���新消息时将显示桌面弹窗通知
{notificationEnabled ? '已开启' : '已关闭'}