assertloom 0.0.0-stage → 0.2.0

package/dist/cli/commands/ci.js

@@ -0,0 +1,258 @@
+import { readFileSync } from "node:fs";
+import { appendFile, mkdir, readdir, readFile, rm, writeFile } from "node:fs/promises";
+import path from "node:path";
+import { Option } from "commander";
+import { createProvider, PROVIDERS } from "../../ai/index.js";
+import { Budget } from "../../ai/pricing.js";
+import { UserError } from "../../errors.js";
+import { MAX_REPAIR_ATTEMPTS } from "../../repair/index.js";
+import { cell, computeTotals, createRedactor, evaluatePolicy, FAIL_ON, renderSummary } from "../../ci/report.js";
+import { runFiles } from "../../ci/run.js";
+import { changedFiles, classify, explicitFiles, repositoryRoot, splitLines, splitList } from "../../ci/select.js";
+const { version } = JSON.parse(readFileSync(new URL("../../../package.json", import.meta.url), "utf8"));
+/** Environment variables that hold provider credentials. Their values are redacted from everything Assertloom writes. */
+export const CREDENTIAL_ENV = ["OPENAI_API_KEY", "ANTHROPIC_API_KEY", "ASSERTLOOM_API_KEY"];
+export const DEFAULT_CI_MAX_FILES = 10;
+export const DEFAULT_CI_MAX_COST_USD = 1;
+const MAX_TIMEOUT_SECONDS = 3600;
+/** Commander passes "" for inputs the action leaves empty; treat them as unset. */
+const given = (value) => (value === undefined || value.trim() === "" ? undefined : value.trim());
+function wholeNumber(value, name, min, max) {
+    const n = Number(value);
+    if (!Number.isInteger(n) || n < min || n > max)
+        throw new UserError(`${name} must be a whole number from ${min} to ${max}.`);
+    return n;
+}
+function noNewlines(value, name) {
+    if (/[\r\n\0]/.test(value))
+        throw new UserError(`${name} must not contain line breaks.`);
+    return value;
+}
+/** Only an empty directory or one holding a previous Assertloom report is reused, so nothing else is ever deleted. */
+async function prepareOutDir(outDir) {
+    const entries = await readdir(outDir).catch(() => undefined);
+    if (entries && entries.length && !entries.includes("report.json")) {
+        throw new UserError(`${outDir} exists and is not an Assertloom report directory. Choose an empty or new directory with --out.`);
+    }
+    for (const name of ["tests", "report.json", "summary.md"])
+        await rm(path.join(outDir, name), { recursive: true, force: true });
+    await mkdir(path.join(outDir, "tests"), { recursive: true });
+}
+export async function runCi(options, deps) {
+    const log = deps.log;
+    const cwd = path.resolve(given(options.workingDirectory) ?? ".");
+    const failOn = (given(options.failOn) ?? "errors");
+    if (!FAIL_ON.includes(failOn))
+        throw new UserError(`--fail-on must be one of: ${FAIL_ON.join(", ")}.`);
+    const maxFiles = wholeNumber(given(options.maxFiles) ?? String(DEFAULT_CI_MAX_FILES), "--max-files", 1, 1000);
+    const maxAttempts = wholeNumber(given(options.maxAttempts) ?? String(MAX_REPAIR_ATTEMPTS), "--max-attempts", 0, MAX_REPAIR_ATTEMPTS);
+    const timeoutSeconds = wholeNumber(given(options.timeout) ?? "120", "--timeout", 1, MAX_TIMEOUT_SECONDS);
+    const maxCost = Number(given(options.maxCost) ?? DEFAULT_CI_MAX_COST_USD);
+    if (!Number.isFinite(maxCost) || maxCost <= 0)
+        throw new UserError("--max-cost must be a positive number of US dollars.");
+    const maxTokensInput = given(options.maxTokens);
+    const maxTokens = maxTokensInput === undefined ? undefined : wholeNumber(maxTokensInput, "--max-tokens", 1, 100_000_000);
+    const outDir = path.resolve(cwd, noNewlines(given(options.out) ?? "assertloom-report", "--out"));
+    const redact = createRedactor(CREDENTIAL_ENV.map((name) => process.env[name]));
+    const base = given(options.base);
+    const fileInput = splitLines(options.files);
+    if (!base && fileInput.length === 0) {
+        throw new UserError("Nothing to select: pass --base (the pull request's base commit) or --files. In GitHub Actions, run on pull_request or set the `files` input.");
+    }
+    await prepareOutDir(outDir);
+    const report = {
+        schemaVersion: 1,
+        assertloom: version,
+        startedAt: new Date().toISOString(),
+        finishedAt: "",
+        mode: fileInput.length ? "files" : "pull-request",
+        ...(fileInput.length ? {} : { base, head: given(options.head) ?? "HEAD" }),
+        provider: options.provider,
+        model: given(options.model) ?? "",
+        limits: { maxFiles, maxCostUsd: maxCost, maxTokens: maxTokens ?? null, maxAttempts, timeoutSeconds },
+        budget: { exhausted: false, costKnown: false, spentUsd: null, tokens: { input: 0, output: 0 } },
+        totals: computeTotals([]),
+        policy: { failOn, failed: false, reasons: [] },
+        files: [],
+    };
+    let budget;
+    const write = async (files) => {
+        report.files = files;
+        report.totals = computeTotals(files);
+        if (budget) {
+            report.budget = {
+                ...report.budget,
+                costKnown: budget.priced,
+                spentUsd: budget.spentUsd ?? null,
+                tokens: { input: budget.usage.inputTokens, output: budget.usage.outputTokens },
+            };
+        }
+        report.finishedAt = new Date().toISOString();
+        await writeFile(path.join(outDir, "report.json"), redact(JSON.stringify(report, null, 2)) + "\n");
+        await writeFile(path.join(outDir, "summary.md"), redact(renderSummary(report)));
+    };
+    let candidates = [];
+    try {
+        const root = await repositoryRoot(cwd);
+        const files = fileInput.length ? explicitFiles(fileInput, root, cwd) : await changedFiles({ cwd: root, base: base, head: given(options.head) });
+        candidates = await classify(files, {
+            root,
+            include: splitList(options.include),
+            exclude: splitList(options.exclude),
+            useDefaultExcludes: options.defaultExcludes,
+        });
+        const eligible = candidates.filter((c) => !c.skip);
+        log(`Selected ${eligible.length} of ${candidates.length} changed files${eligible.length > maxFiles ? ` (processing at most ${maxFiles})` : ""}.`);
+        if (options.dryRun || eligible.length === 0) {
+            if (eligible.length === 0)
+                log("No eligible JavaScript or TypeScript source files changed.");
+            const files = candidates.map((c) => skippedReport(c, "dry-run"));
+            await write(files);
+            return finish(report, failOn, files, write, redact);
+        }
+        const provider = deps.createProvider(options.provider, { model: given(options.model), baseUrl: given(options.baseUrl) });
+        report.model = provider.model;
+        budget = new Budget(provider.model, maxCost, maxTokens);
+        if (!budget.priced && maxTokens === undefined) {
+            throw new UserError(`${provider.model} has no known price, so a dollar limit cannot be enforced. Set a shared token limit with --max-tokens.`);
+        }
+        log(`Using ${provider.name} (${provider.model}); shared limit ${budget.describeLimit()}.`);
+        const result = await runFiles({
+            root,
+            candidates,
+            provider,
+            budget,
+            maxFiles,
+            maxAttempts,
+            timeoutMs: timeoutSeconds * 1000,
+            testsDir: path.join(outDir, "tests"),
+            onProgress: write,
+            runTests: deps.runTests,
+            log: (line) => log(redact(line)),
+        });
+        report.budget.exhausted = result.budgetExhausted;
+        await write(result.files);
+        return finish(report, failOn, result.files, write, redact);
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        const hint = /_API_KEY is not set/.test(message) && process.env.GITHUB_ACTIONS ? " In the GitHub Action, pass the key from a secret with the `api-key` input." : "";
+        report.error = redact(message + hint);
+        await write(report.files.length ? report.files : candidates.map((c) => skippedReport(c, "not-processed")));
+        throw error instanceof UserError ? new UserError(report.error) : error;
+    }
+}
+/** A file that was not processed: skipped by selection, or eligible but not reached (`reason`). */
+function skippedReport(c, reason) {
+    return {
+        path: c.path,
+        ...(c.from ? { from: c.from } : {}),
+        change: c.status,
+        status: "skipped",
+        ...(c.skip
+            ? { skipReason: c.skip.reason, detail: c.skip.detail }
+            : { skipReason: reason, detail: reason === "dry-run" ? "eligible; dry run, not processed" : "eligible; not processed because the run stopped" }),
+        generated: 0,
+        accepted: 0,
+        rejected: 0,
+        repaired: 0,
+        flaggedAsSourceBugs: 0,
+        removed: 0,
+        gaveUp: 0,
+        repairRounds: 0,
+        regressions: 0,
+        durationMs: 0,
+        usage: { inputTokens: 0, outputTokens: 0 },
+        costUsd: null,
+    };
+}
+async function finish(report, failOn, files, write, redact) {
+    report.policy = { failOn, ...evaluatePolicy(failOn, files) };
+    await write(files);
+    // Callers get the same redacted content that was written to disk.
+    return JSON.parse(redact(JSON.stringify(report)));
+}
+/** Writes `key=value` pairs for GitHub Actions step outputs. Values never contain line breaks. */
+async function writeGithubOutputs(file, report, outDir) {
+    const t = report.totals;
+    const outputs = {
+        "report-dir": outDir,
+        "report-path": path.join(outDir, "report.json"),
+        "summary-path": path.join(outDir, "summary.md"),
+        "files-selected": String(t.files.selected),
+        "files-processed": String(t.files.processed),
+        "files-skipped": String(t.files.skipped),
+        "files-failed": String(t.files.failed),
+        "tests-final-passed": String(t.tests.final.passed),
+        "tests-final-failed": String(t.tests.final.failed),
+        "budget-exhausted": String(report.budget.exhausted),
+        result: report.error ? "error" : report.policy.failed ? "failed" : "passed",
+    };
+    const lines = Object.entries(outputs)
+        .filter(([, v]) => !/[\r\n]/.test(v))
+        .map(([k, v]) => `${k}=${v}\n`);
+    await appendFile(file, lines.join(""));
+}
+export function registerCiCommand(program, deps = {}) {
+    program
+        .command("ci")
+        .description("Generate, run, and repair tests for JavaScript/TypeScript files changed in a pull request; write a report and test previews")
+        .option("--base <commit>", "base commit of the pull request; files changed on --head since it are selected")
+        .option("--head <commit>", "head commit (default: HEAD)")
+        .option("--files <list>", "newline-separated files to process instead of a diff (for manual runs)")
+        .option("--include <globs>", "only consider files matching these globs (newline- or comma-separated)")
+        .option("--exclude <globs>", "skip files matching these globs, in addition to the defaults")
+        .option("--no-default-excludes", "do not skip tests, node_modules, build output, declarations, and generated files by default")
+        .option("--max-files <n>", "process at most this many files", String(DEFAULT_CI_MAX_FILES))
+        .option("--max-cost <usd>", "spending limit in US dollars for the whole run, shared by all files and repair rounds", String(DEFAULT_CI_MAX_COST_USD))
+        .option("--max-tokens <n>", "shared token limit for models without a known price (required for them)")
+        .option("--max-attempts <n>", `repair rounds per file, 0-${MAX_REPAIR_ATTEMPTS}`, String(MAX_REPAIR_ATTEMPTS))
+        .option("--timeout <seconds>", "time limit for each test run", "120")
+        .option("--out <dir>", "directory for report.json, summary.md, and tests/", "assertloom-report")
+        .addOption(new Option("--fail-on <policy>", "exit non-zero on: never, errors (files that could not be processed), or tests (also any test not passing)").default("errors"))
+        .addOption(new Option("--provider <name>", "LLM provider").choices(PROVIDERS).default("openai"))
+        .option("--model <model>", "model name (default: $ASSERTLOOM_MODEL or the provider default)")
+        .option("--base-url <url>", "server URL for --provider openai-compatible (default: $ASSERTLOOM_BASE_URL)")
+        .option("--working-directory <dir>", "directory inside the repository to run in (default: current directory)")
+        .option("--dry-run", "select files and write the report without calling a model or running tests")
+        .option("--github-summary <file>", "append the Markdown summary to this file ($GITHUB_STEP_SUMMARY)")
+        .option("--github-output <file>", "append step outputs to this file ($GITHUB_OUTPUT)")
+        .action(async (options) => {
+        const full = { createProvider, log: (line) => console.log(line), ...deps };
+        const outDir = path.resolve(given(options.workingDirectory) ?? ".", given(options.out) ?? "assertloom-report");
+        let report;
+        let failure;
+        try {
+            report = await runCi(options, full);
+        }
+        catch (error) {
+            failure = createRedactor(CREDENTIAL_ENV.map((name) => process.env[name]))(error instanceof Error ? error.message : String(error));
+            throw error;
+        }
+        finally {
+            const summary = await readFile(path.join(outDir, "summary.md"), "utf8").catch(() => undefined);
+            if (given(options.githubSummary)) {
+                // A configuration error stops the run before any report exists; still say why in the job summary.
+                const text = summary ?? `## Assertloom\n\n**Could not run:** ${cell(failure ?? "unknown error")}\n`;
+                await appendFile(options.githubSummary, text + "\n");
+            }
+            if (given(options.githubOutput)) {
+                if (report)
+                    await writeGithubOutputs(options.githubOutput, report, outDir);
+                // Partial results may exist even after an error; expose them so they are uploaded.
+                else
+                    await appendFile(options.githubOutput, `result=error\n${summary !== undefined && !/[\r\n]/.test(outDir) ? `report-dir=${outDir}\n` : ""}`);
+            }
+        }
+        const t = report.totals;
+        console.log(`\n${t.files.processed} processed, ${t.files.skipped} skipped, ${t.files.failed} failed. ` +
+            `Final tests: ${t.tests.final.passed} passed, ${t.tests.final.failed} failed, ${t.tests.final.skipped} skipped.`);
+        console.log(`Report: ${path.join(outDir, "report.json")}`);
+        if (report.policy.failed) {
+            for (const reason of report.policy.reasons)
+                console.error(`✗ ${reason} (--fail-on ${report.policy.failOn})`);
+            process.exitCode = 1;
+        }
+    });
+}
+//# sourceMappingURL=ci.js.map

