weflow-cli 1.5.0

package/src/services/configService.ts

@@ -0,0 +1,197 @@
+import { join } from 'path'
+import { existsSync, readFileSync, writeFileSync, mkdirSync } from 'fs'
+import { homedir, hostname, userInfo } from 'os'
+import crypto from 'crypto'
+import { expandHomePath } from '../utils/pathUtils.js'
+import type { ConfigData } from '../types.js'
+
+const LOCK_PREFIX = 'lock:'
+
+interface CliConfig {
+  dbPath: string
+  wxid: string
+  decryptKey: string
+  decryptKey3x: string
+  dataVersion: string
+  dbPath3x: string
+  ntDbPath: string
+  ntKey: string
+  ntSalt: string
+  contactDbPath: string
+  contactKey: string
+  contactSalt: string
+  // 微信消息通道
+  wechatOcToken: string
+  wechatOcAccountId: string
+  wechatOcBaseUrl: string
+  wechatOcSyncBuf: string
+  // 白名单
+  whitelist: string[]
+  // Vault & Pipeline
+  vaultRepo: string
+  aiEngine: string
+  // 微信读书
+  wereadApiKey: string
+  // AI 引擎 API Key
+  deepseekApiKey: string
+}
+
+/** 需要加密存储的字段 */
+const ENCRYPTED_KEYS: ReadonlySet<keyof CliConfig> = new Set([
+  'decryptKey', 'decryptKey3x', 'ntKey', 'contactKey',
+  'wechatOcToken', 'wereadApiKey', 'deepseekApiKey',
+])
+
+const CONFIG_DIR = join(homedir(), '.weflow-cli')
+const CONFIG_FILE = join(CONFIG_DIR, 'config.json')
+
+export class ConfigService {
+  private config: CliConfig = { dbPath: '', wxid: '', decryptKey: '', decryptKey3x: '', dataVersion: '', dbPath3x: '', ntDbPath: '', ntKey: '', ntSalt: '', contactDbPath: '', contactKey: '', contactSalt: '', wechatOcToken: '', wechatOcAccountId: '', wechatOcBaseUrl: '', wechatOcSyncBuf: '', whitelist: [], vaultRepo: '', aiEngine: 'deepseek', wereadApiKey: '', deepseekApiKey: '' }
+
+  constructor() {
+    this.load()
+  }
+
+  private load(): void {
+    try {
+      if (existsSync(CONFIG_FILE)) {
+        const data = JSON.parse(readFileSync(CONFIG_FILE, 'utf8'))
+        this.config = {
+          dbPath: data.dbPath || '',
+          wxid: data.wxid || '',
+          decryptKey: data.decryptKey || '',
+          decryptKey3x: data.decryptKey3x || '',
+          dataVersion: data.dataVersion || '',
+          dbPath3x: data.dbPath3x || '',
+          ntDbPath: data.ntDbPath || '',
+          ntKey: data.ntKey || '',
+          ntSalt: data.ntSalt || '',
+          contactDbPath: data.contactDbPath || '',
+          contactKey: data.contactKey || '',
+          contactSalt: data.contactSalt || '',
+          wechatOcToken: data.wechatOcToken || '',
+          wechatOcAccountId: data.wechatOcAccountId || '',
+          wechatOcBaseUrl: data.wechatOcBaseUrl || '',
+          wechatOcSyncBuf: data.wechatOcSyncBuf || '',
+          whitelist: Array.isArray(data.whitelist) ? data.whitelist : [],
+          vaultRepo: data.vaultRepo || '',
+          aiEngine: data.aiEngine || 'deepseek',
+          wereadApiKey: data.wereadApiKey || '',
+          deepseekApiKey: data.deepseekApiKey || '',
+        }
+      }
+    } catch (e) {
+      // 配置文件损坏，输出警告并使用默认值
+      console.warn(`[weflow-cli] 配置文件解析失败 (${CONFIG_FILE}): ${e instanceof Error ? e.message : e}，将使用默认配置`)
+    }
+  }
+
+  private save(): void {
+    try {
+      if (!existsSync(CONFIG_DIR)) {
+        mkdirSync(CONFIG_DIR, { recursive: true })
+      }
+      writeFileSync(CONFIG_FILE, JSON.stringify(this.config, null, 2), 'utf8')
+    } catch (e) {
+      console.error('保存配置失败:', e)
+    }
+  }
+
+  get<K extends keyof CliConfig>(key: K): CliConfig[K] {
+    const raw = this.config[key]
+    if (ENCRYPTED_KEYS.has(key) && typeof raw === 'string' && raw.startsWith(LOCK_PREFIX)) {
+      return this.lockDecrypt(raw) as CliConfig[K]
+    }
+    if ((key === 'dbPath' || key === 'dbPath3x' || key === 'ntDbPath') && typeof raw === 'string') {
+      return expandHomePath(raw) as CliConfig[K]
+    }
+    return raw
+  }
+
+  set<K extends keyof CliConfig>(key: K, value: CliConfig[K]): void {
+    if (ENCRYPTED_KEYS.has(key) && typeof value === 'string' && value) {
+      this.config[key] = this.lockEncrypt(value) as CliConfig[K]
+    } else if ((key === 'dbPath' || key === 'dbPath3x' || key === 'ntDbPath') && typeof value === 'string') {
+      this.config[key] = expandHomePath(value) as CliConfig[K]
+    } else {
+      this.config[key] = value
+    }
+    this.save()
+  }
+
+  getAll(): ConfigData {
+    return {
+      dbPath: this.get('dbPath'),
+      wxid: this.get('wxid'),
+      decryptKey: this.get('decryptKey'),
+      decryptKey3x: this.get('decryptKey3x'),
+      dataVersion: (this.config.dataVersion || '4.x') as ConfigData['dataVersion'],
+      dbPath3x: this.get('dbPath3x'),
+      ntDbPath: this.config.ntDbPath,
+      ntKey: this.get('ntKey'),
+      ntSalt: this.config.ntSalt,
+      contactDbPath: this.config.contactDbPath,
+      contactKey: this.get('contactKey'),
+      contactSalt: this.config.contactSalt,
+    }
+  }
+
+  getWhitelist(): string[] {
+    return this.config.whitelist || []
+  }
+
+  setWhitelist(list: string[]): void {
+    this.config.whitelist = list
+    this.save()
+  }
+
+  isConfigured(): boolean {
+    const has4x = !!(this.config.dbPath && this.config.decryptKey)
+    const has3x = !!(this.config.dbPath3x && this.config.decryptKey3x)
+    const hasNt = !!(this.config.ntDbPath && this.config.ntKey && this.config.ntSalt)
+    return has4x || has3x || hasNt
+  }
+
+  clear(): void {
+    this.config = { dbPath: '', wxid: '', decryptKey: '', decryptKey3x: '', dataVersion: '', dbPath3x: '', ntDbPath: '', ntKey: '', ntSalt: '', contactDbPath: '', contactKey: '', contactSalt: '', wechatOcToken: '', wechatOcAccountId: '', wechatOcBaseUrl: '', wechatOcSyncBuf: '', whitelist: [], vaultRepo: '', aiEngine: 'deepseek', wereadApiKey: '', deepseekApiKey: '' }
+    this.save()
+  }
+
+  private lockEncrypt(plaintext: string): string {
+    const machineId = this.getMachineId()
+    const salt = crypto.randomBytes(16)
+    const iv = crypto.randomBytes(12)
+    const derivedKey = crypto.pbkdf2Sync(machineId, salt, 100000, 32, 'sha256')
+    const cipher = crypto.createCipheriv('aes-256-gcm', derivedKey, iv)
+    const encrypted = Buffer.concat([cipher.update(plaintext, 'utf8'), cipher.final()])
+    const authTag = cipher.getAuthTag()
+    const combined = Buffer.concat([salt, iv, authTag, encrypted])
+    return LOCK_PREFIX + combined.toString('base64')
+  }
+
+  private lockDecrypt(stored: string): string {
+    if (!stored || !stored.startsWith(LOCK_PREFIX)) return stored
+    try {
+      const machineId = this.getMachineId()
+      const combined = Buffer.from(stored.slice(LOCK_PREFIX.length), 'base64')
+      const salt = combined.subarray(0, 16)
+      const iv = combined.subarray(16, 28)
+      const authTag = combined.subarray(28, 44)
+      const ciphertext = combined.subarray(44)
+      const derivedKey = crypto.pbkdf2Sync(machineId, salt, 100000, 32, 'sha256')
+      const decipher = crypto.createDecipheriv('aes-256-gcm', derivedKey, iv)
+      decipher.setAuthTag(authTag)
+      const decrypted = Buffer.concat([decipher.update(ciphertext), decipher.final()])
+      return decrypted.toString('utf8')
+    } catch {
+      return ''
+    }
+  }
+
+  private getMachineId(): string {
+    // 使用机器名 + 用户名作为机器标识，足够绑定到单台机器
+    return `${hostname()}-${userInfo().username}-weflow-cli`
+  }
+}
+
+export const configService = new ConfigService()

