From 16300b9a1d855c833d10aceec436d56a31e7e26b Mon Sep 17 00:00:00 2001
From: Visagan Guruparan <103048@smsassist.com>
Date: Thu, 10 Jul 2025 14:22:54 -0500
Subject: [PATCH] Add hook to handle locate element for PlayWright

---
 .../BotSharp.Abstraction/Browsing/IWebDriverHook.cs          | 1 +
 .../PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs    | 5 +++++
 .../UtilFunctions/UtilWebLocateElementFn.cs                  | 1 +
 3 files changed, 7 insertions(+)

diff --git a/src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs b/src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs
index ce5d03b6..346f2f23 100644
--- a/src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs
+++ b/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);
 }
diff --git a/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs b/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs
index e70d7f27..b4b2eb2f 100644
--- a/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs
+++ b/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs
@@ -114,6 +114,11 @@ public partial class PlaywrightWebDriver
             // 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)
         {
diff --git a/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs b/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs
index d30048c8..e2e3d34f 100644
--- a/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs
+++ b/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs
@@ -30,6 +30,7 @@ public class UtilWebLocateElementFn : IFunctionCallback
             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")}. ";