@gajae-code/coding-agent 0.17.1 → 0.17.4

package/src/sdk/transport/socket.ts

@@ -177,20 +177,21 @@ export async function startSocketServe(
 		void task;
 	});
 	server.on("error", error => void close(error).catch(() => undefined));
-	await new Promise<void>((resolve, reject) => {
-		server?.once("error", reject);
-		server?.listen(options.socketPath, () => resolve());
-	});
-	const bound = await fs.lstat(options.socketPath);
-	if (!bound.isSocket()) {
-		await close();
-		throw socketFailure("socket_bind_failed");
-	}
-	ownedIdentity = { dev: bound.dev, ino: bound.ino };
 	try {
+		await new Promise<void>((resolve, reject) => {
+			server?.once("error", reject);
+			server?.listen(options.socketPath, () => resolve());
+		});
+		const bound = await fs.lstat(options.socketPath);
+		if (!bound.isSocket()) throw socketFailure("socket_bind_failed");
+		ownedIdentity = { dev: bound.dev, ino: bound.ino };
 		await fs.chmod(options.socketPath, 0o600);
 	} catch (error) {
-		await close();
+		try {
+			await close();
+		} catch (cleanupError) {
+			throw new AggregateError([error, cleanupError], "socket_serve_startup_cleanup_failed", { cause: error });
+		}
 		throw error;
 	}
 	return { close, done: done.promise };

package/src/sdk/transport/stdio.ts

@@ -1,3 +1,4 @@
+import type { Readable, Writable } from "node:stream";
 import type { ServeHandle, ServeOptions } from "./index";
 import { startRelayPair } from "./relay";
 
@@ -6,11 +7,14 @@ function writeDiagnostic(value: unknown): void {
 }
 
 /** Serves one parent-owned JSONL connection over the process standard streams. */
-export async function startStdioServe(options: ServeOptions): Promise<ServeHandle> {
+export async function startStdioServe(
+	options: ServeOptions,
+	streams: { readonly input?: Readable; readonly output?: Writable } = {},
+): Promise<ServeHandle> {
 	const pair = await startRelayPair({
 		...options,
-		downstream: process.stdin,
-		downstreamSink: process.stdout,
+		downstream: streams.input ?? process.stdin,
+		downstreamSink: streams.output ?? process.stdout,
 		onTransportError: writeDiagnostic,
 	});
 	return { close: pair.close, done: pair.done };