← vibesdk

/**
 * Public type surface for `@space-do/space`.
 *
 * The package ships an esbuild-produced JS bundle (`dist/index.js`). Running
 * `tsc --declaration` against the source currently fails (known errors
 * inherited from the upstream port), so this hand-maintained `.d.ts`
 * describes the subset of the API that consumers actually depend on.
 *
 * Keep in sync with `src/index.ts` exports.
 *
 * Globals referenced (`DurableObjectNamespace`, `DurableObjectState`,
 * `DurableObject`) come from `@cloudflare/workers-types`, which the consumer
 * is expected to include in its tsconfig `types`.
 */

interface DurableObjectConstructor {
	new (state: DurableObjectState, env: unknown): DurableObject;
}

// ── Durable Object classes (DurableObject<Env> subclasses) ──────────────────
// Treated as opaque DO classes; consumers access them via wrangler bindings.
export const SpaceDO: DurableObjectConstructor;

// ── Inspector result types (used by DB-viewer route handlers) ──────────────
export interface AppDatabaseColumn {
	name: string;
	type: string;
	notnull: number;
	pk: number;
}

export interface AppDatabaseTable {
	name: string;
	rowCount: number;
	columns: AppDatabaseColumn[];
}

export interface AppDatabaseReadResult {
	columns: string[];
	rows: Record<string, unknown>[];
	totalCount: number;
}

export interface BranchDeploymentBundle {
	branch: string;
	commitHash: string;
	mainModule: string;
	modules: Record<string, string | Record<string, unknown>>;
	assets: Record<string, string>;
	assetConfig?: {
		html_handling?: 'auto-trailing-slash' | 'drop-trailing-slash' | 'force-trailing-slash' | 'none';
		not_found_handling?: 'single-page-application' | '404-page' | 'none';
	};
	compatibilityDate: string;
}

// ── Env shape (minimum). The host worker's Env satisfies this. ─────────────
export interface Env {
	SPACE_DO: DurableObjectNamespace;
	LOADER?: unknown;
	AI?: unknown;
	ANTHROPIC_API_KEY?: string;
	OPENAI_API_KEY?: string;
	GOOGLE_API_KEY?: string;
	CLOUDFLARE_ACCOUNT_ID?: string;
	CLOUDFLARE_GATEWAY_ID?: string;
	CLOUDFLARE_API_TOKEN?: string;
	CF_AIG_TOKEN?: string;
	SERVER_PASSWORD?: string;
	SERVER_USERNAME?: string;
	ENVIRONMENT?: string;
}

// ── Stored message types (subset, mirrors src/types.ts) ────────────────────
export interface ToolCallInfo {
	id: string;
	name: string;
	arguments: Record<string, unknown>;
}

export interface StoredPart {
	id: string;
	sessionID: string;
	messageID: string;
	type: string;
	text?: string;
	time?: { start: number; end?: number };
	reason?: string;
	cost?: number;
	tokens?: {
		input: number;
		output: number;
		reasoning: number;
		cache: { read: number; write: number };
	};
	callID?: string;
	tool?: string;
	state?: {
		status: string;
		input: Record<string, unknown>;
		raw?: string;
		title?: string;
		output?: string;
		metadata?: Record<string, unknown>;
		time?: { start: number; end?: number };
		error?: string;
	};
	metadata?: Record<string, unknown>;
	snapshot?: string;
}

export interface StoredMessage {
	info: {
		id: string;
		sessionID: string;
		role: 'user' | 'assistant';
		time: { created: number; completed?: number };
		agent: string;
		model?: { providerID: string; modelID: string };
		parentID?: string;
		modelID?: string;
		providerID?: string;
		mode?: string;
		path?: { cwd: string; root: string };
		cost?: number;
		tokens?: {
			total?: number;
			input: number;
			output: number;
			reasoning: number;
			cache: { read: number; write: number };
		};
		finish?: string;
		summary?: { diffs: unknown[] };
		error?: unknown;
	};
	parts: StoredPart[];
}

export type SessionEvent =
	| { type: 'server.connected'; properties: Record<string, unknown> }
	| { type: 'server.heartbeat'; properties: Record<string, unknown> }
	| { type: 'session.created'; properties: { sessionID: string; info: unknown } }
	| { type: 'session.updated'; properties: { sessionID: string; info: unknown } }
	| {
			type: 'message.created';
			properties: { sessionID: string; info: StoredMessage['info'] };
	  }
	| {
			type: 'message.updated';
			properties: { sessionID: string; info: StoredMessage['info'] };
	  }
	| {
			type: 'message.part.updated';
			properties: { sessionID: string; part: StoredPart };
	  }
	| {
			type: 'message.part.delta';
			properties: {
				sessionID: string;
				messageID: string;
				partID: string;
				/** Which field of the part is being extended (e.g. 'text'). */
				field: string;
				/** Incremental chunk to append to `field`. */
				delta: string;
			};
	  }
	| {
			type: 'message.completed';
			properties: { sessionID: string; messageID?: string };
	  }
	| {
			type: 'step.start';
			properties: { sessionID: string; messageID: string };
	  }
	| {
			type: 'tool.called';
			properties: { sessionID: string; messageID: string; tool: ToolCallInfo };
	  }
	| {
			type: 'tool.result';
			properties: { sessionID: string; messageID: string; result: unknown };
	  }
	| { type: 'error'; properties: { sessionID?: string; error: string } }
	| { type: 'done'; properties: { sessionID: string } }
	| { type: string; properties?: Record<string, unknown> };

// ── Convenience info types (subset, opaque) ────────────────────────────────
export interface SessionInfo {
	id: string;
	title: string;
	[key: string]: unknown;
}