weflow-cli 1.5.0

package/src/services/whitelistService.ts

@@ -0,0 +1,72 @@
+/**
+ * 白名单服务 — 控制消息收发权限，仅白名单内 wxid 可收发消息。
+ */
+import { configService } from './configService.js'
+import { resolveTalker } from '../utils/talkerUtils.js'
+
+export interface WhitelistEntry {
+  wxid: string
+  displayName: string
+}
+
+export class WhitelistService {
+  getList(): string[] {
+    return configService.getWhitelist()
+  }
+
+  isAllowed(wxid: string): boolean {
+    return configService.getWhitelist().includes(wxid)
+  }
+
+  /** 直接添加 wxid（不解析昵称），用于 CLI 确认后的最终写入 */
+  addDirect(wxid: string): void {
+    const list = configService.getWhitelist()
+    if (!list.includes(wxid)) {
+      list.push(wxid)
+      configService.setWhitelist(list)
+    }
+  }
+
+  async add(target: string): Promise<{ success: boolean; error?: string; wxid?: string }> {
+    let wxid: string
+
+    // 已知格式直接使用，否则通过 resolveTalker 解析
+    if (target.startsWith('wxid_') || target.includes('@chatroom') || target.includes('@openim')) {
+      wxid = target
+    } else {
+      try {
+        wxid = await resolveTalker(target)
+      } catch (e: any) {
+        return { success: false, error: `无法解析 "${target}": ${e.message}` }
+      }
+    }
+
+    const list = configService.getWhitelist()
+    if (list.includes(wxid)) {
+      return { success: false, error: `"${wxid}" 已在白名单中` }
+    }
+
+    list.push(wxid)
+    configService.setWhitelist(list)
+    return { success: true, wxid }
+  }
+
+  remove(wxid: string): boolean {
+    const list = configService.getWhitelist()
+    const idx = list.indexOf(wxid)
+    if (idx < 0) return false
+    list.splice(idx, 1)
+    configService.setWhitelist(list)
+    return true
+  }
+
+  clear(): void {
+    configService.setWhitelist([])
+  }
+
+  reload(): void {
+    // configService 每次 getWhitelist 都从内存读取，无需主动 reload
+  }
+}
+
+export const whitelistService = new WhitelistService()

package/src/types.ts

@@ -0,0 +1,141 @@
+export interface WxidInfo {
+  wxid: string
+  modifiedTime: number
+  nickname?: string
+  avatarUrl?: string
+}
+
+export interface ChatSession {
+  username: string
+  type: number
+  unreadCount: number
+  summary: string
+  sortTimestamp: number
+  lastTimestamp: number
+  displayName?: string
+  avatarUrl?: string
+  lastMsgType?: number
+}
+
+export interface Message {
+  localId: number
+  serverId: string
+  localType: number
+  createTime: number
+  isSend: number | null
+  senderUsername: string | null
+  content: string
+  rawContent: string
+  parsedContent: string
+  mediaType?: string
+  mediaFileName?: string
+  /** Type=3 图片: 缩略图 base64 数据 */
+  imageBase64?: string
+  /** Type=3 图片: 图片文件名 (UUID.png) */
+  imageFileName?: string
+  /** Type=49 AppMsg: 标题 */
+  appTitle?: string
+  /** Type=49 AppMsg: 描述 */
+  appDescription?: string
+  /** Type=49 AppMsg: URL */
+  appUrl?: string
+}
+
+export interface Contact {
+  username: string
+  displayName: string
+  remark?: string
+  nickname?: string
+  alias?: string
+  avatarUrl?: string
+  type?: string
+}
+
+// === 微信消息通道 ===
+
+export interface WechatOCConfig {
+  baseUrl?: string
+  cdnBaseUrl?: string
+  botType?: string
+  token?: string
+  accountId?: string
+  syncBuf?: string
+  contextTokens?: Record<string, string>
+  longPollTimeoutMs?: number
+  apiTimeoutMs?: number
+  typingKeepaliveIntervalS?: number
+  typingTicketTtlS?: number
+}
+
+export interface WechatLoginSession {
+  sessionKey: string
+  qrcode: string
+  qrcodeImgContent: string
+  startedAt: number
+  status: 'wait' | 'confirmed' | 'expired' | 'scanned'
+  botToken?: string
+  accountId?: string
+  baseUrl?: string
+  userId?: string
+  error?: string
+}
+
+export interface WechatInboundMessage {
+  messageId: string
+  fromUserId: string
+  senderNickname: string
+  timestamp: number
+  timestampMs: number
+  components: WechatMessageComponent[]
+  messageStr: string
+  messageKind: 'text' | 'image' | 'voice' | 'file' | 'video' | 'unknown'
+  rawMessage: Record<string, any>
+  isReply: boolean
+  quotedText?: string
+}
+
+export type WechatMessageComponent =
+  | { type: 'plain'; text: string }
+  | { type: 'image'; filePath: string }
+  | { type: 'record'; filePath: string }
+  | { type: 'file'; name: string; filePath: string }
+  | { type: 'video'; filePath: string }
+
+export interface ExportOptions {
+  talker: string
+  format: 'json' | 'html' | 'txt' | 'excel'
+  output: string
+  limit?: number
+  start?: string
+  end?: string
+}
+
+export interface DbKeyResult {
+  success: boolean
+  key?: string
+  error?: string
+  logs?: string[]
+}
+
+export type DataVersion = '3.x' | '4.x'
+
+export interface ConfigData {
+  dbPath?: string
+  wxid?: string
+  decryptKey?: string
+  decryptKey3x?: string
+  dataVersion?: DataVersion
+  dbPath3x?: string
+  /** NT 格式: 主消息数据库路径 (message_0.db) */
+  ntDbPath?: string
+  /** NT 格式: 64位十六进制密钥 */
+  ntKey?: string
+  /** NT 格式: 32位十六进制盐值 */
+  ntSalt?: string
+  /** NT 格式: 联系人数据库路径 (contact.db) */
+  contactDbPath?: string
+  /** NT 格式: contact.db 64位十六进制密钥 */
+  contactKey?: string
+  /** NT 格式: contact.db 32位十六进制盐值 */
+  contactSalt?: string
+}

