module Pagy::HeadersExtra

def pagy_headers(pagy)
  headers = pagy.vars[:headers]
  pagy_link_header(pagy).tap do |hash|
    hash[headers[:page]]  = pagy.page.to_s if pagy.page && headers[:page]
    hash[headers[:limit]] = pagy.limit.to_s \
        if headers[:limit] && !(defined?(::Pagy::Calendar) && pagy.is_a?(Calendar::Unit))
    return hash if (defined?(::Pagy::Countless) && pagy.is_a?(Countless)) || \
                   (defined?(::Pagy::Keyset) && pagy.is_a?(Keyset))
    hash[headers[:pages]] = pagy.last.to_s if headers[:pages]
    hash[headers[:count]] = pagy.count.to_s if pagy.count && headers[:count] # count may be nil with Calendar
  end
end