codeguard-testgen 1.0.14 → 1.0.16

package/dist/reviewer.js

@@ -0,0 +1,402 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.reviewChangedFiles = reviewChangedFiles;
+exports.generateUnifiedCodeReview = generateUnifiedCodeReview;
+const fs = require("fs/promises");
+const fsSync = require("fs");
+const path = require("path");
+const globals_1 = require("./globals");
+const ai_1 = require("./ai");
+const toolHandlers_1 = require("./toolHandlers");
+const toolDefinitions_1 = require("./toolDefinitions");
+const git_1 = require("./git");
+/**
+ * Review code changes for quality, bugs, performance, and security issues
+ */
+async function reviewChangedFiles() {
+    (0, globals_1.updateSpinner)('🔍 Scanning git changes for review...');
+    try {
+        const { fullDiff, files } = await (0, git_1.getGitDiff)();
+        if (files.length === 0) {
+            (0, globals_1.stopSpinner)();
+            console.log('✅ No changes detected in source files.');
+            console.log('   (Only staged and unstaged changes are checked)');
+            return;
+        }
+        globals_1.g.globalSpinner.text = `Found changes in ${files.length} file(s) to review`;
+        await new Promise(resolve => setTimeout(resolve, 200));
+        const filesToReview = [];
+        for (const fileInfo of files) {
+            const { filePath, diff } = fileInfo;
+            if (!fsSync.existsSync(filePath)) {
+                (0, globals_1.updateSpinner)(`⏭️  Skipping ${filePath} (file not found)`);
+                await new Promise(resolve => setTimeout(resolve, 100));
+                continue;
+            }
+            (0, globals_1.updateSpinner)(`🔄 Analyzing: ${filePath}`);
+            const changedFunctions = await (0, git_1.getChangedFunctionsFromDiff)(filePath, diff);
+            if (changedFunctions.length === 0) {
+                globals_1.g.globalSpinner.text = `⏭️  No exported functions changed in ${path.basename(filePath)}`;
+                await new Promise(resolve => setTimeout(resolve, 150));
+                continue;
+            }
+            globals_1.g.globalSpinner.text = `Found ${changedFunctions.length} changed function(s) in ${path.basename(filePath)}`;
+            await new Promise(resolve => setTimeout(resolve, 200));
+            filesToReview.push({ filePath, diff, functions: changedFunctions });
+        }
+        if (filesToReview.length === 0) {
+            console.log('\n✅ No functions to review.');
+            return;
+        }
+        try {
+            await generateUnifiedCodeReview(filesToReview);
+            (0, globals_1.stopSpinner)();
+            console.log('\n✅ Code review completed!');
+        }
+        catch (error) {
+            console.error(`\n❌ Error during review: ${error.message}`);
+        }
+        console.log('\n' + '='.repeat(60));
+        console.log('📊 Code Review Summary');
+        console.log('='.repeat(60));
+        console.log(`✅ Files reviewed: ${filesToReview.length}`);
+        console.log(`📝 Total functions: ${filesToReview.reduce((sum, f) => sum + f.functions.length, 0)}`);
+        console.log(`📁 Review saved to: reviews/code_review.md`);
+        console.log('='.repeat(60));
+        console.log('\n✨ Done!');
+    }
+    catch (error) {
+        if (error.message === 'Not a git repository') {
+            console.error('❌ Error: Not a git repository');
+            console.error('   Review mode requires git to detect changes.');
+            process.exit(1);
+        }
+        throw error;
+    }
+}
+/**
+ * Build AI prompt for a specific review step based on its ruleset
+ */
+async function buildStepPrompt(step, filesToReview, stepOutputPath) {
+    const filesContext = filesToReview.map(f => `- ${f.filePath}: ${f.functions.join(', ')}`).join('\n');
+    const totalFunctions = filesToReview.reduce((sum, f) => sum + f.functions.length, 0);
+    const rulesetPath = path.join(process.cwd(), 'codeguard-ruleset', step.ruleset);
+    let rulesetContent;
+    try {
+        rulesetContent = await fs.readFile(rulesetPath, 'utf-8');
+        if (globals_1.g.globalSpinner && globals_1.g.globalSpinner.isSpinning) {
+            globals_1.g.globalSpinner.text = `📖 Loaded ruleset: ${step.ruleset}`;
+            await new Promise(resolve => setTimeout(resolve, 100));
+        }
+    }
+    catch (error) {
+        (0, globals_1.stopSpinner)();
+        console.warn(`⚠️  Could not read ruleset file: ${rulesetPath}`);
+        console.warn(`   Using default ruleset message. Error: ${error.message}`);
+        rulesetContent = `Please review the code for ${step.name} following industry best practices.`;
+    }
+    return `You are a senior software engineer conducting a focused code review for: **${step.name}**
+
+## CONTEXT
+Review Focus: ${step.name}
+Category: ${step.category}
+Total files changed: ${filesToReview.length}
+Total functions changed: ${totalFunctions}
+
+Changed files and functions:
+${filesContext}
+
+## GIT DIFFS
+
+${filesToReview.map(f => `### ${f.filePath}
+\`\`\`diff
+${f.diff}
+\`\`\`
+`).join('\n')}
+
+## YOUR TASK
+Conduct a focused code review ONLY for ${step.name}. Use the available tools to analyze the code thoroughly.
+Must complete and write the review within 1 minutes. Your review should be brief and to the point. 
+
+## ANALYSIS STEPS
+
+**Phase 1: Code Analysis (MANDATORY)**
+For each changed file:
+1. analyze_file_ast(filePath) → Get file structure and all functions
+   - OR use analyze_file_ast(filePath, functionName) for specific function reviews (token-efficient)
+2. For each changed function:
+   - get_function_ast(filePath, functionName) → Get implementation details
+3. get_imports_ast(filePath) → Understand dependencies
+4. get_type_definitions(filePath) → Review type safety
+5. Use search_codebase() if needed to understand usage patterns
+6. Use tools to analyze the code thoroughly. -> Where the functions is used and what all dependencies are there that call this function? What can be potential issues with the code?
+
+  **Phase 2: Review Analysis for ${step.name}**
+  
+  Review the code against the following ruleset:
+  
+  ${rulesetContent}
+  
+  Also give attention to any industry specific points that are not covered by the ruleset for ${step.name}.
+
+**Phase 3: Write Review**
+
+Use the write_review tool to save findings to: ${stepOutputPath}
+
+## OUTPUT FORMAT
+
+Your review MUST be in markdown format with the following structure:
+
+\`\`\`markdown
+# ${step.name}
+
+
+## Findings
+
+### 🔴 Critical Issues
+
+#### [Issue Title]
+**File**: \`filePath\`
+**Function**: \`functionName\`
+**Severity**: Critical
+
+**Issue**:
+[Description]
+
+**Current Code**:
+\`\`\`typescript
+// problematic code
+\`\`\`
+
+**Recommended Code**:
+\`\`\`typescript
+// improved code
+\`\`\`
+
+**Rationale**:
+[Why this is important]
+
+---
+
+### 🟠 High Priority Issues
+[Same format as above]
+
+---
+
+### 🟡 Medium Priority Issues
+[Same format as above]
+
+---
+
+## Positive Aspects
+[What was done well in this area]
+
+## Recommendations
+[Specific recommendations for ${step.name}]
+\`\`\`
+
+## CRITICAL REMINDERS
+
+- ALWAYS use tools to analyze code before reviewing
+- Focus ONLY on ${step.name} - do not review other aspects
+- Be specific and actionable
+- Include code examples
+- Use write_review tool to save to: ${stepOutputPath}
+- Must complete within 30 seconds
+- Use avaiable tools extensively to analyze the code.
+- The review should be based on the ruleset, it should be to the point without too much text.
+- Before writing the review, make sure to analyze the code thoroughly using the tools available, how similar code is used in the codebase.
+- Keep the review short and to the point, do not write too much text.
+
+**START**: Begin by calling analyze_file_ast on each changed file.`;
+}
+/**
+ * Execute a single review step with AI
+ */
+async function executeReviewStep(step, filesToReview, stepOutputPath) {
+    try {
+        const prompt = await buildStepPrompt(step, filesToReview, stepOutputPath);
+        const messages = [
+            {
+                role: 'user',
+                content: prompt
+            }
+        ];
+        let iterations = 0;
+        const maxIterations = 30;
+        let reviewWritten = false;
+        while (iterations < maxIterations) {
+            iterations++;
+            const response = await (0, ai_1.callAI)(messages, toolDefinitions_1.TOOLS_FOR_CODE_REVIEW, globals_1.g.CONFIG.aiProvider);
+            if (response.content) {
+                const content = typeof response.content === 'string' ? response.content : JSON.stringify(response.content);
+                messages.push({ role: 'assistant', content });
+                if (typeof content === 'string' && (content.toLowerCase().includes('review complete') ||
+                    content.toLowerCase().includes('review has been written'))) {
+                    if (reviewWritten) {
+                        break;
+                    }
+                }
+            }
+            if (!response.toolCalls || response.toolCalls.length === 0) {
+                if (reviewWritten) {
+                    break;
+                }
+                messages.push({
+                    role: 'user',
+                    content: `Please use the write_review tool NOW to save your ${step.name} review to ${stepOutputPath}. Include all findings with severity levels and code examples.`
+                });
+                continue;
+            }
+            const toolResults = [];
+            for (const toolCall of response.toolCalls) {
+                const toolResult = await (0, toolHandlers_1.executeTool)(toolCall.name, toolCall.input);
+                toolResults.push({ id: toolCall.id, name: toolCall.name, result: toolResult });
+                if (toolCall.name === 'write_review' && toolResult.success) {
+                    reviewWritten = true;
+                }
+            }
+            (0, ai_1.addToolResultsToMessages)(globals_1.g.CONFIG.aiProvider, messages, response.toolCalls, toolResults);
+        }
+        if (!reviewWritten) {
+            throw new Error(`Could not complete ${step.name} review within iteration limit`);
+        }
+        console.log(`   ✅ ${step.name} review completed`);
+        return { success: true, stepId: step.id, stepName: step.name };
+    }
+    catch (error) {
+        console.error(`   ❌ ${step.name} review failed: ${error.message}`);
+        return { success: false, error: error.message, stepId: step.id, stepName: step.name };
+    }
+}
+/**
+ * Merge review results from multiple steps into a single unified review
+ */
+async function mergeReviewResults(stepResults, filesToReview, finalOutputPath) {
+    const timestamp = new Date().toISOString().split('T')[0];
+    const totalFunctions = filesToReview.reduce((sum, f) => sum + f.functions.length, 0);
+    let mergedReview = `# Code Review
+
+**Date**: ${timestamp}
+**Reviewer**: AI Code Review System
+**Files Changed**: ${filesToReview.length}
+**Functions Changed**: ${totalFunctions}
+
+---
+
+## Summary
+
+This review covers ${filesToReview.length} file(s) with ${totalFunctions} changed function(s). The review was conducted across multiple aspects: ${stepResults.map(r => r.stepName).join(', ')}.
+
+---
+
+## Files Changed
+
+${filesToReview.map(f => `- **${f.filePath}**
+  - Functions: ${f.functions.join(', ')}`).join('\n')}
+
+---
+
+`;
+    for (const result of stepResults) {
+        const stepFilePath = path.join('reviews', '.tmp', `step-${result.stepId}.md`);
+        if (result.success) {
+            try {
+                if (fsSync.existsSync(stepFilePath)) {
+                    const stepContent = await fs.readFile(stepFilePath, 'utf-8');
+                    const lines = stepContent.split('\n');
+                    let contentStart = 0;
+                    for (let i = 0; i < lines.length; i++) {
+                        if (lines[i].startsWith('# ')) {
+                            contentStart = i + 1;
+                            break;
+                        }
+                    }
+                    const content = lines.slice(contentStart).join('\n').trim();
+                    mergedReview += `## ${result.stepName}\n\n${content}\n\n---\n\n`;
+                }
+                else {
+                    mergedReview += `## ${result.stepName}\n\n⚠️ Review file not found.\n\n---\n\n`;
+                }
+            }
+            catch (error) {
+                mergedReview += `## ${result.stepName}\n\n⚠️ Error reading review: ${error.message}\n\n---\n\n`;
+            }
+        }
+        else {
+            mergedReview += `## ${result.stepName}\n\n❌ **Review step failed**: ${result.error}\n\n---\n\n`;
+        }
+    }
+    mergedReview += `## Conclusion
+
+This comprehensive review analyzed the code changes from multiple perspectives. Please address the findings based on their severity levels, starting with critical issues.
+
+The review was conducted using AI-powered analysis with access to the full codebase context.
+`;
+    const reviewsDir = path.dirname(finalOutputPath);
+    if (!fsSync.existsSync(reviewsDir)) {
+        await fs.mkdir(reviewsDir, { recursive: true });
+    }
+    await fs.writeFile(finalOutputPath, mergedReview, 'utf-8');
+    console.log(`\n✅ Merged review written to: ${finalOutputPath}`);
+    try {
+        const tmpDir = path.join('reviews', '.tmp');
+        if (fsSync.existsSync(tmpDir)) {
+            const tmpFiles = await fs.readdir(tmpDir);
+            for (const file of tmpFiles) {
+                await fs.unlink(path.join(tmpDir, file));
+            }
+            await fs.rmdir(tmpDir);
+        }
+    }
+    catch (error) {
+        // Ignore cleanup errors
+    }
+}
+/**
+ * Generate unified code review for all changed files
+ */
+async function generateUnifiedCodeReview(filesToReview) {
+    const reviewFilePath = path.join('reviews', 'code_review.md');
+    const enabledSteps = globals_1.g.CONFIG.reviewSteps.filter(step => step.enabled);
+    if (enabledSteps.length === 0) {
+        console.log('\n⚠️  No review steps enabled. Please enable at least one step in codeguard.json');
+        return;
+    }
+    console.log(`\n📋 Running ${enabledSteps.length} review step(s): ${enabledSteps.map(s => s.name).join(', ')}`);
+    const tmpDir = path.join('reviews', '.tmp');
+    if (!fsSync.existsSync(tmpDir)) {
+        await fs.mkdir(tmpDir, { recursive: true });
+    }
+    let stepResults;
+    if (globals_1.g.CONFIG.reviewExecutionMode === 'parallel') {
+        (0, globals_1.updateSpinner)('🔄 Executing review steps in parallel...');
+        const stepPromises = enabledSteps.map(step => {
+            const stepOutputPath = path.join('reviews', '.tmp', `step-${step.id}.md`);
+            return executeReviewStep(step, filesToReview, stepOutputPath);
+        });
+        stepResults = await Promise.all(stepPromises);
+    }
+    else {
+        (0, globals_1.updateSpinner)('🔄 Executing review steps sequentially...');
+        stepResults = [];
+        for (const step of enabledSteps) {
+            if (globals_1.g.globalSpinner) {
+                globals_1.g.globalSpinner.text = `🔍 Running ${step.name} review step...`;
+            }
+            const stepOutputPath = path.join('reviews', '.tmp', `step-${step.id}.md`);
+            const result = await executeReviewStep(step, filesToReview, stepOutputPath);
+            stepResults.push(result);
+            if (globals_1.g.globalSpinner && result.success) {
+                globals_1.g.globalSpinner.text = `${step.name} review completed`;
+                await new Promise(resolve => setTimeout(resolve, 200));
+            }
+        }
+    }
+    await mergeReviewResults(stepResults, filesToReview, reviewFilePath);
+    (0, globals_1.stopSpinner)();
+    const successCount = stepResults.filter(r => r.success).length;
+    const failCount = stepResults.filter(r => !r.success).length;
+    console.log(`\n✅ Review complete: ${successCount} step(s) succeeded${failCount > 0 ? `, ${failCount} failed` : ''}`);
+}
+//# sourceMappingURL=reviewer.js.map

