module Pagy::HeadersExtra

def pagy_headers(pagy)
  pagy_headers_hash(pagy).tap do |hash|
    hash['Link'] = hash['Link'].map { |rel, link| %(<#{link}>; rel="#{rel}") }.join(', ')
  end
end

def pagy_headers_hash(pagy)
  countless             = defined?(Pagy::Countless) && pagy.is_a?(Pagy::Countless)
  rel                   = { 'first' => 1, 'prev' => pagy.prev, 'next' => pagy.next }
  rel['last']           = pagy.last unless countless
  url_str               = pagy_url_for(pagy, PAGE_PLACEHOLDER, absolute: true)
  link                  = rel.map do |r, num| # filter_map if ruby >=2.7
                            next unless num # rubocop:disable Layout/EmptyLineAfterGuardClause
                            [r, url_str.sub(PAGE_PLACEHOLDER, num.to_s)]
                          end.compact.to_h
  hash                  = { 'Link' => link }
  headers               = pagy.vars[:headers]
  hash[headers[:page]]  = pagy.page.to_s         if headers[:page]
  hash[headers[:items]] = pagy.vars[:items].to_s if headers[:items]
  unless countless
    hash[headers[:pages]] = pagy.pages.to_s if headers[:pages]
    hash[headers[:count]] = pagy.count.to_s if headers[:count]
  end
  hash
end

def pagy_headers_merge(pagy)
  response.headers.merge!(pagy_headers(pagy))
end