package/src/utils/errors.ts

@@ -0,0 +1,42 @@
+/**
+ * WeflowError - weflow-cli 统一错误类
+ *
+ * 提供结构化的错误信息，包含：
+ * - message: 人类可读的错误描述
+ * - code: 错误代码（用于程序化判断）
+ * - exitCode: 进程退出码
+ */
+
+export class WeflowError extends Error {
+  constructor(
+    message: string,
+    public code: string = 'UNKNOWN',
+    public exitCode: number = 1,
+  ) {
+    super(message)
+    this.name = 'WeflowError'
+  }
+}
+
+/**
+ * 统一错误处理函数。
+ *
+ * 根据错误类型输出对应的错误信息并以合适的退出码退出进程：
+ * - WeflowError: 使用其自带的 exitCode
+ * - 普通 Error: 视为未预期错误，退出码 1
+ * - 其他: 视为未知错误，退出码 1
+ *
+ * @param e 捕获到的错误对象
+ */
+export function handleError(e: unknown): void {
+  if (e instanceof WeflowError) {
+    console.error(`[weflow-cli] ${e.message}`)
+    process.exit(e.exitCode)
+  } else if (e instanceof Error) {
+    console.error(`[weflow-cli] 未预期的错误: ${e.message}`)
+    process.exit(1)
+  } else {
+    console.error('[weflow-cli] 未知错误')
+    process.exit(1)
+  }
+}

package/src/utils/pathUtils.ts

@@ -0,0 +1,16 @@
+import { homedir } from 'os'
+
+/**
+ * Expand "~" prefix to current user's home directory.
+ */
+export function expandHomePath(inputPath: string): string {
+  const raw = String(inputPath || '').trim()
+  if (!raw) return raw
+
+  if (raw === '~') return homedir()
+  if (/^~[\\/]/.test(raw)) {
+    return `${homedir()}${raw.slice(1)}`
+  }
+
+  return raw
+}

package/src/utils/pythonRunner.ts

