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

namespace BotSharp.Plugin.WebDriver.UtilFunctions;

public class UtilWebCloseBrowserFn : IFunctionCallback
{
    public string Name => "util-web-close_browser";
    public string Indication => "Closing web browser.";
    private readonly IServiceProvider _services;
    private readonly ILogger _logger;

    public UtilWebCloseBrowserFn(
        IServiceProvider services,
        ILogger<UtilWebCloseBrowserFn> logger)
    {
        _services = services;
        _logger = logger;
    }

    public async Task<bool> Execute(RoleDialogModel message)
    {
        var conv = _services.GetRequiredService<IConversationService>();

        var browser = _services.GetRequiredService<IWebBrowser>();
        var msg = new MessageInfo
        {
            AgentId = message.CurrentAgentId,
            MessageId = message.MessageId,
            ContextId = message.CurrentAgentId,
        };

        await browser.CloseBrowser(message.CurrentAgentId);

        message.Content = $"Browser closed.";

        var webDriverService = _services.GetRequiredService<WebDriverService>();
        var path = webDriverService.GetScreenshotFilePath(message.MessageId);

        message.Data = await browser.ScreenshotAsync(msg, path);

        return true;
    }
}