MailUI4Agents/plugins/zcode-mail-bridge/hooks/permission.mjs

#!/usr/bin/env node
/**
 * ZCode `PermissionRequest` 钩子 —— AgentMail 的授权桥。
 *
 * # 它在链路里的位置
 *
 * ZCode 决定某个工具需要授权时会触发本钩子，并把事件 JSON 写到 stdin：
 *
 *   { hook_event_name: "PermissionRequest", tool_name: "Bash",
 *     tool_input: {...}, session_id: "...", permission_mode: "...", ... }
 *
 * 我们在 stdout 回一个结论（这是从 CLI 产物里逆出来的 schema，不是猜的）：
 *
 *   {"decision":"approve"}                     放行
 *   {"decision":"block","reason":"..."}        拒绝（并让模型看到原因）
 *   什么都不输出                                 不表态，退回 ZCode 自己的权限流程
 *
 * schema 是**严格**的：多一个键就会让 ZCode 报
 * "Hook stdout failed HookJSONOutput schema validation"，
 * 所以这里只输出这两个键。
 *
 * # 为什么自己开 SSE，而不是找桥要
 *
 * 决策是人点出来的，通过网关的 `permission_decision` 事件下发。
 * 钩子是**一次性进程**，没有常驻连接可用；而网关的 SSE 是**扇出**的
 * （`clients` 按唯一 id 存，`SendToAgent` 推给该 Agent 的所有客户端），
 * 所以钩子可以自己订阅、拿到自己那条决定、然后退出。
 *
 * 这样做的直接好处：**交互模式下也能用** —— 人自己开着 ZCode 干活时并没有
 * 桥进程在跑，若改成「问桥要结论」，这个功能就只在邮件驱动时才存在。
 *
 * # 失败一律 fail closed（但区分模式）
 *
 * 只有明确同意才放行；看不懂的决策文本一律当拒绝（判定交给共用库）。
 * 暂时性失败（5xx / 网络）分两种：邮件驱动的会话没有本地界面兜底，
 * 所以驳回并说明；交互模式则退回 ZCode 自己的权限流程，让人就地决定。
 */

import { readFileSync } from 'node:fs';
import { GatewayClient } from '../lib/gateway.mjs';
import { createSSEClient } from '../lib/sse-client.js';
import { clampRelayKey, isPermanentFailure } from '../lib/relay-key.js';
import { isApproval, isAlwaysDecision } from '../lib/permission-grants.js';
import {
  decidePolicy,
  describeToolCall,
  PERMISSION_EVENT
} from '../lib/hook-policy.mjs';
import { createFileGrantStore, grantsFilePath } from '../lib/grants-file.mjs';

const log = (...parts) => console.error('[agentmail-hook]', ...parts);

/** 等待人工决策的上限。必须**小于** hooks.json 里的 timeoutMs，
 *  否则会是 ZCode 先把钩子杀掉（报成「钩子失败」），而不是我们给出结论。 */
const WAIT_MS = Number(process.env.AGENTMAIL_PERMISSION_WAIT_MS || 540000);

/** 回一个结论并退出。stdout 只允许出现这一个 JSON 对象。 */
function emit(obj) {
  if (obj !== null) process.stdout.write(`${JSON.stringify(obj)}\n`);
  process.exit(0);
}

const approve = () => emit({ decision: 'approve' });
const block = reason => emit({ decision: 'block', reason });
const noOpinion = () => emit(null);

function readHookInput() {
  try {
    return JSON.parse(readFileSync(0, 'utf8'));
  } catch (e) {
    log('stdin 不是合法 JSON:', e?.message || e);
    return null;
  }
}

/**
 * 等 SSE 建连完成。
 *
 * 必须先连上再发权限请求：反过来会有一个窗口 —— 人恰好在窗口内点了同意，
 * 而事件推送给了当时还不存在的客户端，于是这条决定永远等不到
 * （表现为「明明点了同意，工具还是被拒」）。服务端在 AddClient 时会立刻下发
 * 一个 `connected` 事件，就用它做信号。
 */
function waitConnected(sseState) {
  return new Promise(resolve => {
    const timer = setTimeout(() => {
      log('SSE 建连等待超时，仍然继续（可能错过极早到达的决策）');
      resolve();
    }, 5000);
    sseState.onConnected = () => {
      clearTimeout(timer);
      resolve();
    };
  });
}

async function askHuman({ client, baseURL, input, toolName, relayKey, sessionId }) {
  const sseState = { onConnected: null };
  const decisions = [];
  let waiter = null;

  const sse = createSSEClient({
    authHeaders: () => client.authHeaders(),
    baseURL,
    path: '/api/v1/events/stream',
    log,
    onEvent: (evt, data) => {
      if (evt === 'connected' && sseState.onConnected) sseState.onConnected();
      if (evt !== 'permission_decision') return;
      // 只认自己那条：同一 Agent 可能有多个钩子进程同时在等
      // （模型并行发起两个 Bash），按 relay_key 配对才不会互相拿到对方的决定。
      if (data?.relay_key && data.relay_key !== relayKey) return;
      if (waiter) {
        const w = waiter;
        waiter = null;
        w(data);
      } else {
        decisions.push(data);
      }
    }
  });

  try {
    await waitConnected(sseState);

    // 不传 `to`：决策人由服务端按 会话 owner → 线索里最近的人类 → 409 解析。
    // 插件只有本地上下文，猜不出「这条 Agent 链最初是谁派的活」。
    await client.post('/permission/request', {
      question: `是否允许执行 ${toolName}？`,
      options: ['同意', '一直同意', '拒绝'],
      context: [
        describeToolCall(toolName, input.tool_input),
        process.env.AGENTMAIL_MAIL_SUBJECT
          ? `\n触发任务：${process.env.AGENTMAIL_MAIL_SUBJECT}`
          : '',
        process.env.AGENTMAIL_REPLY_TO ? `任务来自：${process.env.AGENTMAIL_REPLY_TO}` : ''
      ]
        .filter(Boolean)
        .join('\n'),
      session_id: sessionId || '',
      relay_key: relayKey
    });

    const decision = decisions.shift() ?? (await new Promise(resolve => {
      // 挂上等待者；超时后也要把 waiter 摘掉，否则后续事件会去 resolve
      // 一个已经没人听的 promise（并让 sse.stop 之后的日志显得诡异）。
      waiter = resolve;
      setTimeout(() => {
        if (waiter !== resolve) return;
        waiter = null;
        resolve(null);
      }, WAIT_MS).unref?.();
    }));
    return decision;
  } finally {
    sse.stop();
  }
}

