assertloom 0.0.0-stage → 0.1.0

package/dist/generator/importPlan.js

@@ -0,0 +1,62 @@
+import path from "node:path";
+const RESERVED = new Set(("break case catch class const continue debugger default delete do else enum export extends false finally for function if import in " +
+    "instanceof new null return super switch this throw true try typeof var void while with yield let static implements interface " +
+    "package private protected public await arguments eval undefined NaN Infinity").split(" "));
+const isIdentifierName = (name) => /^[A-Za-z_$][\w$]*$/.test(name);
+function camelCase(text) {
+    const words = text.split(/[^A-Za-z0-9]+/).filter(Boolean);
+    const joined = words.map((w, i) => (i === 0 ? w[0].toLowerCase() : w[0].toUpperCase()) + w.slice(1)).join("");
+    return /^[A-Za-z_$]/.test(joined) ? joined : `_${joined}`;
+}
+/** A local binding name for `name` that is a valid, unused, non-reserved identifier. */
+function localName(name, used) {
+    let base = isIdentifierName(name) ? name : camelCase(name);
+    if (RESERVED.has(base))
+        base = `_${base}`;
+    let binding = base;
+    for (let n = 2; used.has(binding); n++)
+        binding = `${base}${n}`;
+    used.add(binding);
+    return binding;
+}
+const exportKey = (name) => (isIdentifierName(name) ? name : JSON.stringify(name));
+/** Module specifier for importing `sourcePath` from a test file at `testPath`. */
+export function importSpecifier(testPath, sourcePath) {
+    let relative = path.relative(path.dirname(testPath), sourcePath).replaceAll("\\", "/");
+    relative = relative.replace(/\.(ts|tsx|js|jsx)$/, "");
+    return relative.startsWith(".") ? relative : `./${relative}`;
+}
+export function planImports(analysis, functions, sourcePath, testPath) {
+    const bindings = [];
+    const used = new Set();
+    for (const fn of functions) {
+        const named = fn.exportedAs.find((name) => name !== "default");
+        if (named)
+            bindings.push({ binding: localName(named, used), exportName: named, fn });
+    }
+    for (const fn of functions) {
+        if (fn.exportedAs.some((name) => name !== "default") || !fn.exportedAs.includes("default"))
+            continue;
+        const base = fn.name !== "default" ? fn.name : camelCase(path.basename(sourcePath).replace(/\.[^.]+$/, ""));
+        bindings.push({ binding: localName(used.has(base) ? `${base}Default` : base, used), exportName: "default", fn });
+    }
+    const specifier = JSON.stringify(importSpecifier(testPath, sourcePath));
+    const defaultBinding = bindings.find((b) => b.exportName === "default")?.binding;
+    const named = bindings.filter((b) => b.exportName !== "default");
+    const statements = [];
+    if (analysis.moduleSystem === "commonjs") {
+        const properties = named.map((b) => (b.binding === b.exportName ? b.binding : `${exportKey(b.exportName)}: ${b.binding}`));
+        if (defaultBinding)
+            statements.push(`const ${defaultBinding} = require(${specifier});`);
+        if (properties.length)
+            statements.push(`const { ${properties.join(", ")} } = require(${specifier});`);
+    }
+    else {
+        const specifiers = named.map((b) => (b.binding === b.exportName ? b.binding : `${exportKey(b.exportName)} as ${b.binding}`));
+        const clauses = [defaultBinding, specifiers.length ? `{ ${specifiers.join(", ")} }` : undefined].filter(Boolean);
+        if (clauses.length)
+            statements.push(`import ${clauses.join(", ")} from ${specifier};`);
+    }
+    return { statements, bindings };
+}
+//# sourceMappingURL=importPlan.js.map

package/dist/generator/importPlan.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"importPlan.js","sourceRoot":"","sources":["../../src/generator/importPlan.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAgB7B,MAAM,QAAQ,GAAG,IAAI,GAAG,CACtB,CAAC,oIAAoI;IACnI,+HAA+H;IAC/H,8EAA8E,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAC7F,CAAC;AAEF,MAAM,gBAAgB,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAE3E,SAAS,SAAS,CAAC,IAAY;IAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC1D,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9G,OAAO,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE,CAAC;AAC5D,CAAC;AAED,wFAAwF;AACxF,SAAS,SAAS,CAAC,IAAY,EAAE,IAAiB;IAChD,IAAI,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC3D,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC;QAAE,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;IAC1C,IAAI,OAAO,GAAG,IAAI,CAAC;IACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;QAAE,OAAO,GAAG,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC;IAChE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAClB,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,SAAS,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;AAE3F,kFAAkF;AAClF,MAAM,UAAU,eAAe,CAAC,QAAgB,EAAE,UAAkB;IAClE,IAAI,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,UAAU,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACvF,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;IACtD,OAAO,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;AAC/D,CAAC;AAED,MAAM,UAAU,WAAW,CACzB,QAAsB,EACtB,SAAyB,EACzB,UAAkB,EAClB,QAAgB;IAEhB,MAAM,QAAQ,GAAc,EAAE,CAAC;IAC/B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAE/B,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;QAC/D,IAAI,KAAK;YAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;IACvF,CAAC;IAED,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;QAC3B,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,SAAS;QACrG,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;QAC5G,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;IACnH,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;IACxE,MAAM,cAAc,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,SAAS,CAAC,EAAE,OAAO,CAAC;IACjF,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC;IACjE,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,IAAI,QAAQ,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;QACzC,MAAM,UAAU,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3H,IAAI,cAAc;YAAE,UAAU,CAAC,IAAI,CAAC,SAAS,cAAc,cAAc,SAAS,IAAI,CAAC,CAAC;QACxF,IAAI,UAAU,CAAC,MAAM;YAAE,UAAU,CAAC,IAAI,CAAC,WAAW,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,SAAS,IAAI,CAAC,CAAC;IACxG,CAAC;SAAM,CAAC;QACN,MAAM,UAAU,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC7H,MAAM,OAAO,GAAG,CAAC,cAAc,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACjH,IAAI,OAAO,CAAC,MAAM;YAAE,UAAU,CAAC,IAAI,CAAC,UAAU,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,SAAS,GAAG,CAAC,CAAC;IACzF,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;AAClC,CAAC"}