package/dist/cli/commands/ci.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ci.js","sourceRoot":"","sources":["../../../src/cli/commands/ci.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvF,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,EAAgB,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,SAAS,EAA0C,MAAM,mBAAmB,CAAC;AACtG,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,cAAc,EAAE,cAAc,EAAE,OAAO,EAAE,aAAa,EAA8B,MAAM,oBAAoB,CAAC;AAC7I,OAAO,EAAE,QAAQ,EAAmB,MAAM,iBAAiB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,cAAc,EAAE,UAAU,EAAE,SAAS,EAAkB,MAAM,oBAAoB,CAAC;AAGlI,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,GAAG,CAAC,uBAAuB,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAwB,CAAC;AAE/H,yHAAyH;AACzH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,gBAAgB,EAAE,mBAAmB,EAAE,oBAAoB,CAAC,CAAC;AAE5F,MAAM,CAAC,MAAM,oBAAoB,GAAG,EAAE,CAAC;AACvC,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC;AACzC,MAAM,mBAAmB,GAAG,IAAI,CAAC;AA+BjC,mFAAmF;AACnF,MAAM,KAAK,GAAG,CAAC,KAAyB,EAAE,EAAE,CAAC,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;AAErH,SAAS,WAAW,CAAC,KAAa,EAAE,IAAY,EAAE,GAAW,EAAE,GAAW;IACxE,MAAM,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACxB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,GAAG;QAAE,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,gCAAgC,GAAG,OAAO,GAAG,GAAG,CAAC,CAAC;IAC7H,OAAO,CAAC,CAAC;AACX,CAAC;AAED,SAAS,UAAU,CAAC,KAAa,EAAE,IAAY;IAC7C,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,GAAG,IAAI,gCAAgC,CAAC,CAAC;IACzF,OAAO,KAAK,CAAC;AACf,CAAC;AAED,sHAAsH;AACtH,KAAK,UAAU,aAAa,CAAC,MAAc;IACzC,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC7D,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,SAAS,CAAC,GAAG,MAAM,iGAAiG,CAAC,CAAC;IAClI,CAAC;IACD,KAAK,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,aAAa,EAAE,YAAY,CAAC;QAAE,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/H,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;AAC/D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,KAAK,CAAC,OAAyB,EAAE,IAAoB;IACzE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;IACrB,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,GAAG,CAAC,CAAC;IACjE,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAW,CAAC;IAC7D,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,6BAA6B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACvG,MAAM,QAAQ,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC,oBAAoB,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IAC9G,MAAM,WAAW,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,MAAM,CAAC,mBAAmB,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,mBAAmB,CAAC,CAAC;IACrI,MAAM,cAAc,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,mBAAmB,CAAC,CAAC;IACzG,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,uBAAuB,CAAC,CAAC;IAC1E,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,IAAI,CAAC;QAAE,MAAM,IAAI,SAAS,CAAC,qDAAqD,CAAC,CAAC;IAC1H,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,cAAc,EAAE,cAAc,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC;IACzH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC;IACjG,MAAM,MAAM,GAAG,cAAc,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAE/E,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACjC,MAAM,SAAS,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC5C,IAAI,CAAC,IAAI,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpC,MAAM,IAAI,SAAS,CACjB,8IAA8I,CAC/I,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,CAAC,MAAM,CAAC,CAAC;IAC5B,MAAM,MAAM,GAAa;QACvB,aAAa,EAAE,CAAC;QAChB,UAAU,EAAE,OAAO;QACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,UAAU,EAAE,EAAE;QACd,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc;QACjD,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,MAAM,EAAE,CAAC;QAC1E,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE;QACjC,MAAM,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,IAAI,IAAI,EAAE,WAAW,EAAE,cAAc,EAAE;QACpG,MAAM,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE;QAC/F,MAAM,EAAE,aAAa,CAAC,EAAE,CAAC;QACzB,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;QAC9C,KAAK,EAAE,EAAE;KACV,CAAC;IAEF,IAAI,MAA0B,CAAC;IAC/B,MAAM,KAAK,GAAG,KAAK,EAAE,KAAmB,EAAE,EAAE;QAC1C,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,MAAM,CAAC,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,MAAM,GAAG;gBACd,GAAG,MAAM,CAAC,MAAM;gBAChB,SAAS,EAAE,MAAM,CAAC,MAAM;gBACxB,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,IAAI;gBACjC,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE;aAC/E,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;QAClG,MAAM,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAClF,CAAC,CAAC;IAEF,IAAI,UAAU,GAAgB,EAAE,CAAC;IACjC,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,SAAS,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,YAAY,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,IAAK,EAAE,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjJ,UAAU,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE;YACjC,IAAI;YACJ,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;YACnC,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC;YACnC,kBAAkB,EAAE,OAAO,CAAC,eAAe;SAC5C,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACnD,GAAG,CAAC,YAAY,QAAQ,CAAC,MAAM,OAAO,UAAU,CAAC,MAAM,iBAAiB,QAAQ,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,wBAAwB,QAAQ,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAElJ,IAAI,OAAO,CAAC,MAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,GAAG,CAAC,4DAA4D,CAAC,CAAC;YAC7F,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAc,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;YAC7E,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;YACnB,OAAO,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACzH,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAC9B,MAAM,GAAG,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC9C,MAAM,IAAI,SAAS,CAAC,GAAG,QAAQ,CAAC,KAAK,wGAAwG,CAAC,CAAC;QACjJ,CAAC;QACD,GAAG,CAAC,SAAS,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,KAAK,mBAAmB,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;QAE3F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC;YAC5B,IAAI;YACJ,UAAU;YACV,QAAQ;YACR,MAAM;YACN,QAAQ;YACR,WAAW;YACX,SAAS,EAAE,cAAc,GAAG,IAAI;YAChC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC;YACpC,UAAU,EAAE,KAAK;YACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;SACjC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,eAAe,CAAC;QACjD,MAAM,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,OAAO,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IAC7D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,IAAI,GAAG,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,6EAA6E,CAAC,CAAC,CAAC,EAAE,CAAC;QACpK,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;QACtC,MAAM,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC;QAC3G,MAAM,KAAK,YAAY,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IACzE,CAAC;AACH,CAAC;AAED,mGAAmG;AACnG,SAAS,aAAa,CAAC,CAAY,EAAE,MAAmC;IACtE,OAAO;QACL,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnC,MAAM,EAAE,CAAC,CAAC,MAAM;QAChB,MAAM,EAAE,SAAS;QACjB,GAAG,CAAC,CAAC,CAAC,IAAI;YACR,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE;YACtD,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,kCAAkC,CAAC,CAAC,CAAC,iDAAiD,EAAE,CAAC;QAClJ,SAAS,EAAE,CAAC;QACZ,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,CAAC;QACX,mBAAmB,EAAE,CAAC;QACtB,OAAO,EAAE,CAAC;QACV,MAAM,EAAE,CAAC;QACT,YAAY,EAAE,CAAC;QACf,WAAW,EAAE,CAAC;QACd,UAAU,EAAE,CAAC;QACb,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;QAC1C,OAAO,EAAE,IAAI;KACd,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,MAAM,CACnB,MAAgB,EAChB,MAAc,EACd,KAAmB,EACnB,KAA6C,EAC7C,MAAgC;IAEhC,MAAM,CAAC,MAAM,GAAG,EAAE,MAAM,EAAE,GAAG,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC;IAC7D,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;IACnB,kEAAkE;IAClE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAa,CAAC;AAChE,CAAC;AAED,kGAAkG;AAClG,KAAK,UAAU,kBAAkB,CAAC,IAAY,EAAE,MAAgB,EAAE,MAAc;IAC9E,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;IACxB,MAAM,OAAO,GAA2B;QACtC,YAAY,EAAE,MAAM;QACpB,aAAa,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC;QAC/C,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC;QAC/C,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC1C,iBAAiB,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC;QAC5C,eAAe,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;QACxC,cAAc,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;QACtC,oBAAoB,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;QAClD,oBAAoB,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;QAClD,kBAAkB,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC;QACnD,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;KAC5E,CAAC;IACF,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;SAClC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACpC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,OAAgB,EAAE,IAAI,GAA4B,EAAE;IACpF,OAAO;SACJ,OAAO,CAAC,IAAI,CAAC;SACb,WAAW,CAAC,6HAA6H,CAAC;SAC1I,MAAM,CAAC,iBAAiB,EAAE,gFAAgF,CAAC;SAC3G,MAAM,CAAC,iBAAiB,EAAE,6BAA6B,CAAC;SACxD,MAAM,CAAC,gBAAgB,EAAE,wEAAwE,CAAC;SAClG,MAAM,CAAC,mBAAmB,EAAE,wEAAwE,CAAC;SACrG,MAAM,CAAC,mBAAmB,EAAE,8DAA8D,CAAC;SAC3F,MAAM,CAAC,uBAAuB,EAAE,6FAA6F,CAAC;SAC9H,MAAM,CAAC,iBAAiB,EAAE,iCAAiC,EAAE,MAAM,CAAC,oBAAoB,CAAC,CAAC;SAC1F,MAAM,CAAC,kBAAkB,EAAE,uFAAuF,EAAE,MAAM,CAAC,uBAAuB,CAAC,CAAC;SACpJ,MAAM,CAAC,kBAAkB,EAAE,yEAAyE,CAAC;SACrG,MAAM,CAAC,oBAAoB,EAAE,6BAA6B,mBAAmB,EAAE,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC;SAC7G,MAAM,CAAC,qBAAqB,EAAE,8BAA8B,EAAE,KAAK,CAAC;SACpE,MAAM,CAAC,aAAa,EAAE,mDAAmD,EAAE,mBAAmB,CAAC;SAC/F,SAAS,CAAC,IAAI,MAAM,CAAC,oBAAoB,EAAE,2GAA2G,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;SAC1K,SAAS,CAAC,IAAI,MAAM,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;SAC/F,MAAM,CAAC,iBAAiB,EAAE,iEAAiE,CAAC;SAC5F,MAAM,CAAC,kBAAkB,EAAE,6EAA6E,CAAC;SACzG,MAAM,CAAC,2BAA2B,EAAE,wEAAwE,CAAC;SAC7G,MAAM,CAAC,WAAW,EAAE,4EAA4E,CAAC;SACjG,MAAM,CAAC,yBAAyB,EAAE,iEAAiE,CAAC;SACpG,MAAM,CAAC,wBAAwB,EAAE,mDAAmD,CAAC;SACrF,MAAM,CAAC,KAAK,EAAE,OAAyB,EAAE,EAAE;QAC1C,MAAM,IAAI,GAAmB,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC;QAC3F,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,mBAAmB,CAAC,CAAC;QAC/G,IAAI,MAA4B,CAAC;QACjC,IAAI,OAA2B,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,GAAG,cAAc,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAClI,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC/F,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,kGAAkG;gBAClG,MAAM,IAAI,GAAG,OAAO,IAAI,uCAAuC,IAAI,CAAC,OAAO,IAAI,eAAe,CAAC,IAAI,CAAC;gBACpG,MAAM,UAAU,CAAC,OAAO,CAAC,aAAc,EAAE,IAAI,GAAG,IAAI,CAAC,CAAC;YACxD,CAAC;YACD,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;gBAChC,IAAI,MAAM;oBAAE,MAAM,kBAAkB,CAAC,OAAO,CAAC,YAAa,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;gBAC5E,mFAAmF;;oBAC9E,MAAM,UAAU,CAAC,OAAO,CAAC,YAAa,EAAE,iBAAiB,OAAO,KAAK,SAAS,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc,MAAM,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnJ,CAAC;QACH,CAAC;QACD,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;QACxB,OAAO,CAAC,GAAG,CACT,KAAK,CAAC,CAAC,KAAK,CAAC,SAAS,eAAe,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC,KAAK,CAAC,MAAM,WAAW;YACxF,gBAAgB,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,YAAY,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,YAAY,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,WAAW,CACnH,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;QAC3D,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACzB,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO;gBAAE,OAAO,CAAC,KAAK,CAAC,KAAK,MAAM,eAAe,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;YAC7G,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/cli/commands/generate.d.ts

