@theihtisham/ai-testgen 1.0.0

package/dist/utils/language.js

@@ -0,0 +1,137 @@
+"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.detectLanguage = detectLanguage;
+exports.detectFramework = detectFramework;
+exports.getTestFileExtension = getTestFileExtension;
+exports.buildTestFilePath = buildTestFilePath;
+exports.isTestFile = isTestFile;
+exports.shouldAnalyze = shouldAnalyze;
+const path = __importStar(require("path"));
+const fs = __importStar(require("fs"));
+const defaults_js_1 = require("../config/defaults.js");
+function detectLanguage(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    const language = defaults_js_1.LANGUAGE_EXTENSIONS[ext];
+    if (!language) {
+        throw new Error(`Unsupported file extension: ${ext}. Supported: ${Object.keys(defaults_js_1.LANGUAGE_EXTENSIONS).join(', ')}`);
+    }
+    return language;
+}
+function detectFramework(language, projectDir) {
+    const frameworks = defaults_js_1.LANGUAGE_FRAMEWORKS[language];
+    if (frameworks.length === 0) {
+        throw new Error(`No test framework support for language: ${language}`);
+    }
+    // Check for framework-specific config files
+    if (language === 'typescript' || language === 'javascript') {
+        const vitestConfig = ['vitest.config.ts', 'vitest.config.js', 'vite.config.ts'].find((f) => fs.existsSync(path.join(projectDir, f)));
+        if (vitestConfig)
+            return 'vitest';
+        const jestConfig = ['jest.config.ts', 'jest.config.js', 'jest.config.mjs'].find((f) => fs.existsSync(path.join(projectDir, f)));
+        if (jestConfig)
+            return 'jest';
+        // Check package.json for dependencies
+        const pkgPath = path.join(projectDir, 'package.json');
+        if (fs.existsSync(pkgPath)) {
+            const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
+            const allDeps = { ...pkg.dependencies, ...pkg.devDependencies };
+            if (allDeps['vitest'])
+                return 'vitest';
+            if (allDeps['jest'])
+                return 'jest';
+        }
+        return 'vitest'; // default for JS/TS
+    }
+    if (language === 'python')
+        return 'pytest';
+    if (language === 'go')
+        return 'go-test';
+    return frameworks[0];
+}
+function getTestFileExtension(language) {
+    const map = {
+        typescript: 'ts',
+        javascript: 'js',
+        python: 'py',
+        go: 'go',
+        rust: 'rs',
+    };
+    return map[language];
+}
+function buildTestFilePath(sourceFilePath, outputDir, language, framework) {
+    const dir = path.dirname(sourceFilePath);
+    const ext = path.extname(sourceFilePath);
+    const baseName = path.basename(sourceFilePath, ext);
+    const testExt = getTestFileExtension(language);
+    let testFileName;
+    if (framework === 'pytest') {
+        testFileName = `test_${baseName}.py`;
+    }
+    else if (framework === 'go-test') {
+        testFileName = `${baseName}_test.go`;
+    }
+    else {
+        testFileName = `${baseName}.test.${testExt}`;
+    }
+    if (outputDir === '__tests__' || outputDir.startsWith('./') || outputDir.startsWith('../')) {
+        const testDir = path.resolve(path.dirname(sourceFilePath), outputDir);
+        return path.join(testDir, testFileName);
+    }
+    return path.join(dir, testFileName);
+}
+function isTestFile(filePath) {
+    const base = path.basename(filePath);
+    return (base.includes('.test.') ||
+        base.includes('.spec.') ||
+        base.startsWith('test_') ||
+        base.endsWith('_test.go'));
+}
+function shouldAnalyze(filePath, excludePatterns) {
+    if (isTestFile(filePath))
+        return false;
+    const normalizedPath = filePath.replace(/\\/g, '/');
+    for (const pattern of excludePatterns) {
+        const regex = pattern
+            .replace(/\*\*/g, '.*')
+            .replace(/\*/g, '[^/]*')
+            .replace(/\?/g, '[^/]');
+        if (new RegExp(regex).test(normalizedPath)) {
+            return false;
+        }
+    }
+    return true;
+}
+//# sourceMappingURL=language.js.map