package/dist/generator/index.d.ts

@@ -0,0 +1,5 @@
+export { generateTests, GenerationError, type GenerationResult } from "./testGenerator.js";
+export { planMocks, PURE_BUILTINS, PURE_PACKAGES, type MockedModule, type MockPlan, type RealModule } from "./mocks.js";
+export { previewPathFor, renderTestFile, testPathFor, writeTestFile, TestFileExistsError } from "./testWriter.js";
+export { unifiedDiff } from "./diff.js";
+export { checkBody, type AcceptedCase, type ValidatedSuite } from "./validate.js";

package/dist/generator/index.js

@@ -0,0 +1,6 @@
+export { generateTests, GenerationError } from "./testGenerator.js";
+export { planMocks, PURE_BUILTINS, PURE_PACKAGES } from "./mocks.js";
+export { previewPathFor, renderTestFile, testPathFor, writeTestFile, TestFileExistsError } from "./testWriter.js";
+export { unifiedDiff } from "./diff.js";
+export { checkBody } from "./validate.js";
+//# sourceMappingURL=index.js.map

package/dist/generator/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/generator/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,eAAe,EAAyB,MAAM,oBAAoB,CAAC;AAC3F,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,aAAa,EAAqD,MAAM,YAAY,CAAC;AACxH,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,WAAW,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAClH,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,SAAS,EAA0C,MAAM,eAAe,CAAC"}

package/dist/generator/mocks.d.ts

@@ -0,0 +1,48 @@
+import { type FileAnalysis, type ImportCategory, type SupportedExtension } from "../types/index.js";
+/** Built-ins that are deterministic and side-effect free, so they stay real. */
+export declare const PURE_BUILTINS: Set<string>;
+/** Utility packages without I/O. Mocking them would break the code under test, so they stay real. */
+export declare const PURE_PACKAGES: Set<string>;
+export interface MockedModule {
+    /** Specifier as written in the source. */
+    source: string;
+    /** Specifier used in the test file. */
+    specifier: string;
+    category: ImportCategory;
+    /** The module is not installed or not found, so Jest must not try to resolve it. */
+    virtual: boolean;
+    reason: string;
+}
+export interface RealModule {
+    source: string;
+    reason: string;
+}
+export interface MockPlan {
+    mocked: MockedModule[];
+    real: RealModule[];
+    globals: string[];
+    /** jest.mock() calls and imports of the mocked modules; they go before the source import. */
+    imports: string[];
+    /** Shared mock objects, global mocks, and reset hooks; they go after all imports. */
+    setup: string[];
+    /** Description of the mocks for the model. */
+    summary: string;
+}
+export interface MockOptions {
+    analysis: FileAnalysis;
+    sourcePath: string;
+    testPath: string;
+    extension: SupportedExtension;
+    /** Names already used in the test file (the source function bindings). */
+    reserved: Iterable<string>;
+    /** Modules to mock even if Assertloom would keep them real. Matched by specifier or package name. */
+    mock?: string[];
+    /** Modules to keep real even if Assertloom would mock them. */
+    keepReal?: string[];
+}
+export declare function resolveLocal(fromFile: string, specifier: string): Promise<string | undefined>;
+/**
+ * Decides which imports to mock and writes the mocks. Assertloom writes every jest.mock() factory
+ * itself from how the source uses each import; the model only configures return values in tests.
+ */
+export declare function planMocks(options: MockOptions): Promise<MockPlan>;

package/dist/generator/mocks.js

