Cristina Poncela Cubeiro
101 lines
2.0 KiB
TypeScript
101 lines
2.0 KiB
TypeScript
import type { InstanceStatus } from "../types.ts";
|
|
|
|
export interface SpawnRequest {
|
|
type: "spawn";
|
|
cwd: string;
|
|
label?: string;
|
|
provider?: string;
|
|
model?: string;
|
|
}
|
|
|
|
export interface ListRequest {
|
|
type: "list";
|
|
}
|
|
|
|
export interface StopRequest {
|
|
type: "stop";
|
|
instanceId: string;
|
|
}
|
|
|
|
export interface StatusRequest {
|
|
type: "status";
|
|
instanceId: string;
|
|
}
|
|
|
|
export interface RequestMap {
|
|
spawn: SpawnRequest;
|
|
list: ListRequest;
|
|
stop: StopRequest;
|
|
status: StatusRequest;
|
|
}
|
|
|
|
export type OrchestratorRequest = RequestMap[keyof RequestMap];
|
|
|
|
export interface InstanceSummary {
|
|
id: string;
|
|
status: InstanceStatus;
|
|
cwd: string;
|
|
label?: string;
|
|
sessionId?: string;
|
|
}
|
|
|
|
export interface ResponseBase {
|
|
ok: boolean;
|
|
error?: string;
|
|
}
|
|
|
|
export interface SpawnResponse extends ResponseBase {
|
|
type: "spawn_result";
|
|
instance?: InstanceSummary;
|
|
}
|
|
|
|
export interface ListResponse extends ResponseBase {
|
|
type: "list_result";
|
|
instances?: InstanceSummary[];
|
|
}
|
|
|
|
export interface StopResponse extends ResponseBase {
|
|
type: "stop_result";
|
|
instanceId?: string;
|
|
}
|
|
|
|
export interface StatusResponse extends ResponseBase {
|
|
type: "status_result";
|
|
instance?: InstanceSummary;
|
|
}
|
|
|
|
export interface ErrorResponse extends ResponseBase {
|
|
type: "error";
|
|
ok: false;
|
|
error: string;
|
|
}
|
|
|
|
export interface ResponseMap {
|
|
spawn: SpawnResponse;
|
|
list: ListResponse;
|
|
stop: StopResponse;
|
|
status: StatusResponse;
|
|
}
|
|
|
|
export type OrchestratorResponse = ResponseMap[keyof ResponseMap] | ErrorResponse;
|
|
|
|
export type ResponseFor<T extends OrchestratorRequest> = T extends { type: infer K }
|
|
? K extends keyof ResponseMap
|
|
? ResponseMap[K] | ErrorResponse
|
|
: ErrorResponse
|
|
: ErrorResponse;
|
|
|
|
export function encodeMessage(message: OrchestratorRequest | OrchestratorResponse): string {
|
|
return `${JSON.stringify(message)}\n`;
|
|
}
|
|
|
|
export function parseRequestLine(line: string): OrchestratorRequest {
|
|
const value = JSON.parse(line) as OrchestratorRequest;
|
|
return value;
|
|
}
|
|
|
|
export function parseResponseLine(line: string): OrchestratorResponse {
|
|
const value = JSON.parse(line) as OrchestratorResponse;
|
|
return value;
|
|
}
|