From a1d89d94ff048ebc5d21172bd2992415ed133a35 Mon Sep 17 00:00:00 2001
From: amhsirak <karishmashuklaa@gmail.com>
Date: Wed, 20 Nov 2024 02:37:19 +0530
Subject: [PATCH] feat: use outerHTMl

---
 .../workflow-management/classes/Generator.ts  | 20 +++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/server/src/workflow-management/classes/Generator.ts b/server/src/workflow-management/classes/Generator.ts
index 1ecfac87..72580fa7 100644
--- a/server/src/workflow-management/classes/Generator.ts
+++ b/server/src/workflow-management/classes/Generator.ts
@@ -543,6 +543,22 @@ export class WorkflowGenerator {
     return bestSelector;
   }
 
+  public getOuterHTML = async (page: Page, parentSelector: string): Promise<string> => {
+    try {
+      const outerHTML = await page.evaluate((parentSelector: string) => {
+        const parentElement = document.querySelector(parentSelector);
+        return parentElement ? parentElement.outerHTML : '';
+      }, parentSelector);
+  
+      console.log(`Outer html: ${outerHTML}`)
+      return outerHTML;
+    } catch (error) {
+      console.error('Error in getOuterHTML:', error);
+      return '';
+    }
+  };
+  
+
   /**
    * Generates data for highlighting the element on client side and emits the
    * highlighter event to the client.
@@ -557,10 +573,10 @@ export class WorkflowGenerator {
     if (rect) {
       if (this.getList === true) {
         if (this.listSelector !== '') {
-          const childSelectors = await getChildSelectors(page, this.listSelector || '');
+          const childSelectors = await getChildSelectors(page, await this.getOuterHTML(page, this.listSelector));
           this.socket.emit('highlighter', { rect, selector: displaySelector, elementInfo, childSelectors })
           console.log(`Child Selectors: ${childSelectors}`)
-          console.log(`List sekector ${this.listSelector}`)
+          console.log(`Parent Selector: ${this.listSelector}`)
         } else {
           this.socket.emit('highlighter', { rect, selector: displaySelector, elementInfo });
         }