From 468f84981c202511e63ee595b1803bd99dea73ef Mon Sep 17 00:00:00 2001
From: Haiping Chen <haiping008@gmail.com>
Date: Mon, 7 Oct 2024 20:20:48 -0500
Subject: [PATCH] ReadInnerHTMLAsBody option

---
 .../Browsing/Models/PageActionArgs.cs                     | 2 ++
 .../PlaywrightDriver/PlaywrightWebDriver.GoToPage.cs      | 8 ++++++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/Infrastructure/BotSharp.Abstraction/Browsing/Models/PageActionArgs.cs b/src/Infrastructure/BotSharp.Abstraction/Browsing/Models/PageActionArgs.cs
index d260a09d..aa4154ff 100644
--- a/src/Infrastructure/BotSharp.Abstraction/Browsing/Models/PageActionArgs.cs
+++ b/src/Infrastructure/BotSharp.Abstraction/Browsing/Models/PageActionArgs.cs
@@ -43,4 +43,6 @@ public class PageActionArgs
     /// Wait time in seconds after page is opened
     /// </summary>
     public int WaitTime { get; set; }
+
+    public bool ReadInnerHTMLAsBody { get; set; } = false;
 }
diff --git a/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.GoToPage.cs b/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.GoToPage.cs
index cc1a0009..108ff433 100644
--- a/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.GoToPage.cs
+++ b/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.GoToPage.cs
@@ -67,9 +67,13 @@ public partial class PlaywrightWebDriver
             result.ResponseStatusCode = response.Status;
             if (response.Status == 200)
             {
-                // Disable this due to performance issue, some page is too large
-                // result.Body = await page.InnerHTMLAsync("body");
                 result.IsSuccess = true;
+
+                // Be careful if page is too large, it will cause performance issue
+                if (args.ReadInnerHTMLAsBody)
+                {
+                    result.Body = await page.InnerHTMLAsync("body");
+                }
             }
             else
             {