using Microsoft.Extensions.Logging;

namespace BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;

public partial class PlaywrightWebDriver : IWebBrowser
{
    private readonly IServiceProvider _services;
    private readonly PlaywrightInstance _instance;
    private readonly ILogger _logger;
    public PlaywrightInstance Instance => _instance;

    public Agent Agent => _agent;
    private Agent _agent;

    public PlaywrightWebDriver(IServiceProvider services, PlaywrightInstance instance, ILogger<PlaywrightWebDriver> logger)
    {
        _services = services;
        _instance = instance;
        _logger = logger;
    }

    public void SetAgent(Agent agent)
    {
        _agent = agent;
    }

    private ILocator Locator(HtmlElementContextOut context)
    {
        ILocator element = default;
        if (!string.IsNullOrEmpty(context.ElementId))
        {
            element = _instance.Page.Locator($"#{context.ElementId}");
        }
        else if (!string.IsNullOrEmpty(context.ElementName))
        {
            var role = context.TagName switch
            {
                "input" => AriaRole.Textbox,
                "textarea" => AriaRole.Textbox,
                "button" => AriaRole.Button,
                _ => AriaRole.Generic
            };
            element = _instance.Page.Locator($"[name='{context.ElementName}']");

            if (element.CountAsync().Result == 0)
            {
                _logger.LogError($"Can't locate element {role} {context.ElementName}");
            }
        }
        else
        {
            if (context.Index < 0)
            {
                throw new Exception($"Can't locate the web element {context.Index}.");
            }
            element = _instance.Page.Locator(context.TagName).Nth(context.Index);
        }

        return element;
    }
}