@@ -0,0 +1,81 @@
+/**
+ * pythonRunner - 统一的 Python 脚本调用工具
+ *
+ * 消除 bin/weflow-cli.ts 中重复出现的 execFile 调用模式，
+ * 统一封装路径解析、参数序列化、错误处理。
+ */
+import { execFile } from 'child_process'
+import { promisify } from 'util'
+import { fileURLToPath } from 'url'
+import { dirname, join } from 'path'
+import { existsSync } from 'fs'
+
+const execFileAsync = promisify(execFile)
+
+export interface PythonScriptResult {
+  stdout: string
+  stderr: string
+  exitCode: number
+}
+
+export interface PythonScriptError extends Error {
+  script: string
+  stderr: string
+  exitCode: number
+}
+
+/** 获取项目根目录 */
+export function getProjectRoot(): string {
+  // 兼容 ESM 和 CJS
+  const possibleDirname = typeof __dirname !== 'undefined'
+    ? __dirname
+    : dirname(fileURLToPath(import.meta.url))
+  // 从 src/utils/ 或 bin/ 向上找到 package.json 所在目录
+  let dir = possibleDirname
+  for (let i = 0; i < 5; i++) {
+    try {
+      if (existsSync(join(dir, 'package.json'))) return dir
+    } catch {}
+    dir = dirname(dir)
+  }
+  return possibleDirname
+}
+
+/**
+ * 运行 Python 脚本，统一封装路径解析、参数序列化、错误处理。
+ *
+ * @param scriptName 脚本文件名（如 'biz_daily.py'），相对于 scripts/ 目录
+ * @param args 命令行参数数组
+ * @param options 可选配置
+ * @returns 脚本执行结果
+ * @throws PythonScriptError 当脚本执行失败时
+ */
+export async function runPythonScript(
+  scriptName: string,
+  args: string[] = [],
+  options: { cwd?: string; timeout?: number; env?: Record<string, string> } = {},
+): Promise<PythonScriptResult> {
+  const projectRoot = getProjectRoot()
+  const scriptPath = join(projectRoot, 'scripts', scriptName)
+  const cwd = options.cwd || projectRoot
+
+  try {
+    const { stdout, stderr } = await execFileAsync('python', [scriptPath, ...args], {
+      cwd,
+      timeout: options.timeout || 300000, // 默认 5 分钟
+      maxBuffer: 10 * 1024 * 1024, // 10MB
+      env: { ...process.env, ...options.env },
+    })
+    return { stdout, stderr, exitCode: 0 }
+  } catch (e: any) {
+    const error: PythonScriptError = Object.assign(
+      new Error(`Python 脚本 ${scriptName} 执行失败: ${e.message}`),
+      {
+        script: scriptName,
+        stderr: e.stderr || '',
+        exitCode: e.code || 1,
+      },
+    )
+    throw error
+  }
+}

package/src/utils/talkerUtils.ts

@@ -0,0 +1,84 @@
+/**
+ * Talker (wxid) resolution utilities.
+ * Supports: wxid / @chatroom / @openim / nickname / remark / index
+ */
+import inquirer from 'inquirer'
+import { chatService } from '../services/chatService.js'
+import type { ChatSession } from '../types.js'
+
+/**
+ * 将用户输入解析为 talker (wxid)。
+ *
+ * 支持: wxid_xxx / 群聊ID / 昵称 / 备注名 / 序号 [N] / 纯数字
+ *
+ * @throws {Error} 当输入无法解析到任何会话时
+ */
+export async function resolveTalker(input: string): Promise<string> {
+  // 1. 已知格式直接返回
+  if (input.startsWith('wxid_') || input.includes('@chatroom') || input.includes('@openim')) {
+    return input
+  }
+
+  // 2. 序号匹配: [N] 或纯数字
+  const numMatch = input.match(/^\[?(\d+)\]?$/)
+  if (numMatch) {
+    const index = parseInt(numMatch[1]) - 1
+    const sessions = await chatService.listSessions()
+    if (index < 0 || index >= sessions.length) {
+      throw new Error(`序号 ${input} 超出范围，共 ${sessions.length} 个会话。运行 weflow-cli sessions 查看列表`)
+    }
+    return sessions[index].username
+  }
+
+  // 3. 昵称/备注搜索
+  const sessions = await chatService.listSessions(input, 50)
+
+  if (sessions.length === 0) {
+    throw new Error(`找不到 "${input}" 对应的会话。运行 weflow-cli sessions 查看所有会话`)
+  }
+
+  // 精确匹配
+  const exact = sessions.find(s =>
+    s.displayName === input || s.username === input,
+  )
+  if (exact) return exact.username
+
+  // 只有一个匹配
+  if (sessions.length === 1) {
+    return sessions[0].username
+  }
+
+  // 多个匹配 — 交互选择
+  const choices = sessions.slice(0, 20).map(s => ({
+    name: `${s.displayName}  (${s.username})`,
+    value: s.username,
+  }))
+
+  const { selected } = await inquirer.prompt([{
+    type: 'select',
+    name: 'selected',
+    message: `"${input}" 匹配到多个会话，请选择:`,
+    choices,
+    loop: false,
+  }])
+
+  return selected
+}
+
+/**
+ * 批量为 wxid 列表查找 displayName。
+ * 调用 chatService.listSessions 获取带昵称的会话列表。
+ */
+export async function lookupDisplayNames(wxids: string[]): Promise<Map<string, string>> {
+  const sessions = await chatService.listSessions()
+  const nameMap = new Map<string, string>()
+  const sessionMap = new Map<string, ChatSession>()
+  for (const s of sessions) {
+    sessionMap.set(s.username, s)
+  }
+  for (const wxid of wxids) {
+    const s = sessionMap.get(wxid)
+    nameMap.set(wxid, s?.displayName || wxid)
+  }
+  return nameMap
+}