@@ -0,0 +1,356 @@
+import { readFile, stat } from "node:fs/promises";
+import { createRequire } from "node:module";
+import path from "node:path";
+import { classifySource, scanImports } from "../analyzer/dependencies.js";
+import { parseSource } from "../analyzer/parser.js";
+import { collectImportUsage, emptyShape } from "../analyzer/usage.js";
+import { SUPPORTED_EXTENSIONS } from "../types/index.js";
+/** Built-ins that are deterministic and side-effect free, so they stay real. */
+export const PURE_BUILTINS = new Set(["assert", "buffer", "path", "path/posix", "path/win32", "querystring", "string_decoder", "url", "util"]);
+/** Utility packages without I/O. Mocking them would break the code under test, so they stay real. */
+export const PURE_PACKAGES = new Set([
+    "@babel/runtime", "big.js", "bignumber.js", "change-case", "classnames", "clsx", "date-fns", "dayjs", "decimal.js",
+    "deepmerge", "fast-deep-equal", "immer", "joi", "lodash", "lodash-es", "ms", "qs", "ramda", "semver", "tslib",
+    "validator", "yup", "zod",
+]);
+const MAX_LOCAL_DEPTH = 5;
+const TYPESCRIPT_EXTENSIONS = [".ts", ".tsx", ".mts", ".cts"];
+async function isFile(file) {
+    try {
+        return (await stat(file)).isFile();
+    }
+    catch {
+        return false;
+    }
+}
+export async function resolveLocal(fromFile, specifier) {
+    if (specifier.startsWith("#"))
+        return undefined;
+    const base = path.resolve(path.dirname(fromFile), specifier);
+    const candidates = [base];
+    const jsExtension = /\.[cm]?jsx?$/.exec(base);
+    if (jsExtension)
+        candidates.push(...TYPESCRIPT_EXTENSIONS.map((ext) => base.slice(0, -jsExtension[0].length) + ext));
+    candidates.push(...SUPPORTED_EXTENSIONS.map((ext) => base + ext), ...SUPPORTED_EXTENSIONS.map((ext) => path.join(base, `index${ext}`)));
+    for (const candidate of candidates)
+        if (await isFile(candidate))
+            return candidate;
+    return undefined;
+}
+function isInstalled(fromFile, specifier) {
+    try {
+        createRequire(fromFile).resolve(specifier);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/** Why a local file has side effects (its first impure dependency), or undefined if it has none. */
+async function localImpurity(file, visited, depth) {
+    if (visited.has(file))
+        return undefined;
+    visited.add(file);
+    const extension = path.extname(file);
+    if (!SUPPORTED_EXTENSIONS.includes(extension))
+        return undefined;
+    if (depth > MAX_LOCAL_DEPTH)
+        return "deeply nested local modules";
+    let ast;
+    try {
+        ast = parseSource(await readFile(file, "utf8"), extension);
+    }
+    catch {
+        return `${path.basename(file)}, which could not be parsed`;
+    }
+    const { globals } = collectImportUsage(ast);
+    if (globals.length)
+        return `global ${globals[0]}`;
+    for (const entry of scanImports(ast).imports) {
+        if (entry.typeOnly)
+            continue;
+        const decision = await decide(entry.source, entry.category, entry.packageName, file, visited, depth + 1);
+        if (decision.mock)
+            return entry.category === "local" ? `${entry.source} (${decision.reason})` : entry.source;
+    }
+    return undefined;
+}
+async function decide(source, category, packageName, fromFile, visited, depth) {
+    if (category === "builtin") {
+        return PURE_BUILTINS.has(source.replace(/^node:/, ""))
+            ? { mock: false, virtual: false, reason: "pure Node built-in" }
+            : { mock: true, virtual: false, reason: "Node built-in with side effects" };
+    }
+    if (category === "external") {
+        if (packageName && PURE_PACKAGES.has(packageName))
+            return { mock: false, virtual: false, reason: "utility library without I/O" };
+        const installed = isInstalled(fromFile, source);
+        return { mock: true, virtual: !installed, reason: installed ? "external package" : "external package, not installed" };
+    }
+    const file = await resolveLocal(fromFile, source);
+    if (!file)
+        return { mock: true, virtual: true, reason: "local module not found" };
+    const impurity = await localImpurity(file, visited, depth);
+    return impurity
+        ? { mock: true, virtual: false, reason: `depends on ${impurity}` }
+        : { mock: false, virtual: false, reason: "local module without side effects" };
+}
+function matches(list, source, packageName) {
+    return !!list?.some((entry) => entry === source || entry === packageName || entry.replace(/^node:/, "") === source.replace(/^node:/, ""));
+}
+function specifierFromTest(source, sourcePath, testPath) {
+    if (!source.startsWith("."))
+        return source;
+    const relative = path.relative(path.dirname(testPath), path.resolve(path.dirname(sourcePath), source)).replaceAll("\\", "/");
+    return relative.startsWith(".") ? relative : `./${relative}`;
+}
+function mergeShapes(a, b) {
+    const merged = a
+        ? { ...a, members: { ...a.members }, resultNames: [...a.resultNames] }
+        : { ...emptyShape(), members: {}, resultNames: [] };
+    merged.called ||= b.called;
+    merged.constructed ||= b.constructed;
+    merged.instanceOf ||= b.instanceOf;
+    merged.createdAtLoad ||= b.createdAtLoad;
+    for (const name of b.resultNames)
+        if (!merged.resultNames.includes(name))
+            merged.resultNames.push(name);
+    for (const [key, value] of Object.entries(b.members))
+        merged.members[key] = mergeShapes(merged.members[key], value);
+    if (b.result)
+        merged.result = mergeShapes(merged.result, b.result);
+    return merged;
+}
+/**
+ * Constructors, and calls made while the module loads (`const client = createClient()`), return one
+ * shared mock object, because tests cannot configure them before they run. Other calls are plain
+ * jest.fn() mocks whose return values tests set themselves.
+ */
+function sharesInstance(shape) {
+    return !!shape.result && (shape.constructed || shape.createdAtLoad);
+}
+const isCallable = (shape) => shape.called || shape.constructed || shape.instanceOf;
+function propertyKey(name) {
+    return /^[A-Za-z_$][\w$]*$/.test(name) ? name : JSON.stringify(name);
+}
+class FactoryWriter {
+    locals = [];
+    count = 0;
+    members(shape) {
+        return Object.entries(shape.members).map(([key, value]) => `${propertyKey(key)}: ${this.value(value)}`);
+    }
+    /** A member that is only read (never called) is still a jest.fn(), the safest stand-in for an unknown value. */
+    value(shape) {
+        const members = this.members(shape);
+        const withMembers = (base) => (members.length ? `Object.assign(${base}, { ${members.join(", ")} })` : base);
+        if (sharesInstance(shape)) {
+            const name = `instance${++this.count}`;
+            this.locals.push(`const ${name} = ${this.object(shape.result)};`);
+            return withMembers(`function () { return ${name}; }`);
+        }
+        if (shape.instanceOf && !shape.called && !shape.constructed)
+            return withMembers("class {}");
+        if (shape.called || shape.constructed || members.length === 0)
+            return withMembers("jest.fn()");
+        return `{ ${members.join(", ")} }`;
+    }
+    object(shape) {
+        const members = this.members(shape);
+        return members.length ? `{ ${members.join(", ")} }` : "{}";
+    }
+}
+function modelModule(usages) {
+    let defaultShape;
+    let moduleShape = emptyShape();
+    for (const usage of usages) {
+        if (usage.imported === "default" && usage.style === "esm")
+            defaultShape = mergeShapes(defaultShape, usage.shape);
+        else if (usage.imported === "*")
+            moduleShape = mergeShapes(moduleShape, usage.shape);
+        else
+            moduleShape.members[usage.imported] = mergeShapes(moduleShape.members[usage.imported], usage.shape);
+    }
+    return { defaultShape, moduleShape, hasEsm: usages.some((u) => u.style === "esm"), callableRoot: !defaultShape && isCallable(moduleShape) };
+}
+function factoryFor(model) {
+    const writer = new FactoryWriter();
+    let value;
+    if (model.callableRoot) {
+        value = writer.value(model.moduleShape);
+    }
+    else {
+        const members = [
+            ...(model.hasEsm ? ["__esModule: true"] : []),
+            ...(model.defaultShape ? [`default: ${writer.value(model.defaultShape)}`] : []),
+            ...writer.members(model.moduleShape),
+        ];
+        value = members.length ? `{ ${members.join(", ")} }` : "{}";
+    }
+    if (writer.locals.length === 0)
+        return `() => (${value})`;
+    return `() => {\n  ${writer.locals.join("\n  ")}\n  return ${value};\n}`;
+}
+function shapeFor(model, usage) {
+    if (usage.imported === "default" && usage.style === "esm")
+        return model.defaultShape;
+    if (usage.imported === "*")
+        return model.moduleShape;
+    return model.moduleShape.members[usage.imported];
+}
+class NameAllocator {
+    used;
+    constructor(reserved) {
+        this.used = new Set(reserved);
+    }
+    has(name) {
+        return this.used.has(name);
+    }
+    claim(preferred) {
+        let name = preferred;
+        for (let n = 2; this.used.has(name); n++)
+            name = `${preferred}Mock${n === 2 ? "" : n}`;
+        this.used.add(name);
+        return name;
+    }
+}
+function describeShape(shape, expr, names, out, isRoot) {
+    if (sharesInstance(shape)) {
+        const lastSegment = expr.split(/[.()\s]/).filter(Boolean).pop() ?? "mock";
+        const name = names.claim(shape.resultNames.find((n) => !names.has(n)) ?? `${lastSegment[0].toLowerCase()}${lastSegment.slice(1)}Instance`);
+        const init = shape.constructed ? `new ${expr}()` : `${expr}()`;
+        out.handles.push({ name, init });
+        out.lines.push(`\`${init}\` returns one shared mock object, available in tests as \`${name}\``);
+        describeShape(shape.result, name, names, out, false);
+    }
+    else if (shape.called || shape.constructed || (!isRoot && Object.keys(shape.members).length === 0)) {
+        out.fns.push(expr);
+    }
+    for (const [key, member] of Object.entries(shape.members)) {
+        const access = /^[A-Za-z_$][\w$]*$/.test(key) ? `${expr}.${key}` : `${expr}[${JSON.stringify(key)}]`;
+        describeShape(member, access, names, out, false);
+    }
+}
+function importStatement(usage, specifier, model, commonjs) {
+    const spec = JSON.stringify(specifier);
+    const local = usage.local;
+    if (commonjs) {
+        if (usage.imported === "default" && usage.style === "esm")
+            return `const ${local} = require(${spec}).default;`;
+        if (usage.imported === "*")
+            return `const ${local} = require(${spec});`;
+        return `const { ${usage.imported === local ? local : `${propertyKey(usage.imported)}: ${local}`} } = require(${spec});`;
+    }
+    if (usage.imported === "default" && usage.style === "esm")
+        return `import ${local} from ${spec};`;
+    if (usage.imported === "*")
+        return model.callableRoot ? `import ${local} from ${spec};` : `import * as ${local} from ${spec};`;
+    return `import { ${usage.imported === local ? local : `${JSON.stringify(usage.imported)} as ${local}`} } from ${spec};`;
+}
+/**
+ * Decides which imports to mock and writes the mocks. Assertloom writes every jest.mock() factory
+ * itself from how the source uses each import; the model only configures return values in tests.
+ */
+export async function planMocks(options) {
+    const { analysis, sourcePath, testPath } = options;
+    const commonjs = analysis.moduleSystem === "commonjs";
+    const typescript = analysis.language === "typescript";
+    const names = new NameAllocator(options.reserved);
+    const mocked = [];
+    const real = [];
+    const seen = new Set();
+    for (const entry of analysis.imports) {
+        if (entry.typeOnly || (entry.kind !== "import" && entry.kind !== "require") || seen.has(entry.source))
+            continue;
+        seen.add(entry.source);
+        let decision;
+        if (matches(options.keepReal, entry.source, entry.packageName)) {
+            decision = { mock: false, virtual: false, reason: "kept real with --keep-real" };
+        }
+        else {
+            decision = await decide(entry.source, entry.category, entry.packageName, sourcePath, new Set([sourcePath]), 1);
+            if (!decision.mock && matches(options.mock, entry.source, entry.packageName)) {
+                const missing = entry.category === "external" ? !isInstalled(sourcePath, entry.source) : entry.category === "local" && !(await resolveLocal(sourcePath, entry.source));
+                decision = { mock: true, virtual: missing, reason: "mocked with --mock" };
+            }
+        }
+        if (decision.mock) {
+            mocked.push({
+                source: entry.source,
+                specifier: specifierFromTest(entry.source, sourcePath, testPath),
+                category: classifySource(entry.source).category,
+                virtual: decision.virtual,
+                reason: decision.reason,
+            });
+        }
+        else {
+            real.push({ source: entry.source, reason: decision.reason });
+        }
+    }
+    const mockCalls = [];
+    const imports = [];
+    const handles = [];
+    const summary = [];
+    for (const module of mocked) {
+        const usages = analysis.importUsage.filter((u) => u.source === module.source);
+        const model = modelModule(usages);
+        const virtual = module.virtual ? ", { virtual: true }" : "";
+        mockCalls.push(`jest.mock(${JSON.stringify(module.specifier)}, ${factoryFor(model)}${virtual});`);
+        const visible = usages.filter((u) => u.local && !names.has(u.local));
+        const locals = [...new Map(visible.map((u) => [u.local, u])).values()];
+        for (const usage of locals) {
+            names.claim(usage.local);
+            imports.push(importStatement(usage, module.specifier, model, commonjs));
+        }
+        const description = { handles: [], lines: [], fns: [] };
+        for (const usage of locals) {
+            const shape = shapeFor(model, usage);
+            if (shape)
+                describeShape(shape, usage.local, names, description, usage.imported === "*" || usage.imported === "default");
+        }
+        handles.push(...description.handles.map((h) => `const ${h.name} = ${h.init};`));
+        const detail = module.virtual ? `${module.reason}; virtual mock` : module.reason;
+        summary.push(`- ${JSON.stringify(module.source)} (${detail}). Imported in the test file as: ${locals.map((u) => `\`${u.local}\``).join(", ") || "(nothing)"}`);
+        for (const line of description.lines)
+            summary.push(`  - ${line}`);
+        if (description.fns.length)
+            summary.push(`  - jest.fn() mocks: ${[...new Set(description.fns)].map((f) => `\`${f}\``).join(", ")}`);
+    }
+    // Pure built-ins such as `path` are also imported into the test, so expected values can be built portably.
+    const helpers = [];
+    for (const module of real) {
+        if (classifySource(module.source).category !== "builtin")
+            continue;
+        const usages = analysis.importUsage.filter((u) => u.source === module.source && u.local && !names.has(u.local));
+        const model = modelModule(usages);
+        for (const usage of [...new Map(usages.map((u) => [u.local, u])).values()]) {
+            names.claim(usage.local);
+            imports.push(importStatement(usage, module.source, model, commonjs));
+            helpers.push(`\`${usage.local}\` (${module.source})`);
+        }
+    }
+    if (helpers.length)
+        summary.push(`- Not mocked, and also imported in the test file for building expected values: ${helpers.join(", ")}`);
+    const otherReal = real.filter((r) => classifySource(r.source).category !== "builtin");
+    if (otherReal.length)
+        summary.push(`- Not mocked (real code runs): ${otherReal.map((r) => JSON.stringify(r.source)).join(", ")}`);
+    const globals = analysis.globals;
+    const globalLines = [];
+    for (const name of globals) {
+        const original = names.claim(`original${name[0].toUpperCase()}${name.slice(1)}`);
+        globalLines.push(`const ${original} = globalThis.${name};`, `globalThis.${name} = jest.fn()${typescript ? ` as unknown as typeof ${name}` : ""};`, `afterAll(() => {\n  globalThis.${name} = ${original};\n});`);
+        summary.push(`- global \`${name}\` is a jest.fn() mock for every test.`);
+    }
+    const setup = [handles.join("\n"), globalLines.join("\n")].filter(Boolean);
+    if (mocked.length > 0 || globals.length > 0) {
+        setup.push("beforeEach(() => {\n  jest.resetAllMocks();\n});", "afterEach(() => {\n  jest.restoreAllMocks();\n});");
+    }
+    return {
+        mocked,
+        real,
+        globals,
+        imports: [...mockCalls, ...(mockCalls.length ? [""] : []), ...imports],
+        setup,
+        summary: summary.join("\n"),
+    };
+}
+//# sourceMappingURL=mocks.js.map

