BotSharp/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.InputUserText.cs

namespace BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;

public partial class PlaywrightWebDriver
{
    public async Task<BrowserActionResult> InputUserText(BrowserActionParams actionParams)
    {
        var result = new BrowserActionResult();
        await _instance.Wait(actionParams.ConversationId);

        var page = _instance.GetPage(actionParams.ConversationId);
        ILocator locator = default;
        int count = 0;

        // try attribute
        if (count == 0 && !string.IsNullOrEmpty(actionParams.Context.AttributeName))
        {
            locator = page.Locator($"[{actionParams.Context.AttributeName}='{actionParams.Context.AttributeValue}']");
            count = await locator.CountAsync();
        }

        // Find by text exactly match
        if (count == 0)
        {
            locator = page.GetByRole(AriaRole.Textbox, new PageGetByRoleOptions
            {
                Name = actionParams.Context.ElementText
            });
            count = await locator.CountAsync();
        }

        if (count == 0)
        {
            locator = page.GetByPlaceholder(actionParams.Context.ElementText);
            count = await locator.CountAsync();
        }

        if (count == 0)
        {
            var driverService = _services.GetRequiredService<WebDriverService>();
            var html = await FilteredInputHtml(actionParams.ConversationId);
            var htmlElementContextOut = await driverService.InferElement(actionParams.Agent,
                html,
                actionParams.Context.ElementText,
                actionParams.MessageId);
            locator = Locator(actionParams.ConversationId, htmlElementContextOut);
            count = await locator.CountAsync();
        }
        else if (count > 0)
        {
            try
            {
                await locator.FillAsync(actionParams.Context.InputText);
                if (actionParams.Context.PressEnter.HasValue && actionParams.Context.PressEnter.Value)
                {
                    await locator.PressAsync("Enter");
                }

                // Triggered ajax
                await _instance.Wait(actionParams.ConversationId);
                result.IsSuccess = true;
            }
            catch (Exception ex)
            {
                result.Message = ex.Message;
                result.StackTrace = ex.StackTrace;
                _logger.LogError(ex.Message);
            }
        }

        return result;
    }

    private async Task<string> FilteredInputHtml(string conversationId)
    {
        var driverService = _services.GetRequiredService<WebDriverService>();

        // Retrieve the page raw html and infer the element path
        var body = await _instance.GetPage(conversationId).QuerySelectorAsync("body");

        var str = new List<string>();
        var inputs = await body.QuerySelectorAllAsync("input");
        foreach (var input in inputs)
        {
            var text = await input.TextContentAsync();
            var id = await input.GetAttributeAsync("id");
            var name = await input.GetAttributeAsync("name");
            var type = await input.GetAttributeAsync("type");
            var placeholder = await input.GetAttributeAsync("placeholder");

            str.Add(driverService.AssembleMarkup("input", new MarkupProperties
            {
                Id = id,
                Name = name,
                Type = type,
                Text = text,
                Placeholder = placeholder
            }));
        }

        inputs = await body.QuerySelectorAllAsync("textarea");
        foreach (var input in inputs)
        {
            var text = await input.TextContentAsync();
            var id = await input.GetAttributeAsync("id");
            var name = await input.GetAttributeAsync("name");
            var type = await input.GetAttributeAsync("type");
            var placeholder = await input.GetAttributeAsync("placeholder");
            str.Add(driverService.AssembleMarkup("textarea", new MarkupProperties
            {
                Id = id,
                Name = name,
                Type = type,
                Text = text,
                Placeholder = placeholder
            }));
        }

        return string.Join("", str);
    }
}