self-bench 0.3.0

package/dist/modal-executor.js

@@ -0,0 +1,152 @@
+import { ModalClient } from "modal";
+import { InactivityTimeoutError, RollingOutput } from "./process.js";
+export class ModalSandboxExecutor {
+    #client;
+    #config;
+    #image;
+    constructor(config, client = new ModalClient()) {
+        this.#config = config;
+        this.#client = client;
+    }
+    async run(request, options = {}) {
+        const app = await this.#client.apps.fromName(this.#config.app, {
+            createIfMissing: true,
+            ...(this.#config.environment ? { environment: this.#config.environment } : {}),
+        });
+        if (!this.#image) {
+            this.#image = this.#buildImage();
+        }
+        const sandbox = await this.#client.sandboxes.create(app, this.#image, {
+            timeoutMs: request.timeoutMs,
+            idleTimeoutMs: Math.min(request.timeoutMs, 10 * 60 * 1000),
+            cpu: request.cpu ?? 4,
+            memoryMiB: request.memoryMiB ?? 8192,
+            workdir: "/work",
+            name: sandboxName(request.runId, request.stage),
+            tags: { run_id: request.runId, stage: request.stage },
+        });
+        const abort = () => {
+            void sandbox.terminate();
+        };
+        options.signal?.addEventListener("abort", abort, { once: true });
+        try {
+            for (const file of request.files ?? []) {
+                if (typeof file.contents === "string") {
+                    await sandbox.filesystem.writeText(file.contents, file.path);
+                }
+                else {
+                    await sandbox.filesystem.writeBytes(file.contents, file.path);
+                }
+            }
+            const secrets = await this.#secrets(request.secrets);
+            const process = await sandbox.exec([...request.command], {
+                env: { ...request.environment },
+                ...(secrets.length > 0 ? { secrets } : {}),
+            });
+            await process.closeStdin();
+            const stdoutOutput = new RollingOutput();
+            const stderrOutput = new RollingOutput();
+            let inactivityTimer;
+            let inactivityError;
+            const clearInactivityTimer = () => {
+                if (inactivityTimer) {
+                    clearTimeout(inactivityTimer);
+                    inactivityTimer = undefined;
+                }
+            };
+            const armInactivityTimer = () => {
+                clearInactivityTimer();
+                if (!request.inactivityTimeoutMs) {
+                    return;
+                }
+                inactivityTimer = setTimeout(() => {
+                    inactivityError = new InactivityTimeoutError(`Modal sandbox ${sandbox.sandboxId} stage ${request.stage}`, request.inactivityTimeoutMs ?? 0);
+                    void sandbox.terminate();
+                }, request.inactivityTimeoutMs);
+                inactivityTimer.unref();
+            };
+            const consume = async (streamName, stream, output) => {
+                const reader = stream.getReader();
+                try {
+                    while (true) {
+                        const { value, done } = await reader.read();
+                        if (done) {
+                            return;
+                        }
+                        if (value !== undefined) {
+                            const chunk = typeof value === "string" ? Buffer.from(value) : Buffer.from(value);
+                            output.push(chunk);
+                            options.onProgress?.({ stream: streamName, bytes: chunk.byteLength });
+                            armInactivityTimer();
+                        }
+                    }
+                }
+                finally {
+                    reader.releaseLock();
+                }
+            };
+            armInactivityTimer();
+            const [exitCode, stdout, stderr] = await Promise.all([
+                process.wait(),
+                consume("stdout", process.stdout, stdoutOutput),
+                consume("stderr", process.stderr, stderrOutput),
+            ])
+                .then(([exitCode]) => {
+                if (inactivityError) {
+                    throw inactivityError;
+                }
+                return [exitCode, stdoutOutput.text(), stderrOutput.text()];
+            })
+                .catch((error) => {
+                throw inactivityError ?? error;
+            })
+                .finally(clearInactivityTimer);
+            const outputs = {};
+            for (const path of request.outputPaths ?? []) {
+                try {
+                    outputs[path] = await sandbox.filesystem.readBytes(path);
+                }
+                catch {
+                    // Callers validate required outputs after persisting stdout/stderr. Returning an
+                    // absent output keeps the model or command failure diagnosable.
+                }
+            }
+            return { sandboxId: sandbox.sandboxId, exitCode, stdout, stderr, outputs };
+        }
+        finally {
+            options.signal?.removeEventListener("abort", abort);
+            await sandbox.terminate().catch(() => undefined);
+        }
+    }
+    close() {
+        this.#client.close();
+    }
+    #buildImage() {
+        return this.#client.images
+            .fromRegistry(this.#config.image)
+            .dockerfileCommands([
+            "USER root",
+            "RUN apt-get update && apt-get install -y --no-install-recommends bash ca-certificates curl git jq ripgrep unzip xz-utils && rm -rf /var/lib/apt/lists/*",
+            "RUN curl -fsSL https://github.com/cli/cli/releases/download/v2.89.0/gh_2.89.0_linux_amd64.tar.gz -o /tmp/gh.tar.gz && echo 'd0422caade520530e76c1c558da47daebaa8e1203d6b7ff10ad7d6faba3490d8  /tmp/gh.tar.gz' | sha256sum -c - && tar -xzf /tmp/gh.tar.gz -C /tmp && mv /tmp/gh_2.89.0_linux_amd64/bin/gh /usr/local/bin/gh && rm -rf /tmp/gh.tar.gz /tmp/gh_2.89.0_linux_amd64",
+            "RUN npm install --global --ignore-scripts @earendil-works/pi-coding-agent@0.84.0",
+            "RUN npm install --global @openai/codex@0.146.1",
+            "ENV PATH=/root/.local/bin:/usr/local/bin:/usr/local/sbin:/usr/bin:/usr/sbin:/bin:/sbin",
+            "WORKDIR /work",
+        ]);
+    }
+    async #secrets(entries) {
+        if (!entries || Object.keys(entries).length === 0) {
+            return [];
+        }
+        return [
+            await this.#client.secrets.fromObject({ ...entries }, {
+                ...(this.#config.environment ? { environment: this.#config.environment } : {}),
+            }),
+        ];
+    }
+}
+function sandboxName(runId, stage) {
+    const suffix = crypto.randomUUID().slice(0, 8);
+    return `${runId.slice(0, 24)}-${stage.slice(0, 16)}-${suffix}`.replace(/[^a-zA-Z0-9._-]/g, "-");
+}
+//# sourceMappingURL=modal-executor.js.map

