Module: T::Collectable

Included in:
CLI, List, Search
Defined in:
lib/t/collectable.rb

Constant Summary collapse

MAX_NUM_RESULTS =
200
MAX_PAGE =
51

Instance Method Summary collapse

Instance Method Details

#collect_with_count(count) ⇒ Object



19
20
21
22
23
24
25
26
27
28
29
30
31
# File 'lib/t/collectable.rb', line 19

def collect_with_count(count)
  opts = {}
  opts[:count] = MAX_NUM_RESULTS
  collect_with_max_id do |max_id|
    opts[:max_id] = max_id unless max_id.nil?
    opts[:count] = count unless count >= MAX_NUM_RESULTS
    if count.positive?
      tweets = yield opts
      count -= tweets.length
      tweets
    end
  end.flatten.compact
end

#collect_with_max_id(collection = [], max_id = nil) ⇒ Object



9
10
11
12
13
14
15
16
17
# File 'lib/t/collectable.rb', line 9

def collect_with_max_id(collection = [], max_id = nil, &)
  tweets = Retryable.retryable(tries: 3, on: Twitter::Error, sleep: 0) do
    yield(max_id)
  end
  return collection if tweets.nil?

  collection += tweets
  tweets.empty? ? collection.flatten : collect_with_max_id(collection, tweets.last.id - 1, &)
end

#collect_with_page(collection = ::Set.new, page = 1, previous = nil) ⇒ Object



33
34
35
36
37
38
39
40
41
# File 'lib/t/collectable.rb', line 33

def collect_with_page(collection = ::Set.new, page = 1, previous = nil, &)
  tweets = Retryable.retryable(tries: 3, on: Twitter::Error, sleep: 0) do
    yield page
  end
  return collection if tweets.nil? || tweets == previous || page >= MAX_PAGE

  collection += tweets
  tweets.empty? ? collection.flatten : collect_with_page(collection, page + 1, tweets, &)
end