@@ -0,0 +1,3 @@
+import { type Command } from "commander";
+export declare const DEFAULT_MAX_COST_USD = 1;
+export declare function registerGenerateCommand(program: Command): void;

package/dist/cli/commands/generate.js

@@ -0,0 +1,201 @@
+import { rm, writeFile } from "node:fs/promises";
+import path from "node:path";
+import { Option } from "commander";
+import { createProvider, PROVIDERS } from "../../ai/index.js";
+import { Budget, estimateCost, formatUsd } from "../../ai/pricing.js";
+import { UserError } from "../../errors.js";
+import { generateTests, testPathFor, TestFileExistsError, writeTestFile } from "../../generator/index.js";
+import { MAX_REPAIR_ATTEMPTS, repairTests, summarize } from "../../repair/index.js";
+import { runJest } from "../../runner/index.js";
+import { formatCoverage, printRunResult } from "../report.js";
+import { validateSourceFile } from "../validateFile.js";
+export const DEFAULT_MAX_COST_USD = 1;
+function tokens(usage, model) {
+    const cost = estimateCost(model, usage);
+    return `${usage.inputTokens.toLocaleString()} in / ${usage.outputTokens.toLocaleString()} out${cost === undefined ? "" : `, ≈${formatUsd(cost)}`}`;
+}
+const collect = (value, previous) => [...previous, value];
+function printMocks(result) {
+    const { mocked, real, globals } = result.mocks;
+    const names = [...mocked.map((m) => m.source), ...globals.map((g) => `global ${g}`)];
+    if (names.length)
+        console.log(`✓ Mocked ${names.length} ${names.length === 1 ? "dependency" : "dependencies"}: ${names.join(", ")}`);
+    for (const m of mocked)
+        console.log(`  - ${m.source}: ${m.reason}${m.virtual ? " (virtual mock)" : ""}`);
+    if (real.length)
+        console.log(`  Kept real: ${real.map((r) => `${r.source} (${r.reason})`).join(", ")}`);
+}
+function relative(file) {
+    return path.relative(process.cwd(), file) || file;
+}
+/** Prints what changes in an existing test file before it is overwritten. Previews are scratch files. */
+function showDiff(file) {
+    if (file.includes(".test.preview."))
+        return undefined;
+    return (diff) => {
+        console.log(`\nOverwriting ${relative(file)}. Changes:`);
+        for (const line of diff)
+            console.log(`  ${line}`);
+    };
+}
+const STOP_MESSAGES = {
+    "all-passing": "all tests pass",
+    "max-attempts": "reached the repair limit",
+    regressed: "a repair broke passing tests and was reverted",
+    "no-valid-repairs": "the model proposed no usable repairs",
+    unrepairable: "the remaining failures cannot be fixed by editing tests",
+    budget: "another repair round could exceed the spending limit",
+};
+function printRepair(repair, model) {
+    for (const attempt of repair.attempts) {
+        const counts = ["fix", "skip", "remove"]
+            .map((action) => [action, attempt.changes.filter((c) => c.action === action).length])
+            .filter(([, n]) => n > 0)
+            .map(([action, n]) => `${n} ${action === "fix" ? "fixed" : action === "skip" ? "flagged as source bugs" : "removed"}`);
+        const after = attempt.after ? ` → ${attempt.after.passed} passed, ${attempt.after.failed} failed` : "";
+        console.log(`${attempt.regressed ? "!" : "✓"} Repair attempt ${attempt.attempt}: ${counts.join(", ") || "no changes"}${after}`);
+        for (const change of attempt.changes)
+            console.log(`  - ${change.action} ${change.functionName} › "${change.name}": ${change.reason}`);
+        for (const invalid of attempt.invalid)
+            console.log(`  ! ignored repair for ${invalid.functionName} › "${invalid.name}": ${invalid.reason}`);
+        if (attempt.regressed)
+            console.log("  ! This attempt broke passing tests; it was reverted.");
+    }
+    for (const change of repair.gaveUp)
+        console.log(`  - skipped ${change.functionName} › "${change.name}": ${change.reason}`);
+    const { final } = repair;
+    const total = final.passed + final.failed + final.skipped;
+    const skipped = final.skipped ? `, ${final.skipped} skipped` : "";
+    console.log(`✓ Final result: ${final.passed}/${total} passing${skipped} (${STOP_MESSAGES[repair.stopReason]})`);
+    if (final.coverage)
+        console.log(`✓ Coverage: ${formatCoverage(final.coverage)}`);
+    console.log(`  Repair tokens: ${tokens(repair.usage, model)}`);
+    if (!repair.sourceUnchanged)
+        console.error("✗ A test run modified the source file. Assertloom restored the original contents.");
+}
+function buildReport(file, generation, run, repair, budget) {
+    return {
+        file,
+        model: generation.model,
+        mocks: { mocked: generation.mocks.mocked, real: generation.mocks.real, globals: generation.mocks.globals },
+        generated: generation.suites.reduce((n, s) => n + s.cases.length, 0),
+        rejected: generation.rejected,
+        initial: repair?.initial ?? summarize(run),
+        attempts: repair?.attempts ?? [],
+        gaveUp: repair?.gaveUp ?? [],
+        final: summarize(run),
+        stopReason: repair?.stopReason ?? (run.status === "passed" ? "all-passing" : "not-repaired"),
+        sourceUnchanged: repair?.sourceUnchanged ?? true,
+        usage: {
+            generation: generation.usage,
+            repair: repair?.usage ?? { inputTokens: 0, outputTokens: 0 },
+        },
+        costUsd: budget.spentUsd ?? null,
+    };
+}
+export function registerGenerateCommand(program) {
+    program
+        .command("generate")
+        .description("Generate Jest tests for a source file, run them, and keep them if they pass")
+        .argument("<file>", "path to a JavaScript or TypeScript source file")
+        .addOption(new Option("--provider <name>", "LLM provider").choices(PROVIDERS).default("openai"))
+        .option("--model <model>", "model name (default: $ASSERTLOOM_MODEL or the provider default)")
+        .option("--base-url <url>", "server URL for --provider openai-compatible (default: $ASSERTLOOM_BASE_URL)")
+        .option("--out <path>", "where to write the tests (default: <file>.test.preview.<ext>, promoted to <file>.test.<ext> when all pass)")
+        .option("--dry-run", "print the generated tests instead of writing or running them")
+        .option("--no-run", "write the preview file without running it")
+        .option("--no-repair", "do not try to repair failing tests")
+        .option("--max-attempts <n>", `repair rounds, 0-${MAX_REPAIR_ATTEMPTS}`, String(MAX_REPAIR_ATTEMPTS))
+        .option("--report <file>", "write before/after results of every run and repair round as JSON")
+        .option("--mock <module>", "mock this import even if Assertloom would keep it real (repeatable)", collect, [])
+        .option("--keep-real <module>", "do not mock this import (repeatable)", collect, [])
+        .option("--allow-network", "let generated tests open network connections (blocked by default)")
+        .option("--max-cost <usd>", "stop before model calls could push spending past this many US dollars", String(DEFAULT_MAX_COST_USD))
+        .option("--force", "overwrite an existing test file that Assertloom did not generate")
+        .action(async (file, options) => {
+        const maxAttempts = Number(options.maxAttempts);
+        if (!Number.isInteger(maxAttempts) || maxAttempts < 0 || maxAttempts > MAX_REPAIR_ATTEMPTS) {
+            throw new UserError(`--max-attempts must be a whole number from 0 to ${MAX_REPAIR_ATTEMPTS}.`);
+        }
+        const maxCost = Number(options.maxCost);
+        if (!Number.isFinite(maxCost) || maxCost <= 0)
+            throw new UserError("--max-cost must be a positive number of US dollars.");
+        const source = await validateSourceFile(file);
+        const provider = createProvider(options.provider, { model: options.model, baseUrl: options.baseUrl });
+        const budget = new Budget(provider.model, maxCost);
+        const testPath = options.out ? path.resolve(options.out) : undefined;
+        const allowNetwork = !!options.allowNetwork;
+        console.log(`Generating tests for ${source.inputPath} with ${provider.name} (${provider.model})...`);
+        const result = await generateTests({ source, provider, testPath, mock: options.mock, keepReal: options.keepReal, budget });
+        const exported = result.analysis.functions.filter((f) => f.exported).length;
+        const caseCount = result.suites.reduce((n, s) => n + s.cases.length, 0);
+        const byCategory = (category) => result.suites.reduce((n, s) => n + s.cases.filter((c) => c.category === category).length, 0);
+        console.log(`✓ Analyzed ${exported} exported functions (${result.suites.length} tested)`);
+        printMocks(result);
+        if (result.redactions)
+            console.log(`✓ Redacted ${result.redactions} likely secrets before sending`);
+        console.log(`✓ Generated ${caseCount} test cases (${byCategory("normal")} normal, ${byCategory("edge")} edge, ${byCategory("error")} error)`);
+        for (const r of result.rejected)
+            console.log(`  ! rejected ${r.functionName} › "${r.name}": ${r.reason}`);
+        console.log(`  Tokens: ${tokens(result.usage, result.model)} (${result.model})`);
+        if (options.dryRun) {
+            console.log(`\n${result.content}`);
+            return;
+        }
+        await writeTestFile(result.testPath, result.content, options.force, showDiff(result.testPath));
+        if (!options.run) {
+            console.log(`\nTests saved to ${relative(result.testPath)} (not run)`);
+            return;
+        }
+        let run = await runJest({ testPath: result.testPath, sourcePath: source.absolutePath, allowNetwork });
+        printRunResult(run);
+        let content = result.content;
+        let repair;
+        if (run.status === "failed" && options.repair) {
+            repair = await repairTests({
+                provider,
+                analysis: result.analysis,
+                promptSource: result.promptSource,
+                sourcePath: source.absolutePath,
+                testPath: result.testPath,
+                extension: source.extension,
+                preamble: result.preamble,
+                mockSummary: result.mocks.summary,
+                suites: result.suites,
+                initialRun: run,
+                maxAttempts,
+                allowNetwork,
+                budget,
+            });
+            printRepair(repair, result.model);
+            run = repair.run;
+            content = repair.content;
+        }
+        if (options.report) {
+            await writeFile(options.report, JSON.stringify(buildReport(source.inputPath, result, run, repair, budget), null, 2));
+            console.log(`  Report written to ${relative(path.resolve(options.report))}`);
+        }
+        if (run.status !== "passed") {
+            console.log(`\nTests kept in ${relative(result.testPath)} for review.`);
+            process.exitCode = 1;
+            return;
+        }
+        if (testPath) {
+            console.log(`\nTests saved to ${relative(result.testPath)}`);
+            return;
+        }
+        const finalPath = testPathFor(source.absolutePath);
+        try {
+            await writeTestFile(finalPath, content, options.force, showDiff(finalPath));
+            await rm(result.testPath);
+            console.log(`\nTests saved to ${relative(finalPath)}`);
+        }
+        catch (error) {
+            if (!(error instanceof TestFileExistsError))
+                throw error;
+            console.log(`\n${relative(finalPath)} already exists and was not generated by Assertloom.`);
+            console.log(`Passing tests kept in ${relative(result.testPath)}; rerun with --force to replace the existing file.`);
+        }
+    });
+}
+//# sourceMappingURL=generate.js.map

