@theihtisham/ai-testgen 1.0.0

package/dist/config/loader.js

@@ -0,0 +1,126 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findConfig = findConfig;
+exports.loadConfig = loadConfig;
+exports.resolveConfig = resolveConfig;
+exports.createSampleConfig = createSampleConfig;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const yaml = __importStar(require("js-yaml"));
+const defaults_js_1 = require("./defaults.js");
+const CONFIG_FILENAMES = ['.aitestgen.yml', '.aitestgen.yaml', '.aitestgen.json'];
+function findConfig(startDir) {
+    let current = path.resolve(startDir);
+    const root = path.parse(current).root;
+    while (current !== root) {
+        for (const filename of CONFIG_FILENAMES) {
+            const configPath = path.join(current, filename);
+            if (fs.existsSync(configPath)) {
+                return configPath;
+            }
+        }
+        current = path.dirname(current);
+    }
+    return null;
+}
+function loadConfig(configPath) {
+    const ext = path.extname(configPath);
+    const raw = fs.readFileSync(configPath, 'utf-8');
+    let parsed;
+    if (ext === '.json') {
+        parsed = JSON.parse(raw);
+    }
+    else {
+        parsed = yaml.load(raw);
+    }
+    return deepMerge(defaults_js_1.DEFAULT_CONFIG, parsed);
+}
+function resolveConfig(cliOverrides, projectDir) {
+    const configPath = cliOverrides['config'];
+    const foundPath = configPath ?? findConfig(projectDir);
+    const base = foundPath ? loadConfig(foundPath) : { ...defaults_js_1.DEFAULT_CONFIG };
+    return deepMerge(base, cliOverrides);
+}
+function deepMerge(target, source) {
+    const result = { ...target };
+    for (const key of Object.keys(source)) {
+        const sourceVal = source[key];
+        const targetVal = target[key];
+        if (isPlainObject(sourceVal) &&
+            isPlainObject(targetVal)) {
+            result[key] = deepMerge(targetVal, sourceVal);
+        }
+        else if (sourceVal !== undefined) {
+            result[key] = sourceVal;
+        }
+    }
+    return result;
+}
+function isPlainObject(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function createSampleConfig(outputPath) {
+    const sample = {
+        version: '1.0.0',
+        language: 'auto',
+        framework: 'auto',
+        outputDir: '__tests__',
+        coverage: { target: 90, strict: false },
+        ai: {
+            enabled: true,
+            provider: 'openai',
+            model: 'gpt-4o',
+            apiKeyEnv: 'OPENAI_API_KEY',
+            maxTokens: 4096,
+            temperature: 0.2,
+            privacyMode: true,
+        },
+        generation: {
+            unitTests: true,
+            integrationTests: true,
+            edgeCaseTests: true,
+            mockGeneration: true,
+            mutationTesting: false,
+            maxTestsPerFunction: 10,
+            includeComments: true,
+        },
+        exclude: ['node_modules/**', 'dist/**', 'coverage/**'],
+        include: ['src/**/*.{ts,tsx,js,jsx}'],
+    };
+    const content = yaml.dump(sample, { indent: 2, lineWidth: 100 });
+    fs.writeFileSync(outputPath, content, 'utf-8');
+}
+//# sourceMappingURL=loader.js.map

package/dist/config/loader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loader.js","sourceRoot":"","sources":["../../src/config/loader.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAQA,gCAcC;AAED,gCAeC;AAED,sCAYC;AA6BD,gDA+BC;AAjHD,uCAAyB;AACzB,2CAA6B;AAC7B,8CAAgC;AAEhC,+CAA+C;AAE/C,MAAM,gBAAgB,GAAG,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;AAElF,SAAgB,UAAU,CAAC,QAAgB;IACzC,IAAI,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACrC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC;IAEtC,OAAO,OAAO,KAAK,IAAI,EAAE,CAAC;QACxB,KAAK,MAAM,QAAQ,IAAI,gBAAgB,EAAE,CAAC;YACxC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAChD,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC9B,OAAO,UAAU,CAAC;YACpB,CAAC;QACH,CAAC;QACD,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAgB,UAAU,CAAC,UAAkB;IAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACrC,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAEjD,IAAI,MAA+B,CAAC;IACpC,IAAI,GAAG,KAAK,OAAO,EAAE,CAAC;QACpB,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4B,CAAC;IACtD,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAA4B,CAAC;IACrD,CAAC;IAED,OAAO,SAAS,CACd,4BAAoD,EACpD,MAAM,CACqB,CAAC;AAChC,CAAC;AAED,SAAgB,aAAa,CAC3B,YAAoC,EACpC,UAAkB;IAElB,MAAM,UAAU,GAAG,YAAY,CAAC,QAAqC,CAAuB,CAAC;IAC7F,MAAM,SAAS,GAAG,UAAU,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,4BAAc,EAAE,CAAC;IAEvE,OAAO,SAAS,CACd,IAA0C,EAC1C,YAAkD,CACvB,CAAC;AAChC,CAAC;AAED,SAAS,SAAS,CAAC,MAA+B,EAAE,MAA+B;IACjF,MAAM,MAAM,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;IAE7B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC9B,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAE9B,IACE,aAAa,CAAC,SAAS,CAAC;YACxB,aAAa,CAAC,SAAS,CAAC,EACxB,CAAC;YACD,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CACrB,SAAoC,EACpC,SAAoC,CACrC,CAAC;QACJ,CAAC;aAAM,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAgB,kBAAkB,CAAC,UAAkB;IACnD,MAAM,MAAM,GAA4B;QACtC,OAAO,EAAE,OAAO;QAChB,QAAQ,EAAE,MAAM;QAChB,SAAS,EAAE,MAAM;QACjB,SAAS,EAAE,WAAW;QACtB,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE;QACvC,EAAE,EAAE;YACF,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,QAAQ;YACf,SAAS,EAAE,gBAAgB;YAC3B,SAAS,EAAE,IAAI;YACf,WAAW,EAAE,GAAG;YAChB,WAAW,EAAE,IAAI;SAClB;QACD,UAAU,EAAE;YACV,SAAS,EAAE,IAAI;YACf,gBAAgB,EAAE,IAAI;YACtB,aAAa,EAAE,IAAI;YACnB,cAAc,EAAE,IAAI;YACpB,eAAe,EAAE,KAAK;YACtB,mBAAmB,EAAE,EAAE;YACvB,eAAe,EAAE,IAAI;SACtB;QACD,OAAO,EAAE,CAAC,iBAAiB,EAAE,SAAS,EAAE,aAAa,CAAC;QACtD,OAAO,EAAE,CAAC,0BAA0B,CAAC;KACtC,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;IACjE,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC"}

package/dist/coverage.d.ts

@@ -0,0 +1,4 @@
+import { SourceAnalysis, CoveragePrediction, TestSuite } from './types.js';
+export declare function predictCoverage(analysis: SourceAnalysis, suite: TestSuite): CoveragePrediction;
+export declare function formatCoverageReport(prediction: CoveragePrediction): string;
+//# sourceMappingURL=coverage.d.ts.map

package/dist/coverage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"coverage.d.ts","sourceRoot":"","sources":["../src/coverage.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,SAAS,EACV,MAAM,YAAY,CAAC;AAEpB,wBAAgB,eAAe,CAAC,QAAQ,EAAE,cAAc,EAAE,KAAK,EAAE,SAAS,GAAG,kBAAkB,CA4F9F;AAED,wBAAgB,oBAAoB,CAAC,UAAU,EAAE,kBAAkB,GAAG,MAAM,CA2B3E"}

package/dist/coverage.js

@@ -0,0 +1,108 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.predictCoverage = predictCoverage;
+exports.formatCoverageReport = formatCoverageReport;
+function predictCoverage(analysis, suite) {
+    const uncoveredPaths = [];
+    const suggestions = [];
+    const testedNames = new Set(suite.testCases.map((tc) => tc.name.split(' ')[0] ?? ''));
+    // Check function coverage
+    let coveredFunctions = 0;
+    for (const fn of analysis.functions) {
+        if (fn.isExported) {
+            const isTested = testedNames.has(fn.name);
+            if (isTested) {
+                coveredFunctions++;
+            }
+            else {
+                uncoveredPaths.push(`Function ${fn.name} is not tested`);
+                suggestions.push(`Add tests for exported function: ${fn.name}`);
+            }
+        }
+    }
+    // Check class method coverage
+    let coveredMethods = 0;
+    let totalMethods = 0;
+    for (const cls of analysis.classes) {
+        for (const method of cls.methods) {
+            totalMethods++;
+            const isTested = testedNames.has(method.name) || Array.from(testedNames).some((n) => n.includes(method.name));
+            if (isTested) {
+                coveredMethods++;
+            }
+            else {
+                uncoveredPaths.push(`${cls.name}.${method.name} is not tested`);
+                suggestions.push(`Add tests for ${cls.name}.${method.name} method`);
+            }
+        }
+    }
+    // Check for uncovered error paths
+    for (const fn of analysis.functions) {
+        if (fn.throws.length > 0) {
+            const hasErrorTest = suite.testCases.some((tc) => tc.name.includes(fn.name) && (tc.tags.includes('error') || tc.tags.includes('edge-case')));
+            if (!hasErrorTest) {
+                uncoveredPaths.push(`${fn.name} error paths not tested`);
+                suggestions.push(`Add error path tests for ${fn.name} (throws: ${fn.throws.join(', ')})`);
+            }
+        }
+    }
+    // Check for untested async error paths
+    for (const fn of analysis.functions) {
+        if (fn.isAsync) {
+            const hasAsyncErrorTest = suite.testCases.some((tc) => tc.name.includes(fn.name) && tc.tags.includes('async'));
+            if (!hasAsyncErrorTest) {
+                suggestions.push(`Add async rejection test for ${fn.name}`);
+            }
+        }
+    }
+    // Check for untested optional params
+    for (const fn of analysis.functions) {
+        const optionalParams = fn.params.filter((p) => p.optional);
+        if (optionalParams.length > 0) {
+            const hasOptionalTest = suite.testCases.some((tc) => tc.name.includes(fn.name) && tc.name.includes('undefined'));
+            if (!hasOptionalTest) {
+                suggestions.push(`Test ${fn.name} without optional params: ${optionalParams.map((p) => p.name).join(', ')}`);
+            }
+        }
+    }
+    // Calculate estimates
+    const totalExported = analysis.functions.filter((f) => f.isExported).length + totalMethods;
+    const covered = coveredFunctions + coveredMethods;
+    const functionCoverage = totalExported > 0 ? Math.round((covered / totalExported) * 100) : 100;
+    const lineCoverage = Math.min(functionCoverage + 10, suite.coverageEstimate);
+    const branchCoverage = Math.max(functionCoverage - 10, 0);
+    return {
+        filePath: analysis.filePath,
+        estimatedLineCoverage: lineCoverage,
+        estimatedBranchCoverage: branchCoverage,
+        estimatedFunctionCoverage: functionCoverage,
+        uncoveredPaths: uncoveredPaths.slice(0, 10),
+        suggestions: suggestions.slice(0, 10),
+    };
+}
+function formatCoverageReport(prediction) {
+    const lines = [];
+    lines.push('');
+    lines.push(`  Coverage Prediction: ${prediction.filePath}`);
+    lines.push('  ----------------------------------------');
+    lines.push(`  Line coverage:     ~${prediction.estimatedLineCoverage}%`);
+    lines.push(`  Branch coverage:   ~${prediction.estimatedBranchCoverage}%`);
+    lines.push(`  Function coverage: ~${prediction.estimatedFunctionCoverage}%`);
+    if (prediction.uncoveredPaths.length > 0) {
+        lines.push('');
+        lines.push('  Uncovered paths:');
+        for (const path of prediction.uncoveredPaths) {
+            lines.push(`    - ${path}`);
+        }
+    }
+    if (prediction.suggestions.length > 0) {
+        lines.push('');
+        lines.push('  Suggestions:');
+        for (const suggestion of prediction.suggestions) {
+            lines.push(`    + ${suggestion}`);
+        }
+    }
+    lines.push('');
+    return lines.join('\n');
+}
+//# sourceMappingURL=coverage.js.map

package/dist/coverage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"coverage.js","sourceRoot":"","sources":["../src/coverage.ts"],"names":[],"mappings":";;AAMA,0CA4FC;AAED,oDA2BC;AAzHD,SAAgB,eAAe,CAAC,QAAwB,EAAE,KAAgB;IACxE,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAEtF,0BAA0B;IAC1B,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACpC,IAAI,EAAE,CAAC,UAAU,EAAE,CAAC;YAClB,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,QAAQ,EAAE,CAAC;gBACb,gBAAgB,EAAE,CAAC;YACrB,CAAC;iBAAM,CAAC;gBACN,cAAc,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,gBAAgB,CAAC,CAAC;gBACzD,WAAW,CAAC,IAAI,CAAC,oCAAoC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAClE,CAAC;QACH,CAAC;IACH,CAAC;IAED,8BAA8B;IAC9B,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;QACnC,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YACjC,YAAY,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;YAC9G,IAAI,QAAQ,EAAE,CAAC;gBACb,cAAc,EAAE,CAAC;YACnB,CAAC;iBAAM,CAAC;gBACN,cAAc,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,gBAAgB,CAAC,CAAC;gBAChE,WAAW,CAAC,IAAI,CAAC,iBAAiB,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,SAAS,CAAC,CAAC;YACtE,CAAC;QACH,CAAC;IACH,CAAC;IAED,kCAAkC;IAClC,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACpC,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,YAAY,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CACvC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAClG,CAAC;YACF,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,cAAc,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,yBAAyB,CAAC,CAAC;gBACzD,WAAW,CAAC,IAAI,CAAC,4BAA4B,EAAE,CAAC,IAAI,aAAa,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5F,CAAC;QACH,CAAC;IACH,CAAC;IAED,uCAAuC;IACvC,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACpC,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;YACf,MAAM,iBAAiB,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAC5C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC/D,CAAC;YACF,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBACvB,WAAW,CAAC,IAAI,CAAC,gCAAgC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;IACH,CAAC;IAED,qCAAqC;IACrC,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACpC,MAAM,cAAc,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC3D,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,eAAe,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAC1C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CACnE,CAAC;YACF,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,WAAW,CAAC,IAAI,CACd,QAAQ,EAAE,CAAC,IAAI,6BAA6B,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC3F,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,sBAAsB;IACtB,MAAM,aAAa,GAAG,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,YAAY,CAAC;IAC3F,MAAM,OAAO,GAAG,gBAAgB,GAAG,cAAc,CAAC;IAClD,MAAM,gBAAgB,GAAG,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,GAAG,aAAa,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IAE/F,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,gBAAgB,GAAG,EAAE,EAAE,KAAK,CAAC,gBAAgB,CAAC,CAAC;IAC7E,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,gBAAgB,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;IAE1D,OAAO;QACL,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,qBAAqB,EAAE,YAAY;QACnC,uBAAuB,EAAE,cAAc;QACvC,yBAAyB,EAAE,gBAAgB;QAC3C,cAAc,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;QAC3C,WAAW,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;KACtC,CAAC;AACJ,CAAC;AAED,SAAgB,oBAAoB,CAAC,UAA8B;IACjE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,0BAA0B,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC5D,KAAK,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;IACzD,KAAK,CAAC,IAAI,CAAC,yBAAyB,UAAU,CAAC,qBAAqB,GAAG,CAAC,CAAC;IACzE,KAAK,CAAC,IAAI,CAAC,yBAAyB,UAAU,CAAC,uBAAuB,GAAG,CAAC,CAAC;IAC3E,KAAK,CAAC,IAAI,CAAC,yBAAyB,UAAU,CAAC,yBAAyB,GAAG,CAAC,CAAC;IAE7E,IAAI,UAAU,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,KAAK,MAAM,IAAI,IAAI,UAAU,CAAC,cAAc,EAAE,CAAC;YAC7C,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,IAAI,UAAU,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC7B,KAAK,MAAM,UAAU,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;YAChD,KAAK,CAAC,IAAI,CAAC,SAAS,UAAU,EAAE,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/generators/ai-generator.d.ts

@@ -0,0 +1,4 @@
+import { SourceAnalysis, TestSuite, TestFramework, TestGenConfig } from '../types.js';
+export declare function generateWithAI(analysis: SourceAnalysis, framework: TestFramework, config: TestGenConfig): Promise<string>;
+export declare function buildAISuite(generatedCode: string, analysis: SourceAnalysis, framework: TestFramework): TestSuite;
+//# sourceMappingURL=ai-generator.d.ts.map

package/dist/generators/ai-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-generator.d.ts","sourceRoot":"","sources":["../../src/generators/ai-generator.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,cAAc,EACd,SAAS,EACT,aAAa,EACb,aAAa,EACd,MAAM,aAAa,CAAC;AAGrB,wBAAsB,cAAc,CAClC,QAAQ,EAAE,cAAc,EACxB,SAAS,EAAE,aAAa,EACxB,MAAM,EAAE,aAAa,GACpB,OAAO,CAAC,MAAM,CAAC,CAwCjB;AAkGD,wBAAgB,YAAY,CAC1B,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,cAAc,EACxB,SAAS,EAAE,aAAa,GACvB,SAAS,CAaX"}

package/dist/generators/ai-generator.js

@@ -0,0 +1,175 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateWithAI = generateWithAI;
+exports.buildAISuite = buildAISuite;
+const fs = __importStar(require("fs"));
+const openai_1 = __importDefault(require("openai"));
+const analyzer_js_1 = require("../analyzers/analyzer.js");
+async function generateWithAI(analysis, framework, config) {
+    if (!config.ai.enabled || config.ai.provider === 'none') {
+        throw new Error('AI generation is not enabled in configuration');
+    }
+    const apiKey = process.env[config.ai.apiKeyEnv];
+    if (!apiKey) {
+        throw new Error(`AI API key not found. Set the ${config.ai.apiKeyEnv} environment variable.`);
+    }
+    const client = new openai_1.default({ apiKey, baseURL: getBaseURL(config) });
+    const prompt = buildPrompt(analysis, framework, config);
+    const response = await client.chat.completions.create({
+        model: config.ai.model,
+        messages: [
+            {
+                role: 'system',
+                content: getSystemPrompt(framework),
+            },
+            {
+                role: 'user',
+                content: prompt,
+            },
+        ],
+        max_tokens: config.ai.maxTokens,
+        temperature: config.ai.temperature,
+    });
+    const content = response.choices[0]?.message?.content;
+    if (!content) {
+        throw new Error('AI returned empty response');
+    }
+    // Extract code from markdown code blocks if present
+    const codeBlockMatch = content.match(/```(?:typescript|javascript|python|go|tsx|jsx)?\n([\s\S]*?)```/);
+    return codeBlockMatch?.[1]?.trim() ?? content.trim();
+}
+function getBaseURL(config) {
+    if (config.ai.provider === 'openai')
+        return undefined; // uses default
+    if (config.ai.provider === 'anthropic')
+        return 'https://api.anthropic.com/v1';
+    return undefined;
+}
+function getSystemPrompt(framework) {
+    const frameworkName = framework === 'go-test' ? 'Go testing' : framework;
+    return `You are an expert test engineer. Generate comprehensive, well-structured test suites using ${frameworkName}.
+Follow these rules:
+- Use the appropriate testing patterns and best practices for ${frameworkName}
+- Include edge cases: null/undefined/nil, empty inputs, boundary values
+- Mock external dependencies properly
+- Include descriptive test names and comments
+- Handle both happy path and error cases
+- Return ONLY the test code, no explanations outside code comments`;
+}
+function buildPrompt(analysis, framework, config) {
+    const parts = [];
+    parts.push(`Generate a complete test suite for the following ${analysis.language} source file.`);
+    parts.push(`Use the ${framework} testing framework.`);
+    parts.push('');
+    if (config.ai.privacyMode) {
+        // Send only analysis metadata, not full source
+        parts.push('## Source Code Analysis');
+        parts.push(`File: ${analysis.filePath}`);
+        parts.push(`Language: ${analysis.language}`);
+        parts.push(`Lines of code: ${analysis.linesOfCode}`);
+        parts.push('');
+        parts.push('## Exported Functions');
+        for (const fn of analysis.functions.filter((f) => f.isExported)) {
+            const params = fn.params.map((p) => `${p.name}${p.type ? ': ' + p.type : ''}`).join(', ');
+            parts.push(`- ${fn.name}(${params})${fn.returnType ? ': ' + fn.returnType : ''}${fn.isAsync ? ' [async]' : ''}`);
+            if (fn.throws.length > 0) {
+                parts.push(`  Throws: ${fn.throws.join(', ')}`);
+            }
+        }
+        parts.push('');
+        parts.push('## Classes');
+        for (const cls of analysis.classes) {
+            parts.push(`- ${cls.name}${cls.extends ? ` extends ${cls.extends}` : ''}`);
+            for (const method of cls.methods) {
+                const params = method.params.map((p) => `${p.name}${p.type ? ': ' + p.type : ''}`).join(', ');
+                parts.push(`  - ${method.name}(${params})${method.returnType ? ': ' + method.returnType : ''}`);
+            }
+        }
+        parts.push('');
+        parts.push('## Interfaces');
+        for (const iface of analysis.interfaces) {
+            parts.push(`- ${iface.name}`);
+            for (const prop of iface.properties) {
+                parts.push(`  - ${prop.name}${prop.type ? ': ' + prop.type : ''}${prop.optional ? '?' : ''}`);
+            }
+        }
+        parts.push('');
+        parts.push('## Imports');
+        for (const imp of analysis.imports) {
+            parts.push(`- ${imp.modulePath}: ${imp.namedImports.join(', ') || 'default'}`);
+        }
+        parts.push('');
+        parts.push('## Edge Cases to Cover');
+        const edgeCases = analysis.functions.flatMap((fn) => (0, analyzer_js_1.detectEdgeCases)(fn)).slice(0, 10);
+        for (const ec of edgeCases) {
+            parts.push(`- ${ec.name}: ${ec.expectedBehavior}`);
+        }
+    }
+    else {
+        // Send full source code (not recommended for proprietary code)
+        parts.push('## Source Code');
+        parts.push('```' + analysis.language);
+        try {
+            const source = fs.readFileSync(analysis.filePath, 'utf-8');
+            parts.push(source);
+        }
+        catch {
+            parts.push('(Could not read source file)');
+        }
+        parts.push('```');
+    }
+    parts.push('');
+    parts.push('Generate tests that achieve at least 90% code coverage.');
+    return parts.join('\n');
+}
+function buildAISuite(generatedCode, analysis, framework) {
+    return {
+        filePath: '',
+        sourceFilePath: analysis.filePath,
+        language: analysis.language,
+        framework,
+        testCases: [],
+        mocks: [],
+        imports: [generatedCode],
+        setupCode: '',
+        teardownCode: '',
+        coverageEstimate: 90, // AI-generated tests target 90%+
+    };
+}
+//# sourceMappingURL=ai-generator.js.map

package/dist/generators/ai-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-generator.js","sourceRoot":"","sources":["../../src/generators/ai-generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAUA,wCA4CC;AAkGD,oCAiBC;AAzKD,uCAAyB;AACzB,oDAA4B;AAO5B,0DAA2D;AAEpD,KAAK,UAAU,cAAc,CAClC,QAAwB,EACxB,SAAwB,EACxB,MAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxD,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;IAChD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CACb,iCAAiC,MAAM,CAAC,EAAE,CAAC,SAAS,wBAAwB,CAC7E,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,gBAAM,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAEnE,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IAExD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;QACpD,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,KAAK;QACtB,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,eAAe,CAAC,SAAS,CAAC;aACpC;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,MAAM;aAChB;SACF;QACD,UAAU,EAAE,MAAM,CAAC,EAAE,CAAC,SAAS;QAC/B,WAAW,EAAE,MAAM,CAAC,EAAE,CAAC,WAAW;KACnC,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC;IACtD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;IAChD,CAAC;IAED,oDAAoD;IACpD,MAAM,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,gEAAgE,CAAC,CAAC;IACvG,OAAO,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;AACvD,CAAC;AAED,SAAS,UAAU,CAAC,MAAqB;IACvC,IAAI,MAAM,CAAC,EAAE,CAAC,QAAQ,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC,CAAC,eAAe;IACtE,IAAI,MAAM,CAAC,EAAE,CAAC,QAAQ,KAAK,WAAW;QAAE,OAAO,8BAA8B,CAAC;IAC9E,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,SAAwB;IAC/C,MAAM,aAAa,GAAG,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;IACzE,OAAO,8FAA8F,aAAa;;gEAEpD,aAAa;;;;;mEAKV,CAAC;AACpE,CAAC;AAED,SAAS,WAAW,CAClB,QAAwB,EACxB,SAAwB,EACxB,MAAqB;IAErB,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,CAAC,IAAI,CAAC,oDAAoD,QAAQ,CAAC,QAAQ,eAAe,CAAC,CAAC;IACjG,KAAK,CAAC,IAAI,CAAC,WAAW,SAAS,qBAAqB,CAAC,CAAC;IACtD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,IAAI,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;QAC1B,+CAA+C;QAC/C,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,SAAS,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QACzC,KAAK,CAAC,IAAI,CAAC,aAAa,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7C,KAAK,CAAC,IAAI,CAAC,kBAAkB,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;QACrD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACpC,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,EAAE,CAAC;YAChE,MAAM,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1F,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,IAAI,IAAI,MAAM,IAAI,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACjH,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,KAAK,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACzB,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnC,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3E,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBACjC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC9F,KAAK,CAAC,IAAI,CAAC,OAAO,MAAM,CAAC,IAAI,IAAI,MAAM,IAAI,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAClG,CAAC;QACH,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC5B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACxC,KAAK,CAAC,IAAI,CAAC,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9B,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAChG,CAAC;QACH,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACzB,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnC,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,SAAS,EAAE,CAAC,CAAC;QACjF,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACrC,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAA,6BAAe,EAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACvF,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;SAAM,CAAC;QACN,+DAA+D;QAC/D,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC7B,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC3D,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,KAAK,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAC7C,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;IAEtE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,SAAgB,YAAY,CAC1B,aAAqB,EACrB,QAAwB,EACxB,SAAwB;IAExB,OAAO;QACL,QAAQ,EAAE,EAAE;QACZ,cAAc,EAAE,QAAQ,CAAC,QAAQ;QACjC,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,SAAS;QACT,SAAS,EAAE,EAAE;QACb,KAAK,EAAE,EAAE;QACT,OAAO,EAAE,CAAC,aAAa,CAAC;QACxB,SAAS,EAAE,EAAE;QACb,YAAY,EAAE,EAAE;QAChB,gBAAgB,EAAE,EAAE,EAAE,iCAAiC;KACxD,CAAC;AACJ,CAAC"}

package/dist/generators/generator.d.ts

@@ -0,0 +1,4 @@
+import { SourceAnalysis, TestFramework, TestGenConfig, GenerationResult, CLIOptions } from '../types.js';
+export declare function generateTestSuite(analysis: SourceAnalysis, framework: TestFramework, config: TestGenConfig, options: CLIOptions): Promise<GenerationResult>;
+export declare function formatResults(results: GenerationResult[]): string;
+//# sourceMappingURL=generator.d.ts.map

package/dist/generators/generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.d.ts","sourceRoot":"","sources":["../../src/generators/generator.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,cAAc,EAEd,aAAa,EACb,aAAa,EACb,gBAAgB,EAChB,UAAU,EACX,MAAM,aAAa,CAAC;AASrB,wBAAsB,iBAAiB,CACrC,QAAQ,EAAE,cAAc,EACxB,SAAS,EAAE,aAAa,EACxB,MAAM,EAAE,aAAa,EACrB,OAAO,EAAE,UAAU,GAClB,OAAO,CAAC,gBAAgB,CAAC,CA2C3B;AAwBD,wBAAgB,aAAa,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG,MAAM,CA4BjE"}

package/dist/generators/generator.js

@@ -0,0 +1,121 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateTestSuite = generateTestSuite;
+exports.formatResults = formatResults;
+const js_ts_generator_js_1 = require("./js-ts-generator.js");
+const python_generator_js_1 = require("./python-generator.js");
+const go_generator_js_1 = require("./go-generator.js");
+const ai_generator_js_1 = require("./ai-generator.js");
+const language_js_1 = require("../utils/language.js");
+const file_js_1 = require("../utils/file.js");
+const path = __importStar(require("path"));
+async function generateTestSuite(analysis, framework, config, options) {
+    const startTime = Date.now();
+    let suite;
+    let usedAI = false;
+    // Try AI generation first if enabled
+    if (config.ai.enabled && config.ai.provider !== 'none' && !options.noAI) {
+        try {
+            const generatedCode = await (0, ai_generator_js_1.generateWithAI)(analysis, framework, config);
+            suite = (0, ai_generator_js_1.buildAISuite)(generatedCode, analysis, framework);
+            usedAI = true;
+        }
+        catch (err) {
+            const errorMessage = err instanceof Error ? err.message : String(err);
+            console.warn(`AI generation failed, falling back to AST-based: ${errorMessage}`);
+            suite = generateLocalSuite(analysis, framework, config);
+        }
+    }
+    else {
+        suite = generateLocalSuite(analysis, framework, config);
+    }
+    // Determine output path
+    const testFilePath = (0, language_js_1.buildTestFilePath)(analysis.filePath, options.output ?? config.outputDir, analysis.language, framework);
+    suite.filePath = testFilePath;
+    // Get the final code
+    const code = suite.imports[0] ?? '';
+    if (!options.dryRun) {
+        (0, file_js_1.writeFile)(testFilePath, code);
+    }
+    return {
+        sourceFile: analysis.filePath,
+        testFile: testFilePath,
+        testSuite: suite,
+        duration: Date.now() - startTime,
+        usedAI,
+    };
+}
+function generateLocalSuite(analysis, framework, config) {
+    const maxTests = config.generation.maxTestsPerFunction;
+    switch (analysis.language) {
+        case 'typescript':
+        case 'javascript':
+            return (0, js_ts_generator_js_1.generateJsTsTestSuite)(analysis, framework, maxTests);
+        case 'python':
+            return (0, python_generator_js_1.generatePythonTestSuite)(analysis, framework, maxTests);
+        case 'go':
+            return (0, go_generator_js_1.generateGoTestSuite)(analysis, framework, maxTests);
+        case 'rust':
+            throw new Error('Rust test generation requires AI mode. Enable AI in your configuration.');
+        default:
+            throw new Error(`Unsupported language: ${analysis.language}`);
+    }
+}
+function formatResults(results) {
+    const lines = [];
+    let totalTests = 0;
+    let totalDuration = 0;
+    let totalCoverage = 0;
+    for (const result of results) {
+        totalTests += result.testSuite.testCases.length;
+        totalDuration += result.duration;
+        totalCoverage += result.testSuite.coverageEstimate;
+    }
+    const avgCoverage = results.length > 0 ? Math.round(totalCoverage / results.length) : 0;
+    lines.push('');
+    lines.push(`  Files analyzed: ${results.length}`);
+    lines.push(`  Tests generated: ${totalTests}`);
+    lines.push(`  Avg coverage estimate: ${avgCoverage}%`);
+    lines.push(`  Total time: ${totalDuration}ms`);
+    lines.push(`  AI assisted: ${results.some((r) => r.usedAI) ? 'Yes' : 'No'}`);
+    lines.push('');
+    for (const result of results) {
+        lines.push(`  ${path.basename(result.testFile)}`);
+        lines.push(`    ${result.testSuite.testCases.length} tests | ~${result.testSuite.coverageEstimate}% coverage | ${result.duration}ms`);
+    }
+    return lines.join('\n');
+}
+//# sourceMappingURL=generator.js.map

package/dist/generators/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../src/generators/generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgBA,8CAgDC;AAwBD,sCA4BC;AA5GD,6DAA6D;AAC7D,+DAAgE;AAChE,uDAAwD;AACxD,uDAAiE;AACjE,sDAAyD;AACzD,8CAA6C;AAC7C,2CAA6B;AAEtB,KAAK,UAAU,iBAAiB,CACrC,QAAwB,EACxB,SAAwB,EACxB,MAAqB,EACrB,OAAmB;IAEnB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,KAAgB,CAAC;IACrB,IAAI,MAAM,GAAG,KAAK,CAAC;IAEnB,qCAAqC;IACrC,IAAI,MAAM,CAAC,EAAE,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC,QAAQ,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACxE,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,IAAA,gCAAc,EAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;YACxE,KAAK,GAAG,IAAA,8BAAY,EAAC,aAAa,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YACzD,MAAM,GAAG,IAAI,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,oDAAoD,YAAY,EAAE,CAAC,CAAC;YACjF,KAAK,GAAG,kBAAkB,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;SAAM,CAAC;QACN,KAAK,GAAG,kBAAkB,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IAC1D,CAAC;IAED,wBAAwB;IACxB,MAAM,YAAY,GAAG,IAAA,+BAAiB,EACpC,QAAQ,CAAC,QAAQ,EACjB,OAAO,CAAC,MAAM,IAAI,MAAM,CAAC,SAAS,EAClC,QAAQ,CAAC,QAAQ,EACjB,SAAS,CACV,CAAC;IACF,KAAK,CAAC,QAAQ,GAAG,YAAY,CAAC;IAE9B,qBAAqB;IACrB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAEpC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACpB,IAAA,mBAAS,EAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,OAAO;QACL,UAAU,EAAE,QAAQ,CAAC,QAAQ;QAC7B,QAAQ,EAAE,YAAY;QACtB,SAAS,EAAE,KAAK;QAChB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;QAChC,MAAM;KACP,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,QAAwB,EACxB,SAAwB,EACxB,MAAqB;IAErB,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC;IAEvD,QAAQ,QAAQ,CAAC,QAAQ,EAAE,CAAC;QAC1B,KAAK,YAAY,CAAC;QAClB,KAAK,YAAY;YACf,OAAO,IAAA,0CAAqB,EAAC,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC9D,KAAK,QAAQ;YACX,OAAO,IAAA,6CAAuB,EAAC,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAChE,KAAK,IAAI;YACP,OAAO,IAAA,qCAAmB,EAAC,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC5D,KAAK,MAAM;YACT,MAAM,IAAI,KAAK,CAAC,yEAAyE,CAAC,CAAC;QAC7F;YACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED,SAAgB,aAAa,CAAC,OAA2B;IACvD,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,aAAa,GAAG,CAAC,CAAC;IAEtB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,UAAU,IAAI,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC;QAChD,aAAa,IAAI,MAAM,CAAC,QAAQ,CAAC;QACjC,aAAa,IAAI,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC;IACrD,CAAC;IAED,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAExF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,qBAAqB,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAClD,KAAK,CAAC,IAAI,CAAC,sBAAsB,UAAU,EAAE,CAAC,CAAC;IAC/C,KAAK,CAAC,IAAI,CAAC,4BAA4B,WAAW,GAAG,CAAC,CAAC;IACvD,KAAK,CAAC,IAAI,CAAC,iBAAiB,aAAa,IAAI,CAAC,CAAC;IAC/C,KAAK,CAAC,IAAI,CAAC,kBAAkB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7E,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAClD,KAAK,CAAC,IAAI,CAAC,OAAO,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,aAAa,MAAM,CAAC,SAAS,CAAC,gBAAgB,gBAAgB,MAAM,CAAC,QAAQ,IAAI,CAAC,CAAC;IACxI,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/generators/go-generator.d.ts

@@ -0,0 +1,3 @@
+import { SourceAnalysis, TestSuite, TestFramework } from '../types.js';
+export declare function generateGoTestSuite(analysis: SourceAnalysis, _framework: TestFramework, maxTestsPerFunction: number): TestSuite;
+//# sourceMappingURL=go-generator.d.ts.map

package/dist/generators/go-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"go-generator.d.ts","sourceRoot":"","sources":["../../src/generators/go-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,cAAc,EACd,SAAS,EAET,aAAa,EAEd,MAAM,aAAa,CAAC;AAGrB,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,cAAc,EACxB,UAAU,EAAE,aAAa,EACzB,mBAAmB,EAAE,MAAM,GAC1B,SAAS,CA6BX"}