require ::File.expand_path("../util/configuration", File.dirname(__FILE__))
require ::File.expand_path("finder_diagnostics", File.dirname(__FILE__))

module Cornucopia
  module Capybara
    module MatcherExtensions
      def assert_selector(*args)
        __cornucopia_assert_selector_function(:assert_selector, *args)
      end

      def assert_no_selector(*args)
        __cornucopia_assert_selector_function(:assert_no_selector, *args)
      end

      def __cornucopia_assert_selector_function(assert_selector_function, *args)
        retry_count = 0
        result      = nil

        support_options = __cornucopia__extract_selector_support_options(*args)

        begin
          retry_count += 1
          result      = send("__cornucopia_orig_#{assert_selector_function}", *args)
        rescue Selenium::WebDriver::Error::StaleElementReferenceError
          retry if __cornucopia__retry_selector(retry_count, support_options)

          result = __cornucopia__analyze_selector(assert_selector_function, support_options, *args)
        rescue Exception
          result = __cornucopia__analyze_selector(assert_selector_function, support_options, *args)
        end

        result
      end

      private

      def __cornucopia__extract_selector_support_options(*args)
        support_options = {}

        if args[-1].is_a?(Hash)
          support_options[:__cornucopia_no_analysis]      = args[-1].delete(:__cornucopia_no_analysis)
          support_options[:__cornucopia_retry_with_found] = args[-1].delete(:__cornucopia_retry_with_found)
        end

        support_options
      end

      def __cornucopia__analyze_selector(function_name, support_options, *args)
        return_value = nil
        error        = $!

        if !support_options[:__cornucopia_no_analysis] &&
            (Cornucopia::Util::Configuration.analyze_selector_exceptions ||
                support_options[:__cornucopia_retry_with_found])
          find_action = Cornucopia::Capybara::FinderDiagnostics::FindAction.new(self,
                                                                                {},
                                                                                support_options,
                                                                                function_name,
                                                                                *args)

          if find_action.perform_analysis(Cornucopia::Util::Configuration.retry_match_with_found ||
                                              support_options[:__cornucopia_retry_with_found])
            return_value = find_action.simple_run rescue nil
            return_value ||= find_action.return_value
          else
            raise error
          end
        else
          raise error
        end

        return_value
      end

      def __cornucopia__retry_selector(retry_count, support_options)
        retry_count <= Cornucopia::Util::Configuration.selenium_cache_retry_count &&
            !support_options[:__cornucopia_no_analysis]
      end
    end
  end
end

load ::File.expand_path("install_matcher_extensions.rb", File.dirname(__FILE__))