module Pagy::Backend

def pagy_elasticsearch_rails(pagy_search_args, vars={})
  model, query_or_payload, options, *called = pagy_search_args
  vars           = pagy_elasticsearch_rails_get_vars(nil, vars)
  options[:size] = vars[:items]
  options[:from] = vars[:items] * (vars[:page] - 1)
  response       = model.search(query_or_payload, **options)
  total          = response.respond_to?(:raw_response) ? response.raw_response['hits']['total'] : response.response['hits']['total']
  vars[:count]   = total.is_a?(Hash) ? total['value'] : total
  pagy = Pagy.new(vars)
  # with :last_page overflow we need to re-run the method in order to get the hits
  return pagy_elasticsearch_rails(pagy_search_args, vars.merge(page: pagy.page)) \
         if defined?(Pagy::UseOverflowExtra) && pagy.overflow? && pagy.vars[:overflow] == :last_page
  [ pagy, called.empty? ? response : response.send(*called) ]
end