deflake 1.2.36 → 1.2.37

package/cli.js

@@ -216,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) {
@@ -270,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/package.json

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