@trim21/personal-pi-extensions 0.0.421 → 0.0.423

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trim21/personal-pi-extensions",
-  "version": "0.0.421",
+  "version": "0.0.423",
   "type": "module",
   "description": "Custom pi coding-agent extensions: bwrap sandbox, workspace guard, opencode edit, and more",
   "keywords": [

package/src/bwrap/approval-rules.ts

@@ -1,13 +1,13 @@
 /**
- * approval-rules —— bash 命令审核组件（对齐 opencode 的权限方法）：
- * 用 tree-sitter 解析命令（含嵌套 `$(...)`），按 BashArity 生成命令模式
- * （`git checkout main` → `git checkout *`），再用通配匹配对 allow/deny
- * 规则求值。接入 bwrap 的 `dangerouslyDisableSandbox` 审批：命中规则自动
+ * approval-rules —— bash 命令匹配引擎（对齐 opencode 的权限方法）：
+ * 用 tree-sitter 解析命令（含嵌套 `$(...)`），对每条命令的原文做通配
+ * 匹配，对 allow/deny 规则求值。规则匹配不经过任何命令归一——BashArity
+ * 归一模式只用于 "allow forever" 的建议规则（见 approval-suggest.ts）。
+ * 接入 bwrap 的 `dangerouslyDisableSandbox` 审批：命中规则自动
  * 放行/拒绝，未命中才弹审批对话框。文件输出重定向（`>` / `>>` / `&>`
  * 等）不会因命令规则自动放行，避免 `echo *` 把 `echo '' > file` 带过。
  *
  * 参考实现：
- * - opencode packages/opencode/src/permission/arity.ts（BashArity 表）
  * - opencode packages/core/src/util/wildcard.ts（通配匹配）
  * - opencode packages/opencode/src/tool/shell.ts（tree-sitter 命令提取）
  */
@@ -173,34 +173,6 @@ export async function parseBashCommands(command: string): Promise<ParsedBash> {
   }
 }
 