package/dist/cli/commands/generate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.js","sourceRoot":"","sources":["../../../src/cli/commands/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,EAAgB,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAEtE,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,mBAAmB,EAAE,aAAa,EAAyB,MAAM,0BAA0B,CAAC;AACjI,OAAO,EAAE,mBAAmB,EAAE,WAAW,EAAE,SAAS,EAAsC,MAAM,uBAAuB,CAAC;AACxH,OAAO,EAAE,OAAO,EAAkB,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAmBxD,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAEtC,SAAS,MAAM,CAAC,KAAiB,EAAE,KAAa;IAC9C,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACxC,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,cAAc,EAAE,SAAS,KAAK,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;AACrJ,CAAC;AAED,MAAM,OAAO,GAAG,CAAC,KAAa,EAAE,QAAkB,EAAE,EAAE,CAAC,CAAC,GAAG,QAAQ,EAAE,KAAK,CAAC,CAAC;AAE5E,SAAS,UAAU,CAAC,MAAwB;IAC1C,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC;IAC/C,MAAM,KAAK,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;IACrF,IAAI,KAAK,CAAC,MAAM;QAAE,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,cAAc,KAAK,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACrI,KAAK,MAAM,CAAC,IAAI,MAAM;QAAE,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACzG,IAAI,IAAI,CAAC,MAAM;QAAE,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC1G,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC;AACpD,CAAC;AAED,yGAAyG;AACzG,SAAS,QAAQ,CAAC,IAAY;IAC5B,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QAAE,OAAO,SAAS,CAAC;IACtD,OAAO,CAAC,IAAI,EAAE,EAAE;QACd,OAAO,CAAC,GAAG,CAAC,iBAAiB,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACzD,KAAK,MAAM,IAAI,IAAI,IAAI;YAAE,OAAO,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,aAAa,GAA+B;IAChD,aAAa,EAAE,gBAAgB;IAC/B,cAAc,EAAE,0BAA0B;IAC1C,SAAS,EAAE,+CAA+C;IAC1D,kBAAkB,EAAE,sCAAsC;IAC1D,YAAY,EAAE,yDAAyD;IACvE,MAAM,EAAE,sDAAsD;CAC/D,CAAC;AAEF,SAAS,WAAW,CAAC,MAAoB,EAAE,KAAa;IACtD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACtC,MAAM,MAAM,GAAI,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAW;aAChD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,MAAM,CAAU,CAAC;aAC7F,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;aACxB,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;QACzH,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,YAAY,OAAO,CAAC,KAAK,CAAC,MAAM,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QACvG,OAAO,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,mBAAmB,OAAO,CAAC,OAAO,KAAK,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,YAAY,GAAG,KAAK,EAAE,CAAC,CAAC;QAChI,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,OAAO;YAAE,OAAO,CAAC,GAAG,CAAC,OAAO,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,YAAY,OAAO,MAAM,CAAC,IAAI,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACtI,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,OAAO;YAAE,OAAO,CAAC,GAAG,CAAC,0BAA0B,OAAO,CAAC,YAAY,OAAO,OAAO,CAAC,IAAI,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5I,IAAI,OAAO,CAAC,SAAS;YAAE,OAAO,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAC;IAC/F,CAAC;IACD,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM;QAAE,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,YAAY,OAAO,MAAM,CAAC,IAAI,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAE3H,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;IACzB,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC;IAC1D,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;IAClE,OAAO,CAAC,GAAG,CAAC,mBAAmB,KAAK,CAAC,MAAM,IAAI,KAAK,WAAW,OAAO,KAAK,aAAa,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IAChH,IAAI,KAAK,CAAC,QAAQ;QAAE,OAAO,CAAC,GAAG,CAAC,eAAe,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjF,OAAO,CAAC,GAAG,CAAC,oBAAoB,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;IAC/D,IAAI,CAAC,MAAM,CAAC,eAAe;QAAE,OAAO,CAAC,KAAK,CAAC,mFAAmF,CAAC,CAAC;AAClI,CAAC;AAED,SAAS,WAAW,CAAC,IAAY,EAAE,UAA4B,EAAE,GAAc,EAAE,MAAgC,EAAE,MAAc;IAC/H,OAAO;QACL,IAAI;QACJ,KAAK,EAAE,UAAU,CAAC,KAAK;QACvB,KAAK,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,UAAU,CAAC,KAAK,CAAC,OAAO,EAAE;QAC1G,SAAS,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QACpE,QAAQ,EAAE,UAAU,CAAC,QAAQ;QAC7B,OAAO,EAAE,MAAM,EAAE,OAAO,IAAI,SAAS,CAAC,GAAG,CAAC;QAC1C,QAAQ,EAAE,MAAM,EAAE,QAAQ,IAAI,EAAE;QAChC,MAAM,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE;QAC5B,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC;QACrB,UAAU,EAAE,MAAM,EAAE,UAAU,IAAI,CAAC,GAAG,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,cAAc,CAAC;QAC5F,eAAe,EAAE,MAAM,EAAE,eAAe,IAAI,IAAI;QAChD,KAAK,EAAE;YACL,UAAU,EAAE,UAAU,CAAC,KAAK;YAC5B,MAAM,EAAE,MAAM,EAAE,KAAK,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;SAC7D;QACD,OAAO,EAAE,MAAM,CAAC,QAAQ,IAAI,IAAI;KACjC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,OAAgB;IACtD,OAAO;SACJ,OAAO,CAAC,UAAU,CAAC;SACnB,WAAW,CAAC,6EAA6E,CAAC;SAC1F,QAAQ,CAAC,QAAQ,EAAE,gDAAgD,CAAC;SACpE,SAAS,CAAC,IAAI,MAAM,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;SAC/F,MAAM,CAAC,iBAAiB,EAAE,iEAAiE,CAAC;SAC5F,MAAM,CAAC,kBAAkB,EAAE,6EAA6E,CAAC;SACzG,MAAM,CAAC,cAAc,EAAE,4GAA4G,CAAC;SACpI,MAAM,CAAC,WAAW,EAAE,8DAA8D,CAAC;SACnF,MAAM,CAAC,UAAU,EAAE,2CAA2C,CAAC;SAC/D,MAAM,CAAC,aAAa,EAAE,oCAAoC,CAAC;SAC3D,MAAM,CAAC,oBAAoB,EAAE,oBAAoB,mBAAmB,EAAE,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC;SACpG,MAAM,CAAC,iBAAiB,EAAE,kEAAkE,CAAC;SAC7F,MAAM,CAAC,iBAAiB,EAAE,qEAAqE,EAAE,OAAO,EAAE,EAAE,CAAC;SAC7G,MAAM,CAAC,sBAAsB,EAAE,sCAAsC,EAAE,OAAO,EAAE,EAAE,CAAC;SACnF,MAAM,CAAC,iBAAiB,EAAE,mEAAmE,CAAC;SAC9F,MAAM,CAAC,kBAAkB,EAAE,uEAAuE,EAAE,MAAM,CAAC,oBAAoB,CAAC,CAAC;SACjI,MAAM,CAAC,SAAS,EAAE,kEAAkE,CAAC;SACrF,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,OAAwB,EAAE,EAAE;QACvD,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC,IAAI,WAAW,GAAG,mBAAmB,EAAE,CAAC;YAC3F,MAAM,IAAI,SAAS,CAAC,mDAAmD,mBAAmB,GAAG,CAAC,CAAC;QACjG,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,IAAI,CAAC;YAAE,MAAM,IAAI,SAAS,CAAC,qDAAqD,CAAC,CAAC;QAC1H,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACtG,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrE,MAAM,YAAY,GAAG,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC;QAE5C,OAAO,CAAC,GAAG,CAAC,wBAAwB,MAAM,CAAC,SAAS,SAAS,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,KAAK,MAAM,CAAC,CAAC;QACrG,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAE3H,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;QAC5E,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACxE,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAE,EAAE,CACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAE/F,OAAO,CAAC,GAAG,CAAC,cAAc,QAAQ,wBAAwB,MAAM,CAAC,MAAM,CAAC,MAAM,UAAU,CAAC,CAAC;QAC1F,UAAU,CAAC,MAAM,CAAC,CAAC;QACnB,IAAI,MAAM,CAAC,UAAU;YAAE,OAAO,CAAC,GAAG,CAAC,cAAc,MAAM,CAAC,UAAU,gCAAgC,CAAC,CAAC;QACpG,OAAO,CAAC,GAAG,CACT,eAAe,SAAS,gBAAgB,UAAU,CAAC,QAAQ,CAAC,YAAY,UAAU,CAAC,MAAM,CAAC,UAAU,UAAU,CAAC,OAAO,CAAC,SAAS,CACjI,CAAC;QACF,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ;YAAE,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,YAAY,OAAO,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1G,OAAO,CAAC,GAAG,CAAC,aAAa,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;QAEjF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;YACnC,OAAO;QACT,CAAC;QAED,MAAM,aAAa,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/F,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,CAAC,oBAAoB,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YACvE,OAAO;QACT,CAAC;QAED,IAAI,GAAG,GAAG,MAAM,OAAO,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,YAAY,EAAE,YAAY,EAAE,CAAC,CAAC;QACtG,cAAc,CAAC,GAAG,CAAC,CAAC;QACpB,IAAI,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC7B,IAAI,MAAgC,CAAC;QAErC,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YAC9C,MAAM,GAAG,MAAM,WAAW,CAAC;gBACzB,QAAQ;gBACR,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,UAAU,EAAE,MAAM,CAAC,YAAY;gBAC/B,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;gBACjC,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,UAAU,EAAE,GAAG;gBACf,WAAW;gBACX,YAAY;gBACZ,MAAM;aACP,CAAC,CAAC;YACH,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YAClC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;YACjB,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC3B,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,SAAS,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACrH,OAAO,CAAC,GAAG,CAAC,uBAAuB,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC;QAC/E,CAAC;QAED,IAAI,GAAG,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,mBAAmB,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YACxE,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;YACrB,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,CAAC,GAAG,CAAC,oBAAoB,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC7D,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC;YACH,MAAM,aAAa,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;YAC5E,MAAM,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC1B,OAAO,CAAC,GAAG,CAAC,oBAAoB,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,CAAC,KAAK,YAAY,mBAAmB,CAAC;gBAAE,MAAM,KAAK,CAAC;YACzD,OAAO,CAAC,GAAG,CAAC,KAAK,QAAQ,CAAC,SAAS,CAAC,sDAAsD,CAAC,CAAC;YAC5F,OAAO,CAAC,GAAG,CAAC,yBAAyB,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,oDAAoD,CAAC,CAAC;QACtH,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/cli/commands/run.d.ts

