deflake 1.1.0 → 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,11 +510,11 @@ async function runDoctor(argv) {
 
     // 2. Framework Detection
     console.log(`${C.BRIGHT}Detecting Frameworks:${C.RESET}`);
-    const framework = DeFlakeClient.detectFramework();
+    const activeFramework = DeFlakeClient.detectFramework();
 
-    if (framework !== 'generic') {
-        const capitalized = framework.charAt(0).toUpperCase() + framework.slice(1);
-        console.log(`  ✅ Found: ${C.GREEN}${capitalized}${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)`);
@@ -541,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}`);

package/package.json

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