deflake 1.2.9 → 1.2.12

package/cli.js

@@ -715,7 +715,6 @@ async function runDoctor(argv) {
 }
 
 async function applySelfHealing(result) {
-
     if (!result.location || !result.location.fullRootPath || !result.fix) return;
 
     try {
@@ -727,28 +726,55 @@ async function applySelfHealing(result) {
             return;
         }
 
-        let fixCode = result.fix;
+        let patches = [];
         try {
             const parsed = JSON.parse(result.fix);
-            if (parsed.code) fixCode = parsed.code;
-        } catch (e) { }
+            if (parsed.patches && Array.isArray(parsed.patches)) {
+                patches = parsed.patches;
+            } else if (parsed.code) {
+                patches = [{
+                    file: filePath,
+                    line: targetLine,
+                    action: 'REPLACE',
+                    new_line: parsed.code
+                }];
+            }
+        } catch (e) {
+            // Fallback for raw string fixes
+            patches = [{
+                file: filePath,
+                line: targetLine,
+                action: 'REPLACE',
+                new_line: result.fix
+            }];
+        }
 
-        const lines = fs.readFileSync(filePath, 'utf8').split('\n');
-        const originalLineIndex = targetLine - 1;
+        for (const patch of patches) {
+            const pFile = patch.file || filePath;
+            const pLine = parseInt(patch.line || targetLine);
+            const pAction = patch.action || 'REPLACE';
+            let pNew = patch.new_line;
 
-        if (originalLineIndex < 0 || originalLineIndex >= lines.length) {
-            console.error(`  ❌ [Self-Healing] Line ${targetLine} out of bounds in ${filePath}`);
-            return;
-        }
+            if (!fs.existsSync(pFile)) continue;
 
-        // Apply fix: Replace the entire line or the specific locator
-        // For robustness, we replace the whole line but keep indentation
-        const originalLine = lines[originalLineIndex];
-        const indentation = originalLine.match(/^\s*/)[0];
-        lines[originalLineIndex] = indentation + fixCode.trim();
+            const lines = fs.readFileSync(pFile, 'utf8').split('\n');
+            const originalLineIndex = pLine - 1;
 
-        fs.writeFileSync(filePath, lines.join('\n'));
-        console.log(`  ✅ ${C.GREEN}[Self-Healing] Successfully patched:${C.RESET} ${path.basename(filePath)}:${targetLine}`);
+            if (originalLineIndex < 0 || originalLineIndex >= lines.length) continue;
+
+            const originalLine = lines[originalLineIndex];
+            const indentation = originalLine.match(/^\s*/)[0];
+
+            if (pAction === 'INSERT_AFTER') {
+                lines.splice(pLine, 0, indentation + pNew.trim());
+                fs.writeFileSync(pFile, lines.join('\n'));
+                console.log(`  ✅ ${C.GREEN}[Self-Healing] Successfully inserted at:${C.RESET} ${path.basename(pFile)}:${pLine}`);
+            } else {
+                lines[originalLineIndex] = indentation + pNew.trim();
+                fs.writeFileSync(pFile, lines.join('\n'));
+                console.log(`  ✅ ${C.GREEN}[Self-Healing] Successfully patched:${C.RESET} ${path.basename(pFile)}:${pLine}`);
+            }
+        }
     } catch (error) {
         console.error(`  ❌ ${C.RED}[Self-Healing] Error patching file:${C.RESET} ${error.message}`);
     }
@@ -981,6 +1007,7 @@ async function runCommand(cmd, args) {
 
 async function main() {
     const command = argv._ || [];
+    let finalExitCode = 0;
 
     // If 'doctor' or 'migrate' was called, don't proceed to wrapper logic
     if (command.includes('doctor') || command.includes('migrate')) return;
@@ -1001,20 +1028,22 @@ async function main() {
             const artifacts = detectAllArtifacts(null, argv.html);
             const fixesApplied = await analyzeFailures(artifacts, output, client);
 
+            let outcomeCode = code;
             // AUTO-VERIFICATION
             if (fixesApplied > 0 && argv.fix) {
                 console.log(`\n${C.BRIGHT}💉 Fixes applied. Re-running tests to verify...${C.RESET}`);
                 const secondRun = await runCommand(cmd, args);
+                outcomeCode = secondRun.code;
                 if (secondRun.code === 0) {
                     console.log(`\n${C.GREEN}${C.BRIGHT}✅ All tests passed after DeFlake healing!${C.RESET}`);
                 } else {
                     console.log(`\n${C.YELLOW}⚠️  Some tests still failing after fixes. Check the report for details.${C.RESET}`);
                 }
             }
-            process.exit(code);
+            finalExitCode = outcomeCode;
         } else {
             console.log("\n🟢 Command passed successfully.");
-            process.exit(0);
+            finalExitCode = 0;
         }
     } else {
         const artifacts = detectAllArtifacts(argv.log, argv.html);
@@ -1023,10 +1052,11 @@ async function main() {
     }
 
     // FINAL REPORT TRIGGER
-    // We wait until the very end so verification runs (if any) are reflected
     if (argv.report) {
         showFrameworkReport();
     }
+
+    process.exit(finalExitCode);
 }
 
 // main() call is now handled by the check above

package/package.json

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