2016-12-07 12:15:01 +00:00
|
|
|
module API
|
|
|
|
module Helpers
|
|
|
|
module Pagination
|
|
|
|
def paginate(relation)
|
2018-04-25 13:19:55 +00:00
|
|
|
DefaultPaginationStrategy.new(self).paginate(relation)
|
2016-12-07 12:15:01 +00:00
|
|
|
end
|
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
class DefaultPaginationStrategy
|
|
|
|
attr_reader :ctx
|
|
|
|
delegate :params, :header, :request, to: :ctx
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
def initialize(ctx)
|
|
|
|
@ctx = ctx
|
|
|
|
end
|
2018-02-06 16:53:42 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
def paginate(relation)
|
|
|
|
relation = add_default_order(relation)
|
2018-02-06 16:53:42 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
relation.page(params[:page]).per(params[:per_page]).tap do |data|
|
|
|
|
add_pagination_headers(data)
|
|
|
|
end
|
|
|
|
end
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
private
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
def add_default_order(relation)
|
|
|
|
if relation.is_a?(ActiveRecord::Relation) && relation.order_values.empty?
|
|
|
|
relation = relation.order(:id)
|
|
|
|
end
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
relation
|
|
|
|
end
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
def add_pagination_headers(paginated_data)
|
|
|
|
header 'X-Per-Page', paginated_data.limit_value.to_s
|
|
|
|
header 'X-Page', paginated_data.current_page.to_s
|
|
|
|
header 'X-Next-Page', paginated_data.next_page.to_s
|
|
|
|
header 'X-Prev-Page', paginated_data.prev_page.to_s
|
|
|
|
header 'Link', pagination_links(paginated_data)
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
return if data_without_counts?(paginated_data)
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
header 'X-Total', paginated_data.total_count.to_s
|
|
|
|
header 'X-Total-Pages', total_pages(paginated_data).to_s
|
2018-02-06 16:53:42 +00:00
|
|
|
end
|
2016-12-07 12:15:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
def pagination_links(paginated_data)
|
|
|
|
request_url = request.url.split('?').first
|
|
|
|
request_params = params.clone
|
|
|
|
request_params[:per_page] = paginated_data.limit_value
|
2017-08-17 15:56:01 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
links = []
|
2017-12-04 09:49:53 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
request_params[:page] = paginated_data.prev_page
|
|
|
|
links << %(<#{request_url}?#{request_params.to_query}>; rel="prev") if request_params[:page]
|
|
|
|
|
|
|
|
request_params[:page] = paginated_data.next_page
|
|
|
|
links << %(<#{request_url}?#{request_params.to_query}>; rel="next") if request_params[:page]
|
|
|
|
|
|
|
|
request_params[:page] = 1
|
|
|
|
links << %(<#{request_url}?#{request_params.to_query}>; rel="first")
|
|
|
|
|
|
|
|
unless data_without_counts?(paginated_data)
|
|
|
|
request_params[:page] = total_pages(paginated_data)
|
|
|
|
links << %(<#{request_url}?#{request_params.to_query}>; rel="last")
|
|
|
|
end
|
|
|
|
|
|
|
|
links.join(', ')
|
2017-12-04 09:49:53 +00:00
|
|
|
end
|
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
def total_pages(paginated_data)
|
|
|
|
# Ensure there is in total at least 1 page
|
|
|
|
[paginated_data.total_pages, 1].max
|
|
|
|
end
|
2018-02-06 16:53:42 +00:00
|
|
|
|
2018-04-25 13:19:55 +00:00
|
|
|
def data_without_counts?(paginated_data)
|
|
|
|
paginated_data.is_a?(Kaminari::PaginatableWithoutCount)
|
|
|
|
end
|
2018-02-06 16:53:42 +00:00
|
|
|
end
|
2016-12-07 12:15:01 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|