forked from kaminari/kaminari
-
Notifications
You must be signed in to change notification settings - Fork 5
/
array_extension.rb
46 lines (39 loc) · 1.43 KB
/
array_extension.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
module Kaminari
# Kind of Array that can paginate
class PaginatableArray < Array
include Kaminari::ConfigurationMethods::ClassMethods
attr_internal_accessor :limit_value, :offset_value
# ==== Options
# * <tt>:limit</tt> - limit
# * <tt>:offset</tt> - offset
# * <tt>:total_count</tt> - total_count
def initialize(original_array, options = {})
@_original_array, @_limit_value, @_offset_value, @_total_count = original_array, (options[:limit] || default_per_page).to_i, options[:offset].to_i, options[:total_count]
super(original_array[@_offset_value, @_limit_value] || [])
end
# items at the specified "page"
def page(num = 1)
offset(limit_value * ([num.to_i, 1].max - 1))
end
# returns another chunk of the original array
def limit(num)
self.class.new @_original_array, :limit => num, :offset => @_offset_value, :total_count => @_total_count
end
# total item numbers of the original array
def total_count
@_total_count || @_original_array.count
end
# returns another chunk of the original array
def offset(num)
arr = self.class.new @_original_array, :limit => @_limit_value, :offset => num, :total_count => @_total_count
class << arr
include Kaminari::PageScopeMethods
end
arr
end
end
# Wrap an Array object to make it paginatable
def self.paginate_array(array)
PaginatableArray.new array
end
end