@whyshu/argent-engine 0.0.0-stage → 0.6.0

package/engine/scripts/archive/feishu-bridge.js

@@ -0,0 +1,246 @@
+#!/usr/bin/env node
+/**
+ * argent-feishu-bridge — 飞书对话消息桥（Argent on DSH）
+ *
+ * 常驻进程：监听飞书 IM 消息 → 交给 DSH headless 会话处理 → 回复。
+ * 与 hermes 版 gateway 等价，走 lark-cli event consume（本地总线）。
+ *
+ * 用法:
+ *   node feishu-bridge.js [--role general|ml-seller] [--dsh-profile headless]
+ *   ARGENT_FEISHU_IDENTITY=bot|user   # 监听身份（默认 bot）
+ *
+ * 契约（lark-event skill）:
+ *   - stderr 出现 `[event] ready event_key=...` 后才开始读 stdout
+ *   - SIGTERM 优雅退出（consume 收到 SIGTERM 清理订阅）
+ *   - 事件为 stdout NDJSON，每行一条（--jq 已投影）
+ */
+
+import { spawn, execFile } from 'node:child_process'
+import { promisify } from 'node:util'
+import readline from 'node:readline'
+import { homedir } from 'node:os'
+import { join } from 'node:path'
+import { mkdirSync, existsSync, readFileSync, appendFileSync, writeFileSync } from 'node:fs'
+
+const execFileAsync = promisify(execFile)
+
+// ── 参数 ───────────────────────────────────────────────────
+const args = process.argv.slice(2)
+// 多实例（改造项7）：bridge 随实例运行——配置按 $DSH_HOME 解析（未设置回退 ~/.dsh 主实例）
+const BRIDGE_HOME = process.env.DSH_HOME || join(homedir(), '.dsh')
+// 飞书凭据：优先 $DSH_HOME/argent/config.json（Web UI 配置），否则 lark-cli 全局配置
+// （readFileSync/join/homedir 已在顶部统一导入）
+let _feishuEnv = null
+try {
+  const cfg = JSON.parse(readFileSync(join(BRIDGE_HOME, 'argent', 'config.json'), 'utf8'))
+  if (cfg.feishuAppId && cfg.feishuAppSecret) {
+    _feishuEnv = { LARK_APP_ID: cfg.feishuAppId, LARK_APP_SECRET: cfg.feishuAppSecret }
+  }
+} catch { /* 未配置则用 lark-cli 全局 */ }
+const bridgeEnv = _feishuEnv ? { ...process.env, ..._feishuEnv } : process.env
+const getArg = (name, def) => {
+  const i = args.indexOf(name)
+  return i >= 0 && args[i + 1] ? args[i + 1] : def
+}
+const ROLE = getArg('--role', 'general')
+const DSH_PROFILE = getArg('--dsh-profile', 'headless')
+const IDENTITY = process.env.ARGENT_FEISHU_IDENTITY || 'bot'
+
+const ROLE_PREFIX = {
+  general: '你是 Argent 通用助手（问述科技）。处理以下飞书消息，简洁中文回复：\n',
+  'ml-seller': '你是 Argent 美客多卖家助手（问述科技）。处理以下飞书消息；涉及货件/店铺查询用 ml_* 工具（如不可用则说明环境未就绪），简洁中文回复：\n',
+  'ml-daily': '你是 Argent 美客多日报助手（问述科技）。只读查询与日报；FULL/发票不属本实例。处理以下飞书消息，简洁中文回复：\n',
+  'ml-fulfillment': '你是 Argent 美客多 FULL/发票助手（问述科技）。涉及 FULL 货件/发票流程；写操作先确认。处理以下飞书消息，简洁中文回复：\n',
+  'pet-dev': '你是 Argent 云宠乐园开发者（问述科技）。处理开发任务与部署请求；任务流转看板为准。处理以下飞书消息，简洁中文回复：\n',
+  'argent-dev': '你是 Argent 开发者（问述科技）。处理 Argent 开发任务。处理以下飞书消息，简洁中文回复：\n',
+  'whyshu-pm': '你是问述科技PM（问述科技）。掌握策划案需求，分发任务给开发者并在看板登记。处理以下飞书消息，简洁中文回复：\n',
+  'lktop-pm': '你是 LKTOPPM（问述科技）。掌握 LKTOP 需求，分发任务给 LKTOPAPP 开发者。处理以下飞书消息，简洁中文回复：\n',
+  'lktopapp-dev': '你是 LKTOPAPP开发者（问述科技）。处理 LKTOP APP 开发任务。处理以下飞书消息，简洁中文回复：\n',
+}
+
+// ── 多轮记忆（每会话上下文注入，方案 B）──────────────────
+// 历史存 $DSH_HOME/argent/bridge-context/<chatId>.jsonl（最多 CONTEXT_MAX 条）
+// 注入到 prompt 供参考；不跨会话共享，天然按聊天隔离
+const CONTEXT_DIR = join(BRIDGE_HOME, 'argent', 'bridge-context')
+const CONTEXT_MAX = 20
+
+const contextFile = (chatId) => join(CONTEXT_DIR, String(chatId).replace(/[^a-zA-Z0-9_-]/g, '') + '.jsonl')
+
+function readContext(chatId) {
+  try {
+    const f = contextFile(chatId)
+    if (!existsSync(f)) return []
+    const out = []
+    for (const l of readFileSync(f, 'utf-8').split('\n')) {
+      if (!l.trim()) continue
+      try {
+        const o = JSON.parse(l)
+        if (o && o.role && o.text) out.push(o)
+      } catch { /* 跳过坏行 */ }
+    }
+    return out.slice(-CONTEXT_MAX)
+  } catch { return [] }
+}
+
+function appendContext(chatId, role, text) {
+  try {
+    mkdirSync(CONTEXT_DIR, { recursive: true })
+    const f = contextFile(chatId)
+    appendFileSync(f, JSON.stringify({ role, text, ts: Date.now() }) + '\n')
+    const lines = readFileSync(f, 'utf-8').split('\n').filter(Boolean)
+    if (lines.length > CONTEXT_MAX * 2) {
+      writeFileSync(f, lines.slice(-CONTEXT_MAX * 2).join('\n') + '\n')
+    }
+  } catch { /* 记忆失败不阻断对话 */ }
+}
+
+function contextBlock(hist) {
+  if (!hist.length) return ''
+  const lines = hist.map((o) => (o.role === 'user' ? '用户' : '助手') + '：' + o.text)
+  return '以下为本会话之前的对话记录（供参考，不要复述历史，直接回答最新问题）：\n' + lines.join('\n') + '\n\n'
+}
+
+// ── 去重（message_id，10 分钟 TTL）────────────────────────
+const seen = new Map()
+const isDuplicate = (id) => {
+  if (!id) return false
+  if (seen.has(id)) return true
+  seen.set(id, Date.now())
+  if (seen.size > 500) {
+    const now = Date.now()
+    for (const [k, t] of seen) if (now - t > 600000) seen.delete(k)
+  }
+  return false
+}
+
+const shellQuote = (s) => `'${String(s ?? '').replace(/'/g, `'\\''`)}'`
+
+// ── 回复飞书 ───────────────────────────────────────────────
+async function reply(chatId, text) {
+  const { stdout } = await execFileAsync(
+    'lark-cli', ['im', '+messages-send', '--chat-id', String(chatId), '--text', String(text), '--as', IDENTITY],
+    { timeout: 30000 },
+  )
+  console.log(`[bridge] 已回复 chat=${chatId} len=${String(text).length}`)
+  return stdout
+}
+
+// ── @ 过滤（2026-08-26 修复：仅回复 @ 自己的消息）──────────
+// 事件带 mentions（被 @ 者 [{id, key, name}]）；按显示名匹配本角色。
+// 平台侧两种配置都兼容：
+//  - 推全部群消息（未关「接收群内所有消息」）：无 @ 或 @ 别人 → 跳过
+//  - 只推 @ 自己的消息（已关）：mentions 必含自己 → 命中
+const MENTION_NAMES = {
+  'pet-dev': ['云宠乐园开发者', 'pet-dev'],
+  'argent-dev': ['Argent开发者', 'Argent 开发者', 'argent-dev'],
+  'whyshu-pm': ['问述科技PM', '问述科技 PM', 'whyshu-pm'],
+  'lktop-pm': ['LKTOPPM', 'LKTOP PM', 'lktop-pm'],
+  'lktopapp-dev': ['LKTOPAPP开发者', 'LKTOPAPP 开发者', 'lktopapp-dev'],
+  general: ['MeU'], // 主 Bot 群内显示名（精确匹配，防误命中其他角色名）
+}
+function shouldHandle(evt) {
+  const mentions = Array.isArray(evt.mentions) ? evt.mentions : []
+  if (!mentions.length) return false // 消息未 @ 任何人 → 不处理
+  const names = MENTION_NAMES[ROLE] || MENTION_NAMES.general
+  return mentions.some((m) => names.some((n) => {
+    if (ROLE === 'general') return String(m.name || '') === n // 主 Bot 精确匹配（防误命中「Argent开发者」）
+    return String(m.name || '').includes(n)
+  }))
+}
+
+// ── 处理一条消息 ───────────────────────────────────────────
+async function handleMessage(evt) {
+  const { chat_id: chatId, message_id: msgId, text } = evt || {}
+  if (!chatId || !text) return
+  if (isDuplicate(msgId)) return
+  if (!shouldHandle(evt)) {
+    console.log(`[bridge] 跳过非 @ 消息 chat=${chatId} msg=${msgId} mentions=${JSON.stringify((evt.mentions || []).map((m) => m.name))}`)
+    return
+  }
+
+  // 机器人自己的消息不回（sender 过滤在 jq 层，见下方提示）
+  const hist = readContext(chatId)
+  const task = (ROLE_PREFIX[ROLE] || ROLE_PREFIX.general) + contextBlock(hist) + '当前问题：' + text
+  console.log(`[bridge] chat=${chatId} msg=${msgId} sender=${evt.sender_id || '?'} len=${text.length} hist=${hist.length} @命中`)
+
+  // 交给 headless 会话（一次性）
+  let answer
+  try {
+    const { stdout } = await execFileAsync('dsh', ['--profile', DSH_PROFILE, task], {
+      timeout: 300000,
+      maxBuffer: 2 * 1024 * 1024,
+    })
+    answer = (stdout || '').trim().slice(0, 1500)
+  } catch (err) {
+    answer = `⚠️ 处理失败：${(err.message || '').slice(0, 200)}`
+    console.error(`[bridge] dsh error: ${err.message}`)
+  }
+  if (!answer) answer = '⚠️ 没有得到回复'
+
+  // 记录本轮（仅记录成功回答，避免把错误回复带入上下文）
+  appendContext(chatId, 'user', text)
+  if (!answer.startsWith('⚠️')) appendContext(chatId, 'assistant', answer)
+
+  try {
+    await reply(chatId, answer)
+  } catch (err) {
+    console.error(`[bridge] reply error: ${err.message}`)
+  }
+}
+
+// ── 启动 consume ───────────────────────────────────────────
+function start() {
+  const jq = 'select(.message_type=="text") | {chat_id, sender_id, message_id, text: .content, mentions: (.mentions // [])}'
+  const child = spawn('lark-cli', ['event', 'consume', 'im.message.receive_v1', '--as', IDENTITY, '--jq', jq], {
+    stdio: ['pipe', 'pipe', 'pipe'],
+    env: bridgeEnv,
+  })
+
+  let ready = false
+  const stderr = readline.createInterface({ input: child.stderr })
+  stderr.on('line', (line) => {
+    if (!ready && line.includes('[event] ready')) {
+      ready = true
+      console.log('[bridge] lark event bus ready')
+      return
+    }
+    // 错误 envelope（JSON）
+    if (line.includes('"ok":false')) {
+      console.error('[bridge] lark error:', line.slice(0, 500))
+      if (line.includes('"type":"auth"') || line.includes('missing_scope')) {
+        console.error('[bridge] 请先完成 lark-cli 登录/授权（lark-cli auth login --as bot）')
+      }
+    }
+  })
+
+  const stdout = readline.createInterface({ input: child.stdout })
+  stdout.on('line', (line) => {
+    if (!line.trim()) return
+    let evt
+    try { evt = JSON.parse(line) } catch { return }
+    handleMessage(evt).catch((err) => console.error('[bridge] handle error:', err.message))
+  })
+
+  child.on('exit', (code, signal) => {
+    console.log(`[bridge] consume exited code=${code} signal=${signal}`)
+    // 意外退出则 3s 后重启（守护）；SIGTERM 传播则不重启
+    if (!shuttingDown && code !== 0) {
+      setTimeout(start, 3000)
+    }
+  })
+
+  return child
+}
+
+let shuttingDown = false
+for (const sig of ['SIGINT', 'SIGTERM']) {
+  process.on(sig, () => {
+    shuttingDown = true
+    console.log(`[bridge] ${sig}，退出中...`)
+    process.exit(0)
+  })
+}
+
+console.log(`[bridge] argent-feishu-bridge 启动（role=${ROLE} dsh=${DSH_PROFILE} identity=${IDENTITY}）`)
+console.log('[bridge] 提示：消费端请用 --as bot；若需要过滤机器人自身消息，在 jq 中加 sender 过滤')
+start()