package/dist/modal-executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modal-executor.js","sourceRoot":"","sources":["../src/modal-executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,WAAW,EAAe,MAAM,OAAO,CAAC;AAE7D,OAAO,EAAE,sBAAsB,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAQrE,MAAM,OAAO,oBAAoB;IACtB,OAAO,CAAc;IACrB,OAAO,CAA2D;IAC3E,MAAM,CAAoB;IAE1B,YACE,MAAgE,EAChE,MAAM,GAAG,IAAI,WAAW,EAAE;QAE1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAuB,EAAE,UAA6B,EAAE;QAChE,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE;YAC7D,eAAe,EAAE,IAAI;YACrB,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC/E,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACnC,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,EAAE;YACpE,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;YAC1D,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;YACrB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;YACpC,OAAO,EAAE,OAAO;YAChB,IAAI,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC;YAC/C,IAAI,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE;SACtD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,GAAG,EAAE;YACjB,KAAK,OAAO,CAAC,SAAS,EAAE,CAAC;QAC3B,CAAC,CAAC;QACF,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC;YACH,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;gBACvC,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBACtC,MAAM,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC/D,CAAC;qBAAM,CAAC;oBACN,MAAM,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,EAAE;gBACvD,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,WAAW,EAAE;gBAC/B,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC3C,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;YAC3B,MAAM,YAAY,GAAG,IAAI,aAAa,EAAE,CAAC;YACzC,MAAM,YAAY,GAAG,IAAI,aAAa,EAAE,CAAC;YACzC,IAAI,eAA0D,CAAC;YAC/D,IAAI,eAAmD,CAAC;YACxD,MAAM,oBAAoB,GAAG,GAAS,EAAE;gBACtC,IAAI,eAAe,EAAE,CAAC;oBACpB,YAAY,CAAC,eAAe,CAAC,CAAC;oBAC9B,eAAe,GAAG,SAAS,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC;YACF,MAAM,kBAAkB,GAAG,GAAS,EAAE;gBACpC,oBAAoB,EAAE,CAAC;gBACvB,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;oBACjC,OAAO;gBACT,CAAC;gBACD,eAAe,GAAG,UAAU,CAAC,GAAG,EAAE;oBAChC,eAAe,GAAG,IAAI,sBAAsB,CAC1C,iBAAiB,OAAO,CAAC,SAAS,UAAU,OAAO,CAAC,KAAK,EAAE,EAC3D,OAAO,CAAC,mBAAmB,IAAI,CAAC,CACjC,CAAC;oBACF,KAAK,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3B,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;gBAChC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC,CAAC;YACF,MAAM,OAAO,GAAG,KAAK,EACnB,UAA+B,EAC/B,MAA2C,EAC3C,MAAqB,EACN,EAAE;gBACjB,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;gBAClC,IAAI,CAAC;oBACH,OAAO,IAAI,EAAE,CAAC;wBACZ,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;wBAC5C,IAAI,IAAI,EAAE,CAAC;4BACT,OAAO;wBACT,CAAC;wBACD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;4BACxB,MAAM,KAAK,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BAClF,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACnB,OAAO,CAAC,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;4BACtE,kBAAkB,EAAE,CAAC;wBACvB,CAAC;oBACH,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,MAAM,CAAC,WAAW,EAAE,CAAC;gBACvB,CAAC;YACH,CAAC,CAAC;YACF,kBAAkB,EAAE,CAAC;YACrB,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACnD,OAAO,CAAC,IAAI,EAAE;gBACd,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC;gBAC/C,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC;aAChD,CAAC;iBACC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE;gBACnB,IAAI,eAAe,EAAE,CAAC;oBACpB,MAAM,eAAe,CAAC;gBACxB,CAAC;gBACD,OAAO,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,EAAE,EAAE,YAAY,CAAC,IAAI,EAAE,CAAU,CAAC;YACvE,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBACxB,MAAM,eAAe,IAAI,KAAK,CAAC;YACjC,CAAC,CAAC;iBACD,OAAO,CAAC,oBAAoB,CAAC,CAAC;YACjC,MAAM,OAAO,GAA+B,EAAE,CAAC;YAC/C,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,WAAW,IAAI,EAAE,EAAE,CAAC;gBAC7C,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBAC3D,CAAC;gBAAC,MAAM,CAAC;oBACP,iFAAiF;oBACjF,gEAAgE;gBAClE,CAAC;YACH,CAAC;YACD,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAC7E,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACpD,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM;aACvB,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;aAChC,kBAAkB,CAAC;YAClB,WAAW;YACX,yJAAyJ;YACzJ,iXAAiX;YACjX,kFAAkF;YAClF,gDAAgD;YAChD,wFAAwF;YACxF,eAAe;SAChB,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAqD;QAClE,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClD,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO;YACL,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CACnC,EAAE,GAAG,OAAO,EAAE,EACd;gBACE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC/E,CACF;SACF,CAAC;IACJ,CAAC;CACF;AAED,SAAS,WAAW,CAAC,KAAa,EAAE,KAAa;IAC/C,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/C,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,MAAM,EAAE,CAAC,OAAO,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC;AAClG,CAAC"}