package/dist/utils/language.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"language.js","sourceRoot":"","sources":["../../src/utils/language.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAKA,wCAOC;AAED,0CAkCC;AAED,oDASC;AAED,8CA0BC;AAED,gCAQC;AAED,sCAcC;AAjHD,2CAA6B;AAC7B,uCAAyB;AAEzB,uDAAiF;AAEjF,SAAgB,cAAc,CAAC,QAAgB;IAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACjD,MAAM,QAAQ,GAAG,iCAAmB,CAAC,GAAG,CAAC,CAAC;IAC1C,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,+BAA+B,GAAG,gBAAgB,MAAM,CAAC,IAAI,CAAC,iCAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAgB,eAAe,CAAC,QAA2B,EAAE,UAAkB;IAC7E,MAAM,UAAU,GAAG,iCAAmB,CAAC,QAAQ,CAAC,CAAC;IACjD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,2CAA2C,QAAQ,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,4CAA4C;IAC5C,IAAI,QAAQ,KAAK,YAAY,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC3D,MAAM,YAAY,GAAG,CAAC,kBAAkB,EAAE,kBAAkB,EAAE,gBAAgB,CAAC,CAAC,IAAI,CAClF,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAC/C,CAAC;QACF,IAAI,YAAY;YAAE,OAAO,QAAQ,CAAC;QAElC,MAAM,UAAU,GAAG,CAAC,gBAAgB,EAAE,gBAAgB,EAAE,iBAAiB,CAAC,CAAC,IAAI,CAC7E,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAC/C,CAAC;QACF,IAAI,UAAU;YAAE,OAAO,MAAM,CAAC;QAE9B,sCAAsC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QACtD,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,EAAE,GAAG,GAAG,CAAC,YAAY,EAAE,GAAG,GAAG,CAAC,eAAe,EAAE,CAAC;YAChE,IAAI,OAAO,CAAC,QAAQ,CAAC;gBAAE,OAAO,QAAQ,CAAC;YACvC,IAAI,OAAO,CAAC,MAAM,CAAC;gBAAE,OAAO,MAAM,CAAC;QACrC,CAAC;QAED,OAAO,QAAQ,CAAC,CAAC,oBAAoB;IACvC,CAAC;IAED,IAAI,QAAQ,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC3C,IAAI,QAAQ,KAAK,IAAI;QAAE,OAAO,SAAS,CAAC;IAExC,OAAO,UAAU,CAAC,CAAC,CAAE,CAAC;AACxB,CAAC;AAED,SAAgB,oBAAoB,CAAC,QAA2B;IAC9D,MAAM,GAAG,GAAsC;QAC7C,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,IAAI;QAChB,MAAM,EAAE,IAAI;QACZ,EAAE,EAAE,IAAI;QACR,IAAI,EAAE,IAAI;KACX,CAAC;IACF,OAAO,GAAG,CAAC,QAAQ,CAAC,CAAC;AACvB,CAAC;AAED,SAAgB,iBAAiB,CAC/B,cAAsB,EACtB,SAAiB,EACjB,QAA2B,EAC3B,SAAwB;IAExB,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACzC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;IACpD,MAAM,OAAO,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IAE/C,IAAI,YAAoB,CAAC;IACzB,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,YAAY,GAAG,QAAQ,QAAQ,KAAK,CAAC;IACvC,CAAC;SAAM,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QACnC,YAAY,GAAG,GAAG,QAAQ,UAAU,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,YAAY,GAAG,GAAG,QAAQ,SAAS,OAAO,EAAE,CAAC;IAC/C,CAAC;IAED,IAAI,SAAS,KAAK,WAAW,IAAI,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3F,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,SAAS,CAAC,CAAC;QACtE,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;IAC1C,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;AACtC,CAAC;AAED,SAAgB,UAAU,CAAC,QAAgB;IACzC,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACrC,OAAO,CACL,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACvB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;QACxB,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAC1B,CAAC;AACJ,CAAC;AAED,SAAgB,aAAa,CAAC,QAAgB,EAAE,eAAyB;IACvE,IAAI,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,KAAK,CAAC;IAEvC,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IACpD,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,OAAO;aAClB,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC;aACtB,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC;aACvB,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;YAC3C,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/utils/logger.d.ts

@@ -0,0 +1,13 @@
+export type LogLevel = 'debug' | 'info' | 'warn' | 'error' | 'success';
+export declare function setLogLevel(level: LogLevel): void;
+export declare const logger: {
+    debug(message: string, ...args: unknown[]): void;
+    info(message: string, ...args: unknown[]): void;
+    warn(message: string, ...args: unknown[]): void;
+    error(message: string, ...args: unknown[]): void;
+    success(message: string, ...args: unknown[]): void;
+    plain(message: string, ...args: unknown[]): void;
+    heading(message: string): void;
+    subheading(message: string): void;
+};
+//# sourceMappingURL=logger.d.ts.map

package/dist/utils/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../src/utils/logger.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,QAAQ,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,CAAC;AAMvE,wBAAgB,WAAW,CAAC,KAAK,EAAE,QAAQ,GAAG,IAAI,CAEjD;AAQD,eAAO,MAAM,MAAM;mBACF,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;kBAMlC,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;kBAMjC,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;mBAMhC,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;qBAM/B,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;mBAMnC,MAAM,WAAW,OAAO,EAAE,GAAG,IAAI;qBAI/B,MAAM,GAAG,IAAI;wBAIV,MAAM,GAAG,IAAI;CAGlC,CAAC"}

package/dist/utils/logger.js

@@ -0,0 +1,57 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.logger = void 0;
+exports.setLogLevel = setLogLevel;
+const chalk_1 = __importDefault(require("chalk"));
+let currentLogLevel = 'info';
+const LOG_ORDER = ['debug', 'info', 'warn', 'error', 'success'];
+function setLogLevel(level) {
+    currentLogLevel = level;
+}
+function shouldLog(level) {
+    if (level === 'success')
+        return true;
+    if (level === 'debug' && currentLogLevel !== 'debug')
+        return false;
+    return LOG_ORDER.indexOf(level) >= LOG_ORDER.indexOf(currentLogLevel);
+}
+exports.logger = {
+    debug(message, ...args) {
+        if (shouldLog('debug')) {
+            console.log(chalk_1.default.gray(`[debug] ${message}`), ...args);
+        }
+    },
+    info(message, ...args) {
+        if (shouldLog('info')) {
+            console.log(chalk_1.default.cyan(`[info] ${message}`), ...args);
+        }
+    },
+    warn(message, ...args) {
+        if (shouldLog('warn')) {
+            console.warn(chalk_1.default.yellow(`[warn] ${message}`), ...args);
+        }
+    },
+    error(message, ...args) {
+        if (shouldLog('error')) {
+            console.error(chalk_1.default.red(`[error] ${message}`), ...args);
+        }
+    },
+    success(message, ...args) {
+        if (shouldLog('success')) {
+            console.log(chalk_1.default.green(`  ${message}`), ...args);
+        }
+    },
+    plain(message, ...args) {
+        console.log(message, ...args);
+    },
+    heading(message) {
+        console.log('\n' + chalk_1.default.bold.blue(`  ${message}`));
+    },
+    subheading(message) {
+        console.log(chalk_1.default.bold.white(`  ${message}`));
+    },
+};
+//# sourceMappingURL=logger.js.map

package/dist/utils/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../src/utils/logger.ts"],"names":[],"mappings":";;;;;;AAQA,kCAEC;AAVD,kDAA0B;AAI1B,IAAI,eAAe,GAAa,MAAM,CAAC;AAEvC,MAAM,SAAS,GAAe,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;AAE5E,SAAgB,WAAW,CAAC,KAAe;IACzC,eAAe,GAAG,KAAK,CAAC;AAC1B,CAAC;AAED,SAAS,SAAS,CAAC,KAAe;IAChC,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACrC,IAAI,KAAK,KAAK,OAAO,IAAI,eAAe,KAAK,OAAO;QAAE,OAAO,KAAK,CAAC;IACnE,OAAO,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;AACxE,CAAC;AAEY,QAAA,MAAM,GAAG;IACpB,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,WAAW,OAAO,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,IAAI,CAAC,OAAe,EAAE,GAAG,IAAe;QACtC,IAAI,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,UAAU,OAAO,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED,IAAI,CAAC,OAAe,EAAE,GAAG,IAAe;QACtC,IAAI,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YACtB,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,MAAM,CAAC,UAAU,OAAO,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YACvB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,WAAW,OAAO,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,GAAG,IAAe;QACzC,IAAI,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,KAAK,OAAO,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,CAAC,OAAe;QACrB,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,eAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,OAAO,EAAE,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,UAAU,CAAC,OAAe;QACxB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,OAAO,EAAE,CAAC,CAAC,CAAC;IAChD,CAAC;CACF,CAAC"}

package/dist/watcher/index.d.ts

@@ -0,0 +1,2 @@
+export { FileWatcher } from './watcher.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/watcher/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/watcher/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC"}

package/dist/watcher/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FileWatcher = void 0;
+var watcher_js_1 = require("./watcher.js");
+Object.defineProperty(exports, "FileWatcher", { enumerable: true, get: function () { return watcher_js_1.FileWatcher; } });
+//# sourceMappingURL=index.js.map

package/dist/watcher/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/watcher/index.ts"],"names":[],"mappings":";;;AAAA,2CAA2C;AAAlC,yGAAA,WAAW,OAAA"}

package/dist/watcher/watcher.d.ts

@@ -0,0 +1,19 @@
+import { FileChangeEvent, TestGenConfig } from '../types.js';
+type FileWatcherCallback = (events: FileChangeEvent[]) => Promise<void>;
+export declare class FileWatcher {
+    private watchers;
+    private pendingEvents;
+    private debounceTimer;
+    private debounceMs;
+    private ignorePatterns;
+    private callback;
+    private running;
+    constructor(config: TestGenConfig, callback: FileWatcherCallback);
+    start(directories: string[]): void;
+    stop(): void;
+    isRunning(): boolean;
+    private shouldIgnore;
+    private debouncedNotify;
+}
+export {};
+//# sourceMappingURL=watcher.d.ts.map

package/dist/watcher/watcher.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"watcher.d.ts","sourceRoot":"","sources":["../../src/watcher/watcher.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG7D,KAAK,mBAAmB,GAAG,CAAC,MAAM,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AAExE,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAwC;IACxD,OAAO,CAAC,aAAa,CAAyB;IAC9C,OAAO,CAAC,aAAa,CAA8C;IACnE,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,cAAc,CAAW;IACjC,OAAO,CAAC,QAAQ,CAAsB;IACtC,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,aAAa,EAAE,QAAQ,EAAE,mBAAmB;IAMhE,KAAK,CAAC,WAAW,EAAE,MAAM,EAAE,GAAG,IAAI;IAuClC,IAAI,IAAI,IAAI;IAaZ,SAAS,IAAI,OAAO;IAIpB,OAAO,CAAC,YAAY;IASpB,OAAO,CAAC,eAAe;CAexB"}

package/dist/watcher/watcher.js

@@ -0,0 +1,122 @@
+"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.FileWatcher = void 0;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const logger_js_1 = require("../utils/logger.js");
+class FileWatcher {
+    watchers = new Map();
+    pendingEvents = [];
+    debounceTimer = null;
+    debounceMs;
+    ignorePatterns;
+    callback;
+    running = false;
+    constructor(config, callback) {
+        this.debounceMs = config.watch.debounceMs;
+        this.ignorePatterns = config.watch.ignorePatterns;
+        this.callback = callback;
+    }
+    start(directories) {
+        if (this.running)
+            return;
+        this.running = true;
+        for (const dir of directories) {
+            const absoluteDir = path.resolve(dir);
+            if (!fs.existsSync(absoluteDir)) {
+                logger_js_1.logger.warn(`Watch directory does not exist: ${absoluteDir}`);
+                continue;
+            }
+            try {
+                const watcher = fs.watch(absoluteDir, { recursive: true }, (eventType, filename) => {
+                    if (!filename)
+                        return;
+                    if (this.shouldIgnore(filename))
+                        return;
+                    const filePath = path.join(absoluteDir, filename);
+                    this.pendingEvents.push({
+                        filePath,
+                        eventType: eventType,
+                        timestamp: Date.now(),
+                    });
+                    this.debouncedNotify();
+                });
+                this.watchers.set(absoluteDir, watcher);
+                logger_js_1.logger.info(`Watching: ${absoluteDir}`);
+            }
+            catch (err) {
+                logger_js_1.logger.error(`Failed to watch ${absoluteDir}: ${err instanceof Error ? err.message : String(err)}`);
+            }
+        }
+    }
+    stop() {
+        this.running = false;
+        for (const [dir, watcher] of this.watchers) {
+            watcher.close();
+            logger_js_1.logger.debug(`Stopped watching: ${dir}`);
+        }
+        this.watchers.clear();
+        if (this.debounceTimer) {
+            clearTimeout(this.debounceTimer);
+            this.debounceTimer = null;
+        }
+    }
+    isRunning() {
+        return this.running;
+    }
+    shouldIgnore(filename) {
+        const normalized = filename.replace(/\\/g, '/');
+        return this.ignorePatterns.some((pattern) => {
+            return normalized.includes(pattern) || normalized.match(new RegExp(pattern.replace(/\*/g, '.*'))) !== null;
+        });
+    }
+    debouncedNotify() {
+        if (this.debounceTimer) {
+            clearTimeout(this.debounceTimer);
+        }
+        this.debounceTimer = setTimeout(() => {
+            const events = [...this.pendingEvents];
+            this.pendingEvents = [];
+            if (events.length > 0) {
+                this.callback(events).catch((err) => {
+                    logger_js_1.logger.error(`Watch callback error: ${err instanceof Error ? err.message : String(err)}`);
+                });
+            }
+        }, this.debounceMs);
+    }
+}
+exports.FileWatcher = FileWatcher;
+//# sourceMappingURL=watcher.js.map

package/dist/watcher/watcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"watcher.js","sourceRoot":"","sources":["../../src/watcher/watcher.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uCAAyB;AACzB,2CAA6B;AAE7B,kDAA4C;AAI5C,MAAa,WAAW;IACd,QAAQ,GAA8B,IAAI,GAAG,EAAE,CAAC;IAChD,aAAa,GAAsB,EAAE,CAAC;IACtC,aAAa,GAAyC,IAAI,CAAC;IAC3D,UAAU,CAAS;IACnB,cAAc,CAAW;IACzB,QAAQ,CAAsB;IAC9B,OAAO,GAAG,KAAK,CAAC;IAExB,YAAY,MAAqB,EAAE,QAA6B;QAC9D,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC;QAC1C,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC;QAClD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,WAAqB;QACzB,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACtC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;gBAChC,kBAAM,CAAC,IAAI,CAAC,mCAAmC,WAAW,EAAE,CAAC,CAAC;gBAC9D,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CACtB,WAAW,EACX,EAAE,SAAS,EAAE,IAAI,EAAE,EACnB,CAAC,SAAS,EAAE,QAAQ,EAAE,EAAE;oBACtB,IAAI,CAAC,QAAQ;wBAAE,OAAO;oBACtB,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC;wBAAE,OAAO;oBAExC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;oBAElD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;wBACtB,QAAQ;wBACR,SAAS,EAAE,SAAyC;wBACpD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACtB,CAAC,CAAC;oBAEH,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,CAAC,CACF,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBACxC,kBAAM,CAAC,IAAI,CAAC,aAAa,WAAW,EAAE,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,kBAAM,CAAC,KAAK,CAAC,mBAAmB,WAAW,KAAK,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtG,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI;QACF,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,KAAK,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC3C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,kBAAM,CAAC,KAAK,CAAC,qBAAqB,GAAG,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACjC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEO,YAAY,CAAC,QAAgB;QACnC,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;YAC1C,OAAO,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,KAAK,CACrD,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CACzC,KAAK,IAAI,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe;QACrB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;YACnC,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC;YACvC,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC;YACxB,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBAClC,kBAAM,CAAC,KAAK,CAAC,yBAAyB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC5F,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IACtB,CAAC;CACF;AA/FD,kCA+FC"}

package/package.json

@@ -0,0 +1,63 @@
+{
+  "name": "@theihtisham/ai-testgen",
+  "version": "1.0.0",
+  "description": "AI-powered test generator that creates comprehensive test suites from source code",
+  "main": "dist/index.js",
+  "bin": {
+    "ai-testgen": "./dist/cli.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "start": "node dist/cli.js",
+    "dev": "ts-node src/cli.ts",
+    "test": "vitest run tests/config.test.ts tests/language.test.ts tests/file-utils.test.ts tests/generators.test.ts tests/coverage.test.ts tests/watcher.test.ts tests/incremental.test.ts tests/mutation.test.ts && vitest run tests/analyzer.test.ts",
+    "test:watch": "vitest",
+    "test:coverage": "vitest run --coverage",
+    "lint": "tsc --noEmit",
+    "clean": "rimraf dist",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "testing",
+    "ai",
+    "test-generator",
+    "unit-tests",
+    "integration-tests",
+    "mutation-testing",
+    "code-coverage",
+    "jest",
+    "vitest",
+    "pytest",
+    "go-test"
+  ],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "chalk": "^4.1.2",
+    "commander": "^12.1.0",
+    "fast-glob": "^3.3.2",
+    "js-yaml": "^4.1.0",
+    "openai": "^4.67.3",
+    "ora": "^5.4.1",
+    "ts-morph": "^21.0.1"
+  },
+  "devDependencies": {
+    "@types/js-yaml": "^4.0.9",
+    "@types/node": "^22.10.0",
+    "cross-env": "^10.1.0",
+    "rimraf": "^6.0.1",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.7.2",
+    "vitest": "^2.1.8"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/theihtisham/ai-testgen"
+  }
+}

