storkit/frontend/src/api/agents.ts

export type AgentStatusValue = "pending" | "running" | "completed" | "failed";

export interface AgentInfo {
	story_id: string;
	agent_name: string;
	status: AgentStatusValue;
	session_id: string | null;
	worktree_path: string | null;
	base_branch: string | null;
	log_session_id: string | null;
}

export interface AgentEvent {
	type: "status" | "output" | "agent_json" | "done" | "error" | "warning";
	story_id?: string;
	agent_name?: string;
	status?: string;
	text?: string;
	data?: unknown;
	session_id?: string | null;
	message?: string;
}

export interface AgentConfigInfo {
	name: string;
	role: string;
	model: string | null;
	allowed_tools: string[] | null;
	max_turns: number | null;
	max_budget_usd: number | null;
}

const DEFAULT_API_BASE = "/api";

function buildApiUrl(path: string, baseUrl = DEFAULT_API_BASE): string {
	return `${baseUrl}${path}`;
}

async function requestJson<T>(
	path: string,
	options: RequestInit = {},
	baseUrl = DEFAULT_API_BASE,
): Promise<T> {
	const res = await fetch(buildApiUrl(path, baseUrl), {
		headers: {
			"Content-Type": "application/json",
			...(options.headers ?? {}),
		},
		...options,
	});

	if (!res.ok) {
		const text = await res.text();
		throw new Error(text || `Request failed (${res.status})`);
	}

	return res.json() as Promise<T>;
}

export const agentsApi = {
	startAgent(storyId: string, agentName?: string, baseUrl?: string) {
		return requestJson<AgentInfo>(
			"/agents/start",
			{
				method: "POST",
				body: JSON.stringify({
					story_id: storyId,
					agent_name: agentName,
				}),
			},
			baseUrl,
		);
	},

	stopAgent(storyId: string, agentName: string, baseUrl?: string) {
		return requestJson<boolean>(
			"/agents/stop",
			{
				method: "POST",
				body: JSON.stringify({
					story_id: storyId,
					agent_name: agentName,
				}),
			},
			baseUrl,
		);
	},

	listAgents(baseUrl?: string) {
		return requestJson<AgentInfo[]>("/agents", {}, baseUrl);
	},

	getAgentConfig(baseUrl?: string) {
		return requestJson<AgentConfigInfo[]>("/agents/config", {}, baseUrl);
	},

	reloadConfig(baseUrl?: string) {
		return requestJson<AgentConfigInfo[]>(
			"/agents/config/reload",
			{ method: "POST" },
			baseUrl,
		);
	},
};

/**
 * Subscribe to SSE events for a running agent.
 * Returns a cleanup function to close the connection.
 */
export function subscribeAgentStream(
	storyId: string,
	agentName: string,
	onEvent: (event: AgentEvent) => void,
	onError?: (error: Event) => void,
): () => void {
	const host = import.meta.env.DEV ? "http://127.0.0.1:3001" : "";
	const url = `${host}/agents/${encodeURIComponent(storyId)}/${encodeURIComponent(agentName)}/stream`;

	const eventSource = new EventSource(url);

	eventSource.onmessage = (e) => {
		try {
			const data = JSON.parse(e.data) as AgentEvent;
			onEvent(data);

			// Close on terminal events
			if (
				data.type === "done" ||
				data.type === "error" ||
				(data.type === "status" && data.status === "stopped")
			) {
				eventSource.close();
			}
		} catch (err) {
			console.error("Failed to parse agent event:", err);
		}
	};

	eventSource.onerror = (e) => {
		onError?.(e);
		eventSource.close();
	};

	return () => {
		eventSource.close();
	};
}