package/dist/parallel.d.ts

@@ -0,0 +1,2 @@
+export declare function parallelMap<T, R>(values: readonly T[], concurrency: number, action: (value: T) => Promise<R>): Promise<R[]>;
+//# sourceMappingURL=parallel.d.ts.map

package/dist/parallel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parallel.d.ts","sourceRoot":"","sources":["../src/parallel.ts"],"names":[],"mappings":"AAAA,wBAAsB,WAAW,CAAC,CAAC,EAAE,CAAC,EACpC,MAAM,EAAE,SAAS,CAAC,EAAE,EACpB,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,GAC/B,OAAO,CAAC,CAAC,EAAE,CAAC,CAmBd"}

package/dist/parallel.js

@@ -0,0 +1,19 @@
+export async function parallelMap(values, concurrency, action) {
+    if (!Number.isInteger(concurrency) || concurrency < 1) {
+        throw new Error("concurrency must be a positive integer");
+    }
+    const output = new Array(values.length);
+    let cursor = 0;
+    await Promise.all(Array.from({ length: Math.min(concurrency, values.length) }, async () => {
+        while (cursor < values.length) {
+            const index = cursor;
+            cursor += 1;
+            const value = values[index];
+            if (value !== undefined) {
+                output[index] = await action(value);
+            }
+        }
+    }));
+    return output;
+}
+//# sourceMappingURL=parallel.js.map