package/dist/generator/mocks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mocks.js","sourceRoot":"","sources":["../../src/generator/mocks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC1E,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACtE,OAAO,EAAE,oBAAoB,EAAsG,MAAM,mBAAmB,CAAC;AAE7J,gFAAgF;AAChF,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;AAE/I,qGAAqG;AACrG,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC;IACnC,gBAAgB,EAAE,QAAQ,EAAE,cAAc,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY;IAClH,WAAW,EAAE,iBAAiB,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO;IAC7G,WAAW,EAAE,KAAK,EAAE,KAAK;CAC1B,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,qBAAqB,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;AAiD9D,KAAK,UAAU,MAAM,CAAC,IAAY;IAChC,IAAI,CAAC;QACH,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAAgB,EAAE,SAAiB;IACpE,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAChD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;IAC7D,MAAM,UAAU,GAAG,CAAC,IAAI,CAAC,CAAC;IAC1B,MAAM,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9C,IAAI,WAAW;QAAE,UAAU,CAAC,IAAI,CAAC,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IACrH,UAAU,CAAC,IAAI,CAAC,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;IACxI,KAAK,MAAM,SAAS,IAAI,UAAU;QAAE,IAAI,MAAM,MAAM,CAAC,SAAS,CAAC;YAAE,OAAO,SAAS,CAAC;IAClF,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,WAAW,CAAC,QAAgB,EAAE,SAAiB;IACtD,IAAI,CAAC;QACH,aAAa,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,oGAAoG;AACpG,KAAK,UAAU,aAAa,CAAC,IAAY,EAAE,OAAoB,EAAE,KAAa;IAC5E,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IACxC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAClB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,IAAI,CAAE,oBAA0C,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IACvF,IAAI,KAAK,GAAG,eAAe;QAAE,OAAO,6BAA6B,CAAC;IAElE,IAAI,GAAG,CAAC;IACR,IAAI,CAAC;QACH,GAAG,GAAG,WAAW,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,SAA+B,CAAC,CAAC;IACnF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,6BAA6B,CAAC;IAC7D,CAAC;IAED,MAAM,EAAE,OAAO,EAAE,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;IAC5C,IAAI,OAAO,CAAC,MAAM;QAAE,OAAO,UAAU,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IAElD,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;QAC7C,IAAI,KAAK,CAAC,QAAQ;YAAE,SAAS;QAC7B,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;QACzG,IAAI,QAAQ,CAAC,IAAI;YAAE,OAAO,KAAK,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;IAC/G,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,KAAK,UAAU,MAAM,CACnB,MAAc,EACd,QAAwB,EACxB,WAA+B,EAC/B,QAAgB,EAChB,OAAoB,EACpB,KAAa;IAEb,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACpD,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,oBAAoB,EAAE;YAC/D,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,iCAAiC,EAAE,CAAC;IAChF,CAAC;IAED,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QAC5B,IAAI,WAAW,IAAI,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC;YAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,6BAA6B,EAAE,CAAC;QACjI,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAChD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,iCAAiC,EAAE,CAAC;IACzH,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAClD,IAAI,CAAC,IAAI;QAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC;IAClF,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;IAC3D,OAAO,QAAQ;QACb,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,QAAQ,EAAE,EAAE;QAClE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,mCAAmC,EAAE,CAAC;AACnF,CAAC;AAED,SAAS,OAAO,CAAC,IAA0B,EAAE,MAAc,EAAE,WAA+B;IAC1F,OAAO,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,KAAK,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5I,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc,EAAE,UAAkB,EAAE,QAAgB;IAC7E,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,MAAM,CAAC;IAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC7H,OAAO,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;AAC/D,CAAC;AAED,SAAS,WAAW,CAAC,CAAyB,EAAE,CAAa;IAC3D,MAAM,MAAM,GAAe,CAAC;QAC1B,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,WAAW,CAAC,EAAE;QACtE,CAAC,CAAC,EAAE,GAAG,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;IACtD,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,CAAC;IAC3B,MAAM,CAAC,WAAW,KAAK,CAAC,CAAC,WAAW,CAAC;IACrC,MAAM,CAAC,UAAU,KAAK,CAAC,CAAC,UAAU,CAAC;IACnC,MAAM,CAAC,aAAa,KAAK,CAAC,CAAC,aAAa,CAAC;IACzC,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,WAAW;QAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxG,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;IACpH,IAAI,CAAC,CAAC,MAAM;QAAE,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;IACnE,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AACH,SAAS,cAAc,CAAC,KAAiB;IACvC,OAAO,CAAC,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;AACtE,CAAC;AAED,MAAM,UAAU,GAAG,CAAC,KAAiB,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,UAAU,CAAC;AAEhG,SAAS,WAAW,CAAC,IAAY;IAC/B,OAAO,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,aAAa;IACR,MAAM,GAAa,EAAE,CAAC;IACvB,KAAK,GAAG,CAAC,CAAC;IAElB,OAAO,CAAC,KAAiB;QACvB,OAAO,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,WAAW,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1G,CAAC;IAED,gHAAgH;IAChH,KAAK,CAAC,KAAiB;QACrB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,WAAW,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB,IAAI,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEpH,IAAI,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,WAAW,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAO,CAAC,GAAG,CAAC,CAAC;YACnE,OAAO,WAAW,CAAC,wBAAwB,IAAI,KAAK,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW;YAAE,OAAO,WAAW,CAAC,UAAU,CAAC,CAAC;QAC5F,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,WAAW,CAAC,WAAW,CAAC,CAAC;QAC/F,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IACrC,CAAC;IAED,MAAM,CAAC,KAAiB;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAC7D,CAAC;CACF;AAUD,SAAS,WAAW,CAAC,MAAqB;IACxC,IAAI,YAAoC,CAAC;IACzC,IAAI,WAAW,GAAG,UAAU,EAAE,CAAC;IAC/B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,KAAK;YAAE,YAAY,GAAG,WAAW,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;aAC5G,IAAI,KAAK,CAAC,QAAQ,KAAK,GAAG;YAAE,WAAW,GAAG,WAAW,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;;YAChF,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;IAC3G,CAAC;IACD,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC,YAAY,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;AAC9I,CAAC;AAED,SAAS,UAAU,CAAC,KAAkB;IACpC,MAAM,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;IACnC,IAAI,KAAa,CAAC;IAClB,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;QACvB,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAC1C,CAAC;SAAM,CAAC;QACN,MAAM,OAAO,GAAG;YACd,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/E,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;SACrC,CAAC;QACF,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAC9D,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,UAAU,KAAK,GAAG,CAAC;IAC1D,OAAO,cAAc,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,KAAK,MAAM,CAAC;AAC3E,CAAC;AAED,SAAS,QAAQ,CAAC,KAAkB,EAAE,KAAkB;IACtD,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,KAAK;QAAE,OAAO,KAAK,CAAC,YAAY,CAAC;IACrF,IAAI,KAAK,CAAC,QAAQ,KAAK,GAAG;QAAE,OAAO,KAAK,CAAC,WAAW,CAAC;IACrD,OAAO,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,aAAa;IACA,IAAI,CAAc;IACnC,YAAY,QAA0B;QACpC,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IACD,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IACD,KAAK,CAAC,SAAiB;QACrB,IAAI,IAAI,GAAG,SAAS,CAAC;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE;YAAE,IAAI,GAAG,GAAG,SAAS,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACvF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAQD,SAAS,aAAa,CAAC,KAAiB,EAAE,IAAY,EAAE,KAAoB,EAAE,GAAgB,EAAE,MAAe;IAC7G,IAAI,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,IAAI,MAAM,CAAC;QAC1E,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAC3I,MAAM,IAAI,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC;QAC/D,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACjC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,8DAA8D,IAAI,IAAI,CAAC,CAAC;QAChG,aAAa,CAAC,KAAK,CAAC,MAAO,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;IACxD,CAAC;SAAM,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;QACrG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IACD,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1D,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC;QACrG,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,KAAkB,EAAE,SAAiB,EAAE,KAAkB,EAAE,QAAiB;IACnG,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACvC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;IAC1B,IAAI,QAAQ,EAAE,CAAC;QACb,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,KAAK;YAAE,OAAO,SAAS,KAAK,cAAc,IAAI,YAAY,CAAC;QAC/G,IAAI,KAAK,CAAC,QAAQ,KAAK,GAAG;YAAE,OAAO,SAAS,KAAK,cAAc,IAAI,IAAI,CAAC;QACxE,OAAO,WAAW,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,KAAK,EAAE,gBAAgB,IAAI,IAAI,CAAC;IAC1H,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,KAAK,KAAK,KAAK;QAAE,OAAO,UAAU,KAAK,SAAS,IAAI,GAAG,CAAC;IAClG,IAAI,KAAK,CAAC,QAAQ,KAAK,GAAG;QAAE,OAAO,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,UAAU,KAAK,SAAS,IAAI,GAAG,CAAC,CAAC,CAAC,eAAe,KAAK,SAAS,IAAI,GAAG,CAAC;IAC/H,OAAO,YAAY,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,KAAK,EAAE,WAAW,IAAI,GAAG,CAAC;AAC1H,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,OAAoB;IAClD,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IACnD,MAAM,QAAQ,GAAG,QAAQ,CAAC,YAAY,KAAK,UAAU,CAAC;IACtD,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,KAAK,YAAY,CAAC;IACtD,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAElD,MAAM,MAAM,GAAmB,EAAE,CAAC;IAClC,MAAM,IAAI,GAAiB,EAAE,CAAC;IAC9B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;QACrC,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC;YAAE,SAAS;QAChH,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAEvB,IAAI,QAAkB,CAAC;QACvB,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC;YAC/D,QAAQ,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,4BAA4B,EAAE,CAAC;QACnF,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/G,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC7E,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,KAAK,OAAO,IAAI,CAAC,CAAC,MAAM,YAAY,CAAC,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;gBACvK,QAAQ,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,oBAAoB,EAAE,CAAC;YAC5E,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC;gBACV,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,SAAS,EAAE,iBAAiB,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,EAAE,QAAQ,CAAC;gBAChE,QAAQ,EAAE,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,QAAQ;gBAC/C,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,MAAM,EAAE,QAAQ,CAAC,MAAM;aACxB,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAa,EAAE,CAAC;IAE7B,KAAK,MAAM,MAAM,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9E,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QAClC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5D,SAAS,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,UAAU,CAAC,KAAK,CAAC,GAAG,OAAO,IAAI,CAAC,CAAC;QAElG,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACvE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACzB,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,SAAS,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,WAAW,GAAgB,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC;QACrE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACrC,IAAI,KAAK;gBAAE,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,QAAQ,KAAK,GAAG,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC;QAC3H,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;QAEhF,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;QACjF,OAAO,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,MAAM,oCAAoC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,WAAW,EAAE,CAAC,CAAC;QAC/J,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,KAAK;YAAE,OAAO,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAClE,IAAI,WAAW,CAAC,GAAG,CAAC,MAAM;YAAE,OAAO,CAAC,IAAI,CAAC,wBAAwB,CAAC,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACtI,CAAC;IAED,2GAA2G;IAC3G,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,KAAK,MAAM,MAAM,IAAI,IAAI,EAAE,CAAC;QAC1B,IAAI,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,KAAK,SAAS;YAAE,SAAS;QACnE,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QAChH,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QAClC,KAAK,MAAM,KAAK,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;YAC3E,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACzB,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;YACrE,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,CAAC,KAAK,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IACD,IAAI,OAAO,CAAC,MAAM;QAAE,OAAO,CAAC,IAAI,CAAC,kFAAkF,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzI,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC;IACtF,IAAI,SAAS,CAAC,MAAM;QAAE,OAAO,CAAC,IAAI,CAAC,kCAAkC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAElI,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;IACjC,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACjF,WAAW,CAAC,IAAI,CACd,SAAS,QAAQ,iBAAiB,IAAI,GAAG,EACzC,cAAc,IAAI,eAAe,UAAU,CAAC,CAAC,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,EACrF,kCAAkC,IAAI,MAAM,QAAQ,QAAQ,CAC7D,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,cAAc,IAAI,wCAAwC,CAAC,CAAC;IAC3E,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3E,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5C,KAAK,CAAC,IAAI,CAAC,kDAAkD,EAAE,mDAAmD,CAAC,CAAC;IACtH,CAAC;IAED,OAAO;QACL,MAAM;QACN,IAAI;QACJ,OAAO;QACP,OAAO,EAAE,CAAC,GAAG,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,OAAO,CAAC;QACtE,KAAK;QACL,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;KAC5B,CAAC;AACJ,CAAC"}

