using OpenQA.Selenium;
using OpenQA.Selenium.Chrome;
using OpenQA.Selenium.Firefox;
using WebDriverManager;
using WebDriverManager.DriverConfigs.Impl;

namespace BrowserHelper.Driver;
public class BrowserDriver : IBrowserDriver
{
    public IWebDriver GetChromeDriver(bool headless = true)
    {
        new DriverManager().SetUpDriver(new ChromeConfig());
        ChromeOptions options = new ChromeOptions();
        if (headless)
            options.AddArguments("--headless");
        return new ChromeDriver(options);
    }

    public IWebDriver GetFirefoxDriver(bool headless = true)
    {
        new DriverManager().SetUpDriver(new FirefoxConfig());
        FirefoxOptions options = new FirefoxOptions();
        if (headless)
            options.AddArguments("--headless");
        return new FirefoxDriver(options);
    }
}

public enum BrowserType
{
    Chrome,
    Firefox
}