72 lines
2.2 KiB
C#
72 lines
2.2 KiB
C#
namespace BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;
|
|
|
|
public partial class PlaywrightWebDriver : IWebBrowser
|
|
{
|
|
private IServiceProvider _services => _instance.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)
|
|
{
|
|
_instance = instance;
|
|
_logger = logger;
|
|
_instance.SetServiceProvider(services);
|
|
}
|
|
|
|
public void SetAgent(Agent agent)
|
|
{
|
|
_agent = agent;
|
|
}
|
|
|
|
private ILocator? Locator(string contextId, HtmlElementContextOut context)
|
|
{
|
|
ILocator element = default;
|
|
if (!string.IsNullOrEmpty(context.ElementId))
|
|
{
|
|
element = _instance.GetPage(contextId).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.GetPage(contextId).Locator($"[name='{context.ElementName}']");
|
|
var count = element.CountAsync().Result;
|
|
if (count == 0)
|
|
{
|
|
_logger.LogError($"Can't locate element {role} {context.ElementName}");
|
|
return null;
|
|
}
|
|
else if (count > 1)
|
|
{
|
|
_logger.LogError($"Located multiple elements {role} {context.ElementName}");
|
|
return null;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (context.Index < 0)
|
|
{
|
|
_logger.LogError($"Can't locate the web element {context.Index}.");
|
|
return null;
|
|
}
|
|
element = _instance.GetPage(contextId).Locator(context.TagName).Nth(context.Index);
|
|
}
|
|
|
|
return element;
|
|
}
|
|
|
|
public void SetServiceProvider(IServiceProvider services)
|
|
{
|
|
_instance.SetServiceProvider(_services);
|
|
}
|
|
}
|