async function main() {
  const input = readHookInput();
  if (!input) return noOpinion();

  const toolName = input.tool_name;
  const mode = process.env.AGENTMAIL_PERMISSION_MODE;
  const policy = decidePolicy({
    event: input.hook_event_name,
    toolName,
    mode
  });
  log(`事件 ${input.hook_event_name} 工具 ${toolName} 档位 ${mode || '(默认)'} → ${policy.action}`);

  if (policy.action === 'none') return noOpinion();
  if (policy.action === 'approve') return approve();
  if (policy.action === 'block') return block(policy.reason);

  // ── 问人 ──
  const client = new GatewayClient(process.env);
  const missing = client.checkConfig();
  if (missing.length) {
    log(`未配置：${missing.join('、')}`);
    // 没配好就无法问人。交互模式下退回本地流程仍然可用；
    // 邮件驱动的会话没有本地界面，必须当场说清楚而不是静默挂住。
    return process.env.AGENTMAIL_SESSION_ID
      ? block(`AgentMail 授权桥未配置（缺少 ${missing.join('、')}），无法征求授权，已拒绝 ${toolName}。`)
      : noOpinion();
  }

  const sessionId = process.env.AGENTMAIL_SESSION_ID || '';
  const relayKey = clampRelayKey(
    `${sessionId || input.session_id || 'zcode'}:${input.tool_use_id || toolName}`
  );

  // 「一直同意」要真的记住：钩子一封一进程，所以授权表落盘。
  const grants = createFileGrantStore(grantsFilePath(process.env));
  const grantScope = sessionId || input.session_id || '';
  if (grants.isGranted(grantScope, toolName)) {
    log(`${toolName} 在本会话已获「一直同意」（${grantsFilePath(process.env)}），直接放行`);
    return approve();
  }

  let decision;
  try {
    decision = await askHuman({ client, baseURL: client.baseURL, input, toolName, relayKey, sessionId });
  } catch (e) {
    // 409 = 服务端判定这条任务链上没有人类，永远不会有人来点头。
    // 永久失败（4xx）同样不会因重试而改变 —— 两者都必须当场拒绝，
    // 让模型从工具报错里看到原因并自己改道（挂死时连重试机会都没有）。
    if (isPermanentFailure(e)) {
      const b = e?.body && typeof e.body === 'object' ? e.body : {};
      const reason = [
        b.error || `权限询问无法送达（HTTP ${e?.status}）`,
        typeof e?.body === 'string' ? e.body : '',
        b.detail || '',
        b.suggestion || ''
      ]
        .filter(Boolean)
        .join('\n');
      log(`权限询问永久失败，当场拒绝 ${relayKey}：${reason}`);
      return block(reason);
    }
    const detail = e?.message || String(e);
    log(`权限询问暂时失败：${detail}`);
    if (process.env.AGENTMAIL_SESSION_ID) {
      // 邮件驱动：没有本地界面兜底，退回本地决策等于守卫消失。
      return block(
        `无法把 ${toolName} 的授权请求送达给人（${detail}）。` +
          `这条会话由邮件驱动、没有本地界面，因此不放行。` +
          `请改用不需要授权的方式完成，或在回信里说明需要人工执行哪一步。`
      );
    }
    return noOpinion();
  }

  if (decision === null) {
    return block(
      `等待授权超时（${Math.round(WAIT_MS / 1000)} 秒内没有人决策），未执行 ${toolName}。`
    );
  }

  const text = decision.decision ?? '';
  if (isApproval(text)) {
    if (isAlwaysDecision(text) && grants.grant(grantScope, toolName, text)) {
      log(`记下「一直同意」：会话 ${grantScope} 的 ${toolName} 后续免批`);
    }
    log(`授权 ${relayKey} 获批（${text}${decision.decided_by ? ` by ${decision.decided_by}` : ''}）`);
    return approve();
  }

  return block(
    [
      `用户拒绝了这次 ${toolName} 调用。`,
      decision.note ? `说明：${decision.note}` : '',
      decision.decided_by ? `（由 ${decision.decided_by} 决定）` : ''
    ]
      .filter(Boolean)
      .join('\n')
  );
}

main().catch(e => {
  // 钩子自身崩了**不能**静默退回 ZCode 的权限流程 —— 那在有本地界面时是
  // 合理兜底，在邮件驱动时等于守卫消失。所以这里区分模式，并把原因写在
  // stderr（进 ZCode 日志）供人排查。
  log('钩子异常:', e?.stack || e);
  if (process.env.AGENTMAIL_SESSION_ID) {
    block(`AgentMail 授权钩子内部错误：${e?.message || e}。未执行工具。`);
  }
  noOpinion();
});