module Capybara::Selenium::Driver::W3CFirefoxDriver

def browser_version
  browser.capabilities[:browser_version].to_f
end

def build_node(native_node, initial_cache = {})
  ::Capybara::Selenium::FirefoxNode.new(self, native_node, initial_cache)
end

def extended(driver)
  require 'capybara/selenium/patches/pause_duration_fix' if pause_broken?(driver.browser)
  driver.options[:native_displayed] = false if driver.options[:native_displayed].nil?
end

def pause_broken?(sel_driver)
  sel_driver.capabilities['moz:geckodriverVersion']&.start_with?('0.22.')
end

def refresh
  # Accept any "will repost content" confirmation that occurs
  accept_modal :confirm, wait: 0.1 do
    super
  end
rescue Capybara::ModalNotFound
  # No modal was opened - page has refreshed - ignore
end

def reset!
  # Use instance variable directly so we avoid starting the browser just to reset the session
  return unless @browser
  if browser_version >= 68
    begin
      # Firefox 68 hangs if we try to switch windows while a modal is visible
      browser.switch_to.alert&.dismiss
    rescue Selenium::WebDriver::Error::NoSuchAlertError
      # Swallow
    end
  end
  switch_to_window(window_handles.first)
  window_handles.slice(1..).each { |win| close_window(win) }
  super
end

def resize_window_to(handle, width, height)
  within_given_window(handle) do
    # Don't set the size if already set - See https://github.com/mozilla/geckodriver/issues/643
    if window_size(handle) == [width, height]
      {}
    else
      super
    end
  end
end

def switch_to_frame(frame)
  return super unless frame == :parent
  # geckodriver/firefox has an issue if the current frame is removed from within it
  # so we have to move to the default_content and iterate back through the frames
  handles = @frame_handles[current_window_handle]
  browser.switch_to.default_content
  handles.tap(&:pop).each { |fh| browser.switch_to.frame(fh.native) }
end