class ActionDispatch::SystemTesting::Browser

def configure
  initialize_options
  yield options if block_given? && options
end

def initialize(name)
  @name = name
  set_default_options
end

def initialize_options
  @options ||=
    case type
    when :chrome
      ::Selenium::WebDriver::Chrome::Options.new
    when :firefox
      ::Selenium::WebDriver::Firefox::Options.new
    end
end

def preload
  case type
  when :chrome
    resolve_driver_path(::Selenium::WebDriver::Chrome)
  when :firefox
    resolve_driver_path(::Selenium::WebDriver::Firefox)
  end
end

def resolve_driver_path(namespace)
  namespace::Service.driver_path = ::Selenium::WebDriver::DriverFinder.path(
    options || namespace::Options.new,
    namespace::Service
  )
end

def set_default_options
  case name
  when :headless_chrome
    set_headless_chrome_browser_options
  when :headless_firefox
    set_headless_firefox_browser_options
  end
end

def set_headless_chrome_browser_options
  configure do |capabilities|
    capabilities.add_argument("--headless")
    capabilities.add_argument("--disable-gpu") if Gem.win_platform?
  end
end

def set_headless_firefox_browser_options
  configure do |capabilities|
    capabilities.add_argument("-headless")
  end
end

def type
  case name
  when :headless_chrome
    :chrome
  when :headless_firefox
    :firefox
  else
    name
  end
end