module Pagy::ElasticsearchRailsExtra::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)
  vars[:count]     = ElasticsearchRailsExtra.total_count(response)
  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::OverflowExtra) && pagy.overflow? && pagy.vars[:overflow] == :last_page
  [pagy, called.empty? ? response : response.send(*called)]
end

def pagy_elasticsearch_rails_get_vars(_collection, vars)
  pagy_set_items_from_params(vars) if defined?(ItemsExtra)
  vars[:items] ||= DEFAULT[:items]
  vars[:page]  ||= (params[vars[:page_param] || DEFAULT[:page_param]] || 1).to_i
  vars
end