@hank-warren/pi-plan-mode 0.1.0 → 1.0.0

package/src/tool-selection.ts

@@ -1,98 +0,0 @@
-import type { ToolInfo } from "@earendil-works/pi-coding-agent";
-import { PLAN_MODE_COMPLETE_TOOL_NAME } from "./completion-tool.js";
-import { PLAN_MODE_QUESTION_TOOL_NAME } from "./question-tool.js";
-import { withRequiredPlanModeTools } from "./required-tools.js";
-import {
-	canSelectToolInPlanMode,
-	classifyPlanModeTool,
-	isBuiltinTool,
-	SAFE_BUILTIN_PLAN_TOOLS,
-} from "./tool-policy.js";
-
-export function toolNameFromLegacyKey(key: string, tools: ToolInfo[]) {
-	const directName = tools.find((tool) => tool.name === key)?.name;
-	if (directName) return directName;
-	const [name] = key.split("\u001f");
-	return tools.find((tool) => tool.name === name) ? name : undefined;
-}
-
-export function compareTools(left: ToolInfo, right: ToolInfo) {
-	const leftBuiltin = isBuiltinTool(left);
-	const rightBuiltin = isBuiltinTool(right);
-	if (leftBuiltin !== rightBuiltin) return leftBuiltin ? -1 : 1;
-	return left.name.localeCompare(right.name);
-}
-
-export function toolPolicyLabel(tool: ToolInfo) {
-	const policy = classifyPlanModeTool(tool);
-	if (policy === "read-only") return "built-in read-only";
-	if (policy === "limited") return "built-in limited";
-	if (policy === "blocked") return "built-in blocked";
-	return `user opt-in: ${toolSourceLabel(tool)}`;
-}
-
-function toolSourceLabel(tool: ToolInfo) {
-	const sourceInfo = tool.sourceInfo;
-	const source = `${sourceInfo.scope}/${sourceInfo.source}`;
-	return sourceInfo.path ? `${source} ${sourceInfo.path}` : source;
-}
-
-export function unique(values: string[]) {
-	return Array.from(new Set(values));
-}
-
-export function filterAvailableSelectedToolNames(names: string[], tools: ToolInfo[]) {
-	const availableNames = new Set(tools.filter(canSelectToolInPlanMode).map((tool) => tool.name));
-	return unique(names.filter((name) => availableNames.has(name)));
-}
-
-export function defaultPlanModeToolNames(tools: ToolInfo[], configuredNames: string[] | undefined) {
-	if (configuredNames !== undefined) {
-		return filterAvailableSelectedToolNames(configuredNames, tools);
-	}
-	return tools
-		.filter((tool) => isBuiltinTool(tool) && SAFE_BUILTIN_PLAN_TOOLS.has(tool.name))
-		.map((tool) => tool.name);
-}
-
-interface PlanModeToolSelectionSnapshot {
-	selectedToolNames?: string[];
-	selectedToolKeys?: string[];
-	defaultPlanTools?: string[];
-}
-
-export function snapshotPlanModeSelectedNames(
-	tools: ToolInfo[],
-	selection: PlanModeToolSelectionSnapshot,
-) {
-	const selectedToolNames =
-		selection.selectedToolNames ??
-		selection.selectedToolKeys
-			?.map((key) => toolNameFromLegacyKey(key, tools))
-			.filter((name): name is string => name !== undefined);
-	return new Set(
-		selectedToolNames === undefined
-			? defaultPlanModeToolNames(tools, selection.defaultPlanTools)
-			: filterAvailableSelectedToolNames(selectedToolNames, tools),
-	);
-}
-
-export function snapshotPlanModeToolNames(
-	tools: ToolInfo[],
-	selectedNames: ReadonlySet<string>,
-	selection: PlanModeToolSelectionSnapshot,
-) {
-	if (
-		tools.length === 0 &&
-		selection.selectedToolNames === undefined &&
-		selection.selectedToolKeys === undefined &&
-		selection.defaultPlanTools === undefined
-	) {
-		return ["read", "bash", PLAN_MODE_QUESTION_TOOL_NAME, PLAN_MODE_COMPLETE_TOOL_NAME];
-	}
-	return withRequiredPlanModeTools(
-		tools
-			.filter((tool) => selectedNames.has(tool.name) && canSelectToolInPlanMode(tool))
-			.map((tool) => tool.name),
-	);
-}