package/dist/reviewer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reviewer.js","sourceRoot":"","sources":["../src/reviewer.ts"],"names":[],"mappings":";;AAcA,gDA2EC;AAkUD,8DA0DC;AArdD,kCAAmC;AACnC,6BAA8B;AAC9B,6BAA8B;AAC9B,uCAA0D;AAC1D,6BAAwD;AACxD,iDAA6C;AAC7C,uDAA0D;AAC1D,+BAAgE;AAIhE;;GAEG;AACI,KAAK,UAAU,kBAAkB;IACtC,IAAA,uBAAa,EAAC,uCAAuC,CAAC,CAAC;IAEvD,IAAI,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,MAAM,IAAA,gBAAU,GAAE,CAAC;QAE/C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,IAAA,qBAAW,GAAE,CAAC;YACd,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;YACtD,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;YACjE,OAAO;QACT,CAAC;QAED,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,oBAAoB,KAAK,CAAC,MAAM,oBAAoB,CAAC;QAC5E,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAEvD,MAAM,aAAa,GAAiE,EAAE,CAAC;QAEvF,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC7B,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;YAEpC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACjC,IAAA,uBAAa,EAAC,gBAAgB,QAAQ,mBAAmB,CAAC,CAAC;gBAC3D,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;gBACvD,SAAS;YACX,CAAC;YAED,IAAA,uBAAa,EAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAC;YAE3C,MAAM,gBAAgB,GAAG,MAAM,IAAA,iCAA2B,EAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAE3E,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClC,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,wCAAwC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACzF,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;gBACvD,SAAS;YACX,CAAC;YAED,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,SAAS,gBAAgB,CAAC,MAAM,2BAA2B,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5G,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,aAAa,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;YAC3C,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,yBAAyB,CAAC,aAAa,CAAC,CAAC;YAE/C,IAAA,qBAAW,GAAE,CAAC;YAEd,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,4BAA4B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,qBAAqB,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;QACzD,OAAO,CAAC,GAAG,CAAC,uBAAuB,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QACpG,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;QAC1D,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAE3B,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,IAAI,KAAK,CAAC,OAAO,KAAK,sBAAsB,EAAE,CAAC;YAC7C,OAAO,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;YAC/C,OAAO,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAChE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,eAAe,CAC5B,IAAgB,EAChB,aAA2E,EAC3E,cAAsB;IAEtB,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CACzC,KAAK,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC7C,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEb,MAAM,cAAc,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAErF,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,mBAAmB,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAChF,IAAI,cAAsB,CAAC;IAE3B,IAAI,CAAC;QACH,cAAc,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACzD,IAAI,WAAC,CAAC,aAAa,IAAI,WAAC,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;YAClD,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,sBAAsB,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5D,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,IAAA,qBAAW,GAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC,oCAAoC,WAAW,EAAE,CAAC,CAAC;QAChE,OAAO,CAAC,IAAI,CAAC,4CAA4C,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1E,cAAc,GAAG,8BAA8B,IAAI,CAAC,IAAI,qCAAqC,CAAC;IAChG,CAAC;IAED,OAAO,8EAA8E,IAAI,CAAC,IAAI;;;gBAGhF,IAAI,CAAC,IAAI;YACb,IAAI,CAAC,QAAQ;uBACF,aAAa,CAAC,MAAM;2BAChB,cAAc;;;EAGvC,YAAY;;;;EAIZ,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,QAAQ;;EAExC,CAAC,CAAC,IAAI;;CAEP,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;yCAG4B,IAAI,CAAC,IAAI;;;;;;;;;;;;;;;;mCAgBf,IAAI,CAAC,IAAI;;;;IAIxC,cAAc;;gGAE8E,IAAI,CAAC,IAAI;;;;iDAIxD,cAAc;;;;;;;IAO3D,IAAI,CAAC,IAAI;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCA4CmB,IAAI,CAAC,IAAI;;;;;;kBAMvB,IAAI,CAAC,IAAI;;;sCAGW,cAAc;;;;;;;mEAOe,CAAC;AACpE,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAC9B,IAAgB,EAChB,aAA2E,EAC3E,cAAsB;IAEtB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,IAAI,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;QAE1E,MAAM,QAAQ,GAAc;YAC1B;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,MAAM;aAChB;SACF,CAAC;QAEF,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,aAAa,GAAG,EAAE,CAAC;QACzB,IAAI,aAAa,GAAG,KAAK,CAAC;QAE1B,OAAO,UAAU,GAAG,aAAa,EAAE,CAAC;YAClC,UAAU,EAAE,CAAC;YAEb,MAAM,QAAQ,GAAG,MAAM,IAAA,WAAM,EAAC,QAAQ,EAAE,uCAAqB,EAAE,WAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAEpF,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrB,MAAM,OAAO,GAAG,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAC3G,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;gBAE9C,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;oBACjF,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC,EAAE,CAAC;oBAC/D,IAAI,aAAa,EAAE,CAAC;wBAClB,MAAM;oBACR,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3D,IAAI,aAAa,EAAE,CAAC;oBAClB,MAAM;gBACR,CAAC;gBAED,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,qDAAqD,IAAI,CAAC,IAAI,cAAc,cAAc,gEAAgE;iBACpK,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YAED,MAAM,WAAW,GAAU,EAAE,CAAC;YAC9B,KAAK,MAAM,QAAQ,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBAC1C,MAAM,UAAU,GAAG,MAAM,IAAA,0BAAW,EAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACpE,WAAW,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;gBAE/E,IAAI,QAAQ,CAAC,IAAI,KAAK,cAAc,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;oBAC3D,aAAa,GAAG,IAAI,CAAC;gBACvB,CAAC;YACH,CAAC;YAED,IAAA,6BAAwB,EAAC,WAAC,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAU,EAAE,WAAW,CAAC,CAAC;QAC5F,CAAC;QAED,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,sBAAsB,IAAI,CAAC,IAAI,gCAAgC,CAAC,CAAC;QACnF,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,CAAC,IAAI,mBAAmB,CAAC,CAAC;QAClD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;IAEjE,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,CAAC,IAAI,mBAAmB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACnE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;IACxF,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,kBAAkB,CAC/B,WAAwF,EACxF,aAA2E,EAC3E,eAAuB;IAEvB,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACzD,MAAM,cAAc,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAErF,IAAI,YAAY,GAAG;;YAET,SAAS;;qBAEA,aAAa,CAAC,MAAM;yBAChB,cAAc;;;;;;qBAMlB,aAAa,CAAC,MAAM,iBAAiB,cAAc,2EAA2E,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;;EAM5L,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,QAAQ;iBACzB,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;CAIpD,CAAC;IAEA,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;QACjC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC;QAE9E,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,IAAI,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;oBACpC,MAAM,WAAW,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;oBAE7D,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACtC,IAAI,YAAY,GAAG,CAAC,CAAC;oBACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;wBACtC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;4BAC9B,YAAY,GAAG,CAAC,GAAG,CAAC,CAAC;4BACrB,MAAM;wBACR,CAAC;oBACH,CAAC;oBACD,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;oBAE5D,YAAY,IAAI,MAAM,MAAM,CAAC,QAAQ,OAAO,OAAO,aAAa,CAAC;gBACnE,CAAC;qBAAM,CAAC;oBACN,YAAY,IAAI,MAAM,MAAM,CAAC,QAAQ,0CAA0C,CAAC;gBAClF,CAAC;YACH,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,YAAY,IAAI,MAAM,MAAM,CAAC,QAAQ,gCAAgC,KAAK,CAAC,OAAO,aAAa,CAAC;YAClG,CAAC;QACH,CAAC;aAAM,CAAC;YACN,YAAY,IAAI,MAAM,MAAM,CAAC,QAAQ,iCAAiC,MAAM,CAAC,KAAK,aAAa,CAAC;QAClG,CAAC;IACH,CAAC;IAED,YAAY,IAAI;;;;;CAKjB,CAAC;IAEA,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACjD,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QACnC,MAAM,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,EAAE,CAAC,SAAS,CAAC,eAAe,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,iCAAiC,eAAe,EAAE,CAAC,CAAC;IAEhE,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC5C,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9B,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC1C,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;gBAC5B,MAAM,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YAC3C,CAAC;YACD,MAAM,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,wBAAwB;IAC1B,CAAC;AACH,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,yBAAyB,CAC7C,aAA2E;IAE3E,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;IAE9D,MAAM,YAAY,GAAG,WAAC,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAEvE,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,CAAC,GAAG,CAAC,kFAAkF,CAAC,CAAC;QAChG,OAAO;IACT,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,gBAAgB,YAAY,CAAC,MAAM,oBAAoB,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAE/G,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC5C,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC/B,MAAM,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,WAAwF,CAAC;IAE7F,IAAI,WAAC,CAAC,MAAM,CAAC,mBAAmB,KAAK,UAAU,EAAE,CAAC;QAChD,IAAA,uBAAa,EAAC,0CAA0C,CAAC,CAAC;QAE1D,MAAM,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;YAC3C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC;YAC1E,OAAO,iBAAiB,CAAC,IAAI,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAChD,CAAC;SAAM,CAAC;QACN,IAAA,uBAAa,EAAC,2CAA2C,CAAC,CAAC;QAE3D,WAAW,GAAG,EAAE,CAAC;QACjB,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;YAChC,IAAI,WAAC,CAAC,aAAa,EAAE,CAAC;gBACpB,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,cAAc,IAAI,CAAC,IAAI,iBAAiB,CAAC;YAClE,CAAC;YAED,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC;YAC1E,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;YAC5E,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEzB,IAAI,WAAC,CAAC,aAAa,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACtC,WAAC,CAAC,aAAa,CAAC,IAAI,GAAG,GAAG,IAAI,CAAC,IAAI,mBAAmB,CAAC;gBACvD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,kBAAkB,CAAC,WAAW,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;IAErE,IAAA,qBAAW,GAAE,CAAC;IAEd,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;IAC/D,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;IAE7D,OAAO,CAAC,GAAG,CAAC,wBAAwB,YAAY,qBAAqB,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,SAAS,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACvH,CAAC"}