package/dist/parallel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parallel.js","sourceRoot":"","sources":["../src/parallel.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,MAAoB,EACpB,WAAmB,EACnB,MAAgC;IAEhC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACtD,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,KAAK,CAAI,MAAM,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;QACtE,OAAO,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,MAAM,CAAC;YACrB,MAAM,IAAI,CAAC,CAAC;YACZ,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CACH,CAAC;IACF,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/process.d.ts

@@ -0,0 +1,27 @@
+export interface CommandResult {
+    readonly stdout: string;
+    readonly stderr: string;
+    readonly exitCode: number;
+}
+export interface CommandOptions {
+    readonly cwd?: string;
+    readonly env?: NodeJS.ProcessEnv;
+    readonly input?: Uint8Array | string;
+    readonly timeoutMs?: number;
+    readonly inactivityTimeoutMs?: number;
+    readonly allowFailure?: boolean;
+    readonly signal?: AbortSignal;
+    readonly onOutput?: (stream: "stdout" | "stderr", chunk: Uint8Array) => void;
+}
+export declare class InactivityTimeoutError extends Error {
+    readonly scope: string;
+    readonly timeoutMs: number;
+    constructor(scope: string, timeoutMs: number);
+}
+export declare function runCommand(command: string, args: readonly string[], options?: CommandOptions): Promise<CommandResult>;
+export declare class RollingOutput {
+    #private;
+    push(chunk: Buffer): void;
+    text(): string;
+}
+//# sourceMappingURL=process.d.ts.map

package/dist/process.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"process.d.ts","sourceRoot":"","sources":["../src/process.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC;IACjC,QAAQ,CAAC,KAAK,CAAC,EAAE,UAAU,GAAG,MAAM,CAAC;IACrC,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IACtC,QAAQ,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC;IAChC,QAAQ,CAAC,MAAM,CAAC,EAAE,WAAW,CAAC;IAC9B,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,QAAQ,GAAG,QAAQ,EAAE,KAAK,EAAE,UAAU,KAAK,IAAI,CAAC;CAC9E;AAED,qBAAa,sBAAuB,SAAQ,KAAK;IAE7C,QAAQ,CAAC,KAAK,EAAE,MAAM;IACtB,QAAQ,CAAC,SAAS,EAAE,MAAM;gBADjB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,MAAM;CAK7B;AAED,wBAAsB,UAAU,CAC9B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,SAAS,MAAM,EAAE,EACvB,OAAO,GAAE,cAAmB,GAC3B,OAAO,CAAC,aAAa,CAAC,CA6FxB;AAED,qBAAa,aAAa;;IAMxB,IAAI,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IA0BzB,IAAI,IAAI,MAAM;CAIf"}

package/dist/process.js

