@post-print/agent-test 0.3.11 → 0.3.13

package/dist/viewer/server.js

@@ -0,0 +1,147 @@
+import { createServer } from "node:http";
+import { DEFAULT_VIEWER_WORKERS } from "../worker-pool.js";
+import { encodeViewerEvent } from "./events.js";
+import { createLiveViewerRunner } from "./live-runner.js";
+import { renderViewerPage } from "./page.js";
+import { createViewerRunController, } from "./run-controller.js";
+const VIEWER_HOST = "127.0.0.1";
+export async function listenViewer(options) {
+    const controller = createViewerRunController({
+        catalog: options.catalog,
+        runner: options.runner ?? createLiveViewerRunner(options),
+        maxParallelAgents: options.workers ?? DEFAULT_VIEWER_WORKERS,
+    });
+    const server = createServer((req, res) => {
+        void handleViewerRequest(req, res, options.catalog, controller);
+    });
+    await new Promise((resolveListen, reject) => {
+        server.once("error", reject);
+        server.listen(options.port ?? 0, VIEWER_HOST, () => resolveListen());
+    });
+    const addr = server.address();
+    if (!addr || typeof addr === "string") {
+        server.close();
+        throw new Error("viewer has no port");
+    }
+    return {
+        url: `http://${VIEWER_HOST}:${addr.port}/`,
+        close: () => new Promise((resolveClose, rejectClose) => {
+            const active = controller.activeRunId();
+            if (active) {
+                controller.cancel(active);
+            }
+            server.close((error) => {
+                if (error) {
+                    rejectClose(error);
+                    return;
+                }
+                resolveClose();
+            });
+        }),
+    };
+}
+async function handleViewerRequest(req, res, catalog, controller) {
+    const url = new URL(req.url ?? "/", "http://127.0.0.1");
+    const path = url.pathname;
+    if (req.method === "GET" && (path === "/" || path === "/index.html")) {
+        write(res, 200, "text/html; charset=utf-8", renderViewerPage(catalog));
+        return;
+    }
+    if (req.method === "GET" && path === "/api/catalog") {
+        writeJson(res, 200, catalog);
+        return;
+    }
+    if (req.method === "POST" && path === "/api/runs") {
+        let request;
+        try {
+            request = (await readJson(req));
+        }
+        catch (error) {
+            writeJson(res, 400, { error: error instanceof Error ? error.message : String(error) });
+            return;
+        }
+        try {
+            const started = controller.start(request);
+            writeJson(res, 202, started);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            const status = message.includes("already in progress") ? 409 : 400;
+            writeJson(res, status, { error: message });
+        }
+        return;
+    }
+    const runMatch = path.match(/^\/api\/runs\/([^/]+)\/events$/);
+    if (req.method === "GET" && runMatch?.[1]) {
+        streamEvents(res, controller, runMatch[1]);
+        return;
+    }
+    const cancelMatch = path.match(/^\/api\/runs\/([^/]+)\/cancel$/);
+    if (req.method === "POST" && cancelMatch?.[1]) {
+        const cancelled = controller.cancel(cancelMatch[1]);
+        writeJson(res, cancelled ? 200 : 404, { cancelled });
+        return;
+    }
+    res.writeHead(404);
+    res.end();
+}
+function streamEvents(res, controller, runId) {
+    const history = controller.history(runId);
+    if (!history) {
+        writeJson(res, 404, { error: "Run not found" });
+        return;
+    }
+    res.writeHead(200, {
+        "content-type": "text/event-stream; charset=utf-8",
+        "cache-control": "no-store",
+        connection: "keep-alive",
+    });
+    const writeEvent = (event) => {
+        res.write(`data: ${encodeViewerEvent(event)}\n\n`);
+        return event.type === "run_finished";
+    };
+    for (const event of history) {
+        if (writeEvent(event)) {
+            res.end();
+            return;
+        }
+    }
+    const unsubscribe = controller.subscribe(runId, (event) => {
+        if (writeEvent(event)) {
+            unsubscribe();
+            res.end();
+        }
+    });
+    reqOnClose(res, unsubscribe);
+}
+function reqOnClose(res, unsubscribe) {
+    res.on("close", unsubscribe);
+}
+async function readJson(req) {
+    const chunks = [];
+    let size = 0;
+    for await (const chunk of req) {
+        const buffer = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk);
+        size += buffer.length;
+        if (size > 1_000_000) {
+            throw new Error("Request body is too large");
+        }
+        chunks.push(buffer);
+    }
+    const raw = Buffer.concat(chunks).toString("utf8").trim();
+    if (!raw) {
+        return {};
+    }
+    return JSON.parse(raw);
+}
+function writeJson(res, status, body) {
+    write(res, status, "application/json; charset=utf-8", `${JSON.stringify(body)}\n`);
+}
+function write(res, status, contentType, body) {
+    res.writeHead(status, {
+        "content-type": contentType,
+        "cache-control": "no-store",
+    });
+    res.end(body);
+}
+//# sourceMappingURL=server.js.map

