module Selenium::WebDriver::Firefox::Driver

def marionette?(opts)
  opts.delete(:marionette) != false &&
    (!opts[:desired_capabilities] || opts[:desired_capabilities][:marionette] != false)
end

def new(**opts)
  if marionette?(opts)
    Firefox::Marionette::Driver.new(opts)
  else
    Firefox::Legacy::Driver.new(opts)
  end
end