@whyshu/argent-engine 0.0.0-stage → 0.6.1

package/engine/scripts/flow-trigger-wake.mjs

@@ -0,0 +1,284 @@
+#!/usr/bin/env node
+/**
+ * flow-trigger-wake —— 唤醒**任意流程**的角色会话并投递「本轮任务」（定时触发的执行端）。
+ *
+ * 引擎泛化版（P0 · 2026-09-16）：本脚本**不认识任何具体流程**。
+ *   preset / cwd / 提示词模板 / 日志 / 会话复用文件 全部来自注册表
+ *   （`flow_config.py --flow <id> --json` ← `flow-requirements.json#flows.<id>`）。
+ *   加一条流程 = 注册表加一条 + 一份 `scripts/flow-prompts/<id>.md`，不改本脚本。
+ *
+ * RPC 形状来自实测（会话/提示词端点，见 `docs/dsh-api-protocol.md`）：
+ *   session/create {request:{agentPreset, cwd}}                      → {result:{value:{sessionId}}}
+ *   session/prompt {request:{requestId, sessionId, mode:'queue', content:[{type:'text',text}]}}
+ *
+ * **会话复用**（2026-09-16 修）：旧版每轮 `session/create` ⇒ 会话数无限膨胀（已 136+）。
+ *   现在把 sessionId 落盘（`paths.sessionIdFile`），下轮直接 `session/prompt` 复用；
+ *   若该会话已不可用（RPC 报错）⇒ 自动新建并改写该文件。
+ *
+ * 两种模式：
+ *   · 定时派单（默认）：--tickets a,b
+ *   · **失败交接（P0.6）**：--reason failure --case <caseId> [--attempt N] [--error "摘要"]
+ *     ⇒ 渲染 <flow>.failure.md，唤醒掌舵者判断「重试 / 等人 / 升级」
+ * 用法：node scripts/flow-trigger-wake.mjs --flow <id> --tickets a,b [--max 2] [--dry-run] [--base URL]
+ * 退出码：0 成功（含 dry-run）；2 配置/鉴权/RPC 失败
+ *
+ * **鉴权来源链（2026-10-08 加固）**：不再只认 `/tmp/dsh-web.log`（那是一次静默故障的根因：
+ * 该文件被系统清理后唤醒 exit 2，而台账/遥测里一条失败都没有）。现在按优先级试多个来源
+ * （显式 token 文件 → **凭据里的 browser-session 密钥现签 cookie，不依赖 /tmp** → DSH_WEB_LOG
+ * → plist 的 StandardOutPath → 稳定目录候选 → /tmp 兜底），每个来源**先验后用**；
+ * 全部失败 ⇒ 文案点名每个来源 + 可执行修法，并写台账 `blocked`（by=flow-wake）+ 遥测。
+ * 唯一实现：`scripts/lib/wake-auth.mjs`（含来源表与失败文案）。
+ */
+import { readFileSync, writeFileSync, appendFileSync, mkdirSync } from 'node:fs'
+import { execFileSync } from 'node:child_process'
+import { dirname, join } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { homedir } from 'node:os'
+// 鉴权来源链的唯一实现（多来源 + 优先级 + 失败文案）——见该文件头的事故记录
+import { authSources, acquireWakeAuth, authFailureText, SOURCE_STATUS_ZH } from './lib/wake-auth.mjs'
+
+const SCRIPTS = dirname(fileURLToPath(import.meta.url))
+const arg = (n, d) => { const i = process.argv.indexOf(n); return i > 0 ? process.argv[i + 1] : d }
+const FLOW = arg('--flow', 'wenyao-invoice')
+const BASE = arg('--base', 'http://127.0.0.1:3080')
+const DRY = process.argv.includes('--dry-run')
+const NOTICE = arg('--notice', '')
+const TICKETS = (arg('--tickets', '') || '').split(',').map((x) => x.trim()).filter(Boolean)
+// P0.6：失败交接模式 —— 引擎（argent-flow 的失败回路）发现"需要判断"的失败时用这个模式唤醒角色
+const REASON = arg('--reason', 'tickets')
+const CASE = arg('--case', '')
+const ATTEMPT = arg('--attempt', '')
+const ERROR = arg('--error', '')
+// 2026-09-18：reason=ruling 用 —— 用户对 blocked 票的裁定原文（逐字转达，勿改写）
+const NOTE = arg('--note', '')
+const PY = (process.env.ARGENT_PY || `${process.env.DSH_HOME || `${process.env.HOME}/.dsh`}/argent/venv/bin/python3`)
+const CLONE = process.env.ARGENT_DISH_HOME || join(homedir(), '.argent-dsh')
+const REGISTRY = arg('--registry', '')          // 覆盖流程注册表（测试/多部署用）
+const REG_ARGS = REGISTRY ? ['--registry', REGISTRY] : []
+
+// ── 失败信号（2026-10-08）：唤醒端此前**只** exit 2 ⇒ 定时器"看着正常"、活没人干 ──
+// 事故：/tmp/dsh-web.log 被系统清理 ⇒ `❌ 取不到 token` ⇒ exit 2 ⇒ 该轮角色会话没起、
+// 表没生成、没推送，**而台账/遥测里一条失败都没有**（静默故障，靠人肉比对才发现）。
+// 现在：任何失败路径都必须留一条**既有通道**上的痕迹（台账 blocked by=flow-wake + 遥测 engine 事件），
+// 且信号写入**尽力而为**（try/catch，语义同 `|| true`）：写不进去只告警，绝不掩盖原始错误、
+// 也不改变判定与退出码。
+function signalFailure(detail) {
+  const d = String(detail).slice(0, 280)
+  let ledger
+  try { ledger = cfg && cfg.paths && cfg.paths.ledger } catch { ledger = undefined }
+  try {
+    execFileSync(PY, [join(SCRIPTS, 'flow_ledger.py'), ...(ledger ? ['--ledger', ledger] : []),
+      '--case-id', `wake__${FLOW}`, '--event', 'blocked', '--by', 'flow-wake', '--detail', d],
+    { encoding: 'utf8', timeout: 20000, stdio: ['ignore', 'ignore', 'ignore'] })
+  } catch (e) {
+    console.error(`[flow-wake] ⚠️ 台账 blocked 写入失败（不影响判定）：${String(e.stderr || e.message).trim().slice(0, 200)}`)
+  }
+  try {
+    execFileSync(PY, [join(SCRIPTS, 'telemetry.py'), '--emit', '--kind', 'engine', '--flow', FLOW,
+      '--error-class', 'needs_human', '--detail', d], { encoding: 'utf8', timeout: 20000, stdio: ['ignore', 'ignore', 'ignore'] })
+    execFileSync(PY, [join(SCRIPTS, 'telemetry.py'), '--flush'], { encoding: 'utf8', timeout: 20000, stdio: ['ignore', 'ignore', 'ignore'] })
+  } catch (e) {
+    console.error(`[flow-wake] ⚠️ 遥测上报未完成（不影响判定）：${String(e.stderr || e.message).trim().slice(0, 200)}`)
+  }
+}
+
+// ── 流程配置（唯一读取点；未知流程由 flow_config.py 以 exit 2 拒绝）──
+let cfg
+try {
+  cfg = JSON.parse(execFileSync(PY, [join(SCRIPTS, 'flow_config.py'), '--flow', FLOW, '--json', ...REG_ARGS], { encoding: 'utf8' }))
+} catch (e) {
+  const m = `读取流程配置失败：${String(e.stderr || e.message).trim()}`
+  console.error(`❌ ${m}`)
+  signalFailure(m)
+  process.exit(2)
+}
+const MAX = Number(arg('--max', '0')) || cfg.concurrency.max
+// 2026-09-18：本轮语义来自注册表 trigger.mode（缺省 shadow）—— 日志如实播报，
+// 不再把"影子模式"写死（FULL 转真实运行后，写死会让操作者误判）。
+const MODE_LABEL = cfg.trigger.mode === 'real' ? '真实写入' : '影子（零副作用）'
+const LOG = cfg.paths.log
+const log = (m) => {
+  const line = `[${new Date().toISOString()}] ${m}`
+  try { mkdirSync(dirname(LOG), { recursive: true }); appendFileSync(LOG, line + '\n') } catch {}
+  console.log(m)
+}
+
+/**
+ * **响亮失败**（本文件的唯一失败出口；exit 2 语义不变）。
+ * 三段都做：① 流程日志 + stdout 的 ❌ 文案（含可执行修法）；② **stderr 显著 warn**
+ * （launchd 的 StandardErrorPath 也收得到）；③ 台账 blocked（by=flow-wake）+ 遥测（见 signalFailure）。
+ * 为什么必须这样：客户机上没人看日志，**只有台账/遥测/飞书**是观测面 —— 静默 = 事故。
+ */
+function failWake(msg, detail = msg) {
+  const text = `❌ ${msg}`
+  log(text)
+  console.error(`[flow-wake] ${text}`)
+  signalFailure(detail)
+  process.exit(2)
+}
+// 兜底：本文件里任何**没被显式接住**的异常（fetch 网络错、RPC 抛错…）也走同一条失败通道，
+// 否则 Node 只打一段栈就 exit 1 —— 观测面上仍是"什么都没发生"。
+process.on('unhandledRejection', (e) => failWake(`唤醒异常（未捕获的 Promise 拒绝）：${(e && (e.message || e)) || e}`))
+process.on('uncaughtException', (e) => failWake(`唤醒异常（未捕获）：${(e && (e.message || e)) || e}`))
+
+// ── 提示词模板 ──────────────────────────────────────────────────
+const TPL_PATH = REASON === 'failure'
+  ? cfg.paths.promptTemplate.replace(/\.md$/, '.failure.md')
+  : cfg.paths.promptTemplate
+// 2026-09-18 新增 reason=ruling：把**用户对 blocked 票的裁定**回传给掌舵者去执行。
+// 为什么不套模板：裁定是"人对人"的短指令（如"保留货件号续跑，不清号"），塞进几百行长 SOP 里
+// 会被淹没；角色此刻缺的不是流程知识（它已经判过一次 blocked），而是**人拍板的那句话**。
+const RULING = REASON === 'ruling'
+let tpl = ''
+if (!RULING) {
+  try {
+    tpl = readFileSync(TPL_PATH, 'utf8')
+  } catch {
+    failWake(`缺提示词模板：${TPL_PATH}（每条流程一份；失败交接模板命名 <flow>.failure.md）`,
+      `缺提示词模板 ${TPL_PATH}`)
+  }
+}
+const vars = {
+  flow: FLOW,
+  flowTitle: cfg.title,
+  tickets: TICKETS.length ? TICKETS.join('、') : `（无认领清单：请自行从 to_dispatch 取最多 ${MAX} 张）`,
+  ticketsCsv: TICKETS.join(','),
+  max: String(MAX),
+  cloneScripts: join(CLONE, 'scripts'),
+  // 2026-09-17：解释器要显式传给角色 —— 系统 python 没有 playwright，
+  // 角色照抄 `python3` 会得到「playwright 未安装」（真实票取证踩到过）。
+  py: (process.env.ARGENT_PY || process.env.ARGENT_PYTHON || `${process.env.DSH_HOME || `${process.env.HOME}/.dsh`}/argent/venv/bin/python3`),
+  ledger: cfg.paths.ledger,
+  // 2026-09-20（迭代2-步骤2）：租约与集成名进模板 —— 提示词里**不许**再写死这些
+  // （曾写"带 10 分钟租约"，而注册表后来统一成 20 分钟 ⇒ 提示词说谎且没人发现）。
+  leaseMin: String((cfg.concurrency || {}).leaseMin ?? ''),
+  integration: String(cfg.integration || ''),
+  reason: REASON,
+  case: CASE,
+  attempt: ATTEMPT || '1',
+  error: ERROR || '（未提供摘要，请读日志）',
+}
+// 裁定提示词：只讲"人拍板了什么 + 你要做什么 + 怎么回填"，不复述流程知识（角色已具备）
+const RULING_PROMPT = [
+  `【用户裁定 · 流程 ${cfg.title}（\`${FLOW}\`）】`,
+  ``,
+  `用户已就这条失败票作出决定，请**按用户裁定执行**，不要再重新论证该不该做：`,
+  ``,
+  `- 票：\`${CASE || '（未给 case，无法执行）'}\``,
+  `- 用户裁定原文（逐字，以此为准）：`,
+  `  > ${NOTE || '（未提供裁定原文）'}`,
+  ``,
+  `执行要求：`,
+  `1) 先按 SOP 的派发契约派 subagent 干活（**一店一批**，命令照抄 \`${join(CLONE, 'scripts')}/flow-prompts/${FLOW}.md\`）；`,
+  `2) 回填台账（\`${cfg.paths.ledger}\`）：`,
+  `   \`${vars.py} ${join(CLONE, 'scripts')}/flow_ledger.py --case-id ${CASE || '<case>'} --event retry --by ${cfg.role} --error "用户裁定：<一句话>"\`，`,
+  `   派发后再记 \`--event dispatched\`（带 attempt）；`,
+  `3) 若你判断用户的裁定**在技术上做不到**（例如票上根本没有可续跑的货件号），`,
+  `   **不要硬跑**：记 \`blocked\` 并把障碍讲清楚，回过头问用户。`,
+].join('\n')
+
+const PROMPT = RULING ? RULING_PROMPT : tpl.replace(/\{\{(\w+)\}\}/g, (m, k) => (k in vars ? String(vars[k]) : m))
+
+// ── 鉴权（**多来源 + 先验后用**；唯一实现在 scripts/lib/wake-auth.mjs）──────
+// 历史形态（2026-10-08 前）：只读 `/tmp/dsh-web.log` 里最后一次 `token=…`。dsh 的启动 token 是
+// **每进程随机**的、只在启动日志里 ⇒ 日志一没（/tmp 被清理）就彻底取不到，且没有任何退路。
+// 现在按优先级试：① 显式 token 文件 ② **凭据里的 browser-session 密钥现签 cookie（不依赖 /tmp）**
+// ③ DSH_WEB_LOG ④ plist 的 StandardOutPath ⑤ 稳定目录候选 ⑥ /tmp（旧机器兜底）。
+// **延迟获取**：dry-run / 纯预览路径不该因为"没 token"而失败（原实现在预览前就取 token）。
+let AUTH = null
+async function auth() {
+  if (AUTH) return AUTH
+  const sources = authSources({
+    env: process.env,
+    home: homedir(),
+    dshHome: process.env.DSH_HOME || join(homedir(), '.dsh'),
+    base: BASE,
+  })
+  const r = await acquireWakeAuth({ sources, base: BASE })
+  if (!r.ok) {
+    const head = r.unreachable
+      ? `dsh web 不可达（${BASE}）：${r.unreachable} —— 唤醒无法进行（服务没起就别谈唤醒）`
+      : authFailureText({ tried: r.tried, base: BASE, home: homedir(), dshHome: process.env.DSH_HOME || join(homedir(), '.dsh') })
+    failWake(head, `唤醒取不到鉴权（尝试 ${r.tried.length} 个来源全失败${r.unreachable ? '；服务不可达' : ''}）`)
+  }
+  log(`🔑 唤醒鉴权来源：${r.source}${(() => {
+    // 跳过摘要：**为什么没用上更靠前的来源**必须看得见（否则"凭据明明在、却走了日志"没人能解释）
+    const skip = (r.tried || []).filter((t) => t.status !== 'ok')
+    if (!skip.length) return ''
+    return `（此前跳过 ${skip.length} 个：${skip.map((t) => `${t.id}=${SOURCE_STATUS_ZH[t.status] || t.status}`).join('；')}）`
+  })()}`)
+  AUTH = r
+  return r
+}
+const rpc = async (method, args) => {
+  const { cookie } = await auth()
+  const r = await fetch(`${BASE}/api/${method}`, {
+    method: 'POST',
+    headers: { 'content-type': 'application/json', cookie: `${cookie.name}=${cookie.value}` },
+    body: JSON.stringify({ type: 'client-request', rpcId: 'trig-' + Math.random().toString(36).slice(2), method, payload: { args } }),
+  })
+  return r.json()
+}
+
+// ── 会话复用 ────────────────────────────────────────────────────
+const readSid = () => { try { return JSON.parse(readFileSync(cfg.paths.sessionIdFile, 'utf8')).sessionId || '' } catch { return '' } }
+const writeSid = (sid) => {
+  try {
+    mkdirSync(dirname(cfg.paths.sessionIdFile), { recursive: true })
+    writeFileSync(cfg.paths.sessionIdFile, JSON.stringify({ sessionId: sid, flow: FLOW, updatedAt: new Date().toISOString() }, null, 2) + '\n')
+  } catch (e) { log(`⚠️ 会话 id 落盘失败（下轮会新建会话）：${e.message}`) }
+}
+async function createSession() {
+  const c = await rpc('session/create', { request: { agentPreset: cfg.dispatch.preset, cwd: cfg.dispatch.cwd.replace(/^~/, homedir()) } })
+  const sid = c?.result?.value?.sessionId
+  if (!sid) failWake(`建会话失败：${JSON.stringify(c).slice(0, 200)}`, '唤醒建会话失败（session/create 未返回 sessionId）')
+  writeSid(sid)
+  return sid
+}
+const promptOf = (sid, text) => rpc('session/prompt', {
+  request: { requestId: 'req-' + Math.random().toString(36).slice(2), sessionId: sid, mode: 'queue', content: [{ type: 'text', text: text || PROMPT }] },
+})
+
+// ── dry-run：只解析配置/模板并预览，**不鉴权、不建会话、不投递** ──
+let sid = readSid()
+
+// ── --notice：投递一条自由文本指令给流程角色（不派单、不改台账）──────────
+// 为什么需要（2026-09-23）：美客多限流时角色仍在重跑手头的票，"让角色知道平台慢/别再试"
+// 此前没有通道 —— 只能等它自己撞够失败。有了它就能一句话叫停/叫等。
+if (NOTICE) {
+  const text = `【流程通知 · ${new Date().toLocaleString('zh-CN')}】\n${NOTICE}`
+  if (DRY) { log(`✅ [dry-run] 将向 ${sid || '（新会话）'} 投递通知：\n${text}`); process.exit(0) }
+  if (!sid) sid = await createSession()
+  const r = await promptOf(sid, text)
+  if (r?.error) failWake(`通知投递失败：${JSON.stringify(r).slice(0, 200)}`, `流程通知投递失败（reason=notice）：${JSON.stringify(r).slice(0, 160)}`)
+  log(`✅ 通知已投递给流程 ${FLOW} 的会话 ${sid}`)
+  process.exit(0)
+}
+
+
+if (DRY) {
+  const what = REASON === 'failure' ? `失败交接 case=${CASE || '（未给）'}`
+    : RULING ? `用户裁定 case=${CASE || '（未给）'}`
+    : `票=${TICKETS.join('、') || '（无）'}`
+  log(`✅ [dry-run] 流程 ${FLOW}：本轮语义=${MODE_LABEL} reason=${REASON} ${what} preset=${cfg.dispatch.preset} 最多 ${MAX} 张`
+    + `｜${sid ? '将复用会话 ' + sid : '将新建会话'}｜模板 ${RULING ? '（ruling 直传，不用模板）' : TPL_PATH}（${PROMPT.length} 字）`
+    + `｜**未建会话、未投递**${RULING ? '\n──── 将投递的提示词 ────\n' + PROMPT : ''}`)
+  process.exit(0)
+}
+
+let created = false
+if (!sid) { sid = await createSession(); created = true }
+
+let p = await promptOf(sid)
+if (!p?.result?.ok && !created) {
+  // 复用的会话已不可用 ⇒ 新建一个再投（旧行为：每轮新建；这里只在必要时新建）
+  log(`⚠️ 复用会话 ${sid} 投递失败 ⇒ 改为新建会话重投`)
+  sid = await createSession(); created = true
+  p = await promptOf(sid)
+}
+if (p?.result?.ok) {
+  log(`✅ 已触发流程 ${FLOW} 的会话 ${sid}${created ? '（新建）' : '（复用）'}`
+    + `（reason=${REASON}${REASON === 'failure' ? ' case=' + CASE : ''}，最多 ${MAX} 张，${MODE_LABEL}）`)
+  process.exit(0)
+}
+failWake(`投递失败：${JSON.stringify(p).slice(0, 200)}`, `唤醒投递失败（session/prompt 未 accepted）：${JSON.stringify(p).slice(0, 160)}`)