deflake 1.0.12 → 1.1.2

package/cli.js

@@ -7,13 +7,29 @@ const fs = require('fs');
 const path = require('path');
 const pkg = require('./package.json');
 
+// --- COLORS ---
+const C = {
+    RESET: "\x1b[0m",
+    BRIGHT: "\x1b[1m",
+    RED: "\x1b[31m",
+    GREEN: "\x1b[32m",
+    YELLOW: "\x1b[33m",
+    CYAN: "\x1b[36m",
+    BLUE: "\x1b[34m",
+    GRAY: "\x1b[90m",
+    WHITE: "\x1b[37m"
+};
+
 // --- DETERMINISTIC COMMAND INTERCEPTION ---
 // Check for diagnostic commands before yargs or main logic even starts
 const rawArgs = process.argv.slice(2);
 if (rawArgs.includes('doctor') || rawArgs.includes('--doctor')) {
     // We run the doctor logic and force exit to prevent any wrapper loops
     const doctorArgv = yargs(hideBin(process.argv)).argv;
-    runDoctor(doctorArgv).then(() => process.exit(0)).catch(() => process.exit(1));
+    runDoctor(doctorArgv).then(() => process.exit(0)).catch((err) => {
+        console.error(`${C.RED}Doctor failed:${C.RESET} ${err.message}`);
+        process.exit(1);
+    });
     return; // Safety for some environments
 }
 // ------------------------------------------
@@ -346,18 +362,7 @@ function extractFailureLocation(logText) {
     return null;
 }
 
