@automatalabs/acp-server 0.0.0 → 0.2.0

package/dist/raw-rpc.js

@@ -0,0 +1,127 @@
+import { RequestError, } from "@agentclientprotocol/sdk";
+/** Minimal bidirectional JSON-RPC peer used only by router-owned discovery connections. */
+export class RawRpcPeer {
+    handler;
+    reader;
+    writer;
+    pending = new Map();
+    incoming = new Map();
+    nextRequestId = 0;
+    writeQueue = Promise.resolve();
+    closeReason;
+    closed;
+    constructor(stream, handler) {
+        this.handler = handler;
+        this.reader = stream.readable.getReader();
+        this.writer = stream.writable.getWriter();
+        this.closed = this.receive().finally(() => {
+            const reason = this.closeReason ?? new Error("ACP JSON-RPC connection closed");
+            for (const pending of this.pending.values())
+                pending.reject(reason);
+            this.pending.clear();
+            for (const controller of this.incoming.values())
+                controller.abort(reason);
+            this.incoming.clear();
+        });
+        this.closed.catch(() => { });
+    }
+    request(method, params, signal) {
+        if (signal?.aborted)
+            return Promise.reject(signal.reason);
+        const id = `agentprism-${(this.nextRequestId += 1)}`;
+        const message = {
+            jsonrpc: "2.0",
+            id,
+            method,
+            ...(params === undefined ? {} : { params }),
+        };
+        const onAbort = () => {
+            void this.notify("$/cancel_request", { requestId: id });
+        };
+        signal?.addEventListener("abort", onAbort, { once: true });
+        return new Promise((resolve, reject) => {
+            this.pending.set(id, { resolve, reject });
+            void this.write(message).catch((error) => {
+                this.pending.delete(id);
+                reject(error);
+            });
+        }).finally(() => signal?.removeEventListener("abort", onAbort));
+    }
+    notify(method, params) {
+        return this.write({
+            jsonrpc: "2.0",
+            method,
+            ...(params === undefined ? {} : { params }),
+        });
+    }
+    async close(reason) {
+        this.closeReason = reason;
+        await Promise.allSettled([
+            this.reader.cancel(reason),
+            this.writeQueue.then(() => this.writer.close()),
+        ]);
+        await this.closed.catch(() => { });
+        try {
+            this.writer.releaseLock();
+        }
+        catch { }
+    }
+    write(message) {
+        const operation = this.writeQueue.then(() => this.writer.write(message));
+        this.writeQueue = operation.catch(() => { });
+        return operation;
+    }
+    async receive() {
+        try {
+            while (true) {
+                const item = await this.reader.read();
+                if (item.done)
+                    return;
+                const message = item.value;
+                if ("result" in message || "error" in message) {
+                    const pending = this.pending.get(message.id);
+                    if (!pending)
+                        continue;
+                    this.pending.delete(message.id);
+                    pending.resolve("result" in message ? { result: message.result } : { error: message.error });
+                    continue;
+                }
+                if (!("id" in message)) {
+                    if (message.method === "$/cancel_request") {
+                        const requestId = cancelRequestId(message.params);
+                        if (requestId !== undefined)
+                            this.incoming.get(requestId)?.abort(RequestError.requestCancelled());
+                        continue;
+                    }
+                    void Promise.resolve(this.handler.notification(message)).catch(() => { });
+                    continue;
+                }
+                this.dispatchRequest(message);
+            }
+        }
+        finally {
+            this.reader.releaseLock();
+        }
+    }
+    dispatchRequest(message) {
+        const controller = new AbortController();
+        this.incoming.set(message.id, controller);
+        void Promise.resolve()
+            .then(() => this.handler.request(message, controller.signal))
+            .catch((error) => ({ error: errorResponse(error) }))
+            .then((result) => this.write({ jsonrpc: "2.0", id: message.id, ...result }))
+            .catch(() => { })
+            .finally(() => this.incoming.delete(message.id));
+    }
+}
+export function errorResponse(error) {
+    if (error instanceof RequestError)
+        return error.toErrorResponse();
+    return RequestError.internalError(undefined, error instanceof Error ? error.message : String(error)).toErrorResponse();
+}
+function cancelRequestId(params) {
+    if (params === null || typeof params !== "object" || Array.isArray(params))
+        return undefined;
+    const value = params.requestId;
+    return typeof value === "string" || typeof value === "number" || value === null ? value : undefined;
+}

