deflake 1.2.35 → 1.2.37

package/cli.js

@@ -127,17 +127,25 @@ async function analyzeAndFix(artifacts, client, argv, capturedOutput = '') {
         try {
             const content = fs.readFileSync(art.htmlPath, 'utf8');
             
+            // ALWAYS extract the relevant error block from captured output for this artifact
+            // This gives us: failing locator string, error type, call log, stack trace
+            const relevantOutput = cleanOutput ? extractRelevantOutput(cleanOutput, art.name) : '';
+            if (relevantOutput) {
+                console.log(`   ${C.GRAY}🔬 Error context: ${relevantOutput.length} chars captured${C.RESET}`);
+            }
+            
             // SMART Location Extraction: try multiple sources
             let loc = extractLoc(content);
             let locSource = 'error-context.md';
             
-            if (!loc && cleanOutput) {
-                // Extract the relevant section from captured output for this artifact
-                const relevantOutput = extractRelevantOutput(cleanOutput, art.name);
-                console.log(`   ${C.GRAY}🔬 Relevant output: ${relevantOutput.length} chars for artifact${C.RESET}`);
-                loc = extractLoc(relevantOutput || cleanOutput);
+            if (!loc && relevantOutput) {
+                loc = extractLoc(relevantOutput);
                 locSource = 'console output';
             }
+            if (!loc && cleanOutput) {
+                loc = extractLoc(cleanOutput);
+                locSource = 'full output';
+            }
             
             // Step 1: Location extraction
             if (loc) {
@@ -150,9 +158,9 @@ async function analyzeAndFix(artifacts, client, argv, capturedOutput = '') {
             const source = loc && fs.existsSync(loc.path) ? fs.readFileSync(loc.path, 'utf8') : null;
             console.log(`   ${C.GRAY}📄 Source:${C.RESET} ${source ? 'Loaded (' + source.split('\n').length + ' lines)' : C.YELLOW + 'Not available' + C.RESET}`);
             
-            // Step 3: API call — send BOTH error-context AND console output for richer diagnosis
+            // Step 3: API call — send BOTH error-context AND console error block for richer diagnosis
             console.log(`   ${C.GRAY}🌐 Calling DeFlake API...${C.RESET}`);
-            const res = await client.heal(null, art.htmlPath, loc, source, argv.fix);
+            const res = await client.heal(null, art.htmlPath, loc, source, argv.fix, relevantOutput);
             
             if (res && res.status === 'success') {
                 console.log(`   ${C.GREEN}✅ API Response: SUCCESS${C.RESET}`);
@@ -208,7 +216,27 @@ async function analyzeAndFix(artifacts, client, argv, capturedOutput = '') {
     }
     console.log(`${C.BRIGHT}━━━ DeFlake Summary: ${count}/${artifacts.length} fix(es) applied ━━━${C.RESET}\n`);
     return count;
+}
 
+function checkBalance(code) {
+    const pairs = { '(': ')', '{': '}', '[': ']' };
+    const stack = [];
+    // Skip strings and comments for accuracy
+    let inString = false, stringChar = '', escaped = false;
+    for (const ch of code) {
+        if (escaped) { escaped = false; continue; }
+        if (ch === '\\') { escaped = true; continue; }
+        if (inString) { if (ch === stringChar) inString = false; continue; }
+        if (ch === '"' || ch === "'" || ch === '`') { inString = true; stringChar = ch; continue; }
+        if (pairs[ch]) stack.push(pairs[ch]);
+        else if (ch === ')' || ch === '}' || ch === ']') {
+            if (stack.length === 0 || stack.pop() !== ch) {
+                return { balanced: false, detail: `unexpected '${ch}'` };
+            }
+        }
+    }
+    if (stack.length > 0) return { balanced: false, detail: `unclosed '${stack[stack.length - 1] === ')' ? '(' : stack[stack.length - 1] === '}' ? '{' : '['}'` };
+    return { balanced: true, detail: 'ok' };
 }
 
 async function applyFix(res, loc) {
@@ -262,6 +290,13 @@ async function applyFix(res, loc) {
             }
         }
         if (appliedCount > 0) {
+            // SAFETY: Bracket balance check — rollback if patches broke syntax
+            const originalBalance = checkBalance(original);
+            const patchedBalance = checkBalance(content);
+            if (originalBalance.balanced && !patchedBalance.balanced) {
+                console.log(`   ${C.YELLOW}⏭️  Rollback: patches broke bracket balance (${patchedBalance.detail}). Reverting file.${C.RESET}`);
+                return false;
+            }
             fs.writeFileSync(loc.path, content);
             return true;
         }

package/client.js

@@ -103,7 +103,7 @@ class DeFlakeClient {
         }
     }
 
-    async heal(logPath, htmlPath, failureLocation = null, sourceCode = null, applyFix = false) {
+    async heal(logPath, htmlPath, failureLocation = null, sourceCode = null, applyFix = false, errorContext = '') {
         try {
             if (!fs.existsSync(htmlPath)) throw new Error(`HTML file not found: ${htmlPath}`);
 
@@ -126,9 +126,9 @@ class DeFlakeClient {
             }
 
             const payload = {
-                error_log: logContent || "",
+                error_log: logContent || errorContext || "",
                 html_snapshot: htmlContent || "",
-                failing_line: failureLocation ? `Line ${failureLocation.rootLine}` : "",
+                failing_line: failureLocation ? `Line ${failureLocation.line}` : "",
                 source_code: sourceCode || "",
                 framework: this.framework,
                 apply_fix: applyFix

package/package.json

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