Merge pull request #1096 from visagang/features/vguruparan

Add hook to handle locate element for PlayWright

Author: iceljc

src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs

@@ -5,4 +5,5 @@ namespace BotSharp.Abstraction.Browsing;
 public interface IWebDriverHook
 {
     Task<List<string>> GetUploadFiles(MessageInfo message);
+    Task OnLocateElement(MessageInfo message, string content);
 }

src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs

@@ -114,6 +114,11 @@ public async Task<BrowserActionResult> LocateElement(MessageInfo message, Elemen
             // fix if html has &
             result.Body = HttpUtility.HtmlDecode(html);
             result.IsSuccess = true;
+            var hooks = _services.GetServices<IWebDriverHook>();
+            foreach (var hook in hooks)
+            {
+                await hook.OnLocateElement(message, result.Body);
+            }
         }
         else if (count > 1)
         {

src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebCloseBrowserFn.cs

@@ -28,7 +28,7 @@ public async Task<bool> Execute(RoleDialogModel message)
             ContextId = webDriverService.GetMessageContext(message)
         };
 
-        await browser.CloseBrowser(message.CurrentAgentId);
+        await browser.CloseBrowser(msg.ContextId);
 
         message.Content = $"Browser closed.";
 

src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs

@@ -30,6 +30,7 @@ public async Task<bool> Execute(RoleDialogModel message)
             MessageId = message.MessageId,
             ContextId = webDriverService.GetMessageContext(message)
         };
+        browser.SetServiceProvider(_services);
         var result = await browser.LocateElement(msg, locatorArgs);
 
         message.Content = $"Locating element {(result.IsSuccess ? "success" : "failed")}. ";