@bastani/atomic 0.9.16-alpha.2 → 0.9.16-alpha.4

package/dist/builtin/workflows/src/extension/mcp.ts

@@ -1,121 +0,0 @@
-/**
- * MCP cooperation helpers.
- *
- * Responsibilities:
- *  1. Per-stage server gating: emit `mcp.scope.set` via pi.events so the
- *     host MCP layer can restrict visible MCP servers while a stage runs.
- *  2. Always clear scope after stage completion/failure.
- *
- * NOTE: This is intentionally event-based. Hosts without `mcp.scope.set`
- * listeners simply ignore the event, so stage execution remains functional
- * without a bundled MCP adapter dependency.
- */
-
-import { isStaleExtensionContextError } from "@bastani/atomic";
-// ---------------------------------------------------------------------------
-// Minimal structural types — no hard imports from host MCP internals
-// ---------------------------------------------------------------------------
-
-/** Minimal pi events bus surface used by this module. */
-export interface PiEventBus {
-	emit: (event: string, payload: Record<string, unknown>) => void;
-}
-
-/** Minimal ExtensionAPI surface expected by mcp integration. */
-export interface PiMcpExtensionAPI {
-	events?: PiEventBus;
-	[key: string]: unknown;
-}
-
-// ---------------------------------------------------------------------------
-// Scope types
-// ---------------------------------------------------------------------------
-
-/**
- * Per-stage MCP server gating config.
- * Mirrors the stage-level `mcpServers` declaration in a workflow definition.
- */
-export interface McpScopeOpts {
-	/**
-	 * Stage ID scoping this restriction. The adapter may use this to
-	 * namespace the scope so concurrent stages don't interfere.
-	 */
-	stageId: string;
-	/**
-	 * Server IDs to allow exclusively. When provided, all other servers are
-	 * implicitly denied for this stage.
-	 */
-	allow?: string[];
-	/**
-	 * Server IDs to deny explicitly. Applied after `allow` when both present.
-	 */
-	deny?: string[];
-}
-
-/** Payload shape emitted on `mcp.scope.set`. */
-export interface McpScopeSetPayload {
-	stageId: string;
-	allow: string[] | null;
-	deny: string[] | null;
-}
-
-// ---------------------------------------------------------------------------
-// Scope helpers
-// ---------------------------------------------------------------------------
-
-/**
- * Emits `mcp.scope.set` so the host MCP layer can restrict MCP server access
- * for the named stage. No-op when `pi.events` is absent or when the host has
- * no scope listener.
- *
- * Usage:
- *   setMcpScope(pi, { stageId, allow: ["github", "fetch"], deny: ["filesystem"] });
- */
-export function setMcpScope(pi: PiMcpExtensionAPI, opts: McpScopeOpts): void {
-	if (!pi.events) return;
-
-	const payload: McpScopeSetPayload = {
-		stageId: opts.stageId,
-		allow: opts.allow ?? null,
-		deny: opts.deny ?? null,
-	};
-
-	try {
-		pi.events.emit("mcp.scope.set", payload as unknown as Record<string, unknown>);
-	} catch (error) {
-		if (!isStaleExtensionContextError(error)) throw error;
-		// Scope events are advisory and a stage can outlive its extension runtime.
-	}
-}
-
-/**
- * Emits `mcp.scope.set` with null allow/deny to restore unrestricted access
- * after a stage completes. No-op when `pi.events` is absent.
- *
- * Usage:
- *   clearMcpScope(pi, stageId);
- */
-export function clearMcpScope(pi: PiMcpExtensionAPI, stageId: string): void {
-	if (!pi.events) return;
-
-	const payload: McpScopeSetPayload = {
-		stageId,
-		allow: null,
-		deny: null,
-	};
-
-	try {
-		pi.events.emit("mcp.scope.set", payload as unknown as Record<string, unknown>);
-	} catch (error) {
-		if (!isStaleExtensionContextError(error)) throw error;
-		// Scope events are advisory and a stage can outlive its extension runtime.
-	}
-}
-
-/**
- * Returns `true` if MCP scope gating is supported (pi.events present).
- * When false, per-stage server restrictions silently have no effect.
- */
-export function isMcpScopeSupported(pi: PiMcpExtensionAPI): boolean {
-	return pi.events !== undefined && pi.events !== null;
-}