package/dist/viewer/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/viewer/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAA6C,MAAM,WAAW,CAAC;AAEpF,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAE3D,OAAO,EAAE,iBAAiB,EAAoB,MAAM,aAAa,CAAC;AAClE,OAAO,EAAE,sBAAsB,EAAgC,MAAM,kBAAkB,CAAC;AACxF,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EACN,yBAAyB,GAGzB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,GAAG,WAAW,CAAC;AAehC,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAA4B;IAC9D,MAAM,UAAU,GAAG,yBAAyB,CAAC;QAC5C,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,sBAAsB,CAAC,OAAO,CAAC;QACzD,iBAAiB,EAAE,OAAO,CAAC,OAAO,IAAI,sBAAsB;KAC5D,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QACxC,KAAK,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IACH,MAAM,IAAI,OAAO,CAAO,CAAC,aAAa,EAAE,MAAM,EAAE,EAAE;QACjD,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;IAC9B,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACvC,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACvC,CAAC;IACD,OAAO;QACN,GAAG,EAAE,UAAU,WAAW,IAAI,IAAI,CAAC,IAAI,GAAG;QAC1C,KAAK,EAAE,GAAG,EAAE,CACX,IAAI,OAAO,CAAC,CAAC,YAAY,EAAE,WAAW,EAAE,EAAE;YACzC,MAAM,MAAM,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;YACxC,IAAI,MAAM,EAAE,CAAC;gBACZ,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACtB,IAAI,KAAK,EAAE,CAAC;oBACX,WAAW,CAAC,KAAK,CAAC,CAAC;oBACnB,OAAO;gBACR,CAAC;gBACD,YAAY,EAAE,CAAC;YAChB,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC;KACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,mBAAmB,CACjC,GAAoB,EACpB,GAAmB,EACnB,OAAsB,EACtB,UAA+B;IAE/B,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,GAAG,EAAE,kBAAkB,CAAC,CAAC;IACxD,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC;IAC1B,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,aAAa,CAAC,EAAE,CAAC;QACtE,KAAK,CAAC,GAAG,EAAE,GAAG,EAAE,0BAA0B,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QACvE,OAAO;IACR,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;QACrD,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAC7B,OAAO;IACR,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACnD,IAAI,OAAyB,CAAC;QAC9B,IAAI,CAAC;YACJ,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAqB,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACvF,OAAO;QACR,CAAC;QACD,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC1C,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;YACnE,SAAS,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO;IACR,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;IAC9D,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,YAAY,CAAC,GAAG,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,OAAO;IACR,CAAC;IACD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACjE,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QACrD,OAAO;IACR,CAAC;IACD,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IACnB,GAAG,CAAC,GAAG,EAAE,CAAC;AACX,CAAC;AAED,SAAS,YAAY,CAAC,GAAmB,EAAE,UAA+B,EAAE,KAAa;IACxF,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;QAChD,OAAO;IACR,CAAC;IACD,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;QAClB,cAAc,EAAE,kCAAkC;QAClD,eAAe,EAAE,UAAU;QAC3B,UAAU,EAAE,YAAY;KACxB,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,CAAC,KAAkB,EAAW,EAAE;QAClD,GAAG,CAAC,KAAK,CAAC,SAAS,iBAAiB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACnD,OAAO,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC;IACtC,CAAC,CAAC;IACF,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACvB,GAAG,CAAC,GAAG,EAAE,CAAC;YACV,OAAO;QACR,CAAC;IACF,CAAC;IACD,MAAM,WAAW,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE;QACzD,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACvB,WAAW,EAAE,CAAC;YACd,GAAG,CAAC,GAAG,EAAE,CAAC;QACX,CAAC;IACF,CAAC,CAAC,CAAC;IACH,UAAU,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,UAAU,CAAC,GAAmB,EAAE,WAAuB;IAC/D,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;AAC9B,CAAC;AAED,KAAK,UAAU,QAAQ,CAAC,GAAoB;IAC3C,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnE,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC;QACtB,IAAI,IAAI,GAAG,SAAS,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;IAC1D,IAAI,CAAC,GAAG,EAAE,CAAC;QACV,OAAO,EAAE,CAAC;IACX,CAAC;IACD,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAY,CAAC;AACnC,CAAC;AAED,SAAS,SAAS,CAAC,GAAmB,EAAE,MAAc,EAAE,IAAa;IACpE,KAAK,CAAC,GAAG,EAAE,MAAM,EAAE,iCAAiC,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACpF,CAAC;AAED,SAAS,KAAK,CAAC,GAAmB,EAAE,MAAc,EAAE,WAAmB,EAAE,IAAY;IACpF,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE;QACrB,cAAc,EAAE,WAAW;QAC3B,eAAe,EAAE,UAAU;KAC3B,CAAC,CAAC;IACH,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACf,CAAC"}