package/dist/generator/schema.d.ts

@@ -0,0 +1,40 @@
+import { z } from "zod";
+export declare const TEST_CATEGORIES: readonly ["normal", "edge", "error"];
+export declare const GeneratedTestCaseSchema: z.ZodObject<{
+    name: z.ZodString;
+    category: z.ZodEnum<{
+        edge: "edge";
+        error: "error";
+        normal: "normal";
+    }>;
+    body: z.ZodString;
+}, z.core.$strip>;
+export declare const GeneratedSuiteSchema: z.ZodObject<{
+    functionName: z.ZodString;
+    cases: z.ZodArray<z.ZodObject<{
+        name: z.ZodString;
+        category: z.ZodEnum<{
+            edge: "edge";
+            error: "error";
+            normal: "normal";
+        }>;
+        body: z.ZodString;
+    }, z.core.$strip>>;
+}, z.core.$strip>;
+export declare const GenerationSchema: z.ZodObject<{
+    suites: z.ZodArray<z.ZodObject<{
+        functionName: z.ZodString;
+        cases: z.ZodArray<z.ZodObject<{
+            name: z.ZodString;
+            category: z.ZodEnum<{
+                edge: "edge";
+                error: "error";
+                normal: "normal";
+            }>;
+            body: z.ZodString;
+        }, z.core.$strip>>;
+    }, z.core.$strip>>;
+}, z.core.$strip>;
+export type GeneratedTestCase = z.infer<typeof GeneratedTestCaseSchema>;
+export type GeneratedSuite = z.infer<typeof GeneratedSuiteSchema>;
+export type Generation = z.infer<typeof GenerationSchema>;

