From d79e3061ad13b6e7443c74fa24a9c66bba8bdaaf Mon Sep 17 00:00:00 2001
From: Cristina Poncela Cubeiro
 <140309543+cristinaponcela@users.noreply.github.com>
Date: Mon, 22 Jun 2026 14:53:07 +0200
Subject: [PATCH] cleanup: one shot vs stream

---
 packages/orchestrator/src/cli.ts          |  2 +-
 packages/orchestrator/src/handler.ts      | 29 ++++++++++++-----------
 packages/orchestrator/src/ipc/protocol.ts | 11 +++++++--
 packages/orchestrator/src/ipc/server.ts   | 12 ++++------
 packages/orchestrator/src/supervisor.ts   |  2 --
 5 files changed, 29 insertions(+), 27 deletions(-)

diff --git a/packages/orchestrator/src/cli.ts b/packages/orchestrator/src/cli.ts
index 88304ed6..856dd658 100644
--- a/packages/orchestrator/src/cli.ts
+++ b/packages/orchestrator/src/cli.ts
@@ -41,7 +41,7 @@ async function rpcStream(instanceId: string): Promise<void> {
 
 	await new Promise<void>((resolve, reject) => {
 		socket.once("connect", () => {
-			socket.write(encodeMessage({ type: "rpc", instanceId }));
+			socket.write(encodeMessage({ type: "rpc_stream", instanceId }));
 			resolve();
 		});
 		socket.once("error", reject);
diff --git a/packages/orchestrator/src/handler.ts b/packages/orchestrator/src/handler.ts
index 1aea0219..89fcb154 100644
--- a/packages/orchestrator/src/handler.ts
+++ b/packages/orchestrator/src/handler.ts
@@ -15,6 +15,7 @@ import type {
 	RpcBridgeResponse,
 	RpcReadyResponse,
 	RpcRequest,
+	RpcStreamRequest,
 	SpawnRequest,
 	SpawnResponse,
 	StatusRequest,
@@ -50,9 +51,8 @@ export async function handleIpcRequest(request: SpawnRequest): Promise<SpawnResp
 export async function handleIpcRequest(request: ListRequest): Promise<ListResponse | ErrorResponse>;
 export async function handleIpcRequest(request: StopRequest): Promise<StopResponse | ErrorResponse>;
 export async function handleIpcRequest(request: StatusRequest): Promise<StatusResponse | ErrorResponse>;
-export async function handleIpcRequest(
-	request: RpcRequest,
-): Promise<RpcBridgeResponse | RpcReadyResponse | ErrorResponse>;
+export async function handleIpcRequest(request: RpcRequest): Promise<RpcBridgeResponse | ErrorResponse>;
+export async function handleIpcRequest(request: RpcStreamRequest): Promise<RpcReadyResponse | ErrorResponse>;
 export async function handleIpcRequest(request: OrchestratorRequest): Promise<OrchestratorResponse>;
 export async function handleIpcRequest(request: OrchestratorRequest): Promise<OrchestratorResponse> {
 	switch (request.type) {
@@ -103,17 +103,6 @@ export async function handleIpcRequest(request: OrchestratorRequest): Promise<Or
 		}
 
 		case "rpc": {
-			if (!request.command) {
-				const instance = supervisor.getInstance(request.instanceId);
-				if (!instance) {
-					return unknownInstanceError(request.instanceId);
-				}
-				return {
-					type: "rpc_ready",
-					ok: true,
-					instance: toInstanceSummary(instance),
-				};
-			}
 			const response = await supervisor.handleRpc(request.instanceId, request.command);
 			if (!response) {
 				return unknownInstanceError(request.instanceId);
@@ -125,6 +114,18 @@ export async function handleIpcRequest(request: OrchestratorRequest): Promise<Or
 				response,
 			};
 		}
+
+		case "rpc_stream": {
+			const instance = supervisor.getInstance(request.instanceId);
+			if (!instance) {
+				return unknownInstanceError(request.instanceId);
+			}
+			return {
+				type: "rpc_ready",
+				ok: true,
+				instance: toInstanceSummary(instance),
+			};
+		}
 	}
 }
 
diff --git a/packages/orchestrator/src/ipc/protocol.ts b/packages/orchestrator/src/ipc/protocol.ts
index ab6af46e..0f9f09ed 100644
--- a/packages/orchestrator/src/ipc/protocol.ts
+++ b/packages/orchestrator/src/ipc/protocol.ts
@@ -32,7 +32,12 @@ export interface StatusRequest {
 export interface RpcRequest {
 	type: "rpc";
 	instanceId: string;
-	command?: RpcCommand;
+	command: RpcCommand;
+}
+
+export interface RpcStreamRequest {
+	type: "rpc_stream";
+	instanceId: string;
 }
 
 export interface RequestMap {
@@ -41,6 +46,7 @@ export interface RequestMap {
 	stop: StopRequest;
 	status: StatusRequest;
 	rpc: RpcRequest;
+	rpc_stream: RpcStreamRequest;
 }
 
 export type OrchestratorRequest = RequestMap[keyof RequestMap];
@@ -101,7 +107,8 @@ export interface ResponseMap {
 	list: ListResponse;
 	stop: StopResponse;
 	status: StatusResponse;
-	rpc: RpcBridgeResponse | RpcReadyResponse;
+	rpc: RpcBridgeResponse;
+	rpc_stream: RpcReadyResponse;
 }
 
 export type OrchestratorResponse = ResponseMap[keyof ResponseMap] | ErrorResponse;
diff --git a/packages/orchestrator/src/ipc/server.ts b/packages/orchestrator/src/ipc/server.ts
index b6a6afab..1c45a6db 100644
--- a/packages/orchestrator/src/ipc/server.ts
+++ b/packages/orchestrator/src/ipc/server.ts
@@ -13,6 +13,7 @@ import {
 	type RpcClientMessage,
 	type RpcReadyResponse,
 	type RpcRequest,
+	type RpcStreamRequest,
 	type SpawnRequest,
 	type SpawnResponse,
 	type StatusRequest,
@@ -26,13 +27,8 @@ export interface IpcRequestHandler {
 	(request: ListRequest): Promise<ListResponse | ErrorResponse> | ListResponse | ErrorResponse;
 	(request: StopRequest): Promise<StopResponse | ErrorResponse> | StopResponse | ErrorResponse;
 	(request: StatusRequest): Promise<StatusResponse | ErrorResponse> | StatusResponse | ErrorResponse;
-	(
-		request: RpcRequest,
-	):
-		| Promise<RpcBridgeResponse | RpcReadyResponse | ErrorResponse>
-		| RpcBridgeResponse
-		| RpcReadyResponse
-		| ErrorResponse;
+	(request: RpcRequest): Promise<RpcBridgeResponse | ErrorResponse> | RpcBridgeResponse | ErrorResponse;
+	(request: RpcStreamRequest): Promise<RpcReadyResponse | ErrorResponse> | RpcReadyResponse | ErrorResponse;
 	(request: OrchestratorRequest): Promise<OrchestratorResponse> | OrchestratorResponse;
 	attach(
 		instanceId: string,
@@ -69,7 +65,7 @@ export async function startIpcServer(handler: IpcRequestHandler): Promise<Server
 
 			try {
 				const request = parseRequestLine(line);
-				if (request.type === "rpc" && request.command === undefined) {
+				if (request.type === "rpc_stream") {
 					const response = await handler(request);
 					if (!response.ok || response.type !== "rpc_ready" || !response.instance) {
 						socket.end(encodeMessage(response));
diff --git a/packages/orchestrator/src/supervisor.ts b/packages/orchestrator/src/supervisor.ts
index 070edadd..bb484f49 100644
--- a/packages/orchestrator/src/supervisor.ts
+++ b/packages/orchestrator/src/supervisor.ts
@@ -94,7 +94,6 @@ export class OrchestratorSupervisor {
 		| {
 				handleRpc(command: RpcCommand): Promise<RpcResponse>;
 				handleUiResponse(response: RpcExtensionUIResponse): void;
-				setHostTheme(theme: unknown): void;
 				close(): void;
 		  }
 		| undefined {
@@ -113,7 +112,6 @@ export class OrchestratorSupervisor {
 			handleUiResponse: (response) => {
 				live.rpc.handleUiResponse(response);
 			},
-			setHostTheme: (_theme) => {},
 			close: () => {
 				if (live.onUiRequest === onUiRequest) {
 					live.onUiRequest = undefined;