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

package/dist/builtin/workflows/src/extension/workflow-status-summary.ts

@@ -1,276 +0,0 @@
-/**
- * Concise per-run summaries for the workflow tool `status` listing.
- *
- * `workflow({ action: "status" })` without a `runId` lists every top-level
- * run in the session. Each run is reduced to an agent-friendly summary that
- * carries status, timing, active stages, and awaiting-input/pending-prompt
- * information plus the identifiers needed to feed `pause`/`resume`/
- * `interrupt`/`quit`/`send` directly.
- *
- * cross-ref:
- *  - src/extension/workflow-tool.ts        `case "status"` (listing path)
- *  - src/extension/workflow-tool-content.ts agent-visible text rendering
- *  - src/extension/workflow-targets.ts      topLevelExpandedSnapshots()
- */
-
-import { effectiveRunStatus } from "../shared/returned-run-status.js";
-import type {
-	PendingPrompt,
-	RunBudgetSnapshot,
-	RunBudgetState,
-	RunSnapshot,
-	RunStatus,
-	StageInputRequest,
-	StageSnapshot,
-	StageStatus,
-	ToolNodeStatus,
-} from "../shared/store-types.js";
-import { elapsedRunMs } from "../shared/timing.js";
-
-/**
- * Status filter accepted by the `status` run listing (and the `stages`
- * action). Run-level statuses match runs directly; `awaiting_input` selects
- * runs with at least one stage awaiting input or pending human prompt;
- * `all` (the default) includes everything.
- */
-export type WorkflowRunStatusFilter = StageStatus | RunStatus | "all";
-
-/** A currently active (running or awaiting-input) stage within a run. */
-export interface WorkflowStatusActiveStage {
-	/** Expanded-graph stage id; valid for stage-scoped send/pause/resume. */
-	readonly stageId: string;
-	readonly name: string;
-	readonly status: StageStatus;
-}
-
-/** One stage (or run-level) pending human-input descriptor. */
-export interface WorkflowStatusAwaitingInput {
-	/** Absent for a run-level HIL prompt. */
-	readonly stageId?: string;
-	readonly stageName?: string;
-	/** Pending prompt id; pass as `promptId` to `send` when answering. */
-	readonly promptId?: string;
-	readonly promptKind?: string;
-	readonly message?: string;
-}
-
-/** Non-chat durable tool-node status included additively in run summaries. */
-export interface WorkflowStatusToolNode {
-	readonly id: string;
-	readonly name: string;
-	readonly status: ToolNodeStatus;
-	readonly ordinal: number;
-	readonly executionOrder?: number;
-	readonly parentIds: readonly string[];
-	readonly startedAt?: number;
-	readonly endedAt?: number;
-	readonly replayed?: boolean;
-	readonly resultSummary?: string;
-	readonly error?: string;
-	readonly attachable: false;
-	/** Owning run identity; emitted for summaries, optional for source compatibility. */
-	readonly runId?: string;
-	readonly runName?: string;
-	readonly depth?: number;
-}
-
-/**
- * Concise, JSON-stable summary of one top-level run for the `status`
- * listing. `runId` feeds pause/resume/interrupt/quit/send directly;
- * `awaitingInput` entries carry the stage/prompt ids that `send` accepts.
- */
-export interface WorkflowRunStatusSummary {
-	readonly runId: string;
-	/** Workflow/run name. */
-	readonly name: string;
-	readonly status: RunStatus;
-	readonly startedAt: number;
-	readonly endedAt?: number;
-	/** Pause-adjusted elapsed milliseconds (prior sessions included). */
-	readonly elapsedMs: number;
-	readonly activeStages: readonly WorkflowStatusActiveStage[];
-	/** Durable, non-attachable ctx.tool graph nodes in authored order. */
-	readonly tools?: readonly WorkflowStatusToolNode[];
-	readonly awaitingInputCount: number;
-	readonly awaitingInput: readonly WorkflowStatusAwaitingInput[];
-	readonly budget?: RunBudgetSnapshot;
-	readonly budgetState?: RunBudgetState;
-	readonly exitReason?: string;
-	readonly error?: string;
-}
-
-/** Filtered, ordered status listing: `runs[i]` summarizes `snapshots[i]`. */
-export interface WorkflowStatusListing {
-	readonly filter: WorkflowRunStatusFilter;
-	readonly runs: WorkflowRunStatusSummary[];
-	readonly snapshots: RunSnapshot[];
-}
-
-const workflowStatusRenderRuns = new WeakMap<object, readonly RunSnapshot[]>();
-
-/** Keep the full point-in-time run collection out of agent-facing result JSON. */
-export function setWorkflowStatusRenderRuns(result: object, runs: readonly RunSnapshot[]): void {
-	workflowStatusRenderRuns.set(result, runs);
-}
-
-export function getWorkflowStatusRenderRuns(result: object): readonly RunSnapshot[] | undefined {
-	return workflowStatusRenderRuns.get(result);
-}
-
-function stageIsActive(stage: StageSnapshot): boolean {
-	return stage.status === "running" || stage.status === "awaiting_input";
-}
-
-function stageAwaitingInput(stage: StageSnapshot): boolean {
-	return (
-		stage.status === "awaiting_input" ||
-		stage.awaitingInputSince !== undefined ||
-		stage.pendingPrompt !== undefined ||
-		stage.inputRequest !== undefined
-	);
-}
-
-function promptMessage(prompt: PendingPrompt | undefined, request: StageInputRequest | undefined): string | undefined {
-	if (prompt !== undefined) return prompt.message;
-	return request?.questions[0]?.question;
-}
-
-function awaitingInputEntries(run: RunSnapshot): WorkflowStatusAwaitingInput[] {
-	const entries: WorkflowStatusAwaitingInput[] = [];
-	if (run.pendingPrompt !== undefined) {
-		entries.push({
-			promptId: run.pendingPrompt.id,
-			promptKind: run.pendingPrompt.kind,
-			message: run.pendingPrompt.message,
-		});
-	}
-	for (const stage of run.stages) {
-		if (!stageAwaitingInput(stage)) continue;
-		const entry: WorkflowStatusAwaitingInput = {
-			stageId: stage.id,
-			stageName: stage.name,
-			promptId: stage.pendingPrompt?.id ?? stage.inputRequest?.id,
-			promptKind: stage.pendingPrompt?.kind ?? stage.inputRequest?.kind,
-			message: promptMessage(stage.pendingPrompt, stage.inputRequest),
-		};
-		entries.push(entry);
-	}
-	return entries;
-}
-
-function budgetReport(dimension: "duration" | "tokens" | "cost", reading: number, ceiling: number) {
-	return { dimension, reading, ceiling, percent: ceiling === 0 ? 0 : (reading / ceiling) * 100 };
-}
-/** Reduce one run snapshot to its concise status summary. */
-export function summarizeRunSnapshot(run: RunSnapshot, now = Date.now()): WorkflowRunStatusSummary {
-	const awaitingInput = awaitingInputEntries(run);
-	const elapsedMs = elapsedRunMs(run, now);
-	const duration =
-		run.budget === undefined
-			? undefined
-			: run.endedAt === undefined
-				? {
-						dimension: "duration" as const,
-						reading: elapsedMs,
-						ceiling: run.budget.maxDurationMs,
-						percent: run.budget.maxDurationMs === 0 ? 0 : (elapsedMs / run.budget.maxDurationMs) * 100,
-					}
-				: (run.budgetState?.duration ?? {
-						dimension: "duration" as const,
-						reading: elapsedMs,
-						ceiling: run.budget.maxDurationMs,
-						percent: run.budget.maxDurationMs === 0 ? 0 : (elapsedMs / run.budget.maxDurationMs) * 100,
-					});
-	const usageBudgetReport = (dimension: "tokens" | "cost", ceiling: number | undefined) => {
-		if (ceiling === undefined || ceiling <= 0) return undefined;
-		return (
-			run.budgetState?.[dimension] ?? budgetReport(dimension, run.budgetState?.accounting?.[dimension] ?? 0, ceiling)
-		);
-	};
-	const tokens = usageBudgetReport("tokens", run.budget?.maxTokens) as RunBudgetState["tokens"];
-	const cost = usageBudgetReport("cost", run.budget?.maxCost) as RunBudgetState["cost"];
-	const budgetState =
-		run.budget === undefined
-			? undefined
-			: {
-					...(run.budgetState ?? {}),
-					duration,
-					...(tokens !== undefined ? { tokens } : {}),
-					...(cost !== undefined ? { cost } : {}),
-				};
-	return {
-		runId: run.id,
-		name: run.name,
-		status: effectiveRunStatus(run),
-		startedAt: run.startedAt,
-		endedAt: run.endedAt,
-		elapsedMs,
-		...(run.budget !== undefined ? { budget: run.budget } : {}),
-		...(budgetState !== undefined ? { budgetState } : {}),
-		activeStages: run.stages.filter(stageIsActive).map((stage) => ({
-			stageId: stage.id,
-			name: stage.name,
-			status: stage.status,
-		})),
-		tools: (run.toolNodes ?? []).map((tool) => {
-			const owner = tool as typeof tool & {
-				readonly runId?: string;
-				readonly runName?: string;
-				readonly depth?: number;
-			};
-			return {
-				id: tool.id,
-				name: tool.name,
-				status: tool.status,
-				ordinal: tool.ordinal,
-				executionOrder: tool.executionOrder,
-				parentIds: [...tool.parentIds],
-				startedAt: tool.startedAt,
-				endedAt: tool.endedAt,
-				replayed: tool.replayed,
-				resultSummary: tool.resultSummary,
-				error: tool.error,
-				attachable: false,
-				runId: owner.runId ?? run.id,
-				runName: owner.runName ?? run.name,
-				depth: owner.depth ?? 0,
-			};
-		}),
-		awaitingInputCount: awaitingInput.length,
-		awaitingInput,
-		exitReason: run.exitReason,
-		error: run.error,
-	};
-}
-
-/** True when the summarized run passes the status filter. */
-export function runMatchesStatusFilter(summary: WorkflowRunStatusSummary, filter: WorkflowRunStatusFilter): boolean {
-	if (filter === "all") return true;
-	if (filter === "awaiting_input") return summary.awaitingInputCount > 0;
-	return summary.status === filter;
-}
-
-/**
- * Build the filtered `status` run listing: summaries and their matching
- * snapshots, in-flight runs first, then each bucket by startedAt descending.
- */
-export function buildWorkflowStatusListing(
-	snapshots: readonly RunSnapshot[],
-	filter: WorkflowRunStatusFilter = "all",
-	now = Date.now(),
-): WorkflowStatusListing {
-	const paired = snapshots
-		.map((snapshot) => ({ snapshot, summary: summarizeRunSnapshot(snapshot, now) }))
-		.filter(({ summary }) => runMatchesStatusFilter(summary, filter));
-	paired.sort((a, b) => {
-		const aEnded = a.snapshot.endedAt === undefined ? 0 : 1;
-		const bEnded = b.snapshot.endedAt === undefined ? 0 : 1;
-		if (aEnded !== bEnded) return aEnded - bEnded;
-		return b.snapshot.startedAt - a.snapshot.startedAt;
-	});
-	return {
-		filter,
-		runs: paired.map((pair) => pair.summary),
-		snapshots: paired.map((pair) => pair.snapshot),
-	};
-}