package/dist/builtin/workflows/src/extension/postmortem-deps.ts

@@ -1,70 +0,0 @@
-/**
- * Extension-side wiring for the shared post-mortem stage-chat resolver.
- *
- * Builds `PostMortemStageChatDeps` from live extension runtime surfaces
- * (stage adapters, durable per-run cwd, and the default stage session dir after
- * a host restart) so both the TUI attach pane and `workflow send` revive an
- * eligible terminal agent stage through the same detached, single-flight
- * resolver instead of process-local handle presence alone.
- *
- * cross-ref:
- *   - src/runs/foreground/postmortem-stage-chat.ts (resolver)
- *   - src/tui/overlay-adapter.ts (attach pane wiring)
- *   - src/extension/workflow-tool-send.ts (send parity)
- */
-
-import { getDurableBackend } from "../durable/factory.js";
-import {
-	type EnsurePostMortemStageHandleResult,
-	ensurePostMortemStageHandle,
-	type PostMortemStageChatDeps,
-} from "../runs/foreground/postmortem-stage-chat.js";
-import { stageControlRegistry } from "../runs/foreground/stage-control-registry.js";
-import type { StageAdapters } from "../runs/foreground/stage-runner.js";
-import { store } from "../shared/store.js";
-import { readGraphStoreSnapshot } from "../shared/store-observation.js";
-
-export interface PostMortemResolverDeps {
-	readonly adapters: StageAdapters;
-	readonly resolveDefaultStageSessionDir: () => string | undefined;
-}
-
-/** Persisted original/resolved cwd for a durable run tree, when still available. */
-function resolveStageCwd(runId: string): string | undefined {
-	try {
-		const backend = getDurableBackend();
-		const owningHandle = backend.getWorkflow(runId);
-		const run = readGraphStoreSnapshot(store).runs.find((candidate) => candidate.id === runId);
-		const rootRunId = run?.rootRunId ?? owningHandle?.rootWorkflowId;
-		const cwdHandle = rootRunId === undefined ? owningHandle : (backend.getWorkflow(rootRunId) ?? owningHandle);
-		return cwdHandle?.workflowCwd ?? cwdHandle?.invocationCwd ?? undefined;
-	} catch {
-		return undefined;
-	}
-}
-
-/** Resolver deps for a specific run, keyed so revived handles use the real run cwd. */
-export function postMortemDepsForRun(runId: string, deps: PostMortemResolverDeps): PostMortemStageChatDeps {
-	return {
-		registry: stageControlRegistry,
-		adapters: deps.adapters,
-		cwd: resolveStageCwd(runId),
-		defaultSessionDir: deps.resolveDefaultStageSessionDir(),
-	};
-}
-
-/**
- * Build a `(runId, stageId) => result | undefined` resolver for the attach pane.
- * Unknown stages return `undefined`; known but non-revivable stages retain the
- * resolver's explicit reason so the pane can explain why chat is unavailable.
- */
-export function createPostMortemHandleResolver(
-	deps: PostMortemResolverDeps,
-): (runId: string, stageId: string) => EnsurePostMortemStageHandleResult | undefined {
-	return (runId, stageId) => {
-		const run = readGraphStoreSnapshot(store).runs.find((candidate) => candidate.id === runId);
-		const stage = run?.stages.find((candidate) => candidate.id === stageId);
-		if (stage === undefined) return undefined;
-		return ensurePostMortemStageHandle(runId, stage, postMortemDepsForRun(runId, deps));
-	};
-}

package/dist/builtin/workflows/src/extension/public-types.ts

