summaryrefslogtreecommitdiff
path: root/lib/github/collection.rb
blob: 014b2038c4b53bcac100ec59e54918ade8d0e127 (plain)
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
module Github
  class Collection
    attr_reader :options

    def initialize(options)
      @options = options
    end

    def fetch(url, query = {})
      return [] if url.blank?

      Enumerator.new do |yielder|
        loop do
          response = client.get(url, query)
          response.body.each { |item| yielder << item }

          raise StopIteration unless response.rels.key?(:next)
          url = response.rels[:next]
        end
      end.lazy
    end

    private

    def client
      @client ||= Github::Client.new(options)
    end
  end
end