@@ -0,0 +1,4 @@
+import type { Command } from "commander";
+/** `math.test.preview.ts` / `math.test.ts` / `math.spec.ts` → `math.ts`. */
+export declare function inferSourcePath(testPath: string): string | undefined;
+export declare function registerRunCommand(program: Command): void;

package/dist/cli/commands/run.js

@@ -0,0 +1,53 @@
+import { stat } from "node:fs/promises";
+import path from "node:path";
+import { FileValidationError, validateSourceFile } from "../validateFile.js";
+import { runJest } from "../../runner/index.js";
+import { printRunResult } from "../report.js";
+/** Node timers overflow above 2^31 - 1 ms and fire immediately. */
+const MAX_TIMEOUT_SECONDS = 2_147_483;
+/** `math.test.preview.ts` / `math.test.ts` / `math.spec.ts` → `math.ts`. */
+export function inferSourcePath(testPath) {
+    const inferred = testPath.replace(/\.(test|spec)(\.preview)?(?=\.[cm]?[jt]sx?$)/, "");
+    return inferred === testPath ? undefined : inferred;
+}
+export function registerRunCommand(program) {
+    program
+        .command("run")
+        .description("Run a test file with Jest and report results and coverage of its source file")
+        .argument("<testFile>", "test file to run, such as math.test.preview.ts")
+        .option("--source <file>", "source file to measure coverage for (default: inferred from the test file name)")
+        .option("--timeout <seconds>", "kill Jest if it runs longer than this", "120")
+        .option("--json", "print the full result as JSON")
+        .option("--allow-network", "let the tests open network connections (blocked by default)")
+        .action(async (testFile, options) => {
+        const testPath = path.resolve(testFile);
+        try {
+            if (!(await stat(testPath)).isFile())
+                throw new Error();
+        }
+        catch {
+            throw new FileValidationError(`Test file not found: ${testFile}`);
+        }
+        const sourceArg = options.source ?? inferSourcePath(testFile);
+        if (!sourceArg)
+            throw new FileValidationError(`Cannot infer the source file from ${testFile}; pass --source.`);
+        const source = await validateSourceFile(sourceArg);
+        const timeoutSeconds = Number(options.timeout);
+        if (!Number.isFinite(timeoutSeconds) || timeoutSeconds <= 0 || timeoutSeconds > MAX_TIMEOUT_SECONDS) {
+            throw new FileValidationError(`--timeout must be a positive number of seconds, at most ${MAX_TIMEOUT_SECONDS}.`);
+        }
+        const result = await runJest({
+            testPath,
+            sourcePath: source.absolutePath,
+            timeoutMs: timeoutSeconds * 1000,
+            allowNetwork: !!options.allowNetwork,
+        });
+        if (options.json)
+            console.log(JSON.stringify(result, null, 2));
+        else
+            printRunResult(result);
+        if (result.status !== "passed")
+            process.exitCode = 1;
+    });
+}
+//# sourceMappingURL=run.js.map