package/dist/server.d.ts

@@ -0,0 +1,18 @@
+import { type Stream } from "@agentclientprotocol/sdk";
+import type { CustomBackendConfig } from "@automatalabs/acp-agents";
+import { type BackendTarget } from "./backends.js";
+export interface ServeAcpServerOptions {
+    /** Outer ACP stream. Defaults to process stdin/stdout. */
+    stream?: Stream;
+    /** Programmatic custom backends merged over AGENTPRISM_BACKENDS. */
+    backends?: Record<string, CustomBackendConfig>;
+    /** Exact backend targets, primarily for embedding and deterministic tests. */
+    targets?: readonly BackendTarget[];
+    /** Package version advertised on discovery connections. */
+    version?: string;
+    /** Closes the active outer and downstream connections when aborted. */
+    signal?: AbortSignal;
+}
+/** Serve one connection-pinned AgentPrism ACP V1 stream until either side closes. */
+export declare function serveAcpServer(options?: ServeAcpServerOptions): Promise<void>;
+//# sourceMappingURL=server.d.ts.map

package/dist/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AACA,OAAO,EAYL,KAAK,MAAM,EACZ,MAAM,0BAA0B,CAAC;AAElC,OAAO,KAAK,EAAE,mBAAmB,EAAwB,MAAM,0BAA0B,CAAC;AAC1F,OAAO,EAAyB,KAAK,aAAa,EAAE,MAAM,eAAe,CAAC;AAgB1E,MAAM,WAAW,qBAAqB;IACpC,0DAA0D;IAC1D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,oEAAoE;IACpE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;IAC/C,8EAA8E;IAC9E,OAAO,CAAC,EAAE,SAAS,aAAa,EAAE,CAAC;IACnC,2DAA2D;IAC3D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,uEAAuE;IACvE,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,qFAAqF;AACrF,wBAAsB,cAAc,CAAC,OAAO,GAAE,qBAA0B,GAAG,OAAO,CAAC,IAAI,CAAC,CAiEvF"}

package/dist/server.js

