67 lines
2.1 KiB
C#
67 lines
2.1 KiB
C#
using BotSharp.Abstraction.Browsing.Settings;
|
|
|
|
namespace BotSharp.Plugin.WebDriver.Functions;
|
|
|
|
public class OpenBrowserFn : IFunctionCallback
|
|
{
|
|
public string Name => "open_browser";
|
|
|
|
private readonly IServiceProvider _services;
|
|
private readonly IWebBrowser _browser;
|
|
private readonly WebBrowsingSettings _webBrowsingSettings;
|
|
|
|
public OpenBrowserFn(IServiceProvider services,
|
|
IWebBrowser browser,
|
|
WebBrowsingSettings webBrowsingSettings)
|
|
{
|
|
_services = services;
|
|
_browser = browser;
|
|
_webBrowsingSettings = webBrowsingSettings;
|
|
}
|
|
|
|
public async Task<bool> Execute(RoleDialogModel message)
|
|
{
|
|
var convService = _services.GetRequiredService<IConversationService>();
|
|
var args = JsonSerializer.Deserialize<BrowsingContextIn>(message.FunctionArgs);
|
|
|
|
var webDriverService = _services.GetRequiredService<WebDriverService>();
|
|
var url = webDriverService.ReplaceToken(args.Url);
|
|
|
|
url = url.Replace("https://https://", "https://");
|
|
var msgInfo = new MessageInfo
|
|
{
|
|
AgentId = message.CurrentAgentId,
|
|
ContextId = convService.ConversationId,
|
|
MessageId = message.MessageId
|
|
};
|
|
var result = await _browser.LaunchBrowser(msgInfo, new BrowserActionArgs
|
|
{
|
|
Headless = _webBrowsingSettings.Headless
|
|
});
|
|
result = await _browser.GoToPage(msgInfo, new PageActionArgs
|
|
{
|
|
Url = url
|
|
});
|
|
|
|
if (result.IsSuccess)
|
|
{
|
|
message.Content = string.IsNullOrEmpty(url) ? $"Launch browser with blank page successfully." : $"Open website {url} successfully.";
|
|
}
|
|
else
|
|
{
|
|
message.Content = $"Launch browser failed. {result.Message}";
|
|
}
|
|
|
|
var path = webDriverService.GetScreenshotFilePath(message.MessageId);
|
|
|
|
message.Data = await _browser.ScreenshotAsync(new MessageInfo
|
|
{
|
|
AgentId = message.CurrentAgentId,
|
|
ContextId = convService.ConversationId,
|
|
MessageId = message.MessageId
|
|
}, path);
|
|
|
|
return result.IsSuccess;
|
|
}
|
|
}
|