package/dist/generator/schema.js

@@ -0,0 +1,15 @@
+import { z } from "zod";
+export const TEST_CATEGORIES = ["normal", "edge", "error"];
+export const GeneratedTestCaseSchema = z.object({
+    name: z.string().describe("Short description of the behavior under test, used as the `it` title"),
+    category: z.enum(TEST_CATEGORIES).describe("normal: typical input; edge: boundaries and unusual input; error: throws or rejects"),
+    body: z.string().describe("Statements inside the test callback, without the it() wrapper"),
+});
+export const GeneratedSuiteSchema = z.object({
+    functionName: z.string().describe("Binding name of the function under test, exactly as listed in the prompt"),
+    cases: z.array(GeneratedTestCaseSchema),
+});
+export const GenerationSchema = z.object({
+    suites: z.array(GeneratedSuiteSchema),
+});
+//# sourceMappingURL=schema.js.map

package/dist/generator/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/generator/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAU,CAAC;AAEpE,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sEAAsE,CAAC;IACjG,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,qFAAqF,CAAC;IACjI,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+DAA+D,CAAC;CAC3F,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0EAA0E,CAAC;IAC7G,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,uBAAuB,CAAC;CACxC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,oBAAoB,CAAC;CACtC,CAAC,CAAC"}

package/dist/generator/testGenerator.d.ts

