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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# frozen_string_literal: true
module Gitlab
# Wrapper class of paginated response.
class PaginatedResponse
attr_accessor :client
def initialize(array)
@array = array
end
def ==(other)
@array == other
end
def inspect
@array.inspect
end
def method_missing(name, *args, &block)
if @array.respond_to?(name)
@array.send(name, *args, &block)
else
super
end
end
def respond_to_missing?(method_name, include_private = false)
super || @array.respond_to?(method_name, include_private)
end
def parse_headers!(headers)
@links = Headers::PageLinks.new headers
@total = Headers::Total.new headers
end
def each_page
current = self
yield current
while current.has_next_page?
current = current.next_page
yield current
end
end
def lazy_paginate
to_enum(:each_page).lazy.flat_map(&:to_ary)
end
def auto_paginate(&block)
return lazy_paginate.to_a unless block
lazy_paginate.each(&block)
end
def paginate_with_limit(limit, &block)
return lazy_paginate.take(limit).to_a unless block
lazy_paginate.take(limit).each(&block)
end
def total
@total.total
end
def last_page?
!(@links.nil? || @links.last.nil?)
end
alias has_last_page? last_page?
def last_page
return nil if @client.nil? || !has_last_page?
@client.get(client_relative_path(@links.last))
end
def first_page?
!(@links.nil? || @links.first.nil?)
end
alias has_first_page? first_page?
def first_page
return nil if @client.nil? || !has_first_page?
@client.get(client_relative_path(@links.first))
end
def next_page?
!(@links.nil? || @links.next.nil?)
end
alias has_next_page? next_page?
def next_page
return nil if @client.nil? || !has_next_page?
@client.get(client_relative_path(@links.next))
end
def prev_page?
!(@links.nil? || @links.prev.nil?)
end
alias has_prev_page? prev_page?
def prev_page
return nil if @client.nil? || !has_prev_page?
@client.get(client_relative_path(@links.prev))
end
def client_relative_path(link)
client_endpoint_path = URI.parse(@client.endpoint).request_uri # api/v4
URI.parse(link).request_uri.sub(client_endpoint_path, '')
end
end
end
|
