feat: supervisor

2026-06-18 13:25:50 +02:00
parent 0cbdc283a3
commit 0d02df760f
4 changed files with 109 additions and 14 deletions
@@ -0,0 +1,99 @@
+import { randomUUID } from "node:crypto";
+import {
+	type AgentSessionRuntime,
+	type CreateAgentSessionRuntimeFactory,
+	createAgentSessionFromServices,
+	createAgentSessionRuntime,
+	createAgentSessionServices,
+	getAgentDir,
+	SessionManager,
+} from "@earendil-works/pi-coding-agent";
+import { getInstance, loadInstances, removeInstance, upsertInstance } from "./storage.ts";
+import type { InstanceRecord } from "./types.ts";
+
+interface LiveInstance {
+	runtime: AgentSessionRuntime;
+	record: InstanceRecord;
+}
+
+function cloneInstance(record: InstanceRecord): InstanceRecord {
+	return { ...record };
+}
+
+async function createRuntime(cwd: string): Promise<AgentSessionRuntime> {
+	const agentDir = getAgentDir();
+	const sessionManager = SessionManager.inMemory(cwd);
+	const runtimeFactory: CreateAgentSessionRuntimeFactory = async ({
+		cwd,
+		agentDir,
+		sessionManager,
+		sessionStartEvent,
+	}) => {
+		const services = await createAgentSessionServices({ cwd, agentDir });
+		const created = await createAgentSessionFromServices({
+			services,
+			sessionManager,
+			sessionStartEvent,
+		});
+		return {
+			...created,
+			services,
+			diagnostics: services.diagnostics,
+		};
+	};
+
+	return createAgentSessionRuntime(runtimeFactory, {
+		cwd,
+		agentDir,
+		sessionManager,
+	});
+}
+
+export class OrchestratorSupervisor {
+	private readonly liveInstances = new Map<string, LiveInstance>();
+
+	listInstances(): InstanceRecord[] {
+		return loadInstances().map(cloneInstance);
+	}
+
+	getInstance(instanceId: string): InstanceRecord | undefined {
+		const live = this.liveInstances.get(instanceId);
+		if (live) {
+			return cloneInstance(live.record);
+		}
+		const stored = getInstance(instanceId);
+		return stored ? cloneInstance(stored) : undefined;
+	}
+
+	async spawnInstance(options: { cwd: string; label?: string }): Promise<InstanceRecord> {
+		const runtime = await createRuntime(options.cwd);
+		const now = new Date().toISOString();
+		const record: InstanceRecord = {
+			id: randomUUID(),
+			status: "online",
+			cwd: options.cwd,
+			createdAt: now,
+			lastSeenAt: now,
+			label: options.label,
+			sessionId: runtime.session.sessionId,
+		};
+
+		this.liveInstances.set(record.id, { runtime, record });
+		upsertInstance(record);
+		return cloneInstance(record);
+	}
+
+	async stopInstance(instanceId: string): Promise<InstanceRecord | undefined> {
+		const live = this.liveInstances.get(instanceId);
+		if (!live) {
+			return undefined;
+		}
+
+		await live.runtime.dispose();
+		this.liveInstances.delete(instanceId);
+		removeInstance(instanceId);
+		return cloneInstance(live.record);
+	}
+}
+
+export const supervisor = new OrchestratorSupervisor();