-// ── BashArity：命令前缀 → token 数（参考 opencode arity.ts）─────────────────
-
-/**
- * 命令前缀 → 定义该命令的 token 数。`git checkout main` → `git` 的 arity 2，
- * 权限模式取前 2 个 token + `*`（`git checkout *`），避免具体参数进规则。
- * 表来自 opencode packages/opencode/src/permission/arity.ts（Apache-2.0），
- * 数据存放在 arity.json（所有 key 带引号）。
- */
-const ARITY = JSON.parse(readFileSync(new URL("arity.json", import.meta.url), "utf8")) as Record<
-  string,
-  number
->;
-
-/**
- * 生成命令的权限模式：BashArity 前缀 + `*`。
- * `git checkout main` → `git checkout *`；未收录的命令 → 命令名 + `*`。
- */
-export function commandPattern(command: BashCommand): string {
-  const tokens = [command.name, ...command.args];
-  for (let len = tokens.length; len > 0; len--) {
-    const prefix = tokens.slice(0, len).join(" ");
-    const arity = ARITY[prefix];
-    if (arity !== undefined) return [...tokens.slice(0, arity), "*"].join(" ");
-  }
-  if (tokens.length === 0) return "*";
-  return [tokens[0], "*"].join(" ");
-}
-
 // ── 通配匹配（参考 opencode wildcard.ts）────────────────────────────────────
 
 /**
@@ -218,31 +190,15 @@ export function matchRule(input: string, pattern: string): boolean {
 
 // ── 规则求值 ────────────────────────────────────────────────────────────────
 
-/**
- * 命令（含所有嵌套命令）的权限模式列表（命令替换里的命令也展开）。
- * 供规则求值与"allow forever"写规则复用。
- */
-function patternsFromCommands(commands: BashCommand[]): string[] {
-  const flat: string[] = [];
-  const visit = (cmd: BashCommand) => {
-    flat.push(commandPattern(cmd));
-    for (const nested of cmd.nested) visit(nested);
-  };
-  for (const cmd of commands) visit(cmd);
-  return flat;
-}
-
-export async function commandPatternsFor(command: string): Promise<string[]> {
-  const parsed = await parseBashCommands(command);
-  return patternsFromCommands(parsed.commands);
-}
-
 /**
  * 对命令（含所有嵌套命令）求值：
  * - deny 优先：任一命令命中 deny 规则即整体拒绝
  * - 文件输出重定向不自动放行：即使命令规则全匹配，也返回 undefined 交人审
  * - allow 需全量：所有命令都命中 allow 规则才整体放行，否则返回
  *   undefined（有命令未命中规则，交给人审），避免未允许的命令被同链放行带过。
+ * - 匹配输入是命令原文（tree-sitter command 节点 text，含嵌套逐条展开），
+ *   对齐 opencode shell.ts 的 patterns；通配规则按字面写，`--` 与普通
+ *   token 无区别。
  * 规则内后写优先（findLast，对齐 opencode PermissionV2）。
  */
 export async function evaluateBashApproval(
@@ -250,14 +206,19 @@ export async function evaluateBashApproval(
   rules: readonly ApprovalRule[],
 ): Promise<ApprovalAction | undefined> {
   const parsed = await parseBashCommands(command);
-  const patterns = patternsFromCommands(parsed.commands);
-  if (patterns.length === 0) return;
+  const raws: string[] = [];
+  const visit = (cmd: BashCommand) => {
+    raws.push(cmd.raw);
+    for (const nested of cmd.nested) visit(nested);
+  };
+  for (const cmd of parsed.commands) visit(cmd);
+  if (raws.length === 0) return;
   let allowed = 0;
-  for (const pattern of patterns) {
-    const rule = rules.findLast((r) => matchRule(pattern, r.pattern));
+  for (const raw of raws) {
+    const rule = rules.findLast((r) => matchRule(raw, r.pattern));
     if (rule?.action === "deny") return "deny";
     if (rule?.action === "allow") allowed++;
   }
   if (parsed.hasFileOutputRedirect) return;
-  return allowed === patterns.length ? "allow" : undefined;
+  return allowed === raws.length ? "allow" : undefined;
 }

package/src/bwrap/approval-suggest.ts

@@ -0,0 +1,53 @@
+/**
+ * approval-suggest —— "allow forever" 的建议规则生成：
+ * 用 BashArity 表把命令归一成模式（`git checkout main` → `git checkout *`），
+ * 供用户选择"永久允许"时写入配置。归一模式不参与规则判定——匹配引擎
+ * （approval-rules）按命令原文做通配匹配。
+ *
+ * 参考实现：opencode packages/opencode/src/permission/arity.ts（Apache-2.0）
+ */
+import { readFileSync } from "node:fs";
+
+import { type BashCommand, parseBashCommands } from "./approval-rules.js";
+
+/**
+ * 命令前缀 → 定义该命令的 token 数。`git checkout main` → `git` 的 arity 2，
+ * 权限模式取前 2 个 token + `*`（`git checkout *`），避免具体参数进规则。
+ * 表来自 opencode packages/opencode/src/permission/arity.ts（Apache-2.0），
+ * 数据存放在 arity.json（所有 key 带引号）。
+ */
+const ARITY = JSON.parse(readFileSync(new URL("arity.json", import.meta.url), "utf8")) as Record<
+  string,
+  number
+>;
+
+/**
+ * 生成命令的 BashArity 建议模式：BashArity 前缀 + `*`。
+ * `git checkout main` → `git checkout *`；未收录的命令 → 命令名 + `*`。
+ */
+export function commandPattern(command: BashCommand): string {
+  const tokens = [command.name, ...command.args];
+  for (let len = tokens.length; len > 0; len--) {
+    const prefix = tokens.slice(0, len).join(" ");
+    const arity = ARITY[prefix];
+    if (arity !== undefined) return [...tokens.slice(0, arity), "*"].join(" ");
+  }
+  if (tokens.length === 0) return "*";
+  return [tokens[0], "*"].join(" ");
+}
+
+/** 命令（含所有嵌套命令）的 BashArity 建议模式列表（命令替换里的命令也展开）。 */
+function patternsFromCommands(commands: BashCommand[]): string[] {
+  const flat: string[] = [];
+  const visit = (cmd: BashCommand) => {
+    flat.push(commandPattern(cmd));
+    for (const nested of cmd.nested) visit(nested);
+  };
+  for (const cmd of commands) visit(cmd);
+  return flat;
+}
+
+export async function commandPatternsFor(command: string): Promise<string[]> {
+  const parsed = await parseBashCommands(command);
+  return patternsFromCommands(parsed.commands);
+}

package/src/bwrap/runtime.ts

@@ -23,12 +23,8 @@ import {
   selectCheckboxActions,
   selectWithOptionalInput,
 } from "../lib/ui.js";
-import {
-  type ApprovalRule,
-  commandPatternsFor,
-  evaluateBashApproval,
-  matchRule,
-} from "./approval-rules.js";
+import { type ApprovalRule, evaluateBashApproval, matchRule } from "./approval-rules.js";
+import { commandPatternsFor } from "./approval-suggest.js";
 import {
   type BwrapMode,
   createBwrapBashOperations,

package/src/lib/lsp/lsp.ts

@@ -175,6 +175,8 @@ interface LspState {
   broken: Set<string>;
   spawning: Map<string, Promise<LspClient | undefined>>;
   closing: boolean;
+  /** /lsp-stop 置 true：所有工具调用不再 spawn 服务器，直到 start/reload。 */
+  disabled: boolean;
   /** root+serverID → 服务器状态，用于 footer status 显示。 */
   servers: Map<string, { serverID: string; root: string; state: "running" | "broken" }>;
 }
@@ -214,6 +216,17 @@ export interface LspService {
     options?: LspRequestOptions,
   ): Promise<{ text: string; errorCount: number; warningCount: number }>;
   shutdownAll(): Promise<void>;
+  /** 停止全部服务器并禁用 LSP：之后工具调用不再 spawn，直到 start/reload。 */
+  stop(): Promise<void>;
+  /** 解除禁用并清空 broken 缓存；服务器在下次工具调用时惰性启动。 */
+  start(): void;
+  /**
+   * 重启指定服务器：关闭其全部 client、清除对应 broken 记录并解除禁用；
+   * 其余服务器不受影响。配置在下次工具调用时重新读取。
+   */
+  reload(serverID: string): Promise<void>;
+  /** 已知服务器 id（running 或 broken 的去重集合），供命令补全与提示。 */
+  serverIDs(): string[];
   /** 注入 status 渲染回调；传入 undefined 表示不再渲染。 */
   attachStatus(render: StatusRenderer | undefined): void;
   /** 用当前服务器状态主动刷新一次 status（agent start/end 等生命周期边界）。 */
@@ -247,6 +260,7 @@ export function createLspService(
     broken: new Set(),
     spawning: new Map(),
     closing: false,
+    disabled: false,
     servers: new Map(),
   };
 
@@ -255,6 +269,10 @@ export function createLspService(
   /** 汇总当前所有 LSP server 状态并渲染到 footer status。 */
   function updateStatusText(): void {
     if (!renderStatus) return;
+    if (state.disabled) {
+      renderStatus("lsp: disabled");
+      return;
+    }
     if (state.servers.size === 0) {
       renderStatus(undefined);
       return;
@@ -275,7 +293,7 @@ export function createLspService(
     cwd: string,
     notify?: ExtensionUIContext["notify"],
   ): Promise<LspClient[]> {
-    if (state.closing) return [];
+    if (state.closing || state.disabled) return [];
     if (!containsPath(file, cwd)) return [];
     const config = await loadLspConfig(cwd, globalConfigPath);
     const timeout = timeoutOptions(config);
@@ -326,7 +344,7 @@ export function createLspService(
             diagnosticsDocumentWaitTimeoutMs:
               adapter.diagnosticsWaitMs ?? timeout.diagnosticsDocumentWaitTimeoutMs,
           });
-          if (state.closing) {
+          if (state.closing || state.disabled) {
             await client.shutdown();
             return;
           }
@@ -426,9 +444,8 @@ export function createLspService(
     return { text: report(normalized, issues), errorCount, warningCount };
   }
 
-  /** 终止全部服务器进程（session_shutdown 时调用）。 */
-  async function shutdownAll(): Promise<void> {
-    if (state.closing) return;
+  /** 关闭全部 client 并清空缓存；closing 置 true 让 in-flight spawn 自行退出。 */
+  async function closeAll(): Promise<void> {
     state.closing = true;
     await Promise.all(state.clients.map((client) => client.shutdown())).catch(() => {
       // 个别进程退出失败不阻止清理流程
@@ -439,11 +456,55 @@ export function createLspService(
     updateStatusText();
   }
 
+  /** 终止全部服务器进程（session_shutdown 时调用，终态）。 */
+  async function shutdownAll(): Promise<void> {
+    await closeAll();
+  }
+
+  async function stop(): Promise<void> {
+    await closeAll();
+    state.closing = false;
+    state.disabled = true;
+    updateStatusText();
+  }
+
+  function start(): void {
+    state.closing = false;
+    state.disabled = false;
+    state.broken.clear();
+    updateStatusText();
+  }
+
+  async function reload(serverID: string): Promise<void> {
+    state.closing = true;
+    const targets = state.clients.filter((client) => client.serverID === serverID);
+    await Promise.all(targets.map((client) => client.shutdown())).catch(() => {
+      // 个别进程退出失败不阻止清理流程
+    });
+    state.clients = state.clients.filter((client) => client.serverID !== serverID);
+    for (const [key, server] of state.servers) {
+      if (server.serverID !== serverID) continue;
+      state.broken.delete(key);
+      state.servers.delete(key);
+    }
+    state.closing = false;
+    state.disabled = false;
+    updateStatusText();
+  }
+
+  function serverIDs(): string[] {
+    return [...new Set([...state.servers.values()].map((server) => server.serverID))];
+  }
+
   return {
     touchFile,
     diagnostics,
     lspDiagnosticsForFile,
     shutdownAll,
+    stop,
+    start,
+    reload,
+    serverIDs,
     attachStatus,
     refreshStatus: updateStatusText,
   };
@@ -476,5 +537,49 @@ export function registerLsp(pi: ExtensionAPI, options?: LspServiceOptions): LspS
   // spawn（首次工具调用），start/end 时保证 footer 反映当前实际状态。
   pi.on?.("agent_start", () => service.refreshStatus());
   pi.on?.("agent_end", () => service.refreshStatus());
+
+  pi.registerCommand?.("lsp-stop", {
+    description: "Stop all LSP servers and disable LSP until /lsp-start or /lsp-reload",
+    handler: async (_args, ctx) => {
+      await service.stop();
+      ctx.ui.notify?.("LSP disabled: all servers stopped", "info");
+    },
+  });
+
+  pi.registerCommand?.("lsp-start", {
+    description: "Re-enable LSP; servers start on the next tool call",
+    handler: (_args, ctx) => {
+      service.start();
+      ctx.ui.notify?.("LSP enabled: servers will start on the next tool call", "info");
+      return Promise.resolve();
+    },
+  });
+
+  pi.registerCommand?.("lsp-reload", {
+    description: "Restart a specific LSP server: /lsp-reload <server-id>",
+    getArgumentCompletions: (prefix) =>
+      service
+        .serverIDs()
+        .toSorted()
+        .filter((id) => id.startsWith(prefix))
+        .map((id) => ({ value: id, label: id })),
+    handler: async (args, ctx) => {
+      const serverID = args.trim();
+      const known = service.serverIDs().toSorted();
+      if (!serverID) {
+        ctx.ui.notify?.(
+          `usage: /lsp-reload <server-id>${known.length > 0 ? ` (known: ${known.join(", ")})` : ""}`,
+          "warning",
+        );
+        return;
+      }
+      await service.reload(serverID);
+      ctx.ui.notify?.(
+        `LSP server "${serverID}" reloaded: will restart on the next tool call`,
+        "info",
+      );
+    },
+  });
+
   return service;
 }