@bastani/atomic 0.7.17-1 → 0.8.0-0

package/dist/builtin/subagents/src/slash/slash-commands.ts

@@ -0,0 +1,528 @@
+import { randomUUID } from "node:crypto";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { Key, matchesKey } from "@earendil-works/pi-tui";
+import { discoverAgents, discoverAgentsAll, type ChainConfig } from "../agents/agents.ts";
+import type { SubagentParamsLike } from "../runs/foreground/subagent-executor.ts";
+import { isParallelStep, type ChainStep } from "../shared/settings.ts";
+import type { SlashSubagentResponse, SlashSubagentUpdate } from "./slash-bridge.ts";
+import {
+	applySlashUpdate,
+	buildSlashInitialResult,
+	failSlashResult,
+	finalizeSlashResult,
+} from "./slash-live-state.ts";
+import {
+	SLASH_RESULT_TYPE,
+	SLASH_SUBAGENT_CANCEL_EVENT,
+	SLASH_SUBAGENT_REQUEST_EVENT,
+	SLASH_SUBAGENT_RESPONSE_EVENT,
+	SLASH_SUBAGENT_STARTED_EVENT,
+	SLASH_SUBAGENT_UPDATE_EVENT,
+	type SingleResult,
+	type SubagentState,
+} from "../shared/types.ts";
+
+interface InlineConfig {
+	output?: string | false;
+	outputMode?: "inline" | "file-only";
+	reads?: string[] | false;
+	model?: string;
+	skill?: string[] | false;
+	progress?: boolean;
+}
+
+const parseInlineConfig = (raw: string): InlineConfig => {
+	const config: InlineConfig = {};
+	for (const part of raw.split(",")) {
+		const trimmed = part.trim();
+		if (!trimmed) continue;
+		const eq = trimmed.indexOf("=");
+		if (eq === -1) {
+			if (trimmed === "progress") config.progress = true;
+			continue;
+		}
+		const key = trimmed.slice(0, eq).trim();
+		const val = trimmed.slice(eq + 1).trim();
+		switch (key) {
+			case "output": config.output = val === "false" ? false : val; break;
+			case "outputMode": if (val === "inline" || val === "file-only") config.outputMode = val; break;
+			case "reads": config.reads = val === "false" ? false : val.split("+").filter(Boolean); break;
+			case "model": config.model = val || undefined; break;
+			case "skill": case "skills": config.skill = val === "false" ? false : val.split("+").filter(Boolean); break;
+			case "progress": config.progress = val !== "false"; break;
+		}
+	}
+	return config;
+};
+
+const parseAgentToken = (token: string): { name: string; config: InlineConfig } => {
+	const bracket = token.indexOf("[");
+	if (bracket === -1) return { name: token, config: {} };
+	const end = token.lastIndexOf("]");
+	return { name: token.slice(0, bracket), config: parseInlineConfig(token.slice(bracket + 1, end !== -1 ? end : undefined)) };
+};
+
+const extractExecutionFlags = (rawArgs: string): { args: string; bg: boolean; fork: boolean } => {
+	let args = rawArgs.trim();
+	let bg = false;
+	let fork = false;
+
+	while (true) {
+		if (args.endsWith(" --bg") || args === "--bg") {
+			bg = true;
+			args = args === "--bg" ? "" : args.slice(0, -5).trim();
+			continue;
+		}
+		if (args.endsWith(" --fork") || args === "--fork") {
+			fork = true;
+			args = args === "--fork" ? "" : args.slice(0, -7).trim();
+			continue;
+		}
+		break;
+	}
+
+	return { args, bg, fork };
+};
+
+const makeAgentCompletions = (state: SubagentState, multiAgent: boolean) => (prefix: string) => {
+	if (!state.baseCwd) return null;
+	const agents = discoverAgents(state.baseCwd, "both").agents;
+	if (!multiAgent) {
+		if (prefix.includes(" ")) return null;
+		return agents.filter((a) => a.name.startsWith(prefix)).map((a) => ({ value: a.name, label: a.name }));
+	}
+
+	const lastArrow = prefix.lastIndexOf(" -> ");
+	const segment = lastArrow !== -1 ? prefix.slice(lastArrow + 4) : prefix;
+	if (segment.includes(" -- ") || segment.includes('"') || segment.includes("'")) return null;
+
+	const lastWord = (prefix.match(/(\S*)$/) || ["", ""])[1];
+	const beforeLastWord = prefix.slice(0, prefix.length - lastWord.length);
+
+	if (lastWord === "->") {
+		return agents.map((a) => ({ value: `${prefix} ${a.name}`, label: a.name }));
+	}
+
+	return agents.filter((a) => a.name.startsWith(lastWord)).map((a) => ({ value: `${beforeLastWord}${a.name}`, label: a.name }));
+};
+
+const discoverSavedChains = (cwd: string): ChainConfig[] => {
+	const chainsByName = new Map<string, ChainConfig>();
+	for (const chain of discoverAgentsAll(cwd).chains) {
+		chainsByName.set(chain.name, chain);
+	}
+	return Array.from(chainsByName.values());
+};
+
+const makeChainCompletions = (state: SubagentState) => (prefix: string) => {
+	if (prefix.includes(" ") || !state.baseCwd) return null;
+	return discoverSavedChains(state.baseCwd)
+		.filter((chain) => chain.name.startsWith(prefix))
+		.map((chain) => ({ value: chain.name, label: chain.name }));
+};
+
+const mapSavedChainSteps = (chain: ChainConfig, worktree = false): ChainStep[] => {
+	return (chain.steps as Array<ChainStep & { skills?: string[] | false }>).map((step) => {
+		if (isParallelStep(step)) return worktree ? { ...step, worktree: true } : { ...step };
+		return {
+			agent: step.agent,
+			task: step.task || undefined,
+			output: step.output,
+			outputMode: step.outputMode,
+			reads: step.reads,
+			progress: step.progress,
+			skill: step.skill ?? step.skills,
+			model: step.model,
+		};
+	});
+};
+
+async function requestSlashRun(
+	pi: ExtensionAPI,
+	ctx: ExtensionContext,
+	requestId: string,
+	params: SubagentParamsLike,
+): Promise<SlashSubagentResponse> {
+	return new Promise((resolve, reject) => {
+		let done = false;
+		let started = false;
+
+		const startTimeoutMs = 15_000;
+		const startTimeout = setTimeout(() => {
+			finish(() => reject(new Error(
+				"Slash subagent bridge did not start within 15s. Ensure the extension is loaded correctly.",
+			)));
+		}, startTimeoutMs);
+
+		const onStarted = (data: unknown) => {
+			if (done || !data || typeof data !== "object") return;
+			if ((data as { requestId?: unknown }).requestId !== requestId) return;
+			started = true;
+			clearTimeout(startTimeout);
+			if (ctx.hasUI) ctx.ui.setStatus("subagent-slash", "running...");
+		};
+
+		const onResponse = (data: unknown) => {
+			if (done || !data || typeof data !== "object") return;
+			const response = data as Partial<SlashSubagentResponse>;
+			if (response.requestId !== requestId) return;
+			clearTimeout(startTimeout);
+			finish(() => resolve(response as SlashSubagentResponse));
+		};
+
+		const onUpdate = (data: unknown) => {
+			if (done || !data || typeof data !== "object") return;
+			const update = data as SlashSubagentUpdate;
+			if (update.requestId !== requestId) return;
+			applySlashUpdate(requestId, update);
+			if (!ctx.hasUI) return;
+			const tool = update.currentTool ? ` ${update.currentTool}` : "";
+			const count = update.toolCount ?? 0;
+			ctx.ui.setStatus("subagent-slash", `${count} tools${tool} | Ctrl+O live detail`);
+		};
+
+		const onTerminalInput = ctx.hasUI
+			? ctx.ui.onTerminalInput((input) => {
+				if (!matchesKey(input, Key.escape)) return undefined;
+				pi.events.emit(SLASH_SUBAGENT_CANCEL_EVENT, { requestId });
+				finish(() => reject(new Error("Cancelled")));
+				return { consume: true };
+			})
+			: undefined;
+
+		const unsubStarted = pi.events.on(SLASH_SUBAGENT_STARTED_EVENT, onStarted);
+		const unsubResponse = pi.events.on(SLASH_SUBAGENT_RESPONSE_EVENT, onResponse);
+		const unsubUpdate = pi.events.on(SLASH_SUBAGENT_UPDATE_EVENT, onUpdate);
+
+		const finish = (next: () => void) => {
+			if (done) return;
+			done = true;
+			clearTimeout(startTimeout);
+			unsubStarted();
+			unsubResponse();
+			unsubUpdate();
+			onTerminalInput?.();
+			next();
+		};
+
+		pi.events.emit(SLASH_SUBAGENT_REQUEST_EVENT, { requestId, params });
+
+		// Bridge emits STARTED synchronously during REQUEST emit.
+		// If not started, no bridge received the request.
+		if (!started && done) return;
+		if (!started) {
+			finish(() => reject(new Error(
+				"No slash subagent bridge responded. Ensure the subagent extension is loaded correctly.",
+			)));
+		}
+	});
+}
+
+function extractSlashMessageText(content: string | Array<{ type?: string; text?: string }>): string {
+	if (typeof content === "string") return content;
+	if (!Array.isArray(content)) return "";
+	return content
+		.filter((part): part is { type: "text"; text: string } => part?.type === "text" && typeof part.text === "string")
+		.map((part) => part.text)
+		.join("\n");
+}
+
+function formatExportPathList(paths: string[]): string {
+	return paths.map((file) => `- \`${file}\``).join("\n");
+}
+
+function collectResultPaths(results: SingleResult[], getPath: (result: SingleResult) => string | undefined): string[] {
+	return results
+		.map(getPath)
+		.filter((file): file is string => typeof file === "string" && file.length > 0);
+}
+
+function buildSlashExportText(response: SlashSubagentResponse): string {
+	const output = extractSlashMessageText(response.result.content) || response.errorText || "(no output)";
+	const results = response.result.details?.results ?? [];
+	const sessionFiles = collectResultPaths(results, (result) => result.sessionFile);
+	const savedOutputs = collectResultPaths(results, (result) => result.savedOutputPath);
+	const artifactOutputs = collectResultPaths(results, (result) => result.artifactPaths?.outputPath);
+	const sections = ["## Subagent result", output];
+	if (sessionFiles.length > 0) sections.push("## Child session exports", formatExportPathList(sessionFiles));
+	if (savedOutputs.length > 0) sections.push("## Saved outputs", formatExportPathList(savedOutputs));
+	if (artifactOutputs.length > 0) sections.push("## Artifact outputs", formatExportPathList(artifactOutputs));
+	return sections.join("\n\n");
+}
+
+function persistSlashSessionSnapshot(ctx: ExtensionContext): void {
+	try {
+		if (!ctx.sessionManager) return;
+		const sessionManager = ctx.sessionManager as typeof ctx.sessionManager & {
+			_rewriteFile?: () => void;
+			flushed?: boolean;
+		};
+		const sessionFile = sessionManager.getSessionFile();
+		if (!sessionFile || typeof sessionManager._rewriteFile !== "function") return;
+		fs.mkdirSync(path.dirname(sessionFile), { recursive: true });
+		sessionManager._rewriteFile();
+		sessionManager.flushed = true;
+	} catch (error) {
+		console.error("Failed to persist slash session snapshot for export:", error);
+	}
+}
+
+async function runSlashSubagent(
+	pi: ExtensionAPI,
+	ctx: ExtensionContext,
+	params: SubagentParamsLike,
+): Promise<void> {
+	if (ctx.hasUI) ctx.ui.setToolsExpanded(false);
+	const requestId = randomUUID();
+	const initialDetails = buildSlashInitialResult(requestId, params);
+	const initialText = extractSlashMessageText(initialDetails.result.content) || "Running subagent...";
+	pi.sendMessage({
+		customType: SLASH_RESULT_TYPE,
+		content: initialText,
+		display: true,
+		details: initialDetails,
+	});
+	persistSlashSessionSnapshot(ctx);
+
+	try {
+		const response = await requestSlashRun(pi, ctx, requestId, params);
+		const finalDetails = finalizeSlashResult(response);
+		pi.sendMessage({
+			customType: SLASH_RESULT_TYPE,
+			content: buildSlashExportText(response),
+			display: true,
+			details: finalDetails,
+		});
+		persistSlashSessionSnapshot(ctx);
+		if (ctx.hasUI) {
+			ctx.ui.setStatus("subagent-slash", undefined);
+		}
+		if (response.isError && ctx.hasUI) {
+			ctx.ui.notify(response.errorText || "Subagent failed", "error");
+		}
+	} catch (error) {
+		const message = error instanceof Error ? error.message : String(error);
+		const failedDetails = failSlashResult(requestId, params, message);
+		pi.sendMessage({
+			customType: SLASH_RESULT_TYPE,
+			content: `## Subagent result\n\n${message}`,
+			display: true,
+			details: failedDetails,
+		});
+		persistSlashSessionSnapshot(ctx);
+		if (ctx.hasUI) {
+			ctx.ui.setStatus("subagent-slash", undefined);
+		}
+		if (message === "Cancelled") {
+			if (ctx.hasUI) ctx.ui.notify("Cancelled", "warning");
+			return;
+		}
+		if (ctx.hasUI) ctx.ui.notify(message, "error");
+	}
+}
+
+
+interface ParsedStep { name: string; config: InlineConfig; task?: string }
+
+const parseAgentArgs = (
+	state: SubagentState,
+	args: string,
+	command: string,
+	ctx: ExtensionContext,
+): { steps: ParsedStep[]; task: string } | null => {
+	const input = args.trim();
+	const usage = `Usage: /${command} agent1 "task1" -> agent2 "task2"`;
+	let steps: ParsedStep[];
+	let sharedTask: string;
+	let perStep = false;
+
+	if (input.includes(" -> ")) {
+		perStep = true;
+		const segments = input.split(" -> ");
+		steps = [];
+		for (const seg of segments) {
+			const trimmed = seg.trim();
+			if (!trimmed) continue;
+			let agentPart: string;
+			let task: string | undefined;
+			const qMatch = trimmed.match(/^(\S+(?:\[[^\]]*\])?)\s+(?:"([^"]*)"|'([^']*)')$/);
+			if (qMatch) {
+				agentPart = qMatch[1]!;
+				task = (qMatch[2] ?? qMatch[3]) || undefined;
+			} else {
+				const dashIdx = trimmed.indexOf(" -- ");
+				if (dashIdx !== -1) {
+					agentPart = trimmed.slice(0, dashIdx).trim();
+					task = trimmed.slice(dashIdx + 4).trim() || undefined;
+				} else {
+					agentPart = trimmed;
+				}
+			}
+			const parsed = parseAgentToken(agentPart);
+			steps.push({ ...parsed, task });
+		}
+		sharedTask = steps.find((s) => s.task)?.task ?? "";
+	} else {
+		const delimiterIndex = input.indexOf(" -- ");
+		if (delimiterIndex === -1) {
+			ctx.ui.notify(usage, "error");
+			return null;
+		}
+		const agentsPart = input.slice(0, delimiterIndex).trim();
+		sharedTask = input.slice(delimiterIndex + 4).trim();
+		if (!agentsPart || !sharedTask) {
+			ctx.ui.notify(usage, "error");
+			return null;
+		}
+		steps = agentsPart.split(/\s+/).filter(Boolean).map((t) => parseAgentToken(t));
+	}
+
+	if (steps.length === 0) {
+		ctx.ui.notify(usage, "error");
+		return null;
+	}
+	if (!state.baseCwd) {
+		ctx.ui.notify("Subagent session cwd is not initialized yet", "error");
+		return null;
+	}
+	const agents = discoverAgents(state.baseCwd, "both").agents;
+	for (const step of steps) {
+		if (!agents.find((a) => a.name === step.name)) {
+			ctx.ui.notify(`Unknown agent: ${step.name}`, "error");
+			return null;
+		}
+	}
+	if (command === "chain" && !steps[0]?.task && (perStep || !sharedTask)) {
+		ctx.ui.notify(`First step must have a task: /chain agent "task" -> agent2`, "error");
+		return null;
+	}
+	if (command === "parallel" && !steps.some((s) => s.task) && !sharedTask) {
+		ctx.ui.notify("At least one step must have a task", "error");
+		return null;
+	}
+	return { steps, task: sharedTask };
+};
+
+export function registerSlashCommands(
+	pi: ExtensionAPI,
+	state: SubagentState,
+): void {
+	pi.registerCommand("run", {
+		description: "Run a subagent directly: /run agent[output=file] [task] [--bg] [--fork]",
+		getArgumentCompletions: makeAgentCompletions(state, false),
+		handler: async (args, ctx) => {
+			const { args: cleanedArgs, bg, fork } = extractExecutionFlags(args);
+			const input = cleanedArgs.trim();
+			const firstSpace = input.indexOf(" ");
+			if (!input) { ctx.ui.notify("Usage: /run <agent> [task] [--bg] [--fork]", "error"); return; }
+			const { name: agentName, config: inline } = parseAgentToken(firstSpace === -1 ? input : input.slice(0, firstSpace));
+			const task = firstSpace === -1 ? "" : input.slice(firstSpace + 1).trim();
+
+			if (!state.baseCwd) { ctx.ui.notify("Subagent session cwd is not initialized yet", "error"); return; }
+			const agents = discoverAgents(state.baseCwd, "both").agents;
+			if (!agents.find((a) => a.name === agentName)) { ctx.ui.notify(`Unknown agent: ${agentName}`, "error"); return; }
+
+			let finalTask = task;
+			if (inline.reads && Array.isArray(inline.reads) && inline.reads.length > 0) {
+				finalTask = `[Read from: ${inline.reads.join(", ")}]\n\n${finalTask}`;
+			}
+			const params: SubagentParamsLike = { agent: agentName, task: finalTask, clarify: false, agentScope: "both" };
+			if (inline.output !== undefined) params.output = inline.output;
+			if (inline.outputMode !== undefined) params.outputMode = inline.outputMode;
+			if (inline.skill !== undefined) params.skill = inline.skill;
+			if (inline.model) params.model = inline.model;
+			if (bg) params.async = true;
+			if (fork) params.context = "fork";
+			await runSlashSubagent(pi, ctx, params);
+		},
+	});
+
+	pi.registerCommand("chain", {
+		description: "Run agents in sequence: /chain scout \"task\" -> planner [--bg] [--fork]",
+		getArgumentCompletions: makeAgentCompletions(state, true),
+		handler: async (args, ctx) => {
+			const { args: cleanedArgs, bg, fork } = extractExecutionFlags(args);
+			const parsed = parseAgentArgs(state, cleanedArgs, "chain", ctx);
+			if (!parsed) return;
+			const chain = parsed.steps.map(({ name, config, task: stepTask }, i) => ({
+				agent: name,
+				...(stepTask ? { task: stepTask } : i === 0 && parsed.task ? { task: parsed.task } : {}),
+				...(config.output !== undefined ? { output: config.output } : {}),
+				...(config.outputMode !== undefined ? { outputMode: config.outputMode } : {}),
+				...(config.reads !== undefined ? { reads: config.reads } : {}),
+				...(config.model ? { model: config.model } : {}),
+				...(config.skill !== undefined ? { skill: config.skill } : {}),
+				...(config.progress !== undefined ? { progress: config.progress } : {}),
+			}));
+			const params: SubagentParamsLike = { chain, task: parsed.task, clarify: false, agentScope: "both" };
+			if (bg) params.async = true;
+			if (fork) params.context = "fork";
+			await runSlashSubagent(pi, ctx, params);
+		},
+	});
+
+	pi.registerCommand("run-chain", {
+		description: "Run a saved chain: /run-chain chainName -- task [--bg] [--fork]",
+		getArgumentCompletions: makeChainCompletions(state),
+		handler: async (args, ctx) => {
+			const { args: cleanedArgs, bg, fork } = extractExecutionFlags(args);
+			const delimiterIndex = cleanedArgs.indexOf(" -- ");
+			const usage = "Usage: /run-chain <chainName> -- <task> [--bg] [--fork]";
+			if (delimiterIndex === -1) {
+				ctx.ui.notify(usage, "error");
+				return;
+			}
+			const chainName = cleanedArgs.slice(0, delimiterIndex).trim();
+			const task = cleanedArgs.slice(delimiterIndex + 4).trim();
+			if (!chainName || !task) {
+				ctx.ui.notify(usage, "error");
+				return;
+			}
+			if (!state.baseCwd) { ctx.ui.notify("Subagent session cwd is not initialized yet", "error"); return; }
+			const chain = discoverSavedChains(state.baseCwd).find((candidate) => candidate.name === chainName);
+			if (!chain) {
+				ctx.ui.notify(`Unknown chain: ${chainName}`, "error");
+				return;
+			}
+			const params: SubagentParamsLike = { chain: mapSavedChainSteps(chain), task, clarify: false, agentScope: "both" };
+			if (bg) params.async = true;
+			if (fork) params.context = "fork";
+			await runSlashSubagent(pi, ctx, params);
+		},
+	});
+
+	pi.registerCommand("parallel", {
+		description: "Run agents in parallel: /parallel scout \"task1\" -> reviewer \"task2\" [--bg] [--fork]",
+		getArgumentCompletions: makeAgentCompletions(state, true),
+		handler: async (args, ctx) => {
+			const { args: cleanedArgs, bg, fork } = extractExecutionFlags(args);
+			const parsed = parseAgentArgs(state, cleanedArgs, "parallel", ctx);
+			if (!parsed) return;
+			const tasks = parsed.steps.map(({ name, config, task: stepTask }) => ({
+				agent: name,
+				task: stepTask ?? parsed.task,
+				...(config.output !== undefined ? { output: config.output } : {}),
+				...(config.outputMode !== undefined ? { outputMode: config.outputMode } : {}),
+				...(config.reads !== undefined ? { reads: config.reads } : {}),
+				...(config.model ? { model: config.model } : {}),
+				...(config.skill !== undefined ? { skill: config.skill } : {}),
+				...(config.progress !== undefined ? { progress: config.progress } : {}),
+			}));
+			const params: SubagentParamsLike = { tasks, clarify: false, agentScope: "both" };
+			if (bg) params.async = true;
+			if (fork) params.context = "fork";
+			await runSlashSubagent(pi, ctx, params);
+		},
+	});
+
+
+	pi.registerCommand("subagents-doctor", {
+		description: "Show subagent diagnostics",
+		handler: async (_args, ctx) => {
+			await runSlashSubagent(pi, ctx, { action: "doctor" });
+		},
+	});
+
+}