@@ -0,0 +1,372 @@
+import { readFileSync } from "node:fs";
+import { PROTOCOL_VERSION, RequestError, methods, ndJsonStream, } from "@agentclientprotocol/sdk";
+import { Readable, Writable } from "node:stream";
+import { resolveBackendTargets } from "./backends.js";
+import { ACP_BACKENDS_PROBE_METHOD, assertSessionBackend, discoveryInitializeResponse, isRecord, mergeBackendInitializeResponse, parseProbeBackendsParams, parseRouterInitialize, } from "./protocol.js";
+import { RawRpcPeer, errorResponse } from "./raw-rpc.js";
+/** Serve one connection-pinned AgentPrism ACP V1 stream until either side closes. */
+export async function serveAcpServer(options = {}) {
+    const stream = options.stream ?? stdioStream();
+    const targets = options.targets ? [...options.targets] : resolveBackendTargets({ backends: options.backends });
+    const targetsById = new Map(targets.map((target) => [target.id, target]));
+    if (targetsById.size !== targets.length)
+        throw new Error("ACP backend target ids must be unique");
+    const outerReader = stream.readable.getReader();
+    const first = await readFirst(outerReader, options.signal);
+    if (!first) {
+        outerReader.releaseLock();
+        return;
+    }
+    const remainingReadable = readableFromReader(outerReader);
+    const firstRequest = asInitializeRequest(first);
+    if (!firstRequest) {
+        await rejectFirstMessage(stream.writable, first);
+        await remainingReadable.cancel();
+        return;
+    }
+    let parsed;
+    try {
+        parsed = parseRouterInitialize(firstRequest.params);
+    }
+    catch (error) {
+        await writeResponseAndClose(stream.writable, {
+            jsonrpc: "2.0",
+            id: firstRequest.id,
+            error: errorResponse(error),
+        });
+        await remainingReadable.cancel();
+        return;
+    }
+    if (parsed.selection.mode === "discovery") {
+        await serveDiscoveryConnection({ readable: remainingReadable, writable: stream.writable }, firstRequest, parsed.request, targets, options.version ?? packageVersion(), options.signal);
+        return;
+    }
+    const target = targetsById.get(parsed.selection.backend);
+    if (!target) {
+        await writeResponseAndClose(stream.writable, {
+            jsonrpc: "2.0",
+            id: firstRequest.id,
+            error: RequestError.invalidParams(undefined, `unknown AgentPrism ACP backend ${JSON.stringify(parsed.selection.backend)}`).toErrorResponse(),
+        });
+        await remainingReadable.cancel();
+        return;
+    }
+    await serveBackendConnection({ readable: remainingReadable, writable: stream.writable }, firstRequest, target, options.signal);
+}
+function stdioStream() {
+    return ndJsonStream(Writable.toWeb(process.stdout), Readable.toWeb(process.stdin));
+}
+async function serveDiscoveryConnection(stream, initializeMessage, initializeRequest, targets, version, signal) {
+    const writer = stream.writable.getWriter();
+    await writer.write({
+        jsonrpc: "2.0",
+        id: initializeMessage.id,
+        result: discoveryInitializeResponse(version),
+    });
+    writer.releaseLock();
+    let peer;
+    const handler = {
+        request: async (message, requestSignal) => {
+            if (message.method !== ACP_BACKENDS_PROBE_METHOD) {
+                return { error: RequestError.methodNotFound(message.method).toErrorResponse() };
+            }
+            try {
+                const params = parseProbeBackendsParams(message.params);
+                return {
+                    result: await probeBackends(targets, initializeRequest, params, peer, requestSignal),
+                };
+            }
+            catch (error) {
+                return { error: errorResponse(error) };
+            }
+        },
+        notification: () => { },
+    };
+    peer = new RawRpcPeer(stream, handler);
+    const onAbort = () => void peer.close(signal?.reason);
+    signal?.addEventListener("abort", onAbort, { once: true });
+    try {
+        await peer.closed;
+    }
+    finally {
+        signal?.removeEventListener("abort", onAbort);
+        await peer.close();
+    }
+}
+async function probeBackends(targets, initializeRequest, params, outer, signal) {
+    const backends = await Promise.all(targets.map((target) => probeBackend(target, initializeRequest, params, outer, signal)));
+    return { backends };
+}
+async function probeBackend(target, initializeRequest, params, outer, signal) {
+    let downstream;
+    let peer;
+    let sessionId;
+    let stage = "initialize";
+    try {
+        signal.throwIfAborted();
+        downstream = await target.open();
+        peer = new RawRpcPeer(downstream.stream, {
+            request: (message, requestSignal) => outer.request(message.method, message.params, requestSignal),
+            notification: (message) => outer.notify(message.method, message.params),
+        });
+        const initialized = await peer.request(methods.agent.initialize, initializeRequest, signal);
+        if ("error" in initialized)
+            throw new RequestError(initialized.error.code, initialized.error.message, initialized.error.data);
+        const initializeResponse = requireInitializeResponse(initialized.result);
+        stage = "session/new";
+        const created = await peer.request(methods.agent.session.new, params, signal);
+        if ("error" in created)
+            throw new RequestError(created.error.code, created.error.message, created.error.data);
+        const sessionResponse = requireNewSessionResponse(created.result);
+        sessionId = sessionResponse.sessionId;
+        return {
+            id: target.id,
+            name: target.name,
+            available: true,
+            ...(initializeResponse.agentInfo === undefined ? {} : { agentInfo: initializeResponse.agentInfo }),
+            ...(initializeResponse.agentCapabilities === undefined
+                ? {}
+                : { agentCapabilities: initializeResponse.agentCapabilities }),
+            ...(sessionResponse.modes === undefined ? {} : { modes: sessionResponse.modes }),
+            ...(sessionResponse.configOptions === undefined ? {} : { configOptions: sessionResponse.configOptions }),
+            ...(initializeResponse._meta === undefined ? {} : { initializeMeta: initializeResponse._meta }),
+            ...(sessionResponse._meta === undefined ? {} : { sessionMeta: sessionResponse._meta }),
+        };
+    }
+    catch (error) {
+        return {
+            id: target.id,
+            name: target.name,
+            available: false,
+            stage,
+            error: error instanceof Error ? error.message : String(error),
+        };
+    }
+    finally {
+        if (peer && sessionId) {
+            await peer.request(methods.agent.session.close, { sessionId }).catch(() => undefined);
+        }
+        await peer?.close();
+        await downstream?.close();
+    }
+}
+async function serveBackendConnection(outerStream, initializeMessage, target, signal) {
+    let downstream;
+    const outerWriter = outerStream.writable.getWriter();
+    const outerReader = outerStream.readable.getReader();
+    let innerWriter;
+    let innerReader;
+    let initializeResponded = false;
+    const onAbort = () => {
+        void outerReader.cancel(signal?.reason);
+        void innerReader?.cancel(signal?.reason);
+    };
+    signal?.addEventListener("abort", onAbort, { once: true });
+    try {
+        downstream = await target.open();
+        innerWriter = downstream.stream.writable.getWriter();
+        innerReader = downstream.stream.readable.getReader();
+        await innerWriter.write(initializeMessage);
+        const initialized = await readInitializeResponse(innerReader, outerWriter, initializeMessage.id);
+        if ("error" in initialized) {
+            await outerWriter.write(initialized);
+            initializeResponded = true;
+            return;
+        }
+        const response = requireInitializeResponse(initialized.result);
+        await outerWriter.write({
+            ...initialized,
+            result: mergeBackendInitializeResponse(response, target.id),
+        });
+        initializeResponded = true;
+        const clientToBackend = pumpClientToBackend(outerReader, innerWriter, outerWriter, target.id, signal);
+        const backendToClient = pumpUnchanged(innerReader, outerWriter, signal);
+        await Promise.race([clientToBackend, backendToClient]);
+        await Promise.allSettled([
+            outerReader.cancel(),
+            innerReader.cancel(),
+            clientToBackend,
+            backendToClient,
+        ]);
+    }
+    catch (error) {
+        if (!initializeResponded && !signal?.aborted) {
+            await outerWriter.write({
+                jsonrpc: "2.0",
+                id: initializeMessage.id,
+                error: errorResponse(error),
+            }).catch(() => undefined);
+            initializeResponded = true;
+        }
+        else if (!signal?.aborted) {
+            throw error;
+        }
+    }
+    finally {
+        signal?.removeEventListener("abort", onAbort);
+        await Promise.allSettled([
+            innerWriter?.close() ?? Promise.resolve(),
+            outerWriter.close(),
+        ]);
+        try {
+            innerReader?.releaseLock();
+        }
+        catch { }
+        try {
+            outerReader.releaseLock();
+        }
+        catch { }
+        try {
+            innerWriter?.releaseLock();
+        }
+        catch { }
+        try {
+            outerWriter.releaseLock();
+        }
+        catch { }
+        await downstream?.close();
+    }
+}
+async function readInitializeResponse(reader, outerWriter, initializeId) {
+    while (true) {
+        const item = await reader.read();
+        if (item.done)
+            throw new Error("ACP backend closed before initialize completed");
+        const message = item.value;
+        if (("result" in message || "error" in message) && message.id === initializeId)
+            return message;
+        await outerWriter.write(message);
+    }
+}
+async function pumpClientToBackend(reader, backendWriter, clientWriter, backendId, signal) {
+    while (!signal?.aborted) {
+        const item = await reader.read();
+        if (item.done)
+            return;
+        const message = item.value;
+        if ("method" in message && message.method === methods.agent.session.new && "id" in message) {
+            try {
+                assertSessionBackend(message.params, backendId);
+            }
+            catch (error) {
+                await clientWriter.write({ jsonrpc: "2.0", id: message.id, error: errorResponse(error) });
+                continue;
+            }
+        }
+        if ("method" in message && message.method.startsWith("_automatalabs/agentprism/")) {
+            if ("id" in message) {
+                await clientWriter.write({
+                    jsonrpc: "2.0",
+                    id: message.id,
+                    error: RequestError.methodNotFound(message.method).toErrorResponse(),
+                });
+            }
+            continue;
+        }
+        await backendWriter.write(message);
+    }
+}
+async function pumpUnchanged(reader, writer, signal) {
+    while (!signal?.aborted) {
+        const item = await reader.read();
+        if (item.done)
+            return;
+        await writer.write(item.value);
+    }
+}
+function requireInitializeResponse(value) {
+    if (!isRecord(value) || value.protocolVersion !== PROTOCOL_VERSION) {
+        throw new Error(`ACP backend returned an invalid protocol-${PROTOCOL_VERSION} initialize response`);
+    }
+    return value;
+}
+function requireNewSessionResponse(value) {
+    if (!isRecord(value) || typeof value.sessionId !== "string" || value.sessionId.length === 0) {
+        throw new Error("ACP backend returned an invalid session/new response");
+    }
+    return value;
+}
+function asInitializeRequest(message) {
+    return "method" in message && "id" in message && message.method === methods.agent.initialize
+        ? message
+        : undefined;
+}
+async function rejectFirstMessage(writable, message) {
+    if ("method" in message && "id" in message) {
+        await writeResponseAndClose(writable, {
+            jsonrpc: "2.0",
+            id: message.id,
+            error: RequestError.invalidRequest(undefined, "first ACP request must be initialize").toErrorResponse(),
+        });
+        return;
+    }
+    const writer = writable.getWriter();
+    await writer.close();
+    writer.releaseLock();
+}
+async function writeResponseAndClose(writable, response) {
+    const writer = writable.getWriter();
+    try {
+        await writer.write(response);
+        await writer.close();
+    }
+    finally {
+        writer.releaseLock();
+    }
+}
+async function readFirst(reader, signal) {
+    if (!signal) {
+        const item = await reader.read();
+        return item.done ? undefined : item.value;
+    }
+    if (signal.aborted)
+        return undefined;
+    return new Promise((resolve, reject) => {
+        const onAbort = () => {
+            void reader.cancel(signal.reason);
+            resolve(undefined);
+        };
+        signal.addEventListener("abort", onAbort, { once: true });
+        reader.read().then((item) => {
+            signal.removeEventListener("abort", onAbort);
+            resolve(item.done ? undefined : item.value);
+        }, (error) => {
+            signal.removeEventListener("abort", onAbort);
+            reject(error);
+        });
+    });
+}
+function readableFromReader(reader) {
+    return new ReadableStream({
+        async pull(controller) {
+            try {
+                const item = await reader.read();
+                if (item.done) {
+                    controller.close();
+                    reader.releaseLock();
+                }
+                else {
+                    controller.enqueue(item.value);
+                }
+            }
+            catch (error) {
+                controller.error(error);
+                try {
+                    reader.releaseLock();
+                }
+                catch { }
+            }
+        },
+        async cancel(reason) {
+            await reader.cancel(reason);
+            try {
+                reader.releaseLock();
+            }
+            catch { }
+        },
+    });
+}
+function packageVersion() {
+    const manifest = JSON.parse(readFileSync(new URL("../package.json", import.meta.url), "utf8"));
+    return typeof manifest.version === "string" ? manifest.version : "0.0.0";
+}

