enhance self-heal selector extraction with context information

Author: tech-sushant

src/tools/selfheal-utils/selfheal.ts

@@ -4,6 +4,10 @@ import config from "../../config.js";
 interface SelectorMapping {
   originalSelector: string;
   healedSelector: string;
+  context: {
+    before: string;
+    after: string;
+  };
 }
 
 export async function getSelfHealSelectors(sessionId: string) {
@@ -24,20 +28,40 @@ export async function getSelfHealSelectors(sessionId: string) {
 }
 
 function extractHealedSelectors(logText: string): SelectorMapping[] {
-  // Pattern to match SELFHEAL entries with healed selectors
+  // Split log text into lines for easier context handling
+  const logLines = logText.split("\n");
+
+  // Pattern to match successful SELFHEAL entries only
   const selfhealPattern =
-    /SELFHEAL\s*{\s*"status":"true",\s*"data":\s*{\s*"using":"css selector",\s*"value":"(.*?)"}/g;
+    /SELFHEAL\s*{\s*"status":"true",\s*"data":\s*{\s*"using":"css selector",\s*"value":"(.*?)"}/;
 
   // Pattern to match preceding selector requests
   const requestPattern =
     /POST \/session\/[^/]+\/element.*?"using":"css selector","value":"(.*?)"/g;
 
-  // Find all healed selectors
-  const healedSelectors: string[] = [];
-  let healedMatch;
-  while ((healedMatch = selfhealPattern.exec(logText)) !== null) {
-    healedSelectors.push(healedMatch[1]);
-  }
+  // Find all successful healed selectors with their line numbers and context
+  const healedSelectors: Array<{
+    selector: string;
+    lineNumber: number;
+    context: { before: string; after: string };
+  }> = [];
+
+  logLines.forEach((line, index) => {
+    const match = line.match(selfhealPattern);
+    if (match) {
+      const beforeLine = index > 0 ? logLines[index - 1] : "";
+      const afterLine = index < logLines.length - 1 ? logLines[index + 1] : "";
+
+      healedSelectors.push({
+        selector: match[1],
+        lineNumber: index,
+        context: {
+          before: beforeLine,
+          after: afterLine,
+        },
+      });
+    }
+  });
 
   // Find all selector requests
   const selectorRequests: string[] = [];
@@ -53,7 +77,8 @@ function extractHealedSelectors(logText: string): SelectorMapping[] {
   for (let i = 0; i < minLength; i++) {
     healedMappings.push({
       originalSelector: selectorRequests[i],
-      healedSelector: healedSelectors[i],
+      healedSelector: healedSelectors[i].selector,
+      context: healedSelectors[i].context,
     });
   }