@@ -0,0 +1,136 @@
+import { spawn } from "node:child_process";
+export class InactivityTimeoutError extends Error {
+    scope;
+    timeoutMs;
+    constructor(scope, timeoutMs) {
+        super(`${scope} produced no output for ${timeoutMs}ms`);
+        this.scope = scope;
+        this.timeoutMs = timeoutMs;
+        this.name = "InactivityTimeoutError";
+    }
+}
+export async function runCommand(command, args, options = {}) {
+    return await new Promise((resolve, reject) => {
+        const child = spawn(command, args, {
+            cwd: options.cwd,
+            env: options.env ?? process.env,
+            stdio: [options.input === undefined ? "ignore" : "pipe", "pipe", "pipe"],
+        });
+        const stdout = new RollingOutput();
+        const stderr = new RollingOutput();
+        if (!child.stdout || !child.stderr) {
+            reject(new Error(`failed to capture ${command} output`));
+            return;
+        }
+        let inactivityTimer;
+        let forceKillTimer;
+        let inactivityError;
+        const clearInactivityTimer = () => {
+            if (inactivityTimer) {
+                clearTimeout(inactivityTimer);
+                inactivityTimer = undefined;
+            }
+        };
+        const terminate = () => {
+            child.kill("SIGTERM");
+            forceKillTimer ??= setTimeout(() => child.kill("SIGKILL"), 5_000);
+            forceKillTimer.unref();
+        };
+        const armInactivityTimer = () => {
+            clearInactivityTimer();
+            if (!options.inactivityTimeoutMs) {
+                return;
+            }
+            inactivityTimer = setTimeout(() => {
+                inactivityError = new InactivityTimeoutError(`${command} ${args.join(" ")}`, options.inactivityTimeoutMs ?? 0);
+                terminate();
+            }, options.inactivityTimeoutMs);
+            inactivityTimer.unref();
+        };
+        const capture = (stream, output, chunk) => {
+            output.push(chunk);
+            options.onOutput?.(stream, chunk);
+            armInactivityTimer();
+        };
+        child.stdout.on("data", (chunk) => capture("stdout", stdout, chunk));
+        child.stderr.on("data", (chunk) => capture("stderr", stderr, chunk));
+        child.on("error", reject);
+        const abort = () => terminate();
+        if (options.signal?.aborted) {
+            abort();
+        }
+        else {
+            options.signal?.addEventListener("abort", abort, { once: true });
+        }
+        const timeout = options.timeoutMs
+            ? setTimeout(() => terminate(), options.timeoutMs)
+            : undefined;
+        timeout?.unref();
+        armInactivityTimer();
+        child.on("close", (exitCode, signal) => {
+            options.signal?.removeEventListener("abort", abort);
+            clearInactivityTimer();
+            if (forceKillTimer) {
+                clearTimeout(forceKillTimer);
+            }
+            if (timeout) {
+                clearTimeout(timeout);
+            }
+            const result = {
+                stdout: stdout.text(),
+                stderr: stderr.text(),
+                exitCode: exitCode ?? (signal ? 128 : 1),
+            };
+            if (inactivityError) {
+                reject(inactivityError);
+                return;
+            }
+            if (result.exitCode !== 0 && !options.allowFailure) {
+                const detail = result.stderr.trim() || result.stdout.trim();
+                reject(new Error(`${command} exited ${result.exitCode}${detail ? `: ${detail}` : ""}`));
+                return;
+            }
+            resolve(result);
+        });
+        if (child.stdin && options.input !== undefined) {
+            child.stdin.end(options.input);
+        }
+    });
+}
+export class RollingOutput {
+    #limit = 8 * 1024 * 1024;
+    #chunks = [];
+    #size = 0;
+    #truncated = false;
+    push(chunk) {
+        if (chunk.byteLength >= this.#limit) {
+            this.#chunks.length = 0;
+            this.#chunks.push(chunk.subarray(chunk.byteLength - this.#limit));
+            this.#size = this.#limit;
+            this.#truncated = true;
+            return;
+        }
+        this.#chunks.push(chunk);
+        this.#size += chunk.byteLength;
+        while (this.#size > this.#limit) {
+            const first = this.#chunks.shift();
+            if (!first) {
+                break;
+            }
+            const overflow = this.#size - this.#limit;
+            if (first.byteLength > overflow) {
+                this.#chunks.unshift(first.subarray(overflow));
+                this.#size -= overflow;
+            }
+            else {
+                this.#size -= first.byteLength;
+            }
+            this.#truncated = true;
+        }
+    }
+    text() {
+        const value = Buffer.concat(this.#chunks, this.#size).toString("utf8");
+        return this.#truncated ? `[selfbench: earlier output truncated]\n${value}` : value;
+    }
+}
+//# sourceMappingURL=process.js.map

package/dist/process.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"process.js","sourceRoot":"","sources":["../src/process.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAmB3C,MAAM,OAAO,sBAAuB,SAAQ,KAAK;IAEpC;IACA;IAFX,YACW,KAAa,EACb,SAAiB;QAE1B,KAAK,CAAC,GAAG,KAAK,2BAA2B,SAAS,IAAI,CAAC,CAAC;QAH/C,UAAK,GAAL,KAAK,CAAQ;QACb,cAAS,GAAT,SAAS,CAAQ;QAG1B,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;IACvC,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAe,EACf,IAAuB,EACvB,UAA0B,EAAE;IAE5B,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE;YACjC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG;YAC/B,KAAK,EAAE,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SACzE,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,KAAK,CAAC,qBAAqB,OAAO,SAAS,CAAC,CAAC,CAAC;YACzD,OAAO;QACT,CAAC;QACD,IAAI,eAA2C,CAAC;QAChD,IAAI,cAA0C,CAAC;QAC/C,IAAI,eAAmD,CAAC;QACxD,MAAM,oBAAoB,GAAG,GAAS,EAAE;YACtC,IAAI,eAAe,EAAE,CAAC;gBACpB,YAAY,CAAC,eAAe,CAAC,CAAC;gBAC9B,eAAe,GAAG,SAAS,CAAC;YAC9B,CAAC;QACH,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,GAAS,EAAE;YAC3B,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtB,cAAc,KAAK,UAAU,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;YAClE,cAAc,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC,CAAC;QACF,MAAM,kBAAkB,GAAG,GAAS,EAAE;YACpC,oBAAoB,EAAE,CAAC;YACvB,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;gBACjC,OAAO;YACT,CAAC;YACD,eAAe,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,eAAe,GAAG,IAAI,sBAAsB,CAC1C,GAAG,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAC9B,OAAO,CAAC,mBAAmB,IAAI,CAAC,CACjC,CAAC;gBACF,SAAS,EAAE,CAAC;YACd,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;YAChC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,CAAC,MAA2B,EAAE,MAAqB,EAAE,KAAa,EAAQ,EAAE;YAC1F,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnB,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAClC,kBAAkB,EAAE,CAAC;QACvB,CAAC,CAAC;QACF,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;QAC7E,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;QAC7E,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE1B,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,SAAS,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;YAC5B,KAAK,EAAE,CAAC;QACV,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,SAAS;YAC/B,CAAC,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,SAAS,EAAE,EAAE,OAAO,CAAC,SAAS,CAAC;YAClD,CAAC,CAAC,SAAS,CAAC;QACd,OAAO,EAAE,KAAK,EAAE,CAAC;QACjB,kBAAkB,EAAE,CAAC;QAErB,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE;YACrC,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACpD,oBAAoB,EAAE,CAAC;YACvB,IAAI,cAAc,EAAE,CAAC;gBACnB,YAAY,CAAC,cAAc,CAAC,CAAC;YAC/B,CAAC;YACD,IAAI,OAAO,EAAE,CAAC;gBACZ,YAAY,CAAC,OAAO,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,MAAM,GAAG;gBACb,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;gBACrB,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;gBACrB,QAAQ,EAAE,QAAQ,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;aACzC,CAAC;YACF,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,CAAC,eAAe,CAAC,CAAC;gBACxB,OAAO;YACT,CAAC;YACD,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;gBACnD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5D,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,OAAO,WAAW,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBACxF,OAAO;YACT,CAAC;YACD,OAAO,CAAC,MAAM,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,IAAI,KAAK,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC/C,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACjC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,OAAO,aAAa;IACf,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;IACzB,OAAO,GAAa,EAAE,CAAC;IAChC,KAAK,GAAG,CAAC,CAAC;IACV,UAAU,GAAG,KAAK,CAAC;IAEnB,IAAI,CAAC,KAAa;QAChB,IAAI,KAAK,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACpC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;YACzB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC;QAC/B,OAAO,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM;YACR,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;YAC1C,IAAI,KAAK,CAAC,UAAU,GAAG,QAAQ,EAAE,CAAC;gBAChC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAC/C,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC;YACzB,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC;YACjC,CAAC;YACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;IACH,CAAC;IAED,IAAI;QACF,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACvE,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,0CAA0C,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IACrF,CAAC;CACF"}

package/dist/provenance.d.ts

@@ -0,0 +1,22 @@
+export type SessionProvenanceFormat = "codex" | "claude-code" | "pi" | "generic";
+export type ProvenanceFormat = SessionProvenanceFormat | "github-pull-request";
+interface ProvenanceMessageBase {
+    readonly sessionId: string;
+    readonly messageIndex: number;
+    readonly content: string;
+}
+export type ProvenanceMessage = (ProvenanceMessageBase & {
+    readonly sourceType: SessionProvenanceFormat;
+}) | (ProvenanceMessageBase & {
+    readonly sourceType: "github-pull-request";
+    readonly sourcePr: number;
+    readonly sourceUrl: string;
+});
+export declare function collectRepositoryProvenance(repositoryPath: string, homeDirectory: string): Promise<ProvenanceMessage[]>;
+export declare function collectGitHubPullRequestProvenance(repositoryUrl: string): Promise<ProvenanceMessage[]>;
+export declare function extractGitHubPullRequestProvenance(raw: string, repositoryUrl: string): ProvenanceMessage[];
+export declare function assertProvenanceMatchesPullRequest(message: ProvenanceMessage, sourcePr: number, sourceUrl: string): void;
+export declare function extractProvenanceMessages(raw: string, format?: SessionProvenanceFormat | "auto", fallbackSessionId?: string): ProvenanceMessage[];
+export declare function redactSecrets(value: string): string;
+export {};
+//# sourceMappingURL=provenance.d.ts.map

package/dist/provenance.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provenance.d.ts","sourceRoot":"","sources":["../src/provenance.ts"],"names":[],"mappings":"AAKA,MAAM,MAAM,uBAAuB,GAAG,OAAO,GAAG,aAAa,GAAG,IAAI,GAAG,SAAS,CAAC;AACjF,MAAM,MAAM,gBAAgB,GAAG,uBAAuB,GAAG,qBAAqB,CAAC;AAE/E,UAAU,qBAAqB;IAC7B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,MAAM,iBAAiB,GACzB,CAAC,qBAAqB,GAAG;IAAE,QAAQ,CAAC,UAAU,EAAE,uBAAuB,CAAA;CAAE,CAAC,GAC1E,CAAC,qBAAqB,GAAG;IACvB,QAAQ,CAAC,UAAU,EAAE,qBAAqB,CAAC;IAC3C,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;CAC5B,CAAC,CAAC;AA8BP,wBAAsB,2BAA2B,CAC/C,cAAc,EAAE,MAAM,EACtB,aAAa,EAAE,MAAM,GACpB,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAoB9B;AAED,wBAAsB,kCAAkC,CACtD,aAAa,EAAE,MAAM,GACpB,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAe9B;AAED,wBAAgB,kCAAkC,CAChD,GAAG,EAAE,MAAM,EACX,aAAa,EAAE,MAAM,GACpB,iBAAiB,EAAE,CAkCrB;AAED,wBAAgB,kCAAkC,CAChD,OAAO,EAAE,iBAAiB,EAC1B,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,GAChB,IAAI,CASN;AAED,wBAAgB,yBAAyB,CACvC,GAAG,EAAE,MAAM,EACX,MAAM,GAAE,uBAAuB,GAAG,MAAe,EACjD,iBAAiB,SAAY,GAC5B,iBAAiB,EAAE,CAmBrB;AAmND,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAyBnD"}