package/dist/testGenerator.d.ts

@@ -0,0 +1,24 @@
+export declare function listFilesRecursive(dir: string, fileList?: string[]): Promise<string[]>;
+export declare function getTestFilePath(sourceFile: string): string;
+export declare function generateTests(sourceFile: string): Promise<string>;
+export declare function promptUser(question: string): Promise<string>;
+export declare function generateTestsForFolder(): Promise<void>;
+/**
+ * Generate tests for a single function
+ * @returns true if tests passed, false if legitimate failure reported
+ */
+export declare function generateTestForSingleFunction(sourceFile: string, functionName: string, testFilePath: string, testFileExists: boolean): Promise<boolean>;
+/**
+ * Smart validation that fixes failing tests
+ */
+export declare function smartValidateTestSuite(sourceFile: string, testFilePath: string, functionNames: string[]): Promise<void>;
+/**
+ * Fix failing tests using AI
+ */
+export declare function fixFailingTests(sourceFile: string, testFilePath: string, functionNames: string[], failingTests: string[], fullSuiteOutput: string): Promise<void>;
+/**
+ * Generate tests for multiple functions, one at a time
+ */
+export declare function generateTestsForFunctions(sourceFile: string, functionNames: string[]): Promise<string>;
+export declare function generateTestsForFunction(): Promise<void>;
+//# sourceMappingURL=testGenerator.d.ts.map