package/dist/cli/commands/run.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run.js","sourceRoot":"","sources":["../../../src/cli/commands/run.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACxC,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAE,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,mEAAmE;AACnE,MAAM,mBAAmB,GAAG,SAAS,CAAC;AAStC,4EAA4E;AAC5E,MAAM,UAAU,eAAe,CAAC,QAAgB;IAC9C,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,8CAA8C,EAAE,EAAE,CAAC,CAAC;IACtF,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,OAAgB;IACjD,OAAO;SACJ,OAAO,CAAC,KAAK,CAAC;SACd,WAAW,CAAC,8EAA8E,CAAC;SAC3F,QAAQ,CAAC,YAAY,EAAE,gDAAgD,CAAC;SACxE,MAAM,CAAC,iBAAiB,EAAE,iFAAiF,CAAC;SAC5G,MAAM,CAAC,qBAAqB,EAAE,uCAAuC,EAAE,KAAK,CAAC;SAC7E,MAAM,CAAC,QAAQ,EAAE,+BAA+B,CAAC;SACjD,MAAM,CAAC,iBAAiB,EAAE,6DAA6D,CAAC;SACxF,MAAM,CAAC,KAAK,EAAE,QAAgB,EAAE,OAA0B,EAAE,EAAE;QAC7D,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC;YACH,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE;gBAAE,MAAM,IAAI,KAAK,EAAE,CAAC;QAC1D,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,mBAAmB,CAAC,wBAAwB,QAAQ,EAAE,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,IAAI,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC9D,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,mBAAmB,CAAC,qCAAqC,QAAQ,kBAAkB,CAAC,CAAC;QAC/G,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAEnD,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,cAAc,IAAI,CAAC,IAAI,cAAc,GAAG,mBAAmB,EAAE,CAAC;YACpG,MAAM,IAAI,mBAAmB,CAAC,2DAA2D,mBAAmB,GAAG,CAAC,CAAC;QACnH,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;YAC3B,QAAQ;YACR,UAAU,EAAE,MAAM,CAAC,YAAY;YAC/B,SAAS,EAAE,cAAc,GAAG,IAAI;YAChC,YAAY,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY;SACrC,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,IAAI;YAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;;YAC1D,cAAc,CAAC,MAAM,CAAC,CAAC;QAE5B,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ;YAAE,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/cli/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/cli/index.js