package/package.json

@@ -1,21 +1,45 @@
 {
   "name": "@automatalabs/acp-server",
-  "version": "0.0.0",
-  "description": "Bootstrap package record for AgentPrism's ACP server.",
+  "version": "0.2.0",
   "license": "Apache-2.0",
+  "engines": {
+    "node": ">=22"
+  },
   "repository": {
     "type": "git",
     "url": "git+https://github.com/agentprism/agentprism-workflows.git",
     "directory": "packages/acp-server"
   },
-  "homepage": "https://github.com/agentprism/agentprism-workflows/tree/main/packages/acp-server#readme",
-  "bugs": {
-    "url": "https://github.com/agentprism/agentprism-workflows/issues"
+  "type": "module",
+  "bin": {
+    "agentprism-acp-server": "./dist/cli.js"
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
+    }
   },
   "files": [
-    "README.md"
+    "dist"
   ],
   "publishConfig": {
     "access": "public"
+  },
+  "dependencies": {
+    "@agentclientprotocol/sdk": "^1.4.0",
+    "ws": "^8.21.3",
+    "@automatalabs/acp-agents": "1.1.0"
+  },
+  "devDependencies": {
+    "@types/ws": "^8.18.1"
+  },
+  "scripts": {
+    "build": "tsc -b",
+    "typecheck": "tsc --noEmit",
+    "test": "tsx --test \"test/**/*.test.ts\""
   }
-}
+}