deflake 1.1.9 → 1.2.1

package/cli.js

@@ -220,7 +220,7 @@ function parseCypressLogs(fullLog) {
     return specBlocks;
 }
 
-async function runHealer(logContent, htmlPath, apiUrl, testName) {
+async function runHealer(logContent, htmlPath, apiUrl, testName, applyFix = false) {
     // Check file size limit (Basic check to avoid 429s on huge files)
     if (htmlPath && fs.existsSync(htmlPath)) {
         const stats = fs.statSync(htmlPath);
@@ -282,7 +282,7 @@ async function runHealer(logContent, htmlPath, apiUrl, testName) {
     }
 
     try {
-        const result = await client.heal(finalLogPath, htmlPath, failureLoc, sourceCodeContent);
+        const result = await client.heal(finalLogPath, htmlPath, failureLoc, sourceCodeContent, applyFix);
 
         if (result && result.status === 'success') {
             // Return structured object for grouping
@@ -640,7 +640,10 @@ async function runDoctor(argv) {
         if (result.status === 'success') {
             const usage = result.data;
             process.stdout.write(`\r  ✅ API Connected! (Tier: ${C.GREEN}${usage.tier.toUpperCase()}${C.RESET})                              \n`);
-            console.log(`     Quota: ${usage.usage}/${usage.limit}`);
+            const u = usage;
+            const remaining = u.limit - u.usage;
+            console.log(`     Fix Quota:  ${C.GREEN}${remaining}/${u.limit} remaining${C.RESET}`);
+            console.log(`     Analysis:   ${C.GREEN}FREE${C.RESET} (Pay-per-fix model enabled)`);
         } else {
             process.stdout.write(`\r  ❌ ${C.RED}API Connectivity Failed${C.RESET}       \n`);
             if (result.code === 401 || result.code === 403) {
@@ -731,6 +734,9 @@ async function analyzeFailures(artifacts, fullLog, client) {
     }
 
     const framework = artifacts.length > 0 && artifacts[0].htmlPath?.includes('cypress') ? 'Cypress' : 'Playwright';
+    const results = [];
+    const processLimit = Math.min(artifacts.length, limit);
+    const batchArtifacts = artifacts.slice(0, processLimit);
 
     console.log(`🔍 Analyzing ${batchArtifacts.length} failure(s)...`);
 
@@ -768,7 +774,7 @@ async function analyzeFailures(artifacts, fullLog, client) {
         process.stdout.write(`\r⏳ Analyzing ${displayName}...          `);
 
         try {
-            const result = await runHealer(specificLog, art.htmlPath, argv.apiUrl, art.name);
+            const result = await runHealer(specificLog, art.htmlPath, argv.apiUrl, art.name, argv.fix);
 
             if (result && result.status === 'success') {
                 results.push(result);
@@ -824,7 +830,7 @@ async function analyzeFailures(artifacts, fullLog, client) {
         if (updatedUsage.status === 'success') {
             const u = updatedUsage.data;
             const remaining = u.limit - u.usage;
-            console.log(`  - Monthly Quota:     ${C.GREEN}${remaining}/${u.limit} remaining${C.RESET}`);
+            console.log(`  - Fix Quota:         ${C.GREEN}${remaining}/${u.limit} remaining (Analysis is FREE)${C.RESET}`);
         }
     } catch (e) {
         // usage fetch failed is non-critical

package/client.js

@@ -94,7 +94,7 @@ class DeFlakeClient {
         }
     }
 
-    async heal(logPath, htmlPath, failureLocation = null, sourceCode = null) {
+    async heal(logPath, htmlPath, failureLocation = null, sourceCode = null, applyFix = false) {
         try {
             // ... (rest of the check logic) ...
             if (!fs.existsSync(logPath)) throw new Error(`Log file not found: ${logPath}`);
@@ -119,7 +119,8 @@ class DeFlakeClient {
                 html_snapshot: htmlContent || "",
                 failing_line: failureLocation ? `Line ${failureLocation.rootLine}` : "",
                 source_code: sourceCode || "",
-                framework: this.framework
+                framework: this.framework,
+                apply_fix: applyFix
             };
 
             const response = await axios.post(this.apiUrl, payload, {

package/package.json

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