dsh-archify-projects 0.0.0-stage → 0.1.0

package/lib/host/policy.js

@@ -0,0 +1,124 @@
+/**
+ * 会话策略：这个 agent 能碰什么。
+ *
+ * 三件事集中在这里，因为它们必须一致：
+ *   1. **工具白名单**（`tools.restrict` 的 allow，模型看得见什么）；
+ *   2. **执行层守卫**（`tools.guard`，拦不在名单里的工具、拦非 archify 的 skill）；
+ *   3. **skill 收口口径**（唯一允许的名字）。
+ *
+ * 为什么必须在 `agent.ctx` 上装：`tools.restrict` 的实现第一行是
+ * `scopeOf(this.ctx)` —— 根 ctx 直接抛「requires a scoped context (agent.ctx)」。
+ * 在 runner 里、会话刚建好、第一条 prompt 之前装，顺带拿到「只影响本插件起的会话」
+ * 这个性质：别的 preset 起的 agent 一个都不动。
+ */
+/** 唯一的 skill 名。 */
+export const ARCHIFY_SKILL_NAME = 'archify';
+/**
+ * 基础工具白名单（不含 shell，shell 按平台补）。
+ *
+ * `todo_write` 必须在内：restrict 是按名字收口的，漏掉它模型就「看不到」这个工具
+ * （实测 agent 会在回复里写「I don't have a todo_write tool」），页面也就没有进度。
+ * 它只写自己的计划，不扩大任何能力面。
+ */
+const BASE_ALLOW = ['read', 'write', 'edit', 'glob', 'grep', 'skill', 'todo_write'];
+/**
+ * 问答会话的白名单：只读代码 + 加载 archify（不需要出图，但保留 skill 方便看口径）。
+ * 没有 shell、没有写工具 —— 用户要求「会话只剩 archify 和读代码的能力」。
+ */
+const READONLY_ALLOW = ['read', 'glob', 'grep', 'skill', 'todo_write'];
+/**
+ * 这个受限 agent 允许使用的工具全集。
+ *
+ * shell 按平台二选一：宿主在 Windows 上只注册 `pwsh`，写 `bash` 会让
+ * `restrict` 因为「未知工具名」直接抛。
+ */
+export function allowedToolNames(platform = process.platform) {
+    return [...BASE_ALLOW, platform === 'win32' ? 'pwsh' : 'bash'];
+}
+/** 只读会话（提问）允许的工具：读文件、找文件、加载 archify，别的都没有。 */
+export function readonlyToolNames() {
+    return [...READONLY_ALLOW];
+}
+/**
+ * 执行层守卫：单调、同步、返回字符串即拒绝。
+ *
+ * 两道检查合在一个函数里，因为 guard 只能注册一次，且必须同步给结论：
+ *   - `skill` 只认 archify；
+ *   - 其余工具必须在白名单里。
+ *
+ * 白名单已经收过一遍了，这里仍再查一次 —— restrict 可能因为「未知工具名」降级，
+ * 而 guard 不看注册表，是最后一道能真正拦住执行的闸。
+ */
+export function makeGuard(allow) {
+    const allowed = new Set(allow);
+    return (exec) => {
+        if (exec.name === 'skill') {
+            const requested = readSkillName(exec.arguments);
+            if (requested !== ARCHIFY_SKILL_NAME) {
+                return `这个会话只允许加载 archify skill（请求的是 ${requested ?? '(未指定)'}）`;
+            }
+            return undefined;
+        }
+        if (!allowed.has(exec.name)) {
+            return `工具 "${exec.name}" 在这个会话里被禁用：本会话只允许 ${allow.join(' / ')}`;
+        }
+        return undefined;
+    };
+}
+function readSkillName(args) {
+    if (typeof args !== 'object' || args === null)
+        return null;
+    const name = args.name;
+    return typeof name === 'string' ? name : null;
+}
+/**
+ * 装白名单 + 守卫。
+ *
+ * 两步各自容错：`restrict` 对未注册的名字会抛，抛了也继续装 guard ——
+ * 少了哪一层都要在日志里说清楚，不静默降级。
+ */
+export function tightenAgent(scope, allow, warn) {
+    if (scope === undefined || scope.tools === undefined) {
+        const message = 'agent 作用域没有 tools 服务：白名单未生效';
+        warn(message);
+        return { restricted: false, guarded: false, message };
+    }
+    const tools = scope.tools;
+    let restricted = false;
+    let guarded = false;
+    let message = null;
+    try {
+        run(scope, () => tools.restrict?.({ allow }) ?? undefined, 'dsh-archify-projects: tool allowlist');
+        restricted = true;
+    }
+    catch (err) {
+        message = `工具白名单未生效：${describe(err)}`;
+        warn(message);
+    }
+    try {
+        run(scope, () => tools.guard?.(makeGuard(allow)) ?? undefined, 'dsh-archify-projects: tool guard');
+        guarded = true;
+    }
+    catch (err) {
+        const text = `工具守卫未生效：${describe(err)}`;
+        warn(text);
+        message = message === null ? text : `${message}；${text}`;
+    }
+    return { restricted, guarded, message };
+}
+/** 装只读白名单 + 守卫（问答会话用）。 */
+export function tightenReadOnlyAgent(scope, warn) {
+    return tightenAgent(scope, READONLY_ALLOW, warn);
+}
+/** 有 `effect` 就交给它管生命周期，没有就直接装。 */
+function run(scope, install, label) {
+    if (typeof scope.effect === 'function') {
+        scope.effect(() => install() ?? undefined, label);
+        return;
+    }
+    install();
+}
+function describe(err) {
+    return err instanceof Error ? err.message : String(err);
+}
+//# sourceMappingURL=policy.js.map