@@ -1,234 +0,0 @@
-import type { CreateAgentSessionOptions, DefaultResourceLoaderInheritanceSnapshot } from "@bastani/atomic";
-import type { Api, Model } from "@bastani/pi-ai/compat";
-import type { StageSessionRuntime } from "../runs/foreground/stage-runner.js";
-import type { SessionManager } from "../shared/persistence-restore.js";
-import type { RunStatus, StageStatus } from "../shared/store-types.js";
-import type { WorkflowBudget, WorkflowInputValues } from "../shared/types.js";
-import type { WidgetFactory } from "../tui/store-widget-installer.js";
-import type { RenderResultOpts, WorkflowRegisteredToolResult } from "./render-result.js";
-import type { PiUISurface } from "./wiring.js";
-
-export type PiTheme = Record<string, string>;
-
-export interface PiRenderContext {
-	state?: {
-		runId?: string;
-		stages?: unknown[];
-	};
-	invalidate?: () => void;
-}
-
-export interface PiRenderResultOpts extends RenderResultOpts {}
-
-export interface PiRenderComponent {
-	render(width: number): string[];
-	invalidate?: () => void;
-	includes(searchString: string): boolean;
-}
-
-export interface PiMessageRenderComponent {
-	render(width: number): string[];
-	invalidate?: () => void;
-}
-
-export interface PiMessageRenderOptions {
-	expanded: boolean;
-}
-
-export type PiMessageRendererResult = string | PiMessageRenderComponent | null | undefined;
-export type PiMessageRenderer = (
-	payload: unknown,
-	options?: PiMessageRenderOptions,
-	theme?: unknown,
-) => PiMessageRendererResult;
-
-export interface PiArgumentCompletion {
-	value: string;
-	label: string;
-	description?: string;
-}
-
-export type PiArgumentCompletionResult = PiArgumentCompletion[] | null;
-
-export interface PiCommandOptions {
-	description: string;
-	handler: (args: string, ctx: PiCommandContext) => Promise<void> | void;
-	getArgumentCompletions?: (partial: string) => PiArgumentCompletionResult | Promise<PiArgumentCompletionResult>;
-}
-
-export type PiRuntimeModel = Model<Api>;
-
-export interface PiRuntimeModelRegistry {
-	getAvailable(): PiRuntimeModel[];
-}
-
-export interface PiModelContext {
-	readonly model?: PiRuntimeModel;
-	readonly modelRegistry?: PiRuntimeModelRegistry;
-}
-
-export interface PiCommandContext extends PiModelContext {
-	ui: {
-		notify: (message: string, type?: "info" | "warning" | "error") => void;
-	} & PiUISurface;
-	hasUI?: boolean;
-}
-
-export interface PiFlagNamedOpts {
-	description: string;
-	type?: "string" | "boolean";
-	default?: unknown;
-}
-
-export interface PiAgentToolResult<TDetails> {
-	content: Array<{ type: "text"; text: string } | { type: "image"; [key: string]: unknown }>;
-	details: TDetails;
-	terminate?: boolean;
-}
-
-export interface PiToolOpts<TArgs, TDetails> {
-	name: string;
-	label: string;
-	description: string;
-	parameters: unknown;
-	promptGuidelines?: string[];
-	renderShell?: "default" | "self";
-	execute: (
-		toolCallId: string,
-		params: TArgs,
-		signal: AbortSignal | undefined,
-		onUpdate: ((partial: PiAgentToolResult<TDetails>) => void) | undefined,
-		ctx: PiExecuteContext,
-	) => Promise<PiAgentToolResult<TDetails>>;
-	renderCall?: (args: TArgs, theme: PiTheme, context: PiRenderContext) => PiRenderComponent | string;
-	renderResult?: (
-		result: PiAgentToolResult<TDetails>,
-		opts: PiRenderResultOpts,
-		theme: PiTheme,
-		context: PiRenderContext,
-	) => PiRenderComponent | string;
-}
-
-export interface PiExecuteContext extends PiModelContext {
-	sessionId?: string;
-	ui?: PiUISurface;
-	hasUI?: boolean;
-	orchestrationContext?: CreateAgentSessionOptions["orchestrationContext"];
-	sessionManager?: SessionManager & {
-		getSessionFile?: () => string | undefined;
-	};
-	[key: string]: unknown;
-}
-
-export interface WorkflowResourceInfo {
-	readonly path: string;
-	readonly enabled: boolean;
-	readonly metadata?: {
-		readonly source?: string;
-		readonly scope?: string;
-		readonly origin?: string;
-		readonly baseDir?: string;
-	};
-}
-
-type StageLateMessageRouter = NonNullable<
-	NonNullable<CreateAgentSessionOptions["orchestrationContext"]>["lateMessageRouter"]
->;
-
-export interface ExtensionAPI {
-	registerTool?: <TArgs, TResult>(opts: PiToolOpts<TArgs, TResult>) => void;
-	registerCommand?: (name: string, options: PiCommandOptions) => void;
-	registerMessageRenderer?: (event: string, renderer: PiMessageRenderer) => void;
-	sendMessage?: StageLateMessageRouter["routeMessage"];
-	sendMessages?: StageLateMessageRouter["routeMessages"];
-	registerFlag?: (name: string, opts: PiFlagNamedOpts) => void;
-	getWorkflowResources?: () => readonly WorkflowResourceInfo[];
-	refreshWorkflowResources?: () => Promise<readonly WorkflowResourceInfo[]>;
-	getResourceLoaderInheritanceSnapshot?: () => DefaultResourceLoaderInheritanceSnapshot | undefined;
-	registerShortcut?: (
-		key: string,
-		opts: {
-			description: string;
-			handler: (ctx?: PiCommandContext) => void | Promise<void>;
-		},
-	) => void;
-	getActiveTools?: () => string[];
-	setActiveTools?: (toolNames: string[]) => void;
-	setSessionName?: (name: string) => void | Promise<void>;
-	events?: {
-		emit?: (event: string, payload: Record<string, unknown>) => void;
-		on?: (event: string, handler: (payload: unknown) => void) => void;
-	};
-	exec?: (
-		command: string,
-		args: string[],
-		opts?: { signal?: AbortSignal; timeout?: number },
-	) => Promise<{
-		stdout: string;
-		stderr: string;
-		code: number;
-		killed: boolean;
-	}>;
-	createAgentSession?: (options?: CreateAgentSessionOptions) => Promise<{ session: StageSessionRuntime }>;
-	disableAsyncDiscovery?: boolean;
-	appendEntry?: (type: string, payload: Record<string, unknown>) => string | undefined;
-	setLabel?: (entryId: string, label: string) => void;
-	appendCustomMessageEntry?: (content: string, meta?: Record<string, unknown>) => string | undefined;
-	on?: (
-		event: string,
-		handler: (
-			event?: unknown,
-			ctx?: PiCommandContext & {
-				sessionManager?: SessionManager;
-				hasUI?: boolean;
-			},
-			// `unknown`: a handler may return a result object, a promise of one, or
-			// nothing at all. The host narrows what it actually consumes.
-		) => unknown,
-	) => void;
-	sessionManager?: SessionManager;
-	ui?: {
-		setWidget?: (key: string, factory: WidgetFactory | undefined, opts?: { placement?: string }) => void;
-		custom?: PiUISurface["custom"];
-	} & PiUISurface;
-	[key: string]: unknown;
-}
-
-export interface WorkflowToolArgs {
-	workflow?: string;
-	inputs?: WorkflowInputValues;
-	/** Per-run budget override for action "run". */
-	budget?: WorkflowBudget;
-	action?:
-		| "models"
-		| "run"
-		| "list"
-		| "get"
-		| "status"
-		| "stages"
-		| "stage"
-		| "transcript"
-		| "send"
-		| "pause"
-		| "interrupt"
-		| "quit"
-		| "resume"
-		| "reload"
-		| "inputs";
-	runId?: string;
-	all?: boolean;
-	stageId?: string;
-	message?: string;
-	statusFilter?: StageStatus | RunStatus | "all";
-	format?: "text" | "json";
-	limit?: number;
-	tail?: number;
-	includeToolOutput?: boolean;
-	text?: string;
-	response?: unknown;
-	delivery?: "auto" | "answer" | "prompt" | "steer" | "followUp" | "resume";
-	promptId?: string;
-	reason?: string;
-}
-
-export type WorkflowExecuteToolResult = PiAgentToolResult<WorkflowRegisteredToolResult>;