package/dist/worker-pool.d.ts

@@ -0,0 +1,8 @@
+export declare const DEFAULT_VIEWER_WORKERS = 4;
+export declare const DEFAULT_CLI_WORKERS = 1;
+export declare const MAX_WORKERS = 32;
+/** Parse `--workers` or `AGENT_TEST_WORKERS`. Range is 1-32. */
+export declare function parseWorkerCount(value: string, label?: string): number;
+/** Run items with a fixed number of workers. Extra items wait in the queue. */
+export declare function runWorkerPool<T>(items: readonly T[], workers: number, worker: (item: T) => Promise<void>, signal?: AbortSignal): Promise<void>;
+//# sourceMappingURL=worker-pool.d.ts.map

package/dist/worker-pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-pool.d.ts","sourceRoot":"","sources":["../src/worker-pool.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,sBAAsB,IAAI,CAAC;AACxC,eAAO,MAAM,mBAAmB,IAAI,CAAC;AACrC,eAAO,MAAM,WAAW,KAAK,CAAC;AAE9B,gEAAgE;AAChE,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,SAAc,GAAG,MAAM,CAM3E;AAED,+EAA+E;AAC/E,wBAAsB,aAAa,CAAC,CAAC,EACpC,KAAK,EAAE,SAAS,CAAC,EAAE,EACnB,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,EAClC,MAAM,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,IAAI,CAAC,CAcf"}

package/dist/worker-pool.js

@@ -0,0 +1,26 @@
+export const DEFAULT_VIEWER_WORKERS = 4;
+export const DEFAULT_CLI_WORKERS = 1;
+export const MAX_WORKERS = 32;
+/** Parse `--workers` or `AGENT_TEST_WORKERS`. Range is 1-32. */
+export function parseWorkerCount(value, label = "--workers") {
+    const parsed = Number(value);
+    if (!Number.isInteger(parsed) || parsed < 1 || parsed > MAX_WORKERS) {
+        throw new Error(`${label} must be an integer 1-${MAX_WORKERS}`);
+    }
+    return parsed;
+}
+/** Run items with a fixed number of workers. Extra items wait in the queue. */
+export async function runWorkerPool(items, workers, worker, signal) {
+    const queue = [...items];
+    const workerCount = Math.max(1, Math.min(workers, Math.max(items.length, 1)));
+    await Promise.all(Array.from({ length: workerCount }, async () => {
+        while (!signal?.aborted) {
+            const item = queue.shift();
+            if (item === undefined) {
+                return;
+            }
+            await worker(item);
+        }
+    }));
+}
+//# sourceMappingURL=worker-pool.js.map

package/dist/worker-pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-pool.js","sourceRoot":"","sources":["../src/worker-pool.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC;AACxC,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AACrC,MAAM,CAAC,MAAM,WAAW,GAAG,EAAE,CAAC;AAE9B,gEAAgE;AAChE,MAAM,UAAU,gBAAgB,CAAC,KAAa,EAAE,KAAK,GAAG,WAAW;IAClE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,IAAI,MAAM,GAAG,WAAW,EAAE,CAAC;QACrE,MAAM,IAAI,KAAK,CAAC,GAAG,KAAK,yBAAyB,WAAW,EAAE,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC;AAED,+EAA+E;AAC/E,MAAM,CAAC,KAAK,UAAU,aAAa,CAClC,KAAmB,EACnB,OAAe,EACf,MAAkC,EAClC,MAAoB;IAEpB,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;IACzB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,MAAM,OAAO,CAAC,GAAG,CAChB,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,KAAK,IAAI,EAAE;QAC9C,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;YAC3B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACxB,OAAO;YACR,CAAC;YACD,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;IACF,CAAC,CAAC,CACF,CAAC;AACH,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@post-print/agent-test",
-	"version": "0.3.11",
+	"version": "0.3.13",
 	"type": "module",
 	"bin": {
 		"agent-test": "./dist/cli.js"
@@ -25,7 +25,7 @@
 		"clean": "rm -rf dist tsconfig.tsbuildinfo"
 	},
 	"dependencies": {
-		"@post-print/agent-harness": "0.3.11",
+		"@post-print/agent-harness": "0.3.13",
 		"chalk": "6"
 	},
 	"devDependencies": {