class Selenium::WebDriver::Options

def ==(other)
  return false unless other.is_a? self.class
  as_json == other.as_json
end

def add_option(name, value = nil)
  name, value = name.first if value.nil? && name.is_a?(Hash)
  @options[name] = value
end

def as_json(*)
  options = @options.dup
  downloads = options.delete(:enable_downloads)
  options['se:downloadsEnabled'] = downloads unless downloads.nil?
  w3c_options = process_w3c_options(options)
  browser_options = self.class::CAPABILITIES.each_with_object({}) do |(capability_alias, capability_name), hash|
    capability_value = options.delete(capability_alias)
    hash[capability_name] = capability_value unless capability_value.nil?
  end
  raise Error::WebDriverError, "These options are not w3c compliant: #{options}" unless options.empty?
  browser_options = {self.class::KEY => browser_options} if defined?(self.class::KEY)
  process_browser_options(browser_options)
  generate_as_json(w3c_options.merge(browser_options))
end

def camel_case(str)
  str.gsub(/_([a-z])/) { Regexp.last_match(1)&.upcase }
end

def camelize?(_key)
  true
end

def chrome(**)
  Chrome::Options.new(**)
end

def convert_json_key(key, camelize: true)
  key = key.to_s if key.is_a?(Symbol)
  key = camel_case(key) if camelize
  return key if key.is_a?(String)
  raise TypeError, "expected String or Symbol, got #{key.inspect}:#{key.class}"
end

def edge(**)
  Edge::Options.new(**)
end

def firefox(**)
  Firefox::Options.new(**)
end

def generate_as_json(value, camelize_keys: true)
  if value.is_a?(Hash)
    process_json_hash(value, camelize_keys)
  elsif value.respond_to?(:as_json)
    value.as_json
  elsif value.is_a?(Array)
    value.map { |val| generate_as_json(val, camelize_keys: camelize_keys) }
  elsif value.is_a?(Symbol)
    value.to_s
  else
    value
  end
end

def ie(**)
  IE::Options.new(**)
end

def initialize(**opts)
  self.class.set_capabilities
  @options = opts
  @options[:browser_name] = self.class::BROWSER
end

def process_browser_options(_browser_options)
  nil
end

def process_json_hash(value, camelize_keys)
  value.each_with_object({}) do |(key, val), hash|
    next if val.respond_to?(:empty?) && val.empty?
    camelize = camelize_keys ? camelize?(key) : false
    key = convert_json_key(key, camelize: camelize)
    hash[key] = generate_as_json(val, camelize_keys: camelize)
  end
end

def process_w3c_options(options)
  w3c_options = options.select { |key, val| w3c?(key) && !val.nil? }
  w3c_options[:unhandled_prompt_behavior] &&= w3c_options[:unhandled_prompt_behavior]&.to_s&.tr('_', ' ')
  options.delete_if { |key, _val| w3c?(key) }
  w3c_options
end

def safari(**)
  Safari::Options.new(**)
end

def set_capabilities
  (W3C_OPTIONS + GRID_OPTIONS + self::CAPABILITIES.keys).each do |key|
    next if method_defined? key
    define_method key do
      @options[key]
    end
    define_method :"#{key}=" do |value|
      @options[key] = value
    end
  end
end

def w3c?(key)
  W3C_OPTIONS.include?(key) || key.to_s.include?(':')
end