package/dist/builtin/workflows/src/extension/render-call.ts

@@ -1,115 +0,0 @@
-/**
- * Render the workflow tool call as a compact string for display in chat.
- * cross-ref: pi-subagents src/extension/index.ts renderCall slot
- */
-
-import type { WorkflowInputValues } from "../shared/types.js";
-import { truncateToWidth } from "../tui/text-helpers.js";
-
-/** Renderer-only subset of the canonical WorkflowToolArgs from index.ts. */
-export interface WorkflowToolArgs {
-	workflow?: string;
-	inputs?: WorkflowInputValues;
-	action?:
-		| "models"
-		| "run"
-		| "list"
-		| "get"
-		| "status"
-		| "stages"
-		| "stage"
-		| "transcript"
-		| "send"
-		| "pause"
-		| "interrupt"
-		| "quit"
-		| "resume"
-		| "reload"
-		| "inputs";
-	runId?: string;
-}
-
-function runTarget(args: WorkflowToolArgs): string | undefined {
-	if (args.workflow !== undefined && args.workflow.trim().length > 0) return args.workflow;
-	if (args.runId !== undefined && args.runId.trim().length > 0) return args.runId;
-	return undefined;
-}
-
-function quoted(name: string | undefined): string {
-	return name === undefined ? "" : `"${name}"`;
-}
-
-export interface RenderCallOpts {
-	/** Optional host render width in terminal cells. */
-	width?: number;
-}
-
-function fitLine(line: string, width?: number): string {
-	if (width === undefined || width <= 0) return line;
-	return truncateToWidth(line, width, "…");
-}
-
-/**
- * Returns a compact human-readable string describing the tool invocation.
- * Used in the renderCall slot of the workflow tool registration.
- */
-export function renderCall(args: WorkflowToolArgs, opts: RenderCallOpts = {}): string {
-	const action = args.action ?? "run";
-	const name = runTarget(args);
-
-	let line: string;
-	switch (action) {
-		case "list":
-			line = "workflow: list registered workflows";
-			break;
-		case "status":
-			line = "workflow: list retained runs";
-			break;
-		case "inputs":
-			line = name === undefined ? "workflow: show inputs" : `workflow: show inputs for ${quoted(name)}`;
-			break;
-		case "run":
-			line = name === undefined ? "workflow: run" : `workflow: run ${quoted(name)}`;
-			break;
-		case "stages":
-			line = name === undefined ? "workflow: list stages" : `workflow: list stages for ${quoted(name)}`;
-			break;
-		case "stage":
-			line = name === undefined ? "workflow: inspect stage" : `workflow: inspect stage in ${quoted(name)}`;
-			break;
-		case "transcript":
-			line =
-				name === undefined
-					? "workflow: read stage transcript"
-					: `workflow: read stage transcript in ${quoted(name)}`;
-			break;
-		case "send":
-			line = name === undefined ? "workflow: send to stage" : `workflow: send to stage in ${quoted(name)}`;
-			break;
-		case "pause":
-			line = name === undefined ? "workflow: pause run" : `workflow: pause run ${quoted(name)}`;
-			break;
-		case "reload":
-			line = "workflow: reload runtime";
-			break;
-		case "interrupt":
-			line = name === undefined ? "workflow: interrupt run" : `workflow: interrupt run ${quoted(name)}`;
-			break;
-		case "quit":
-			line = name === undefined ? "workflow: quit run" : `workflow: quit run ${quoted(name)}`;
-			break;
-		case "resume":
-			line = name === undefined ? "workflow: resume run" : `workflow: resume run ${quoted(name)}`;
-			break;
-		case "get":
-			line = name === undefined ? "workflow: get" : `workflow: get ${quoted(name)}`;
-			break;
-		case "models":
-			line = "workflow: list configured models";
-			break;
-		default:
-			line = name === undefined ? `workflow: ${action}` : `workflow: ${action} ${quoted(name)}`;
-			break;
-	}
-	return fitLine(line, opts.width);
-}

package/dist/builtin/workflows/src/extension/render-component.ts

@@ -1,12 +0,0 @@
-import type { PiRenderComponent } from "./public-types.js";
-
-export function dynamicTextRenderComponent(renderText: (width: number) => string): PiRenderComponent {
-	return {
-		render(width: number): string[] {
-			return renderText(width).split("\n");
-		},
-		includes(searchString: string): boolean {
-			return renderText(120).includes(searchString);
-		},
-	};
-}