module Pagy::Backend

def pagy_elasticsearch_rails(pagy_search_args, vars={})
  model, search_args, _block, *called = pagy_search_args
  vars                   = pagy_elasticsearch_rails_get_vars(nil, vars)
  search_args[-1][:size] = vars[:items]
  search_args[-1][:from] = vars[:items] * (vars[:page] - 1)
  response               = model.search(*search_args)
  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
  if defined?(OVERFLOW) && pagy.overflow? && pagy.vars[:overflow] == :last_page
    return pagy_elasticsearch_rails(pagy_search_args, vars.merge(page: pagy.page))
  end
  return pagy, called.empty? ? response : response.send(*called)
end