package/dist/testGenerator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"testGenerator.d.ts","sourceRoot":"","sources":["../src/testGenerator.ts"],"names":[],"mappings":"AAeA,wBAAsB,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAM,EAAO,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAmChG;AAGD,wBAAgB,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CA0B1D;AAMD,wBAAsB,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAqBvE;AAMD,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAYlE;AAoBD,wBAAsB,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC,CAuD5D;AAMD;;;GAGG;AACH,wBAAsB,6BAA6B,CACjD,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,EACpB,YAAY,EAAE,MAAM,EACpB,cAAc,EAAE,OAAO,GACtB,OAAO,CAAC,OAAO,CAAC,CAkvBlB;AAED;;GAEG;AACH,wBAAsB,sBAAsB,CAC1C,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,EACpB,aAAa,EAAE,MAAM,EAAE,GACtB,OAAO,CAAC,IAAI,CAAC,CAkCf;AAED;;GAEG;AACH,wBAAsB,eAAe,CACnC,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,EACpB,aAAa,EAAE,MAAM,EAAE,EACvB,YAAY,EAAE,MAAM,EAAE,EACtB,eAAe,EAAE,MAAM,GACtB,OAAO,CAAC,IAAI,CAAC,CAoFf;AAED;;GAEG;AACH,wBAAsB,yBAAyB,CAAC,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAqD5G;AAED,wBAAsB,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC,CAmE9D"}