126 lines
4.2 KiB
C#
126 lines
4.2 KiB
C#
namespace BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;
|
|
|
|
public partial class PlaywrightWebDriver
|
|
{
|
|
public async Task<BrowserActionResult> GoToPage(MessageInfo message, PageActionArgs args)
|
|
{
|
|
var result = new BrowserActionResult();
|
|
var context = await _instance.GetContext(message.ContextId);
|
|
try
|
|
{
|
|
IPage? page = null;
|
|
if (!args.OpenNewTab)
|
|
{
|
|
page = _instance.Contexts[message.ContextId].Pages.LastOrDefault();
|
|
|
|
if (page != null)
|
|
{
|
|
if (!args.OpenBlankPage)
|
|
{
|
|
if (!_instance.Pages[message.ContextId].Contains(page))
|
|
{
|
|
_instance.Pages[message.ContextId].Add(page);
|
|
}
|
|
}
|
|
if (args.OpenBlankPage && page.Url != "about:blank")
|
|
{
|
|
await page.EvaluateAsync(@"() => {
|
|
window.open('', '_blank');
|
|
}");
|
|
}
|
|
|
|
if (args.EnableResponseCallback)
|
|
{
|
|
page.Request += async (sender, e) =>
|
|
{
|
|
await _instance.HandleFetchRequest(e, message, args);
|
|
};
|
|
|
|
page.Response += async (sender, e) =>
|
|
{
|
|
await _instance.HandleFetchResponse(e, message, args);
|
|
};
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
page = await _instance.NewPage(message, args);
|
|
}
|
|
|
|
if (page == null)
|
|
{
|
|
page = await _instance.NewPage(message, args);
|
|
}
|
|
|
|
// Active current tab
|
|
await page.BringToFrontAsync();
|
|
var response = await page.GotoAsync(args.Url, new PageGotoOptions
|
|
{
|
|
Timeout = args.Timeout > 0 ? args.Timeout : 30000
|
|
});
|
|
|
|
if (args.Selectors != null)
|
|
{
|
|
// 使用传入的选择器列表进行并行等待
|
|
var tasks = args.Selectors.Select(selector =>
|
|
page.WaitForSelectorAsync(selector, new PageWaitForSelectorOptions
|
|
{
|
|
Timeout = args.Timeout > 0 ? args.Timeout : 30000
|
|
})
|
|
).ToArray();
|
|
|
|
await Task.WhenAll(tasks);
|
|
|
|
// 在此处提取所有选择器的 HTML 内容
|
|
var contentTasks = args.Selectors.Select(selector => page.InnerHTMLAsync(selector)).ToArray();
|
|
var contents = await Task.WhenAll(contentTasks);
|
|
|
|
result.IsSuccess = true;
|
|
result.Body = string.Join(", ", contents.Select((content, index) => $"{args.Selectors[index]}: {content}"));
|
|
|
|
return result;
|
|
}
|
|
|
|
await page.WaitForLoadStateAsync(LoadState.DOMContentLoaded);
|
|
if (args.WaitForNetworkIdle)
|
|
{
|
|
await page.WaitForLoadStateAsync(LoadState.NetworkIdle, new PageWaitForLoadStateOptions
|
|
{
|
|
Timeout = args.Timeout > 0 ? args.Timeout : 30000
|
|
});
|
|
}
|
|
|
|
if (args.WaitTime > 0)
|
|
{
|
|
await Task.Delay(args.WaitTime * 1000);
|
|
}
|
|
|
|
result.ResponseStatusCode = response.Status;
|
|
result.UrlAfterAction = page.Url;
|
|
if (response.Status == 200)
|
|
{
|
|
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
|
|
{
|
|
result.Message = response.StatusText;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
result.Message = ex.Message;
|
|
result.StackTrace = ex.StackTrace;
|
|
_logger.LogError(ex.Message);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
}
|