package/lib/host/policy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policy.js","sourceRoot":"","sources":["../../src/host/policy.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAIH,mBAAmB;AACnB,MAAM,CAAC,MAAM,kBAAkB,GAAG,SAAS,CAAA;AAE3C;;;;;;GAMG;AACH,MAAM,UAAU,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,CAAU,CAAA;AAE5F;;;GAGG;AACH,MAAM,cAAc,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,CAAU,CAAA;AAE/E;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,WAAmB,OAAO,CAAC,QAAQ;IAClE,OAAO,CAAC,GAAG,UAAU,EAAE,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAA;AAChE,CAAC;AAED,8CAA8C;AAC9C,MAAM,UAAU,iBAAiB;IAC/B,OAAO,CAAC,GAAG,cAAc,CAAC,CAAA;AAC5B,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,SAAS,CACvB,KAAwB;IAExB,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAA;IAC9B,OAAO,CAAC,IAAuB,EAAsB,EAAE;QACrD,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC1B,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YAC/C,IAAI,SAAS,KAAK,kBAAkB,EAAE,CAAC;gBACrC,OAAO,gCAAgC,SAAS,IAAI,OAAO,GAAG,CAAA;YAChE,CAAC;YACD,OAAO,SAAS,CAAA;QAClB,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,OAAO,OAAO,IAAI,CAAC,IAAI,sBAAsB,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAA;QAClE,CAAC;QACD,OAAO,SAAS,CAAA;IAClB,CAAC,CAAA;AACH,CAAC;AAED,SAAS,aAAa,CAAC,IAAa;IAClC,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI;QAAE,OAAO,IAAI,CAAA;IAC1D,MAAM,IAAI,GAAI,IAA2B,CAAC,IAAI,CAAA;IAC9C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAA;AAC/C,CAAC;AASD;;;;;GAKG;AACH,MAAM,UAAU,YAAY,CAC1B,KAAiC,EACjC,KAAwB,EACxB,IAA+B;IAE/B,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACrD,MAAM,OAAO,GAAG,6BAA6B,CAAA;QAC7C,IAAI,CAAC,OAAO,CAAC,CAAA;QACb,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;IACvD,CAAC;IACD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAA;IACzB,IAAI,UAAU,GAAG,KAAK,CAAA;IACtB,IAAI,OAAO,GAAG,KAAK,CAAA;IACnB,IAAI,OAAO,GAAkB,IAAI,CAAA;IAEjC,IAAI,CAAC;QACH,GAAG,CACD,KAAK,EACL,GAAG,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,SAAS,EAC9C,sCAAsC,CACvC,CAAA;QACD,UAAU,GAAG,IAAI,CAAA;IACnB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,GAAG,YAAY,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAA;QACrC,IAAI,CAAC,OAAO,CAAC,CAAA;IACf,CAAC;IAED,IAAI,CAAC;QACH,GAAG,CACD,KAAK,EACL,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,SAAS,EAClD,kCAAkC,CACnC,CAAA;QACD,OAAO,GAAG,IAAI,CAAA;IAChB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,GAAG,WAAW,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAA;QACvC,IAAI,CAAC,IAAI,CAAC,CAAA;QACV,OAAO,GAAG,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,IAAI,IAAI,EAAE,CAAA;IAC1D,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,CAAA;AACzC,CAAC;AAED,0BAA0B;AAC1B,MAAM,UAAU,oBAAoB,CAClC,KAAiC,EACjC,IAA+B;IAE/B,OAAO,YAAY,CAAC,KAAK,EAAE,cAAc,EAAE,IAAI,CAAC,CAAA;AAClD,CAAC;AAED,mCAAmC;AACnC,SAAS,GAAG,CAAC,KAAqB,EAAE,OAAuC,EAAE,KAAa;IACxF,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;QACvC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,IAAI,SAAS,EAAE,KAAK,CAAC,CAAA;QACjD,OAAM;IACR,CAAC;IACD,OAAO,EAAE,CAAA;AACX,CAAC;AAED,SAAS,QAAQ,CAAC,GAAY;IAC5B,OAAO,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;AACzD,CAAC"}