@@ -0,0 +1,33 @@
+import { type Budget } from "../ai/pricing.js";
+import type { LLMProvider, TokenUsage } from "../ai/provider.js";
+import { UserError } from "../errors.js";
+import type { FileAnalysis, SourceFile } from "../types/index.js";
+import { type MockPlan } from "./mocks.js";
+import { type RejectedCase, type ValidatedSuite } from "./validate.js";
+export declare class GenerationError extends UserError {
+}
+export interface GenerationResult {
+    analysis: FileAnalysis;
+    testPath: string;
+    content: string;
+    /** Mocks, imports, and setup that precede the describe blocks. */
+    preamble: string[];
+    mocks: MockPlan;
+    suites: ValidatedSuite[];
+    /** The source as sent to the model, with secrets redacted. */
+    promptSource: string;
+    rejected: RejectedCase[];
+    model: string;
+    usage: TokenUsage;
+    redactions: number;
+}
+export declare function generateTests(options: {
+    source: SourceFile;
+    provider: LLMProvider;
+    testPath?: string;
+    /** Modules to mock even if Assertloom would keep them real. */
+    mock?: string[];
+    /** Modules to keep real even if Assertloom would mock them. */
+    keepReal?: string[];
+    budget?: Budget;
+}): Promise<GenerationResult>;