@zhushanwen/pi-permission 1.3.4 → 1.4.1

package/src/ast/analyzer.ts

@@ -102,90 +102,100 @@ function parseRawString(node: Node): string | null {
 }
 
 /**
- * 从一个 command 节点提取 argv（对应 Rust parse_plain_command_from_node）。
- *
- * 5 分支处理 named children：
- *  1. command_name → 其第一个 named child 必须是 word，取文本
- *  2. word | number → 直接取文本
- *  3. string → parseDoubleQuotedString（含 expansion 则 null）
- *  4. raw_string → parseRawString
- *  5. concatenation（如 -g"*.py"）→ 递归拼接各 part（word/number/string/raw_string）
- *
- * 任一分支失败 → 返回 null（该 command 被跳过，但 clean 已在 DFS 阶段被置 false）。
+ * command_name 分支：其第一个 named child 必须是 word，取文本；否则 null。
  */
-function parsePlainCommand(cmd: Node): string[] | null {
-	if (cmd.type !== "command") {
+function parseCommandName(child: Node): string | null {
+	const wordNode = child.namedChild(0);
+	if (!wordNode || wordNode.type !== "word") {
 		return null;
 	}
+	return wordNode.text;
+}
 
-	const words: string[] = [];
-	for (const child of cmd.namedChildren) {
-		switch (child.type) {
-			case "command_name": {
-				const wordNode = child.namedChild(0);
-				if (!wordNode || wordNode.type !== "word") {
-					return null;
-				}
-				words.push(wordNode.text);
-				break;
-			}
+/**
+ * concatenation 分支（如 -g"*.py" 或 '/usr'"/"'local'/bin）：
+ * 递归拼接各 part（word/number/string/raw_string），空拼接结果 → null。
+ */
+function parseConcatenation(child: Node): string | null {
+	let concatenated = "";
+	for (const part of child.namedChildren) {
+		switch (part.type) {
 			case "word":
-			case "number": {
-				words.push(child.text);
+			case "number":
+				concatenated += part.text;
 				break;
-			}
 			case "string": {
-				const parsed = parseDoubleQuotedString(child);
-				if (parsed === null) {
-					return null;
-				}
-				words.push(parsed);
+				const parsed = parseDoubleQuotedString(part);
+				if (parsed === null) return null;
+				concatenated += parsed;
 				break;
 			}
 			case "raw_string": {
-				const parsed = parseRawString(child);
-				if (parsed === null) {
-					return null;
-				}
-				words.push(parsed);
-				break;
-			}
-			case "concatenation": {
-				// 拼接如 -g"*.py" 或 '/usr'"/"'local'/bin
-				let concatenated = "";
-				for (const part of child.namedChildren) {
-					switch (part.type) {
-						case "word":
-						case "number":
-							concatenated += part.text;
-							break;
-						case "string": {
-							const parsed = parseDoubleQuotedString(part);
-							if (parsed === null) return null;
-							concatenated += parsed;
-							break;
-						}
-						case "raw_string": {
-							const parsed = parseRawString(part);
-							if (parsed === null) return null;
-							concatenated += parsed;
-							break;
-						}
-						default:
-							return null;
-					}
-				}
-				if (concatenated.length === 0) {
-					return null;
-				}
-				words.push(concatenated);
+				const parsed = parseRawString(part);
+				if (parsed === null) return null;
+				concatenated += parsed;
 				break;
 			}
 			default:
-				// 未知 named child（不应发生 —— DFS 阶段已拒，但保险起见 fail-closed）
 				return null;
 		}
 	}
+	if (concatenated.length === 0) {
+		return null;
+	}
+	return concatenated;
+}
+
+/**
+ * 解析 command 的单个 named child 为一个 argv token（分发表）。
+ *
+ * 5 分支：
+ *  1. command_name → parseCommandName（第一个 named child 必须是 word）
+ *  2. word | number → 直接取文本
+ *  3. string → parseDoubleQuotedString（含 expansion 则 null）
+ *  4. raw_string → parseRawString
+ *  5. concatenation → parseConcatenation
+ *
+ * 未知 named child（不应发生 —— DFS 阶段已拒，但保险起见 fail-closed）→ null。
+ * 失败 → 调用方让整个 command 返回 null（该 command 被跳过，但 clean 已在 DFS 阶段被置 false）。
+ */
+function parseChildToken(child: Node): string | null {
+	switch (child.type) {
+		case "command_name":
+			return parseCommandName(child);
+		case "word":
+		case "number":
+			return child.text;
+		case "string":
+			return parseDoubleQuotedString(child);
+		case "raw_string":
+			return parseRawString(child);
+		case "concatenation":
+			return parseConcatenation(child);
+		default:
+			return null;
+	}
+}
+
+/**
+ * 从一个 command 节点提取 argv（对应 Rust parse_plain_command_from_node）。
+ *
+ * 主函数编排：类型守卫 → 逐 named child 分发（parseChildToken）→ 组装 words。
+ * 任一 child 解析失败 → 返回 null。
+ */
+function parsePlainCommand(cmd: Node): string[] | null {
+	if (cmd.type !== "command") {
+		return null;
+	}
+
+	const words: string[] = [];
+	for (const child of cmd.namedChildren) {
+		const token = parseChildToken(child);
+		if (token === null) {
+			return null;
+		}
+		words.push(token);
+	}
 	return words;
 }
 

package/src/classifier/__tests__/classifier.test.ts

@@ -60,7 +60,7 @@ function okResult(content: string): CallLLMResult {
 }
 
 function failResult(error: string, stopReason?: "error" | "aborted"): CallLLMResult {
-	return { ok: false, error, recoverable: true, ...(stopReason !== undefined ? { stopReason } : {}) };
+	return { ok: false, error, ...(stopReason !== undefined ? { stopReason } : {}) };
 }
 
 /** 构造「永不 resolve」的 mock callLLM（用于 timeout/abort 测试） */

package/src/classifier/classifier.ts

@@ -8,7 +8,7 @@
  * 关键设计：
  *   - 依赖注入 ClassifierDeps（callLLM 注入，便于测试 mock，不直接 import）
  *   - C1a 收口：LLM 调用走 llm-shared callLLM（内部完成凭证 getApiKeyAndHeaders +
- *     completeSimple + stopReason 归一化：error/aborted → {ok:false, recoverable:true,
+ *     completeSimple + stopReason 归一化：error/aborted → {ok:false,
  *     stopReason 透传}），不再走 production.ts 的 streamSimple
  *   - G3 修正（由 callLLM 承接）：completeSimple 的 EventStream.result() 只 resolve 不 reject，
  *     error/aborted 也 resolve（带 stopReason）。callLLM 已把 error/aborted 归一为 ok:false +
@@ -90,7 +90,7 @@ function raceResultWithDeadline(
 			(result) => ({ kind: "ok" as const, result }),
 			(error) => ({
 				kind: "ok" as const,
-				result: { ok: false as const, error: error instanceof Error ? error.message : String(error), recoverable: true },
+				result: { ok: false as const, error: error instanceof Error ? error.message : String(error) },
 			}),
 		),
 	];
@@ -119,6 +119,78 @@ function raceResultWithDeadline(
 	});
 }
 
+// ──────────────────────── classifyRisk 分步 helper ────────────────────────
+
+/**
+ * 构造 CallLLMOptions（timeout 秒→毫秒的换算由调用方完成后传入；
+ * signal 透传 callLLM → completeSimple，abort 时 reject 或 resolve(aborted)
+ * 都会归一为 ok:false）。
+ */
+function buildCallOptions(
+	model: Model<Api>,
+	ctx: ToolInvocationContext,
+	config: ClassifierConfig,
+	timeoutMs: number | undefined,
+	signal: AbortSignal | undefined,
+): CallLLMOptions {
+	return {
+		model,
+		systemPrompt: CLASSIFIER_SYSTEM_PROMPT,
+		messages: buildMessages(ctx),
+		...(timeoutMs !== undefined ? { timeoutMs } : {}),
+		...(signal !== undefined ? { signal } : {}),
+		// thinkingLevel 直接透传（含 "off"）；llm-shared 内部会把 "off" 映射为不传 reasoning（provider 默认）
+		reasoning: config.thinkingLevel,
+	};
+}
+
+/** 记录 LLM 调用失败日志（G3 语义：abort 与 error 分开记，即使行为上两者都 fallback 无差别）。 */
+function logFailedCall(
+	result: Extract<CallLLMResult, { ok: false }>,
+	onLog?: (msg: string) => void,
+): void {
+	if (result.stopReason === "aborted") {
+		onLog?.(`[pi-permission] classifier: LLM call aborted (stopReason=aborted), returning fallback`);
+	} else {
+		const stopDetail = result.stopReason !== undefined ? ` (stopReason=${result.stopReason})` : "";
+		onLog?.(`[pi-permission] classifier: LLM call failed: ${result.error}${stopDetail}, returning fallback`);
+	}
+}
+
+/** raceWithDeadline 已收窄的结果形态（timeout/aborted 无 result 字段）。 */
+type SettledRace =
+	| { kind: "ok"; result: CallLLMResult }
+	| { kind: "timeout" }
+	| { kind: "aborted" };
+
+/** settleRacedResult 的输出：fallback = fail-closed ask；ok = 继续解析文本。 */
+type SettledOutcome =
+	| { kind: "fallback"; result: ClassifierResult }
+	| { kind: "ok"; result: Extract<CallLLMResult, { ok: true }> };
+
+/**
+ * race 结果归一：timeout / aborted / ok:false → fail-closed fallback（保留日志区分），
+ * 成功原样透传（调用方收窄后取 content）。
+ */
+function settleRacedResult(settled: SettledRace, timeoutMs: number | undefined, onLog?: (msg: string) => void): SettledOutcome {
+	if (settled.kind === "timeout") {
+		onLog?.(`[pi-permission] classifier: timed out after ${timeoutMs}ms`);
+		return { kind: "fallback", result: { ...CLASSIFY_FALLBACK_RESULT } };
+	}
+	if (settled.kind === "aborted") {
+		onLog?.(`[pi-permission] classifier: aborted by signal`);
+		return { kind: "fallback", result: { ...CLASSIFY_FALLBACK_RESULT } };
+	}
+
+	// ok:false → fallback（fail-closed）。stopReason 独立透传字段保留日志区分
+	const result = settled.result;
+	if (!result.ok) {
+		logFailedCall(result, onLog);
+		return { kind: "fallback", result: { ...CLASSIFY_FALLBACK_RESULT } };
+	}
+	return { kind: "ok", result };
+}
+
 // ──────────────────────── createClassifier ────────────────────────
 
 /**
@@ -148,21 +220,12 @@ export function createClassifier(deps: ClassifierDeps): {
 		// 实证 classifier 真实用到 OAuth/配置的模型（而非 fail-closed 降级）。
 		onLog?.(`[pi-permission] classifier: using model ${model.id}`);
 
-		// 2. 构造 CallLLMOptions（timeout 秒→毫秒；signal 透传 callLLM → completeSimple，
-		//    abort 时 reject 或 resolve(aborted) 都会归一为 ok:false）
+		// 2. 构造 CallLLMOptions
 		const timeoutMs = config.timeout > 0 ? config.timeout * MILLIS_PER_SECOND : undefined;
-		const callPromise = callLLM({
-			model,
-			systemPrompt: CLASSIFIER_SYSTEM_PROMPT,
-			messages: buildMessages(ctx),
-			...(timeoutMs !== undefined ? { timeoutMs } : {}),
-			...(signal !== undefined ? { signal } : {}),
-			// thinkingLevel 直接透传（含 "off"）；llm-shared 内部会把 "off" 映射为不传 reasoning（provider 默认）
-			reasoning: config.thinkingLevel,
-		});
+		const callPromise = callLLM(buildCallOptions(model, ctx, config, timeoutMs, signal));
 
 		// 3. 等待 callLLM + 外层超时/中止兜底（provider 不支持 timeoutMs/signal 时防永挂）
-		let settled: { kind: "ok"; result: CallLLMResult } | { kind: "timeout" } | { kind: "aborted" };
+		let settled: SettledRace;
 		try {
 			settled = await raceResultWithDeadline(callPromise, timeoutMs, signal);
 		} catch (error) {
@@ -171,30 +234,14 @@ export function createClassifier(deps: ClassifierDeps): {
 			return { ...CLASSIFY_FALLBACK_RESULT };
 		}
 
-		if (settled.kind === "timeout") {
-			onLog?.(`[pi-permission] classifier: timed out after ${timeoutMs}ms`);
-			return { ...CLASSIFY_FALLBACK_RESULT };
-		}
-		if (settled.kind === "aborted") {
-			onLog?.(`[pi-permission] classifier: aborted by signal`);
-			return { ...CLASSIFY_FALLBACK_RESULT };
-		}
-
-		// 4. ok:false → fallback（fail-closed）。stopReason 独立透传字段保留日志区分
-		//    （G3 语义：abort 与 error 分开记，即使行为上两者都 fallback 无差别）。
-		const result = settled.result;
-		if (!result.ok) {
-			if (result.stopReason === "aborted") {
-				onLog?.(`[pi-permission] classifier: LLM call aborted (stopReason=aborted), returning fallback`);
-			} else {
-				const stopDetail = result.stopReason !== undefined ? ` (stopReason=${result.stopReason})` : "";
-				onLog?.(`[pi-permission] classifier: LLM call failed: ${result.error}${stopDetail}, returning fallback`);
-			}
-			return { ...CLASSIFY_FALLBACK_RESULT };
+		// 4. timeout / aborted / ok:false → fail-closed fallback；成功 → 收窄透传
+		const settledOutcome = settleRacedResult(settled, timeoutMs, onLog);
+		if (settledOutcome.kind === "fallback") {
+			return settledOutcome.result;
 		}
 
 		// 5. 提取文本 → 三层容错解析
-		return parseClassifierResponse(result.content);
+		return parseClassifierResponse(settledOutcome.result.content);
 	}
 
 	return { classifyRisk };

package/src/classifier/model-resolver.ts

@@ -64,14 +64,11 @@ export interface ResolvedModelEntry {
  * 「无可选模型」提示）。
  *
  * @param ctx 含 modelRegistry 的上下文（model 列表 + auth 判定）
- * @param onWarning 保留参数位（契约兼容；E2 后无文件读取，无警告源）
  * @returns Map<providerName, ResolvedModelEntry[]>
  */
 export function listAvailableModels(
 	ctx: ListAvailableModelsCtx,
-	onWarning?: (msg: string) => void,
 ): Map<string, ResolvedModelEntry[]> {
-	void onWarning; // 无警告源（不再读文件），保留参数位兼容调用方
 	const entries: ResolvedModelEntry[] = [];
 	for (const m of ctx.modelRegistry.getAll()) {
 		if (!ctx.modelRegistry.hasConfiguredAuth(m)) continue;

package/src/index.ts

@@ -18,6 +18,7 @@ import {
 	type ExtensionContext,
 	getAgentDir,
 } from "@earendil-works/pi-coding-agent";
+import { oncePerProcess } from "@zhushanwen/pi-ext-guards";
 import { getLogger } from "@zhushanwen/pi-extension-logger";
 import { migrateLegacyConfig } from "@zhushanwen/pi-llm-shared";
 
@@ -44,8 +45,9 @@ import { paletteFromTheme, type PermissionPalette } from "./statusline-palette.j
 // Added in v1.0.0. Remove after v2.0.0 (one major past).
 // session_start 首次触发时把旧路径 permission-config.json 迁到 config/permission-ext-config.json。
 // 幂等、best-effort（migrateLegacyConfig 见 @zhushanwen/pi-llm-shared）。
-// 模块级 once flag 防同进程重复触发；agentDir 由 getAgentDir() 推导（尊重 PI_CODING_AGENT_DIR）。
-let configMigrationChecked = false;
+// 写 agentDir 全局配置文件属跨 session 副作用——oncePerProcess 守卫防 factory 二调/handler
+// 累积双跑（u-audit-fix 收编原内联 once flag；key 带包名前缀避撞，见 pi-ext-guards JSDoc）。
+// agentDir 由 getAgentDir() 推导（尊重 PI_CODING_AGENT_DIR）。
 
 /** 默认配置 warning 回调（loadAndWatchConfig 共用，透传 logger.warn）。 */
 const defaultConfigWarn = (msg: string): void => logger.warn(msg);
@@ -86,10 +88,8 @@ const _UI_OPTIONS_PARAM_INDEX = 2;
  */
 export default function permissionExtension(pi: ExtensionAPI): void {
 	// W7：注入 listAvailableModels 真实实现（model-picker.ts 默认返回空 Map）。
-	// E2 签名：(ctx, onWarning?) → ctx.modelRegistry.getAll() + hasConfiguredAuth 过滤；warning 透传到 console.warn。
-	setDefaultListAvailableModels((ctx, onWarning) =>
-		listAvailableModels(ctx, onWarning ?? defaultConfigWarn),
-	);
+	// E2 签名：(ctx) → ctx.modelRegistry.getAll() + hasConfiguredAuth 过滤。
+	setDefaultListAvailableModels((ctx) => listAvailableModels(ctx));
 
 	// ──────────────────────── 配置读取（读时刷新，回归 llm-shared 框架） ────────────────────────
 	// 不持有跨调用缓存：每次需要配置直接 loadAndWatchConfig()，llm-shared 内部 mtime+size 去重，
@@ -100,10 +100,9 @@ export default function permissionExtension(pi: ExtensionAPI): void {
 	// ──────────────────────── session_start：迁移旧路径配置 ────────────────────────
 	pi.on("session_start", (_event: unknown, ctx: ExtensionContext) => {
 		// [MIGRATION] Added in v1.0.0. Remove after v2.0.0.
-		if (!configMigrationChecked) {
-			configMigrationChecked = true;
+		oncePerProcess("permission:migrate-legacy-config", () => {
 			migrateLegacyConfig(getAgentDir(), "permission-config.json", "config/permission-ext-config.json");
-		}
+		});
 		// 无需手动刷新：去闭包后每次 loadAndWatchConfig 读时刷新；迁移改写文件后下次读取自动生效。
 		// 注册 footer line renderer（consumer 端握手，pi-statusline 是 canonical owner）。
 		// renderer 无状态：render 时用 statusline 传入的 theme + loadAndWatchConfig 读最新 config。
@@ -195,7 +194,7 @@ export default function permissionExtension(pi: ExtensionAPI): void {
 					},
 					config,
 					{
-						listModels: (pickerCtx) => listAvailableModels(pickerCtx, (m) => logger.warn(m)),
+						listModels: (pickerCtx) => listAvailableModels(pickerCtx),
 						save: (newConfig) => {
 							const r = saveConfig(newConfig);
 							if (r.success) requestFooterRender();

package/src/model-picker.ts

@@ -23,6 +23,10 @@ import { type Component, Container, type SelectItem, SelectList, type SelectList
 import type { Api, Model } from "@earendil-works/pi-ai";
 
 import type { ResolvedModelEntry } from "./classifier/model-resolver.js";
+import { DEFAULT_SELECT_THEME } from "./select-theme.js";
+
+// 保持既有 public 名（model-picker.test.ts MPT7 直接消费此路径）。
+export { DEFAULT_SELECT_THEME };
 
 // ──────────────────────── 类型 ────────────────────────
 
@@ -31,22 +35,6 @@ export type SelectionResult =
 	| { kind: "auto" }
 	| { kind: "specific"; provider: string; modelId: string };
 
-// ──────────────────────── DEFAULT_SELECT_THEME（G2 修正） ────────────────────────
-
-/**
- * SelectList 默认主题。
- *
- * G2/WR2 修正：selectedPrefix 实现为 `(t) => '▶ ' + t`（非 identity），
- * 选中行有视觉区分（▶ 前缀）。
- */
-export const DEFAULT_SELECT_THEME: SelectListTheme = {
-	selectedPrefix: (t: string): string => "\u25B6 " + t,
-	selectedText: (t: string): string => t,
-	description: (t: string): string => t,
-	scrollInfo: (t: string): string => t,
-	noMatch: (t: string): string => t,
-};
-
 /** provider stage 的 'auto' 项 value（用特殊标记区分真实 provider 名）。 */
 const AUTO_VALUE = "__auto__";
 
@@ -189,7 +177,7 @@ export class ProviderModelSelectorComponent extends Container {
 	/** provider stage items：'Auto' + 各 provider（含 model 数量）。 */
 	private buildProviderItems(): SelectItem[] {
 		const items: SelectItem[] = [
-			{ value: AUTO_VALUE, label: "Auto", description: "auto-select cheapest available model" },
+			{ value: AUTO_VALUE, label: "Auto", description: "select first available model" },
 		];
 		for (const provider of this.providers) {
 			const models = this.modelsByProvider.get(provider) ?? [];
@@ -297,7 +285,6 @@ export interface ModelPickerContext {
 /** listAvailableModels 注入签名（E2 改走 ctx.modelRegistry，签名带 ctx；便于测试 mock）。 */
 export type ListAvailableModelsFn = (
 	ctx: ModelPickerContext,
-	onWarning?: (msg: string) => void,
 ) => Map<string, ResolvedModelEntry[]>;
 
 /**