wezard 1.3.13 → 1.3.16

package/dist/daemon/approval.js

@@ -1093,6 +1093,151 @@ export const runMirrorPlanFlow = async ({ log, client, sessionId, chatKey, cwd,
     log.info({ reqId, sessionId, approve }, "mirror plan resolved via send-keys");
     await note(approve ? "✅ 已同意，退出 plan mode 开始执行…" : "✏️ 已选择继续完善计划。");
 };
+const mirrorPickerSlots = new Map();
+const MIRROR_PICKER_SLOT_TTL_MS = 12 * 3600_000; // 与 approval.longPollSec 对齐
+const freshPickerSlot = (sessionId) => {
+    const s = mirrorPickerSlots.get(sessionId);
+    if (!s)
+        return undefined;
+    if (Date.now() - s.at > MIRROR_PICKER_SLOT_TTL_MS) {
+        mirrorPickerSlots.delete(sessionId);
+        return undefined;
+    }
+    return s;
+};
+export const hasMirrorPicker = (sessionId) => freshPickerSlot(sessionId) !== undefined;
+export const mootMirrorPicker = (sessionId) => {
+    const s = freshPickerSlot(sessionId);
+    if (!s)
+        return;
+    if (s.reqId)
+        resolvePending(s.reqId, "moot");
+    mirrorPickerSlots.delete(sessionId);
+};
+// ── picker 代按 → 迟到 hook 请求的去重 ─────────────────────────────────
+// 这类调用的 hook 在本地确认框被按掉**之后**才到 (CodeBuddy「本地面板先行」,
+// 与 askq 同源)。代按 Yes 的那一下就是人工审批结论 —— hook 随后送来的同一调用
+// 不该再发第二张卡。一次性消费 + 短 TTL: 只豁免紧跟着代按的那一次上报, 不构成
+// 任何持久放行。
+const recentPickerAllows = new Map(); // `${sessionId}|${toolName}` → at
+const RECENT_PICKER_TTL_MS = 60_000;
+const recordPickerAllow = (sessionId, toolName) => {
+    recentPickerAllows.set(`${sessionId}|${toolName}`, Date.now());
+};
+const consumePickerAllow = (sessionId, toolName) => {
+    const k = `${sessionId}|${toolName}`;
+    const at = recentPickerAllows.get(k);
+    if (at === undefined)
+        return false;
+    recentPickerAllows.delete(k);
+    return Date.now() - at <= RECENT_PICKER_TTL_MS;
+};
+export const runMirrorPickerFlow = async ({ log, client, cfg, sessionId, chatKey, cwd, jsonlPath, toolName, toolInput, agent, danger, voteTimeoutMs, press }) => {
+    if (!client.isConnected)
+        return;
+    const target = targetChatId(chatKey);
+    const note = async (content) => {
+        try {
+            await client.sendMessage(target, { msgtype: "markdown", markdown: { content: withTagHeader(chatKey, content) } });
+        }
+        catch { /* best-effort */ }
+    };
+    const tail = agent ? `子代理 ${agent} 的原生权限确认` : "CLI 原生权限确认";
+    const sessionName = sessionNameFor(chatKey, jsonlPath, sessionId);
+    const { reqId, promise } = createPending({
+        meta: {
+            kind: "generic",
+            createdAt: Date.now(),
+            toolName,
+            toolInput,
+            cwd,
+            sessionId,
+            chatKey,
+            transcriptTail: tail,
+            danger,
+            forceSingle: Boolean(danger),
+            sessionName,
+        },
+        timeoutMs: voteTimeoutMs,
+    });
+    mirrorPickerSlots.set(sessionId, { reqId, at: Date.now() });
+    try {
+        await client.sendMessage(target, {
+            msgtype: "template_card",
+            template_card: buildCard({
+                reqId,
+                toolName,
+                toolInput,
+                toolInputStr: "",
+                cwd,
+                sessionShort: sessionId.slice(-8),
+                sessionId,
+                chatKey,
+                transcriptTail: tail,
+                windowMinutes: cfg.approval.windowMinutes,
+                danger,
+                forceSingle: Boolean(danger),
+                sessionName,
+            }),
+        });
+        log.info({ reqId, chatKey, toolName, agent }, "mirror picker card sent");
+    }
+    catch (e) {
+        log.error({ err: e.message }, "mirror picker card send failed");
+        resolvePending(reqId, "deny");
+        mirrorPickerSlots.delete(sessionId);
+        return;
+    }
+    let raw;
+    try {
+        raw = (await promise);
+    }
+    catch {
+        mirrorPickerSlots.delete(sessionId);
+        await note(`⌛ 权限确认卡已超时（${toolName}），请在 CLI 中处理。`);
+        return;
+    }
+    // 本地先按掉 (哨兵读到框消失触发的 moot) — 槽位已由 mootMirrorPicker 清除。
+    if (raw === "moot") {
+        log.info({ reqId, sessionId }, "mirror picker mooted by local answer");
+        return;
+    }
+    // 槽位保持到代按完成 — 哨兵靠它跳过在途 pane; 提前清会与代按赛跑出双按键。
+    try {
+        const d = raw;
+        if (d === "deny") {
+            const r = await press("no");
+            await note(r.ok
+                ? `🚫 已替你拒绝 CLI 权限确认（${toolName}）→ ${r.label ?? "Escape"}`
+                : `⚠️ 已记录拒绝，但代按失败（${r.reason ?? "未知"}），请回 CLI 处理。`);
+            return;
+        }
+        if (d === "allow_window" && cfg.approval.windowMinutes > 0) {
+            setAutoWindow(chatKey, cfg.approval.windowMinutes * 60_000, { toolName, toolInput, cwd, transcriptTail: tail });
+            // 与 handler 的 allow_window 同款 sweep: 同 chat 其它 pending 卡一并放行。
+            const swept = resolvePendingsByChat(chatKey, "allow_window", reqId);
+            log.info({ chatKey, minutes: cfg.approval.windowMinutes, swept: swept.length }, "auto-window opened via picker card");
+            if (swept.length > 0)
+                await note(`⚡ 已批量自动放行其他 ${swept.length} 个并发请求。`);
+        }
+        const wantSession = d === "allow_always" || d === "allow_session";
+        const r = await press(wantSession ? "yes_session" : "yes");
+        if (!r.ok) {
+            await note(`⚠️ 已记录同意，但代按失败（${r.reason ?? "未知"}），请回 CLI 手动确认。`);
+            return;
+        }
+        // 代按成功 → CLI 才真正派发工具 → hook 此刻才会上报同一调用; 预先记账让
+        // handler 直接放行, 不再发第二张卡 (consumePickerAllow)。
+        recordPickerAllow(sessionId, toolName);
+        await note(wantSession
+            ? `🔓 已代按「本会话不再询问」（${toolName}）→ ${r.label ?? ""}\n原生确认框不过 hook，「总是」规则帮不上它，已翻译成 CLI 的会话级免问。`
+            : `🔓 已代按 CLI 权限确认（${toolName}）→ ${r.label ?? "Yes"}${d === "allow_window" ? `，并开启 ${cfg.approval.windowMinutes} 分钟自动放行窗口` : ""}`);
+        log.info({ reqId, sessionId, decision: d, pressed: r.label }, "mirror picker resolved via send-keys");
+    }
+    finally {
+        mirrorPickerSlots.delete(sessionId);
+    }
+};
 // 多问题: 逐题顺序发卡 → 收答 → 下一题。任一题选「CLI」整体转 CLI,选「聊聊」
 // 整体转讨论; 全部答完合并成单个 deny+reason 注入。卡不会一次性轰炸 N 张。
 const handleAskUserQuestion = async ({ cfg, log, client, body, getMirrorTarget, flushBeforeCard, clientGone }) => {
@@ -1256,20 +1401,45 @@ const resolveApprover = (cfg, sessionId, getMirrorTarget) => {
 // 只剩 transcript_path + agent 标记能对回父会话。session_id 无镜像绑定时, 按
 // transcript_path 反推父会话并路由过去 —— 否则这类请求落到 ask 兜底, 在镜像
 // pane 里变成远程无人可点的原生确认框 (卡住、无卡片), skipAll/卡片全部失联。
+const SUBAGENT_TRANSCRIPT_RE = /\/([^/]+)\/subagents\/agent-[^/]+\.jsonl$/;
 const subagentParentOf = (transcriptPath, agentId, agentType) => {
-    if (!transcriptPath || (!agentId && !agentType))
+    if (!transcriptPath)
         return undefined;
     const norm = transcriptPath.replace(/\\/g, "/");
-    // `<projectDir>/<encodedCwd>/<parentSid>/subagents/agent-*.jsonl` — 父在上一级
-    const sub = norm.match(/\/([^/]+)\/subagents\/agent-[^/]+\.jsonl$/);
+    // `<projectDir>/<encodedCwd>/<parentSid>/subagents/agent-*.jsonl` — 父在上一级。
+    // 布局本身就是子代理证据, 不要求 agent 标记 —— 部分 CodeBuddy 版本的子代理
+    // hook 不带 agent_id/agent_type, 只认标记会让这类请求丢失父归属。
+    const sub = norm.match(SUBAGENT_TRANSCRIPT_RE);
     if (sub)
         return sub[1];
     // 直接落在 `<projectDir>/<uuid>.jsonl` 的主转录布局 (CC 子代理 hook 给的是父转录)
+    // 给不出子代理证据, 必须有 agent 标记佐证才敢当父 sid 用。
+    if (!agentId && !agentType)
+        return undefined;
     const main = norm.match(/\/([0-9a-fA-F-]{36})\.jsonl$/);
     if (main)
         return main[1];
     return undefined;
 };
+// ── DeferExecuteTool 下钻 ──────────────────────────────────────────────
+// CodeBuddy 的延迟加载工具 (ToolSearch → DeferExecuteTool) 把真实工具包进
+// tool_input.toolName / .toolInput。审批链 (matcher/danger/allow/deny/卡片/缓存)
+// 一律按内层工具裁决 —— 外层名匹配不到任何用户规则, 卡上也读不出语义。hook 侧
+// (pre-tool-use.sh) 已做同样下钻; 这里再剥一层是给旧版 hook / 直连调用兜底。
+const unwrapDeferredTool = (toolName, toolInput) => {
+    if (toolName !== "DeferExecuteTool" || typeof toolInput !== "object" || toolInput === null)
+        return { toolName, toolInput };
+    const o = toolInput;
+    const inner = [o.toolName, o.tool_name, o.name].find((v) => typeof v === "string" && v.length > 0);
+    if (!inner)
+        return { toolName, toolInput };
+    // 内层入参字段名各版本有出入, 依次探测; 都没有则剥掉包装字段取余下对象。
+    const args = [o.toolInput, o.tool_input, o.arguments, o.args, o.input].find((v) => v !== undefined);
+    if (args !== undefined)
+        return { toolName: inner, toolInput: args };
+    const { toolName: _tn, tool_name: _tn2, name: _n, ...rest } = o;
+    return { toolName: inner, toolInput: rest };
+};
 export const makeApproveHandler = ({ cfg, log, client, sourcePath, getMirrorTarget, flushBeforeCard, nativeModal }) => {
     setCardQuoteMax(cfg.approval.cardQuoteMaxChars);
     const detailUrlFor = (id, approver) => buildDetailUrl(cfg.daemon.detailPublicBase, cfg.daemon.host, cfg.daemon.port, id, approver ? targetChatId(approver) : undefined);
@@ -1448,8 +1618,7 @@ export const makeApproveHandler = ({ cfg, log, client, sourcePath, getMirrorTarg
         }
         const body = (await readBody(req));
         let sessionId = body.session_id ?? "";
-        const toolName = body.tool_name ?? "";
-        const toolInput = body.tool_input ?? {};
+        const { toolName, toolInput } = unwrapDeferredTool(body.tool_name ?? "", body.tool_input ?? {});
         const cwd = body.cwd ?? "";
         const transcriptTail = body.transcript_tail ?? "";
         if (!toolName) {
@@ -1462,7 +1631,10 @@ export const makeApproveHandler = ({ cfg, log, client, sourcePath, getMirrorTarg
         // 主会话 / 已解析到归属的子代理请求 (CC/CodeBuddy 都上报父 id) 完全不受影响。
         // 不 gate 在 mirror: headless 下 getMirrorTarget 同样是「session → 归属 chat」的解析
         // (由 index.ts 注入 sessions store 反查), 父归属逻辑对两种模式统一生效。
-        const fromSubagent = Boolean(body.agent_id || body.agent_type);
+        // agent 标记之外, subagents/ 转录布局同样是子代理证据 (部分 CodeBuddy 版本
+        // 两个标记都不带) —— 否则这类请求按主会话处理, no-approver 时挂进无人能点的 ask。
+        const fromSubagent = Boolean(body.agent_id || body.agent_type)
+            || SUBAGENT_TRANSCRIPT_RE.test((body.transcript_path ?? "").replace(/\\/g, "/"));
         if (fromSubagent && sessionId && !getMirrorTarget?.(sessionId)) {
             const parent = subagentParentOf(body.transcript_path ?? "", body.agent_id ?? "", body.agent_type ?? "");
             if (parent && parent !== sessionId) {
@@ -1620,6 +1792,15 @@ export const makeApproveHandler = ({ cfg, log, client, sourcePath, getMirrorTarg
             settleGuard();
             return;
         }
+        // picker 代按后的迟到 hook 上报 (CodeBuddy「本地面板先行」): 刚在审批卡上点过
+        // ✅ 并被代按 Yes 的调用, 不再发第二张卡。一次性消费, 压过 mustCard —— 危险卡
+        // 那次也是人刚点过的。
+        if (consumePickerAllow(sessionId, toolName)) {
+            log.info({ toolName, sessionId }, "picker-card allow replay — auto allow");
+            json(res, 200, { decision: "allow", reason: "picker_card_allow" });
+            settleGuard();
+            return;
+        }
         // danger.skip / danger 模式的早退。第三个参数是「除 danger 外还有没有别的必发卡
         // 理由」—— askRules 不能被 danger 的开关顺带关掉。守卫不在其列: 它要的是事后按框,
         // 早退照样能给 (settleGuard), 拿它挡早退等于把 danger 的开关废掉。