package/src/analyzers/analyzer.ts

@@ -0,0 +1,180 @@
+import {
+  SourceAnalysis,
+  SupportedLanguage,
+  AnalyzedFunction,
+  TestCase,
+  MockDefinition,
+} from '../types.js';
+import { analyzeTsSource } from './js-ts-analyzer.js';
+import { analyzePythonSource } from './python-analyzer.js';
+import { analyzeGoSource } from './go-analyzer.js';
+
+export function analyzeSource(filePath: string, language: SupportedLanguage): SourceAnalysis {
+  switch (language) {
+    case 'typescript':
+    case 'javascript':
+      return analyzeTsSource(filePath, language);
+    case 'python':
+      return analyzePythonSource(filePath);
+    case 'go':
+      return analyzeGoSource(filePath);
+    case 'rust':
+      throw new Error('Rust analysis is not yet supported. Use AI mode for Rust files.');
+    default:
+      throw new Error(`Unsupported language: ${language}`);
+  }
+}
+
+export function detectEdgeCases(fn: AnalyzedFunction): TestCase[] {
+  const cases: TestCase[] = [];
+
+  for (const param of fn.params) {
+    if (param.optional) {
+      cases.push({
+        name: `${fn.name} handles undefined ${param.name}`,
+        type: 'edge-case',
+        description: `Test that ${fn.name} gracefully handles undefined ${param.name}`,
+        code: '',
+        expectedBehavior: 'Should handle undefined parameter without throwing',
+        inputDescription: `${param.name} is undefined`,
+        tags: ['edge-case', 'undefined', param.name],
+      });
+    }
+
+    if (param.type && (param.type.includes('string') || param.type.includes('String'))) {
+      cases.push(
+        createEdgeCase(fn.name, param.name, 'empty string', '""', 'Should handle empty string input'),
+        createEdgeCase(fn.name, param.name, 'very long string', '"a".repeat(10000)', 'Should handle very long string'),
+        createEdgeCase(fn.name, param.name, 'string with special characters', '"\\n\\t\\0\\r"', 'Should handle special characters'),
+      );
+    }
+
+    if (param.type && (param.type.includes('number') || param.type.includes('Number'))) {
+      cases.push(
+        createEdgeCase(fn.name, param.name, 'zero', '0', 'Should handle zero value'),
+        createEdgeCase(fn.name, param.name, 'negative number', '-1', 'Should handle negative numbers'),
+        createEdgeCase(fn.name, param.name, 'very large number', 'Number.MAX_SAFE_INTEGER', 'Should handle large numbers'),
+        createEdgeCase(fn.name, param.name, 'NaN', 'NaN', 'Should handle NaN'),
+        createEdgeCase(fn.name, param.name, 'Infinity', 'Infinity', 'Should handle Infinity'),
+      );
+    }
+
+    if (param.type && (param.type.includes('Array') || param.type.includes('[]'))) {
+      cases.push(
+        createEdgeCase(fn.name, param.name, 'empty array', '[]', 'Should handle empty array'),
+        createEdgeCase(fn.name, param.name, 'array with single element', '[item]', 'Should handle single-element array'),
+        createEdgeCase(fn.name, param.name, 'very large array', 'Array(10000).fill(item)', 'Should handle large arrays'),
+      );
+    }
+
+    if (param.type && (param.type.includes('object') || param.type.includes('Object') || param.type.includes('{'))) {
+      cases.push(
+        createEdgeCase(fn.name, param.name, 'empty object', '{}', 'Should handle empty object'),
+        createEdgeCase(fn.name, param.name, 'null', 'null', 'Should handle null'),
+      );
+    }
+  }
+
+  if (fn.throws.length > 0) {
+    for (const throws of fn.throws) {
+      cases.push({
+        name: `${fn.name} throws for invalid input`,
+        type: 'edge-case',
+        description: `Test that ${fn.name} throws when given invalid input`,
+        code: '',
+        expectedBehavior: `Should throw ${throws}`,
+        inputDescription: 'Invalid input that triggers error',
+        tags: ['edge-case', 'error-path'],
+      });
+    }
+  }
+
+  if (fn.isAsync) {
+    cases.push({
+      name: `${fn.name} handles rejection`,
+      type: 'edge-case',
+      description: `Test that ${fn.name} properly handles promise rejection`,
+      code: '',
+      expectedBehavior: 'Should handle promise rejection gracefully',
+      inputDescription: 'Input that causes promise rejection',
+      tags: ['edge-case', 'async'],
+    });
+  }
+
+  return cases;
+}
+
+function createEdgeCase(
+  fnName: string,
+  paramName: string,
+  caseName: string,
+  value: string,
+  expected: string,
+): TestCase {
+  return {
+    name: `${fnName} handles ${caseName} for ${paramName}`,
+    type: 'edge-case',
+    description: `Test ${fnName} with ${caseName} for parameter ${paramName}`,
+    code: '',
+    expectedBehavior: expected,
+    inputDescription: `${paramName} = ${value}`,
+    tags: ['edge-case', caseName.replace(/\s+/g, '-')],
+  };
+}
+
+export function detectMocks(analysis: SourceAnalysis): MockDefinition[] {
+  const mocks: MockDefinition[] = [];
+
+  for (const imp of analysis.imports) {
+    if (imp.isTypeOnly) continue;
+
+    if (!imp.modulePath.startsWith('.') && imp.namedImports.length > 0) {
+      for (const named of imp.namedImports) {
+        const isLikelyDependency =
+          !imp.modulePath.startsWith('node:') &&
+          !['path', 'fs', 'os', 'util', 'events', 'stream', 'http', 'https', 'crypto'].includes(
+            imp.modulePath,
+          );
+
+        if (isLikelyDependency) {
+          mocks.push({
+            moduleName: imp.modulePath,
+            mockName: `mock${named.charAt(0).toUpperCase() + named.slice(1)}`,
+            setup: `jest.mock('${imp.modulePath}');`,
+            teardown: null,
+            implementations: {
+              [named]: `jest.fn()`,
+            },
+          });
+        }
+      }
+    }
+  }
+
+  return mocks;
+}
+
+export interface DependencyGraph {
+  files: Map<string, string[]>;
+  reverse: Map<string, string[]>;
+}
+
+export function buildDependencyGraph(analyses: SourceAnalysis[]): DependencyGraph {
+  const files = new Map<string, string[]>();
+  const reverse = new Map<string, string[]>();
+
+  for (const analysis of analyses) {
+    const deps: string[] = [];
+    for (const dep of analysis.dependencies) {
+      if (dep.startsWith('.')) {
+        deps.push(dep);
+        const rev = reverse.get(dep) ?? [];
+        rev.push(analysis.filePath);
+        reverse.set(dep, rev);
+      }
+    }
+    files.set(analysis.filePath, deps);
+  }
+
+  return { files, reverse };
+}