@@ -0,0 +1,16 @@
+#!/usr/bin/env node
+import { UserError } from "../errors.js";
+import { createProgram } from "./program.js";
+try {
+    await createProgram().parseAsync(process.argv);
+}
+catch (error) {
+    if (error instanceof UserError) {
+        console.error(`✗ ${error.message}`);
+    }
+    else {
+        console.error(error);
+    }
+    process.exitCode = 1;
+}
+//# sourceMappingURL=index.js.map

package/dist/cli/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,IAAI,CAAC;IACH,MAAM,aAAa,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;AACjD,CAAC;AAAC,OAAO,KAAK,EAAE,CAAC;IACf,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;QAC/B,OAAO,CAAC,KAAK,CAAC,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IACtC,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IACD,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;AACvB,CAAC"}

package/dist/cli/program.d.ts

@@ -0,0 +1,2 @@
+import { Command } from "commander";
+export declare function createProgram(): Command;

package/dist/cli/program.js

@@ -0,0 +1,20 @@
+import { readFileSync } from "node:fs";
+import { Command } from "commander";
+import { registerAnalyzeCommand } from "./commands/analyze.js";
+import { registerCiCommand } from "./commands/ci.js";
+import { registerGenerateCommand } from "./commands/generate.js";
+import { registerRunCommand } from "./commands/run.js";
+const { version } = JSON.parse(readFileSync(new URL("../../package.json", import.meta.url), "utf8"));
+export function createProgram() {
+    const program = new Command()
+        .name("assertloom")
+        .description("Assertloom: AI-generated Jest tests that are executed, verified, and repaired.")
+        .version(version)
+        .showHelpAfterError();
+    registerAnalyzeCommand(program);
+    registerGenerateCommand(program);
+    registerRunCommand(program);
+    registerCiCommand(program);
+    return program;
+}
+//# sourceMappingURL=program.js.map