package/src/services/exportService.ts

@@ -0,0 +1,305 @@
+import { join, basename, dirname } from 'path'
+import { existsSync, mkdirSync, writeFileSync } from 'fs'
+import { execFile } from 'child_process'
+import { promisify } from 'util'
+import { fileURLToPath } from 'url'
+import { chatService } from './chatService.js'
+import { configService } from './configService.js'
+import type { Message } from '../types.js'
+
+const execFileAsync = promisify(execFile)
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+
+export class ExportService {
+  async exportJson(talker: string, outputDir: string, limit = 0): Promise<{ success: boolean; path?: string; error?: string }> {
+    try {
+      const messages = await chatService.getMessages(talker, limit)
+      if (messages.length === 0) {
+        return { success: false, error: '未找到消息' }
+      }
+
+      const dir = this.ensureOutputDir(outputDir)
+      const filePath = join(dir, `${talker}_messages.json`)
+      writeFileSync(filePath, JSON.stringify(messages, null, 2), 'utf8')
+      return { success: true, path: filePath }
+    } catch (e) {
+      return { success: false, error: String(e) }
+    }
+  }
+
+  async exportTxt(talker: string, outputDir: string, limit = 0): Promise<{ success: boolean; path?: string; error?: string }> {
+    try {
+      const messages = await chatService.getMessages(talker, limit)
+      if (messages.length === 0) {
+        return { success: false, error: '未找到消息' }
+      }
+
+      const lines = messages.map(m => {
+        const time = new Date(m.createTime * 1000).toLocaleString('zh-CN')
+        const sender = m.isSend ? '我' : (m.senderUsername || talker)
+        const content = m.parsedContent || m.rawContent || ''
+        return `[${time}] ${sender}: ${content}`
+      })
+
+      const dir = this.ensureOutputDir(outputDir)
+      const filePath = join(dir, `${talker}_messages.txt`)
+      writeFileSync(filePath, lines.join('\n'), 'utf8')
+      return { success: true, path: filePath }
+    } catch (e) {
+      return { success: false, error: String(e) }
+    }
+  }
+
+  async exportHtml(talker: string, outputDir: string, limit = 0): Promise<{ success: boolean; path?: string; error?: string }> {
+    try {
+      // Use Python export_chat_html.py for rich HTML with images
+      const cfg = configService.getAll()
+      const db = cfg.ntDbPath
+      const key = cfg.ntKey
+      const salt = cfg.ntSalt
+
+      if (!db || !key || !salt) {
+        // Fallback: basic HTML
+        return this.exportHtmlBasic(talker, outputDir, limit)
+      }
+
+      // Resolve Python script path (dist/src/services → package root)
+      const pkgRoot = join(__dirname, '..', '..', '..')
+      const script = join(pkgRoot, 'scripts', 'export_chat_html.py')
+
+      // Cache dir for image thumbnails
+      const cacheDir = cfg.contactDbPath
+        ? join(dirname(dirname(cfg.contactDbPath)), 'cache')
+        : ''
+
+      // Resolve display name for filename (prefer remark/nickname over wxid)
+      let displayName = ''
+      try {
+        const sessions = await chatService.listSessions()
+        const match = sessions.find(s => s.username === talker)
+        if (match?.displayName && match.displayName !== talker) {
+          displayName = match.displayName
+        }
+      } catch {}
+
+      const args: string[] = [
+        script,
+        '--db', db,
+        '--key', key,
+        '--salt', salt,
+        '--talker', talker,
+        '--out', outputDir || `./output`,
+        '--single',
+        '--parts', '1',
+      ]
+      if (displayName) {
+        args.push('--name', displayName)
+      }
+      if (cacheDir && existsSync(cacheDir)) {
+        args.push('--cache-dir', cacheDir)
+      }
+
+      console.log(`  Exporting HTML via Python...`)
+      const { stdout } = await execFileAsync('python', args, {
+        timeout: 300_000,
+        maxBuffer: 50 * 1024 * 1024,
+      })
+
+      // Parse JSON result from Python output
+      const lines = stdout.split('\n').filter((l: string) => l.trim())
+      for (let i = lines.length - 1; i >= 0; i--) {
+        try {
+          const result = JSON.parse(lines[i])
+          if (result.success && result.files?.length > 0) {
+            return { success: true, path: result.files[0] }
+          }
+        } catch {}
+      }
+
+      return { success: false, error: 'Python export succeeded but no output found' }
+    } catch (e: any) {
+      console.error(`  Python export failed: ${e.message}`)
+      // Fallback to basic HTML
+      return this.exportHtmlBasic(talker, outputDir, limit)
+    }
+  }
+
+  private async exportHtmlBasic(talker: string, outputDir: string, limit = 0): Promise<{ success: boolean; path?: string; error?: string }> {
+    const messages = await chatService.getMessages(talker, limit)
+    if (messages.length === 0) {
+      return { success: false, error: '未找到消息' }
+    }
+
+    const html = this.buildHtml(talker, messages)
+    const dir = this.ensureOutputDir(outputDir)
+    const filePath = join(dir, `${talker}_messages.html`)
+    writeFileSync(filePath, html, 'utf8')
+    return { success: true, path: filePath }
+  }
+
+  async exportExcel(talker: string, outputDir: string, limit = 0): Promise<{ success: boolean; path?: string; error?: string }> {
+    try {
+      const ExcelJS = await import('exceljs')
+      const messages = await chatService.getMessages(talker, limit)
+      if (messages.length === 0) {
+        return { success: false, error: '未找到消息' }
+      }
+
+      const workbook = new ExcelJS.Workbook()
+      const sheet = workbook.addWorksheet('聊天记录')
+
+      sheet.columns = [
+        { header: '时间', key: 'time', width: 20 },
+        { header: '发送者', key: 'sender', width: 15 },
+        { header: '类型', key: 'type', width: 10 },
+        { header: '内容', key: 'content', width: 80 },
+        { header: '消息ID', key: 'msgId', width: 15 }
+      ]
+
+      for (const m of messages) {
+        sheet.addRow({
+          time: new Date(m.createTime * 1000).toLocaleString('zh-CN'),
+          sender: m.isSend ? '我' : (m.senderUsername || talker),
+          type: this.getMessageTypeName(m.localType),
+          content: m.parsedContent || m.rawContent || '',
+          msgId: m.localId
+        })
+      }
+
+      const dir = this.ensureOutputDir(outputDir)
+      const filePath = join(dir, `${talker}_messages.xlsx`)
+      await workbook.xlsx.writeFile(filePath)
+      return { success: true, path: filePath }
+    } catch (e) {
+      return { success: false, error: String(e) }
+    }
+  }
+
+  private buildHtml(talker: string, messages: Message[]): string {
+    const rows = messages.map(m => {
+      const time = new Date(m.createTime * 1000).toLocaleString('zh-CN')
+      const sender = m.isSend ? '我' : this.escapeHtml(m.senderUsername || talker)
+      const contentType = m.localType
+      const contentHtml = this.renderMessageContent(m)
+      const align = m.isSend ? 'right' : 'left'
+      const bgColor = m.isSend ? '#95ec69' : '#ffffff'
+
+      return `<div style="text-align:${align};margin:8px 0;">
+        <div style="display:inline-block;background:${bgColor};padding:8px 12px;border-radius:8px;max-width:80%;">
+          <div style="font-size:12px;color:#999;">${sender} · ${time}</div>
+          <div style="margin-top:4px;white-space:pre-wrap;word-break:break-all;">${contentHtml}</div>
+        </div>
+      </div>`
+    }).join('\n')
+
+    return `<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+  <meta charset="UTF-8">
+  <title>聊天记录 - ${this.escapeHtml(talker)}</title>
+  <style>
+    body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif; background: #f0f0f0; padding: 20px; }
+    .container { max-width: 820px; margin: 0 auto; background: #f5f5f5; padding: 20px; border-radius: 12px; }
+    .msg-image { color:#888; }
+    .msg-image img { max-width:240px; max-height:240px; border-radius:4px; margin-top:6px; display:block; }
+    .msg-app { margin:0; }
+    .msg-app .app-title { font-size:14px; font-weight:600; color:#333; }
+    .msg-app .app-desc { font-size:12px; color:#999; margin-top:2px; }
+    .msg-app .app-url { display:block; margin-top:6px; padding:6px 10px; background:#f0f0f0; border-left:3px solid #07c160; text-decoration:none; color:#576b95; border-radius:0 4px 4px 0; font-size:13px; }
+    .msg-app .app-file { color:#07c160; font-weight:600; }
+    .msg-sys { color:#bbb; font-size:13px; }
+    .msg-media { color:#888; }
+  </style>
+</head>
+<body>
+  <div class="container">
+    <h2 style="text-align:center;">聊天记录 - ${this.escapeHtml(talker)}</h2>
+    <p style="text-align:center;color:#999;">共 ${messages.length} 条消息</p>
+    ${rows}
+  </div>
+</body>
+</html>`
+  }
+
+  private renderMessageContent(m: Message): string {
+    switch (m.localType) {
+      case 1:
+        return this.escapeHtml(m.parsedContent || m.rawContent || '')
+
+      case 3: {
+        let html = `<span class="msg-media">[图片]</span>`
+        if (m.imageFileName) {
+          html = `<span class="msg-media">[图片: ${this.escapeHtml(m.imageFileName)}]</span>`
+        }
+        if (m.imageBase64) {
+          html += `<br><img src="data:image/jpeg;base64,${m.imageBase64}" />`
+        }
+        return html
+      }
+
+      case 34:
+        return '<span class="msg-media">[语音]</span>'
+
+      case 43:
+        return '<span class="msg-media">[视频]</span>'
+
+      case 47:
+        return this.escapeHtml(m.parsedContent || '[表情]')
+
+      case 49: {
+        let html = '<div class="msg-app">'
+        if (m.appTitle) {
+          html += `<div class="app-title">${this.escapeHtml(m.appTitle)}</div>`
+        }
+        if (m.appDescription) {
+          html += `<div class="app-desc">${this.escapeHtml(m.appDescription)}</div>`
+        }
+        if (m.appUrl) {
+          html += `<a class="app-url" href="${this.escapeHtml(m.appUrl)}" target="_blank">${this.escapeHtml(m.appUrl)}</a>`
+        }
+        html += '</div>'
+        if (!m.appTitle && !m.appDescription && !m.appUrl) {
+          html = this.escapeHtml(m.parsedContent || '[链接/文件]')
+        }
+        return html
+      }
+
+      case 50:
+        return '<span class="msg-media">[语音通话]</span>'
+
+      case 10000:
+        return `<span class="msg-sys">${this.escapeHtml(m.parsedContent || '')}</span>`
+
+      default:
+        return this.escapeHtml(m.parsedContent || m.rawContent || '')
+    }
+  }
+
+  private escapeHtml(str: string): string {
+    return str
+      .replace(/&/g, '&amp;')
+      .replace(/</g, '&lt;')
+      .replace(/>/g, '&gt;')
+      .replace(/"/g, '&quot;')
+  }
+
+  private getMessageTypeName(type: number): string {
+    const map: Record<number, string> = {
+      1: '文本', 3: '图片', 34: '语音', 42: '名片',
+      43: '视频', 47: '表情', 48: '位置', 49: '链接/文件',
+      50: '语音通话', 10000: '系统', 10002: '引用'
+    }
+    return map[type] || `类型${type}`
+  }
+
+  private ensureOutputDir(dir: string): string {
+    const resolved = dir || './output'
+    if (!existsSync(resolved)) {
+      mkdirSync(resolved, { recursive: true })
+    }
+    return resolved
+  }
+}
+
+export const exportService = new ExportService()