deflake 1.2.43 → 1.2.45

package/cli.js

@@ -7,6 +7,104 @@ const fs = require('fs');
 const path = require('path');
 const pkg = require('./package.json');
 
+// --- AX TREE PARSER: Extract available selectors from Playwright Accessibility Tree ---
+function parseAxTreeSelectors(axTreeContent) {
+    const selectors = [];
+    const seen = new Set();
+    const lines = axTreeContent.split('\n');
+    
+    for (const line of lines) {
+        // Match: - role "name" or - role 'name'
+        // Examples:
+        //   - heading "People" [level=3]
+        //   - button "Add a New Employee" [ref=e144]
+        //   - searchbox "Search by Name" [ref=e163]
+        //   - link "Manage Employees" [ref=e128]
+        //   - textbox "Leave Start Date*" [ref=e130]
+        //   - tab "Basic Information" [selected]
+        //   - combobox "Leave Type*"
+        
+        const match = line.match(/- (heading|button|link|searchbox|textbox|combobox|tab|switch|checkbox|radio|slider)\s+["']([^"']+)["']/i);
+        if (match) {
+            const [, role, name] = match;
+            const key = `${role}:${name}`;
+            if (seen.has(key)) continue;
+            seen.add(key);
+            
+            let selector;
+            switch (role.toLowerCase()) {
+                case 'heading':
+                    selector = `page.getByRole('heading', { name: '${name}' })`;
+                    break;
+                case 'button':
+                    // Clean button names that include icon names like "user-add Add a New Employee"
+                    const cleanName = name.replace(/^[a-z-]+ /i, '').trim();
+                    if (cleanName !== name && cleanName.length > 2) {
+                        selector = `page.getByRole('button', { name: '${cleanName}' })`;
+                    } else {
+                        selector = `page.getByRole('button', { name: '${name}' })`;
+                    }
+                    break;
+                case 'link':
+                    const linkName = name.trim();
+                    if (linkName.length > 1) {
+                        selector = `page.getByRole('link', { name: '${linkName}' })`;
+                    }
+                    break;
+                case 'searchbox':
+                    selector = `page.getByPlaceholder('${name}')`;
+                    break;
+                case 'textbox':
+                    selector = `page.getByLabel('${name}')`;
+                    break;
+                case 'combobox':
+                    selector = `page.getByLabel('${name}')`;
+                    break;
+                case 'tab':
+                    selector = `page.getByRole('tab', { name: '${name}' })`;
+                    break;
+                default:
+                    selector = `page.getByRole('${role}', { name: '${name}' })`;
+            }
+            if (selector) selectors.push(selector);
+        }
+        
+        // Also extract text content from generic elements with cursor=pointer (action links)
+        const genericTextMatch = line.match(/- generic.*\[cursor=pointer\]/);
+        if (genericTextMatch) {
+            // Look for the text child in the next few lines
+            const lineIdx = lines.indexOf(line);
+            for (let j = lineIdx + 1; j < Math.min(lineIdx + 4, lines.length); j++) {
+                const textMatch = lines[j].match(/- text: (.+)/);
+                if (textMatch) {
+                    const text = textMatch[1].trim();
+                    if (text.length > 1 && text.length < 50) {
+                        const key = `text:${text}`;
+                        if (!seen.has(key)) {
+                            seen.add(key);
+                            selectors.push(`page.getByText('${text}', { exact: true }).first()`);
+                        }
+                    }
+                    break;
+                }
+            }
+        }
+        
+        // Extract placeholder from textbox entries  
+        const placeholderMatch = line.match(/\/placeholder: (.+)/);
+        if (placeholderMatch) {
+            const ph = placeholderMatch[1].trim();
+            const key = `placeholder:${ph}`;
+            if (!seen.has(key)) {
+                seen.add(key);
+                selectors.push(`page.getByPlaceholder('${ph}')`);
+            }
+        }
+    }
+    
+    return selectors;
+}
+
 // --- PREMIUM COLORS ---
 const C = {
     RESET: "\x1b[0m",
@@ -224,9 +322,20 @@ async function analyzeAndFix(artifacts, client, argv, capturedOutput = '', backu
             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 error block for richer diagnosis
+            // Step 3: Parse AX Tree for available selectors
+            const axSelectors = parseAxTreeSelectors(content);
+            if (axSelectors.length > 0) {
+                console.log(`   ${C.GRAY}🎯 Available selectors: ${axSelectors.length} elements found in AX Tree${C.RESET}`);
+            }
+            
+            // Step 4: API call — send error-context, console error block, AND available selectors
+            const selectorsContext = axSelectors.length > 0 
+                ? `\n\n=== AVAILABLE SELECTORS (pre-computed from page) ===\nThese selectors are VERIFIED to exist on the page. Pick from this list:\n${axSelectors.map((s, i) => `${i+1}. ${s}`).join('\n')}\n=== END AVAILABLE SELECTORS ===`
+                : '';
+            const enrichedOutput = (relevantOutput || '') + selectorsContext;
+            
             console.log(`   ${C.GRAY}🌐 Calling DeFlake API...${C.RESET}`);
-            const res = await client.heal(null, art.htmlPath, loc, source, argv.fix, relevantOutput);
+            const res = await client.heal(null, art.htmlPath, loc, source, argv.fix, enrichedOutput);
             
             if (res && res.status === 'success') {
                 console.log(`   ${C.GREEN}✅ API Response: SUCCESS${C.RESET}`);

package/package.json

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