deflake 1.2.40 → 1.2.42

package/cli.js

@@ -101,6 +101,18 @@ async function main() {
                     break;
                 }
                 
+                // Detect compilation errors: exit code != 0 but no test artifacts
+                // This means the code has syntax errors and tests didn't even run
+                const postArtifacts = detectFailures();
+                if (postArtifacts.length === 0 && rerun.code !== 0) {
+                    console.log(`${C.RED}⚠️  Compilation error detected (tests did not run). Rolling back all changes...${C.RESET}`);
+                    for (const [filePath, originalContent] of backups) {
+                        fs.writeFileSync(filePath, originalContent);
+                        console.log(`   ${C.YELLOW}↩️  Restored: ${path.basename(filePath)}${C.RESET}`);
+                    }
+                    break;
+                }
+                
                 if (attempt === MAX_ATTEMPTS) {
                     console.log(`\n${C.YELLOW}⚠️  Max attempts (${MAX_ATTEMPTS}) reached. Some failures may persist.${C.RESET}`);
                 }
@@ -341,16 +353,13 @@ async function applyFix(res, loc, backups = new Map()) {
                 } else {
                     // REPLACE action
                     // SMART: Handle multi-line expression replacement
-                    // If the original line starts a multi-line expression (has unclosed parens)
-                    // and the replacement is a complete statement, remove continuation lines
                     const origOpen = (originalLine.match(/\(/g) || []).length;
                     const origClose = (originalLine.match(/\)/g) || []).length;
                     const replOpen = (cleanLine.match(/\(/g) || []).length;
                     const replClose = (cleanLine.match(/\)/g) || []).length;
                     
+                    // Case A: Target line is the OPENER of a multi-line expression
                     if (origOpen > origClose && replOpen <= replClose) {
-                        // Original has unclosed parens, replacement is complete
-                        // Find the closing of the multi-line expression
                         let depth = origOpen - origClose;
                         let endIdx = idx;
                         for (let j = idx + 1; j < lines.length && depth > 0; j++) {
@@ -360,8 +369,7 @@ async function applyFix(res, loc, backups = new Map()) {
                             endIdx = j;
                         }
                         if (endIdx > idx) {
-                            const removedLines = endIdx - idx;
-                            console.log(`   ${C.GRAY}🔧 Expanding REPLACE to cover ${removedLines + 1}-line expression (lines ${p.line}-${endIdx + 1})${C.RESET}`);
+                            console.log(`   ${C.GRAY}🔧 Expanding REPLACE to cover ${endIdx - idx + 1}-line expression (lines ${p.line}-${endIdx + 1})${C.RESET}`);
                             for (let j = idx; j <= endIdx; j++) {
                                 console.log(`   ${C.RED}-  Line ${j + 1}: ${lines[j].trim()}${C.RESET}`);
                             }
@@ -372,6 +380,37 @@ async function applyFix(res, loc, backups = new Map()) {
                             console.log(`   ${C.RED}-  Line ${p.line}: ${originalLine.trim()}${C.RESET}`);
                             console.log(`   ${C.GREEN}+  Line ${p.line}: ${cleanLine.trim()}${C.RESET}`);
                         }
+                    // Case B: Target line is a MIDDLE/CONTINUATION line of a multi-line expression
+                    // Detect: replacement is a complete assignment (this.x = ...; ) but current line is just a string
+                    } else if (/^\s*this\.\w+\s*=/.test(cleanLine) && cleanLine.trim().endsWith(';') && !/^\s*this\./.test(originalLine)) {
+                        // Search BACKWARD for the opening line (this.xxx = page.locator()
+                        let startIdx = idx;
+                        for (let j = idx - 1; j >= 0; j--) {
+                            startIdx = j;
+                            const lo = (lines[j].match(/\(/g) || []).length;
+                            const lc = (lines[j].match(/\)/g) || []).length;
+                            if (lo > lc) break; // Found the opener
+                        }
+                        // Search FORWARD for the closing );  
+                        let endIdx = idx;
+                        let depth = 0;
+                        for (let j = startIdx; j < lines.length; j++) {
+                            const lo = (lines[j].match(/\(/g) || []).length;
+                            const lc = (lines[j].match(/\)/g) || []).length;
+                            depth += lo - lc;
+                            endIdx = j;
+                            if (depth <= 0 && j >= idx) break;
+                        }
+                        const totalLines = endIdx - startIdx + 1;
+                        console.log(`   ${C.GRAY}🔧 Expanding REPLACE to cover ${totalLines}-line expression (lines ${startIdx + 1}-${endIdx + 1})${C.RESET}`);
+                        for (let j = startIdx; j <= endIdx; j++) {
+                            console.log(`   ${C.RED}-  Line ${j + 1}: ${lines[j].trim()}${C.RESET}`);
+                        }
+                        // Use indentation from the opener line
+                        const openerIndent = lines[startIdx].match(/^\s*/)[0];
+                        const expandedClean = openerIndent + cleanLine.trim();
+                        lines.splice(startIdx, totalLines, expandedClean);
+                        console.log(`   ${C.GREEN}+  Line ${startIdx + 1}: ${expandedClean.trim()}${C.RESET}`);
                     } else {
                         lines[idx] = cleanLine;
                         console.log(`   ${C.RED}-  Line ${p.line}: ${originalLine.trim()}${C.RESET}`);

package/package.json

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