package/dist/builtin/workflows/src/extension/workflow-targets.ts

@@ -1,210 +0,0 @@
-import { getEnvValue, WORKFLOW_STAGE_SUBAGENT_GUARD_ENV } from "@bastani/atomic";
-import {
-	type ExpandedWorkflowStage,
-	expandedStageLabel,
-	expandWorkflowGraph,
-	stageMatchesExpandedIdentifier,
-} from "../shared/expanded-workflow-graph.js";
-import { isFullRunId, malformedRunIdMessage, RUN_ID_LENGTH } from "../shared/run-id.js";
-import { topLevelWorkflowRuns } from "../shared/run-visibility.js";
-import { type Store, store } from "../shared/store.js";
-import { readGraphStoreSnapshot } from "../shared/store-observation.js";
-import type { RunStatus } from "../shared/store-types.js";
-import type { OverlayPiSurface } from "../tui/overlay-adapter.js";
-import type { PiExecuteContext, WorkflowToolArgs } from "./public-types.js";
-import type { PiUISurface } from "./wiring.js";
-
-export function formatAlreadyEndedRetainedMessage(runId: string): string {
-	return `Run ${runId} already ended; retained for inspection.`;
-}
-
-export function stageFailureMessage(runId: string, resultReason: string, action: "pause" | "interrupt"): string {
-	switch (resultReason) {
-		case "not_found":
-			return `Run not found: ${runId}`;
-		case "already_ended":
-			return `Run already ended: ${runId}`;
-		case "stage_not_found":
-			return `Stage not found for run: ${runId}`;
-		default:
-			return `No active stages to ${action} for run: ${runId}`;
-	}
-}
-
-export function inFlightRunCount(): number {
-	return topLevelWorkflowRuns(store.runs()).filter((run) => run.endedAt === undefined).length;
-}
-
-export function topLevelExpandedSnapshots() {
-	const snapshot = store.snapshot();
-	return topLevelWorkflowRuns(snapshot.runs).map((run) => {
-		const graph = expandWorkflowGraph(snapshot, run.id);
-		return {
-			...structuredClone(run),
-			stages: graph.stages.map((stage) => structuredClone(stage)),
-			toolNodes: graph.tools.map((tool) => structuredClone(tool)),
-		};
-	});
-}
-
-export function allStageConflictMessage(action: "pause" | "interrupt" | "quit"): string {
-	return `Cannot ${action} --all with a stageId; omit stageId or target a single run.`;
-}
-
-export function reloadFailureMessage(error: unknown): string {
-	return `Reload failed: ${error instanceof Error ? error.message : String(error)}`;
-}
-
-function hasWorkflowStageSubagentGuardEnv(): boolean {
-	return getEnvValue(WORKFLOW_STAGE_SUBAGENT_GUARD_ENV) === "1";
-}
-
-export function isWorkflowStageToolContext(ctx: PiExecuteContext): boolean {
-	return hasWorkflowStageSubagentGuardEnv() || ctx.orchestrationContext?.kind === "workflow-stage";
-}
-
-export function isRunStatus(value: string): value is RunStatus {
-	switch (value) {
-		case "pending":
-		case "running":
-		case "paused":
-		case "completed":
-		case "skipped":
-		case "cancelled":
-		case "blocked":
-		case "failed":
-		case "killed":
-			return true;
-		default:
-			return false;
-	}
-}
-
-export { isFullRunId, malformedRunIdMessage, RUN_ID_LENGTH };
-
-export type RunIdResolution =
-	| { kind: "exact"; runId: string }
-	| { kind: "malformed"; message: string }
-	| { kind: "not_found" };
-
-export function resolveRunId(target: string, activeStore: Store = store): RunIdResolution {
-	if (!isFullRunId(target)) return { kind: "malformed", message: malformedRunIdMessage(target) };
-	const exact = activeStore.runs().find((r) => r.id === target);
-	if (exact) return { kind: "exact", runId: exact.id };
-	return { kind: "not_found" };
-}
-
-export type ToolRunTarget =
-	| { kind: "all" }
-	| { kind: "run"; runId: string }
-	| { kind: "malformed"; target: string; message: string }
-	| { kind: "not_found"; target: string; message: string };
-
-export function resolveToolRunTarget(
-	args: WorkflowToolArgs,
-	emptyMessage: string,
-	activeStore: Store = store,
-): ToolRunTarget {
-	const rawTarget = args.runId?.trim() ?? "";
-	if (args.all === true || rawTarget === "--all") return { kind: "all" };
-	const target = rawTarget || activeStore.activeRunId() || "";
-	if (!target) return { kind: "not_found", target: rawTarget, message: emptyMessage };
-	const resolved = resolveRunId(target, activeStore);
-	if (resolved.kind === "exact") return { kind: "run", runId: resolved.runId };
-	if (resolved.kind === "malformed") return { kind: "malformed", target, message: resolved.message };
-	return { kind: "not_found", target, message: `Run not found: ${target}` };
-}
-
-export type ToolStageTarget = { ok: true; runId?: string; stageId?: string } | { ok: false; message: string };
-
-export function resolveStageTarget(runId: string, stageTarget?: string, activeStore: Store = store): ToolStageTarget {
-	const target = stageTarget?.trim();
-	if (!target) return { ok: true, runId };
-	const graph = expandWorkflowGraph(readGraphStoreSnapshot(activeStore), runId);
-	const exactVirtualIds = graph.stages.filter((stage) => stage.id === target);
-	if (exactVirtualIds.length === 1) return resolvedStageTarget(exactVirtualIds[0]!);
-	if (exactVirtualIds.length > 1) return ambiguousStageTarget(target, exactVirtualIds);
-	const exactLocalIds = graph.stages.filter((stage) => stage.workflowGraphTarget.stageId === target);
-	if (exactLocalIds.length === 1) return resolvedStageTarget(exactLocalIds[0]!);
-	if (exactLocalIds.length > 1) return ambiguousStageTarget(target, exactLocalIds);
-	const exactNames = graph.stages.filter((stage) => stage.name === target);
-	if (exactNames.length === 1) return resolvedStageTarget(exactNames[0]!);
-	if (exactNames.length > 1) return ambiguousStageTarget(target, exactNames);
-	const matches = graph.stages.filter((stage) => stageMatchesExpandedIdentifier(stage, target));
-	if (matches.length === 0) return { ok: false, message: `Stage not found in run ${runId}: ${target}` };
-	if (matches.length > 1) return ambiguousStageTarget(target, matches);
-	return resolvedStageTarget(matches[0]!);
-}
-
-function resolvedStageTarget(stage: ExpandedWorkflowStage): ToolStageTarget {
-	return {
-		ok: true,
-		runId: stage.workflowGraphTarget.runId,
-		stageId: stage.workflowGraphTarget.stageId,
-	};
-}
-
-function ambiguousStageTarget(target: string, stages: readonly ExpandedWorkflowStage[]): ToolStageTarget {
-	return {
-		ok: false,
-		message: `Ambiguous stage identifier "${target}" matches: ${stages.map(expandedStageLabel).join(", ")}`,
-	};
-}
-
-export function resolveToolStageTarget(
-	runId: string,
-	stageTarget?: string,
-	activeStore: Store = store,
-): ToolStageTarget {
-	return resolveStageTarget(runId, stageTarget, activeStore);
-}
-
-/**
- * Control target resolved across stages *and* tool nodes.
- *
- * Tool nodes are abort-only control targets: `quit` and `interrupt` may name
- * them by expanded id, local `tool:<argsHash>` id, or tool name, with the same
- * ambiguity handling stages get. They are never chat/attach targets.
- */
-export type ControlNodeTarget =
-	| { ok: true; kind: "run" }
-	| { ok: true; kind: "stage"; runId: string; stageId: string }
-	| { ok: true; kind: "tool"; runId: string; nodeId: string; name: string }
-	| { ok: false; message: string };
-
-export function resolveControlNodeTarget(runId: string, stageTarget?: string): ControlNodeTarget {
-	const target = stageTarget?.trim();
-	if (!target) return { ok: true, kind: "run" };
-	const graph = expandWorkflowGraph(readGraphStoreSnapshot(store), runId);
-	const nodes = graph.renderStages;
-	const candidates: Array<readonly ExpandedWorkflowStage[]> = [
-		nodes.filter((node) => node.id === target),
-		nodes.filter((node) => node.workflowGraphTarget.stageId === target),
-		nodes.filter((node) => node.name === target),
-		nodes.filter((node) => stageMatchesExpandedIdentifier(node, target)),
-	];
-	for (const matches of candidates) {
-		if (matches.length === 1) return resolvedControlNodeTarget(matches[0]!);
-		if (matches.length > 1)
-			return {
-				ok: false,
-				message: `Ambiguous stage identifier "${target}" matches: ${matches.map(expandedStageLabel).join(", ")}`,
-			};
-	}
-	return { ok: false, message: `Stage not found in run ${runId}: ${target}` };
-}
-
-function resolvedControlNodeTarget(node: ExpandedWorkflowStage): ControlNodeTarget {
-	const graphTarget = node.workflowGraphTarget;
-	return node.nodeKind === "tool"
-		? { ok: true, kind: "tool", runId: graphTarget.runId, nodeId: graphTarget.stageId, name: node.name }
-		: { ok: true, kind: "stage", runId: graphTarget.runId, stageId: graphTarget.stageId };
-}
-
-export function toolNodePauseRejectionMessage(name: string, nodeId: string): string {
-	return `Tool nodes cannot be paused; ctx.tool ${name} (${nodeId}) has no turn boundary. Use interrupt or quit to abort it.`;
-}
-
-export function overlaySurfaceFromContext(ctx?: { ui?: PiUISurface }): OverlayPiSurface | undefined {
-	return typeof ctx?.ui?.custom === "function" ? { ui: ctx.ui } : undefined;
-}