class Capybara::Result

def [](*args)
  idx, length = args
  max_idx = case idx
  when Integer
    if !idx.negative?
      length.nil? ? idx : idx + length - 1
    else
      nil
    end
  when Range
    idx.max
  end
  if max_idx.nil?
    full_results[*args]
  else
    load_up_to(max_idx + 1)
    @result_cache[*args]
  end
end

def compare_count
  # Only check filters for as many elements as necessary to determine result
  if (count = @query.options[:count])
    count = Integer(count)
    return load_up_to(count + 1) <=> count
  end
  if (min = @query.options[:minimum])
    min = Integer(min)
    return -1 if load_up_to(min) < min
  end
  if (max = @query.options[:maximum])
    max = Integer(max)
    return 1 if load_up_to(max + 1) > max
  end
  if (between = @query.options[:between])
    min, max = between.minmax
    size = load_up_to(max + 1)
    return 0 if between.include? size
    return size <=> min
  end
  0
end

def each(&block)
  return enum_for(:each) unless block_given?
  @result_cache.each(&block)
  loop do
    next_result = @results_enum.next
    @result_cache << next_result
    yield next_result
  end
  self
end

def empty?
  !any?
end

def failure_message
  message = @query.failure_message
  if count.zero?
    message << ' but there were no matches'
  else
    message << ", found #{count} #{Capybara::Helpers.declension('match', 'matches', count)}: " << full_results.map(&:text).map(&:inspect).join(', ')
  end
  unless rest.empty?
    elements = rest.map { |el| el.text rescue '<<ERROR>>' }.map(&:inspect).join(', ') # rubocop:disable Style/RescueModifier
    message << '. Also found ' << elements << ', which matched the selector but not all filters.'
  end
  message
end

def full_results
  loop { @result_cache << @results_enum.next }
  @result_cache
end

def initialize(elements, query)
  @elements = elements
  @result_cache = []
  @results_enum = lazy_select_elements { |node| query.matches_filters?(node) }
  @query = query
end

def lazy_select_elements(&block)
  # JRuby has an issue with lazy enumerators which
  # causes a concurrency issue with network requests here
  # https://github.com/jruby/jruby/issues/4212
  if RUBY_PLATFORM == 'java'
    @elements.select(&block).to_enum # non-lazy evaluation
  else
    @elements.lazy.select(&block)
  end
end

def load_up_to(num)
  loop do
    break if @result_cache.size >= num
    @result_cache << @results_enum.next
  end
  @result_cache.size
end

def matches_count?
  compare_count.zero?
end

def negative_failure_message
  failure_message.sub(/(to find)/, 'not \1')
end

def rest
  @rest ||= @elements - full_results
end

def unfiltered_size
  @elements.length
end