-// --- COLORS ---
-const C = {
-    RESET: "\x1b[0m",
-    BRIGHT: "\x1b[1m",
-    RED: "\x1b[31m",
-    GREEN: "\x1b[32m",
-    YELLOW: "\x1b[33m",
-    CYAN: "\x1b[36m",
-    BLUE: "\x1b[34m",
-    GRAY: "\x1b[90m",
-    WHITE: "\x1b[37m"
-};
+// --- COLORS Moved to top ---
 
 function printDetailedFix(fixText, location, sourceCode = null, isApplied = false) {
 
@@ -505,13 +510,11 @@ async function runDoctor(argv) {
 
     // 2. Framework Detection
     console.log(`${C.BRIGHT}Detecting Frameworks:${C.RESET}`);
-    const frameworks = [];
-    if (fs.existsSync('playwright.config.ts') || fs.existsSync('playwright.config.js')) frameworks.push('Playwright');
-    if (fs.existsSync('cypress.config.ts') || fs.existsSync('cypress.config.js') || fs.existsSync('cypress.json')) frameworks.push('Cypress');
-    if (fs.existsSync('wdio.conf.ts') || fs.existsSync('wdio.conf.js')) frameworks.push('WebdriverIO');
+    const activeFramework = DeFlakeClient.detectFramework();
 
-    if (frameworks.length > 0) {
-        console.log(`  ✅ Found: ${C.GREEN}${frameworks.join(', ')}${C.RESET}`);
+    if (activeFramework !== 'generic') {
+        const capitalized = activeFramework.charAt(0).toUpperCase() + activeFramework.slice(1);
+        console.log(`  ✅ Active:    ${C.GREEN}${C.BRIGHT}${capitalized}${C.RESET}`);
     } else {
         console.log(`  ⚠️  ${C.YELLOW}No supported frameworks detected in the current directory.${C.RESET}`);
         console.log(`     (Checked for Playwright, Cypress, WebdriverIO files)`);
@@ -543,12 +546,18 @@ async function runDoctor(argv) {
             const packageJson = JSON.parse(fs.readFileSync('package.json', 'utf8'));
             const deps = { ...packageJson.dependencies, ...packageJson.devDependencies };
             const detectedDeps = [];
-            if (deps['@playwright/test']) detectedDeps.push('@playwright/test');
-            if (deps['cypress']) detectedDeps.push('cypress');
-            if (deps['webdriverio']) detectedDeps.push('webdriverio');
+
+            // Core frameworks
+            if (deps['@playwright/test']) detectedDeps.push(`${C.CYAN}@playwright/test${C.RESET}`);
+            if (deps['cypress']) detectedDeps.push(`${C.CYAN}cypress${C.RESET}`);
+            if (deps['webdriverio'] || deps['@wdio/cli']) detectedDeps.push(`${C.CYAN}webdriverio${C.RESET}`);
+
+            // Common plugins
+            if (deps['dotenv']) detectedDeps.push('dotenv');
+            if (deps['typescript']) detectedDeps.push('typescript');
 
             if (detectedDeps.length > 0) {
-                console.log(`  ✅ Installed: ${C.GREEN}${detectedDeps.join(', ')}${C.RESET}`);
+                console.log(`  📦 Installed: ${detectedDeps.join(', ')}`);
             }
         } catch (e) {
             console.log(`  ❌ ${C.RED}Error reading package.json${C.RESET}`);
@@ -816,8 +825,10 @@ async function main() {
     // If 'doctor' was called, don't proceed to wrapper logic
     if (command.includes('doctor')) return;
 
-    console.log("🚑 DeFlake JS Client (Batch Mode)");
+    console.log(`🚑 DeFlake JS Client (Batch Mode) - ${C.CYAN}v${pkg.version}${C.RESET}`);
     const client = new DeFlakeClient(argv.apiUrl);
+    const fw = client.framework.charAt(0).toUpperCase() + client.framework.slice(1);
+    console.log(`🔍 Detected Framework: ${C.GREEN}${fw}${C.RESET}`);
 
     if (command.length > 0) {
         const cmd = command[0];

package/client.js

@@ -10,8 +10,8 @@ class DeFlakeClient {
         this.productionUrl = 'https://deflake-api.up.railway.app/api/deflake';
         this.apiUrl = apiUrl || process.env.DEFLAKE_API_URL || this.productionUrl;
         this.apiKey = apiKey || process.env.DEFLAKE_API_KEY;
-        this.projectName = this.detectProjectName();
-        this.framework = this.detectFramework();
+        this.projectName = DeFlakeClient.detectProjectName();
+        this.framework = DeFlakeClient.detectFramework();
 
         if (!this.apiKey) {
             // We no longer exit here to allow diagnostic tools (like 'doctor') to run.
@@ -19,8 +19,10 @@ class DeFlakeClient {
         }
     }
 
-    detectProjectName() {
+    static detectProjectName() {
         try {
+            const fs = require('fs');
+            const path = require('path');
             // 1. Try package.json
             if (fs.existsSync('package.json')) {
                 const pkg = JSON.parse(fs.readFileSync('package.json', 'utf8'));
@@ -40,14 +42,32 @@ class DeFlakeClient {
         }
     }
 
-    detectFramework() {
+    static detectFramework() {
+        const fs = require('fs');
         // 1. Manual override via environment variable
         if (process.env.DEFLAKE_FRAMEWORK) return process.env.DEFLAKE_FRAMEWORK.toLowerCase();
 
         // 2. Automated detection via configuration files
-        if (fs.existsSync('cypress.config.js') || fs.existsSync('cypress.config.ts') || fs.existsSync('cypress')) return 'cypress';
-        if (fs.existsSync('playwright.config.js') || fs.existsSync('playwright.config.ts')) return 'playwright';
-        if (fs.existsSync('wdio.conf.js') || fs.existsSync('wdio.conf.ts')) return 'webdriverio';
+        const hasFile = (pattern) => {
+            const files = fs.readdirSync('.');
+            return files.some(f => f.match(pattern));
+        };
+
+        if (hasFile(/^cypress\.config\.(js|ts|mjs|cjs)$/) || fs.existsSync('cypress.json') || fs.existsSync('cypress')) return 'cypress';
+        if (hasFile(/^playwright\.config\.(js|ts|mjs|cjs)$/)) return 'playwright';
+        if (hasFile(/^wdio\.conf\.(js|ts|mjs|cjs)$/)) return 'webdriverio';
+
+        // 3. Fallback to package.json dependencies
+        if (fs.existsSync('package.json')) {
+            try {
+                const pkg = JSON.parse(fs.readFileSync('package.json', 'utf8'));
+                const deps = { ...pkg.dependencies, ...pkg.devDependencies };
+                if (deps['cypress']) return 'cypress';
+                if (deps['@playwright/test']) return 'playwright';
+                if (deps['webdriverio'] || deps['@wdio/cli']) return 'webdriverio';
+            } catch (e) { }
+        }
+
         return 'generic';
     }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "deflake",
-  "version": "1.0.12",
+  "version": "1.1.2",
   "description": "AI-powered self-healing tool for Playwright, Cypress, and WebdriverIO tests.",
   "main": "client.js",
   "bin": {