@clinebot/core 0.0.0

package/src/session/session-service.ts

@@ -0,0 +1,610 @@
+import {
+	appendFileSync,
+	existsSync,
+	mkdirSync,
+	readFileSync,
+	renameSync,
+	unlinkSync,
+	writeFileSync,
+} from "node:fs";
+import { join } from "node:path";
+import type {
+	AgentTeamsRuntime,
+	TeamEvent,
+	TeamTeammateSpec,
+} from "@clinebot/agents";
+import { resolveTeamDataDir } from "@clinebot/shared/storage";
+import type { SqliteSessionStore } from "../storage/sqlite-session-store";
+import type { SessionSource, SessionStatus } from "../types/common";
+import { nowIso } from "./session-artifacts";
+import type { SessionManifest } from "./session-manifest";
+import type {
+	PersistedSessionUpdateInput,
+	SessionPersistenceAdapter,
+} from "./unified-session-persistence-service";
+import { UnifiedSessionPersistenceService } from "./unified-session-persistence-service";
+
+export interface SessionRowShape {
+	session_id: string;
+	source: string;
+	pid: number;
+	started_at: string;
+	ended_at?: string | null;
+	exit_code?: number | null;
+	status: SessionStatus;
+	status_lock?: number;
+	interactive: number;
+	provider: string;
+	model: string;
+	cwd: string;
+	workspace_root: string;
+	team_name?: string | null;
+	enable_tools: number;
+	enable_spawn: number;
+	enable_teams: number;
+	parent_session_id?: string | null;
+	parent_agent_id?: string | null;
+	agent_id?: string | null;
+	conversation_id?: string | null;
+	is_subagent: number;
+	prompt?: string | null;
+	metadata_json?: string | null;
+	transcript_path: string;
+	hook_path: string;
+	messages_path?: string | null;
+	updated_at?: string;
+}
+
+export interface CreateRootSessionInput {
+	sessionId: string;
+	source: SessionSource;
+	pid: number;
+	startedAt: string;
+	interactive: boolean;
+	provider: string;
+	model: string;
+	cwd: string;
+	workspaceRoot: string;
+	teamName?: string;
+	enableTools: boolean;
+	enableSpawn: boolean;
+	enableTeams: boolean;
+	prompt?: string;
+	metadata?: Record<string, unknown>;
+	transcriptPath: string;
+	hookPath: string;
+	messagesPath: string;
+}
+
+export interface CreateRootSessionWithArtifactsInput {
+	sessionId: string;
+	source: SessionSource;
+	pid: number;
+	interactive: boolean;
+	provider: string;
+	model: string;
+	cwd: string;
+	workspaceRoot: string;
+	teamName?: string;
+	enableTools: boolean;
+	enableSpawn: boolean;
+	enableTeams: boolean;
+	prompt?: string;
+	metadata?: Record<string, unknown>;
+	startedAt?: string;
+}
+
+export interface RootSessionArtifacts {
+	manifestPath: string;
+	transcriptPath: string;
+	hookPath: string;
+	messagesPath: string;
+	manifest: SessionManifest;
+}
+
+export interface UpsertSubagentInput {
+	agentId: string;
+	parentAgentId: string;
+	conversationId: string;
+	prompt?: string;
+	rootSessionId?: string;
+}
+
+function reviveTeamStateDates(state: TeamRuntimeState): TeamRuntimeState {
+	return {
+		...state,
+		tasks: state.tasks.map((task) => ({
+			...task,
+			createdAt: new Date(task.createdAt),
+			updatedAt: new Date(task.updatedAt),
+		})),
+		mailbox: state.mailbox.map((message) => ({
+			...message,
+			sentAt: new Date(message.sentAt),
+			readAt: message.readAt ? new Date(message.readAt) : undefined,
+		})),
+		missionLog: state.missionLog.map((entry) => ({
+			...entry,
+			ts: new Date(entry.ts),
+		})),
+		runs: (state.runs ?? []).map((run) => ({
+			...run,
+			startedAt: new Date(run.startedAt),
+			endedAt: run.endedAt ? new Date(run.endedAt) : undefined,
+			nextAttemptAt: run.nextAttemptAt
+				? new Date(run.nextAttemptAt)
+				: undefined,
+			heartbeatAt: run.heartbeatAt ? new Date(run.heartbeatAt) : undefined,
+		})),
+		outcomes: (state.outcomes ?? []).map((outcome) => ({
+			...outcome,
+			createdAt: new Date(outcome.createdAt),
+			finalizedAt: outcome.finalizedAt
+				? new Date(outcome.finalizedAt)
+				: undefined,
+		})),
+		outcomeFragments: (state.outcomeFragments ?? []).map((fragment) => ({
+			...fragment,
+			createdAt: new Date(fragment.createdAt),
+			reviewedAt: fragment.reviewedAt
+				? new Date(fragment.reviewedAt)
+				: undefined,
+		})),
+	};
+}
+
+function sanitizeTeamName(name: string): string {
+	return name
+		.toLowerCase()
+		.replace(/[^a-z0-9._-]+/g, "-")
+		.replace(/^-+|-+$/g, "");
+}
+
+interface PersistedTeamEnvelope {
+	version: 1;
+	updatedAt: string;
+	teamState: TeamRuntimeState;
+	teammates: TeamTeammateSpec[];
+}
+
+type TeamRuntimeState = ReturnType<AgentTeamsRuntime["exportState"]>;
+
+export interface FileTeamPersistenceStoreOptions {
+	teamName: string;
+	baseDir?: string;
+}
+
+export class FileTeamPersistenceStore {
+	private readonly dirPath: string;
+	private readonly statePath: string;
+	private readonly taskHistoryPath: string;
+	private readonly teammateSpecs: Map<string, TeamTeammateSpec> = new Map();
+
+	constructor(options: FileTeamPersistenceStoreOptions) {
+		const safeTeamName = sanitizeTeamName(options.teamName);
+		const baseDir = options.baseDir?.trim() || resolveTeamDataDir();
+		this.dirPath = join(baseDir, safeTeamName);
+		this.statePath = join(this.dirPath, "state.json");
+		this.taskHistoryPath = join(this.dirPath, "task-history.jsonl");
+	}
+
+	loadState(): TeamRuntimeState | undefined {
+		if (!existsSync(this.statePath)) {
+			return undefined;
+		}
+		try {
+			const raw = readFileSync(this.statePath, "utf8");
+			const parsed = JSON.parse(raw) as PersistedTeamEnvelope;
+			if (parsed.version !== 1 || !parsed.teamState) {
+				return undefined;
+			}
+			for (const spec of parsed.teammates ?? []) {
+				this.teammateSpecs.set(spec.agentId, spec);
+			}
+			return reviveTeamStateDates(parsed.teamState);
+		} catch {
+			return undefined;
+		}
+	}
+
+	getTeammateSpecs(): TeamTeammateSpec[] {
+		return Array.from(this.teammateSpecs.values());
+	}
+
+	upsertTeammateSpec(spec: TeamTeammateSpec): void {
+		this.teammateSpecs.set(spec.agentId, spec);
+	}
+
+	removeTeammateSpec(agentId: string): void {
+		this.teammateSpecs.delete(agentId);
+	}
+
+	persist(runtime: AgentTeamsRuntime): void {
+		if (!this.hasPersistableState(runtime)) {
+			this.clearPersistedState();
+			return;
+		}
+		this.ensureDir();
+		const envelope: PersistedTeamEnvelope = {
+			version: 1,
+			updatedAt: new Date().toISOString(),
+			teamState: runtime.exportState(),
+			teammates: Array.from(this.teammateSpecs.values()),
+		};
+		const tmpPath = `${this.statePath}.tmp`;
+		writeFileSync(tmpPath, `${JSON.stringify(envelope, null, 2)}\n`, "utf8");
+		renameSync(tmpPath, this.statePath);
+	}
+
+	appendTaskHistory(event: TeamEvent): void {
+		let task: Record<string, unknown> = {};
+		switch (event.type) {
+			case "team_task_updated":
+				task = event.task as unknown as Record<string, unknown>;
+				break;
+			case "team_message":
+				task = {
+					agentId: event.message.fromAgentId,
+					toAgentId: event.message.toAgentId,
+					subject: event.message.subject,
+					taskId: event.message.taskId,
+				};
+				break;
+			case "team_mission_log":
+				task = {
+					agentId: event.entry.agentId,
+					kind: event.entry.kind,
+					summary: event.entry.summary,
+					taskId: event.entry.taskId,
+				};
+				break;
+			case "teammate_spawned":
+			case "teammate_shutdown":
+			case "task_start":
+				task = {
+					agentId: event.agentId,
+					message: "message" in event ? event.message : undefined,
+				};
+				break;
+			case "task_end":
+				task = {
+					agentId: event.agentId,
+					finishReason: event.result?.finishReason,
+					error: event.error?.message,
+				};
+				break;
+			case "agent_event":
+				task = {
+					agentId: event.agentId,
+					eventType: event.event.type,
+				};
+				break;
+		}
+		this.ensureDir();
+		appendFileSync(
+			this.taskHistoryPath,
+			`${JSON.stringify({
+				ts: new Date().toISOString(),
+				type: event.type,
+				task,
+			})}\n`,
+			"utf8",
+		);
+	}
+
+	private ensureDir(): void {
+		if (!existsSync(this.dirPath)) {
+			mkdirSync(this.dirPath, { recursive: true });
+		}
+	}
+
+	private hasPersistableState(runtime: AgentTeamsRuntime): boolean {
+		const state = runtime.exportState();
+		if (this.teammateSpecs.size > 0) {
+			return true;
+		}
+		if (state.members.some((member) => member.role === "teammate")) {
+			return true;
+		}
+		return (
+			state.tasks.length > 0 ||
+			state.mailbox.length > 0 ||
+			state.missionLog.length > 0
+		);
+	}
+
+	private clearPersistedState(): void {
+		if (existsSync(this.statePath)) {
+			unlinkSync(this.statePath);
+		}
+	}
+}
+
+class LocalSessionPersistenceAdapter implements SessionPersistenceAdapter {
+	constructor(private readonly store: SqliteSessionStore) {}
+
+	ensureSessionsDir(): string {
+		return this.store.ensureSessionsDir();
+	}
+
+	async upsertSession(row: SessionRowShape): Promise<void> {
+		this.store.run(
+			`INSERT OR REPLACE INTO sessions (
+				session_id, source, pid, started_at, ended_at, exit_code, status, status_lock, interactive,
+				provider, model, cwd, workspace_root, team_name, enable_tools, enable_spawn, enable_teams,
+				parent_session_id, parent_agent_id, agent_id, conversation_id, is_subagent, prompt,
+				metadata_json, transcript_path, hook_path, messages_path, updated_at
+			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+			[
+				row.session_id,
+				row.source,
+				row.pid,
+				row.started_at,
+				row.ended_at ?? null,
+				row.exit_code ?? null,
+				row.status,
+				typeof row.status_lock === "number" ? row.status_lock : 0,
+				row.interactive,
+				row.provider,
+				row.model,
+				row.cwd,
+				row.workspace_root,
+				row.team_name ?? null,
+				row.enable_tools,
+				row.enable_spawn,
+				row.enable_teams,
+				row.parent_session_id ?? null,
+				row.parent_agent_id ?? null,
+				row.agent_id ?? null,
+				row.conversation_id ?? null,
+				row.is_subagent,
+				row.prompt ?? null,
+				row.metadata_json ?? null,
+				row.transcript_path,
+				row.hook_path,
+				row.messages_path ?? null,
+				row.updated_at ?? nowIso(),
+			],
+		);
+	}
+
+	async getSession(sessionId: string): Promise<SessionRowShape | undefined> {
+		const row = this.store.queryOne<SessionRowShape>(
+			`SELECT session_id, source, pid, started_at, ended_at, exit_code, status, status_lock, interactive,
+					provider, model, cwd, workspace_root, team_name, enable_tools, enable_spawn, enable_teams,
+					parent_session_id, parent_agent_id, agent_id, conversation_id, is_subagent, prompt,
+					metadata_json, transcript_path, hook_path, messages_path, updated_at
+				 FROM sessions WHERE session_id = ?`,
+			[sessionId],
+		);
+		return row ?? undefined;
+	}
+
+	async listSessions(options: {
+		limit: number;
+		parentSessionId?: string;
+		status?: string;
+	}): Promise<SessionRowShape[]> {
+		const whereClauses: string[] = [];
+		const params: unknown[] = [];
+		if (options.parentSessionId) {
+			whereClauses.push("parent_session_id = ?");
+			params.push(options.parentSessionId);
+		}
+		if (options.status) {
+			whereClauses.push("status = ?");
+			params.push(options.status);
+		}
+		const where =
+			whereClauses.length > 0 ? `WHERE ${whereClauses.join(" AND ")}` : "";
+		return this.store.queryAll<SessionRowShape>(
+			`SELECT session_id, source, pid, started_at, ended_at, exit_code, status, status_lock, interactive,
+					provider, model, cwd, workspace_root, team_name, enable_tools, enable_spawn, enable_teams,
+					parent_session_id, parent_agent_id, agent_id, conversation_id, is_subagent, prompt,
+					metadata_json, transcript_path, hook_path, messages_path, updated_at
+				 FROM sessions
+				 ${where}
+				 ORDER BY started_at DESC
+				 LIMIT ?`,
+			[...params, options.limit],
+		);
+	}
+
+	async updateSession(
+		input: PersistedSessionUpdateInput,
+	): Promise<{ updated: boolean; statusLock: number }> {
+		if (input.setRunning) {
+			if (input.expectedStatusLock === undefined) {
+				return { updated: false, statusLock: 0 };
+			}
+			const changed = this.store.run(
+				`UPDATE sessions
+				 SET status = 'running', ended_at = NULL, exit_code = NULL, updated_at = ?, status_lock = ?,
+					 parent_session_id = ?, parent_agent_id = ?, agent_id = ?, conversation_id = ?, is_subagent = 1,
+					 prompt = COALESCE(prompt, ?)
+				 WHERE session_id = ? AND status_lock = ?`,
+				[
+					nowIso(),
+					input.expectedStatusLock + 1,
+					input.parentSessionId ?? null,
+					input.parentAgentId ?? null,
+					input.agentId ?? null,
+					input.conversationId ?? null,
+					input.prompt ?? null,
+					input.sessionId,
+					input.expectedStatusLock,
+				],
+			);
+			return {
+				updated: (changed.changes ?? 0) > 0,
+				statusLock: input.expectedStatusLock + 1,
+			};
+		}
+
+		const fields: string[] = [];
+		const params: unknown[] = [];
+		if (input.status !== undefined) {
+			fields.push("status = ?");
+			params.push(input.status);
+		}
+		if (input.endedAt !== undefined) {
+			fields.push("ended_at = ?");
+			params.push(input.endedAt);
+		}
+		if (input.exitCode !== undefined) {
+			fields.push("exit_code = ?");
+			params.push(input.exitCode);
+		}
+		if (input.prompt !== undefined) {
+			fields.push("prompt = ?");
+			params.push(input.prompt ?? null);
+		}
+		if (input.metadataJson !== undefined) {
+			fields.push("metadata_json = ?");
+			params.push(input.metadataJson ?? null);
+		}
+		if (input.parentSessionId !== undefined) {
+			fields.push("parent_session_id = ?");
+			params.push(input.parentSessionId ?? null);
+		}
+		if (input.parentAgentId !== undefined) {
+			fields.push("parent_agent_id = ?");
+			params.push(input.parentAgentId ?? null);
+		}
+		if (input.agentId !== undefined) {
+			fields.push("agent_id = ?");
+			params.push(input.agentId ?? null);
+		}
+		if (input.conversationId !== undefined) {
+			fields.push("conversation_id = ?");
+			params.push(input.conversationId ?? null);
+		}
+		if (fields.length === 0) {
+			const row = await this.getSession(input.sessionId);
+			return { updated: !!row, statusLock: row?.status_lock ?? 0 };
+		}
+
+		let statusLock = 0;
+		if (input.expectedStatusLock !== undefined) {
+			statusLock = input.expectedStatusLock + 1;
+			fields.push("status_lock = ?");
+			params.push(statusLock);
+		}
+		fields.push("updated_at = ?");
+		params.push(nowIso());
+
+		let sql = `UPDATE sessions SET ${fields.join(", ")} WHERE session_id = ?`;
+		params.push(input.sessionId);
+		if (input.expectedStatusLock !== undefined) {
+			sql += " AND status_lock = ?";
+			params.push(input.expectedStatusLock);
+		}
+		const changed = this.store.run(sql, params);
+		if ((changed.changes ?? 0) === 0) {
+			return { updated: false, statusLock: 0 };
+		}
+		if (input.expectedStatusLock === undefined) {
+			const row = await this.getSession(input.sessionId);
+			statusLock = row?.status_lock ?? 0;
+		}
+		return { updated: true, statusLock };
+	}
+
+	async deleteSession(sessionId: string, cascade: boolean): Promise<boolean> {
+		const changed =
+			this.store.run(`DELETE FROM sessions WHERE session_id = ?`, [sessionId])
+				.changes ?? 0;
+		if (cascade) {
+			this.store.run(`DELETE FROM sessions WHERE parent_session_id = ?`, [
+				sessionId,
+			]);
+		}
+		return changed > 0;
+	}
+
+	async enqueueSpawnRequest(input: {
+		rootSessionId: string;
+		parentAgentId: string;
+		task?: string;
+		systemPrompt?: string;
+	}): Promise<void> {
+		this.store.run(
+			`INSERT INTO subagent_spawn_queue (root_session_id, parent_agent_id, task, system_prompt, created_at, consumed_at)
+			 VALUES (?, ?, ?, ?, ?, NULL)`,
+			[
+				input.rootSessionId,
+				input.parentAgentId,
+				input.task ?? null,
+				input.systemPrompt ?? null,
+				nowIso(),
+			],
+		);
+	}
+
+	async claimSpawnRequest(
+		rootSessionId: string,
+		parentAgentId: string,
+	): Promise<string | undefined> {
+		const row = this.store.queryOne<{ id?: number; task?: string | null }>(
+			`SELECT id, task FROM subagent_spawn_queue
+			 WHERE root_session_id = ? AND parent_agent_id = ? AND consumed_at IS NULL
+			 ORDER BY id ASC LIMIT 1`,
+			[rootSessionId, parentAgentId],
+		);
+		if (!row || typeof row.id !== "number") {
+			return undefined;
+		}
+		this.store.run(
+			`UPDATE subagent_spawn_queue SET consumed_at = ? WHERE id = ?`,
+			[nowIso(), row.id],
+		);
+		return row.task ?? undefined;
+	}
+}
+
+export class CoreSessionService extends UnifiedSessionPersistenceService {
+	constructor(private readonly store: SqliteSessionStore) {
+		super(new LocalSessionPersistenceAdapter(store));
+	}
+
+	createRootSession(input: CreateRootSessionInput): void {
+		this.store.run(
+			`INSERT OR REPLACE INTO sessions (
+				session_id, source, pid, started_at, ended_at, exit_code, status, status_lock, interactive,
+				provider, model, cwd, workspace_root, team_name, enable_tools, enable_spawn, enable_teams,
+				parent_session_id, parent_agent_id, agent_id, conversation_id, is_subagent, prompt,
+				metadata_json, transcript_path, hook_path, messages_path, updated_at
+			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+			[
+				input.sessionId,
+				input.source,
+				input.pid,
+				input.startedAt,
+				null,
+				null,
+				"running",
+				0,
+				input.interactive ? 1 : 0,
+				input.provider,
+				input.model,
+				input.cwd,
+				input.workspaceRoot,
+				input.teamName ?? null,
+				input.enableTools ? 1 : 0,
+				input.enableSpawn ? 1 : 0,
+				input.enableTeams ? 1 : 0,
+				null,
+				null,
+				null,
+				null,
+				0,
+				input.prompt ?? null,
+				input.metadata ? JSON.stringify(input.metadata) : null,
+				input.transcriptPath,
+				input.hookPath,
+				input.messagesPath,
+				nowIso(),
+			],
+		);
+	}
+}