Mercurial > rsstweet
annotate app/models/tweet.rb @ 172:df4be896ab8b
Improved escaping and truncating
| author | nanaya <me@nanaya.pro> |
|---|---|
| date | Wed, 08 Aug 2018 20:37:29 +0900 |
| parents | 5af9b537db86 |
| children | 08cc4a4d8a5f |
| rev | line source |
|---|---|
| 15 | 1 class Tweet |
| 137 | 2 TIMELINE_OPTIONS = { |
| 3 :count => 100, | |
| 4 :exclude_replies => false, | |
| 5 :include_rts => true, | |
| 6 :tweet_mode => :extended, | |
| 7 } | |
| 8 | |
|
158
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
9 def self.cache_expires_time |
|
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
10 (15 + rand(15)).minutes |
|
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
11 end |
|
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
12 |
| 135 | 13 def initialize(twitter_id) |
| 14 @clients = {} | |
| 165 | 15 @twitter_id = twitter_id |
| 135 | 16 end |
| 17 | |
|
158
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
18 def id |
|
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
19 user.id |
| 152 | 20 end |
| 21 | |
|
86
5bfc986200db
The caching becomes a bit confusing because of parameters
nanaya <me@myconan.net>
parents:
78
diff
changeset
|
22 def timeline |
| 145 | 23 if @timeline.nil? |
|
158
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
24 cache_key = "timeline:v2:#{id}/#{Base64.urlsafe_encode64 id.to_s}" |
|
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
25 raw = Rails.cache.fetch(cache_key, :expires_in => self.class.cache_expires_time) do |
|
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
26 client_try(:user_timeline, id, TIMELINE_OPTIONS).tap do |data| |
| 145 | 27 if data[:result] == :ok |
|
158
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
28 if data[:data].any? && data[:data].first.user.id != id |
|
146
7ca8aeba1a63
Try tracing where the failures happen
nanaya <me@nanaya.pro>
parents:
145
diff
changeset
|
29 wrong_user = data[:data].first.user |
| 160 | 30 throw "Wrong timeline data. Requested: #{id}, got: #{wrong_user.id} (#{wrong_user.screen_name.printable})" |
|
146
7ca8aeba1a63
Try tracing where the failures happen
nanaya <me@nanaya.pro>
parents:
145
diff
changeset
|
31 end |
|
7ca8aeba1a63
Try tracing where the failures happen
nanaya <me@nanaya.pro>
parents:
145
diff
changeset
|
32 |
| 145 | 33 data[:data] = data[:data].select do |tweet| |
| 34 tweet.retweeted_status.nil? || tweet.user.id != tweet.retweeted_status.user.id | |
| 35 end.map { |tweet| tweet.to_h } | |
|
134
3646b3e319c7
Only try/catch the tweet downloading part
nanaya <me@nanaya.pro>
parents:
108
diff
changeset
|
36 end |
|
76
0c023d35cd80
Allows usage of multiple twitter keys
nanaya <me@myconan.net>
parents:
73
diff
changeset
|
37 end |
| 145 | 38 end |
| 141 | 39 |
| 145 | 40 raise Twitter::Error::NotFound if raw[:result] == :not_found |
| 144 | 41 |
| 145 | 42 @timeline = raw[:data].map { |tweet_hash| Twitter::Tweet.new(tweet_hash) } |
| 43 end | |
| 44 | |
| 45 @timeline | |
| 15 | 46 end |
| 47 | |
|
48
8983c426e256
Prevent exploding on empty timeline.
nanaya <me@myconan.net>
parents:
47
diff
changeset
|
48 def user |
| 145 | 49 if @user.nil? |
| 165 | 50 cache_key = "user:v1:#{@twitter_id.is_a?(Integer) ? 'id' : 'lookup'}:#{@twitter_id}" |
| 51 raw = Rails.cache.fetch(cache_key, :expires_in => self.class.cache_expires_time) do | |
| 160 | 52 client_try(:user, @twitter_id).tap do |data| |
| 53 if data[:result] == :ok | |
| 54 user = data[:data] | |
| 55 | |
| 165 | 56 if user.id != @twitter_id && user.screen_name.downcase != @twitter_id |
| 160 | 57 throw "Wrong user data. Requested: #{@twitter_id}, got: #{user.id} (#{user.screen_name.printable})" |
| 58 end | |
| 59 end | |
| 60 end | |
|
48
8983c426e256
Prevent exploding on empty timeline.
nanaya <me@myconan.net>
parents:
47
diff
changeset
|
61 end |
| 145 | 62 |
| 63 raise Twitter::Error::NotFound if raw[:result] == :not_found | |
| 64 | |
| 65 @user = raw[:data] | |
| 66 end | |
| 67 | |
| 68 @user | |
|
48
8983c426e256
Prevent exploding on empty timeline.
nanaya <me@myconan.net>
parents:
47
diff
changeset
|
69 end |
|
8983c426e256
Prevent exploding on empty timeline.
nanaya <me@myconan.net>
parents:
47
diff
changeset
|
70 |
| 135 | 71 def client |
| 72 @clients[client_config_id] ||= | |
| 73 Twitter::REST::Client.new do |config| | |
| 74 $cfg[:twitter][client_config_id].each do |cfg_key, cfg_value| | |
| 138 | 75 config.public_send(:"#{cfg_key}=", cfg_value) |
| 135 | 76 end |
| 77 end | |
|
76
0c023d35cd80
Allows usage of multiple twitter keys
nanaya <me@myconan.net>
parents:
73
diff
changeset
|
78 end |
|
0c023d35cd80
Allows usage of multiple twitter keys
nanaya <me@myconan.net>
parents:
73
diff
changeset
|
79 |
| 145 | 80 def client_try(method, *args) |
| 81 initial_config_id = client_config_id | |
| 82 | |
| 83 begin | |
| 84 data = client.public_send method, *args | |
| 85 rescue Twitter::Error::TooManyRequests | |
| 86 @client_config_id += 1 | |
| 87 | |
| 88 if initial_config_id == client_config_id | |
| 89 raise | |
| 90 else | |
| 91 retry | |
| 92 end | |
| 93 rescue Twitter::Error::NotFound | |
| 94 return { :result => :not_found } | |
| 95 end | |
| 96 | |
| 97 { :result => :ok, :data => data } | |
| 98 end | |
| 99 | |
| 135 | 100 def client_config_id |
|
158
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
101 @client_config_count ||= $cfg[:twitter].size |
|
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
102 @client_config_id ||= rand(@client_config_count) |
|
76
0c023d35cd80
Allows usage of multiple twitter keys
nanaya <me@myconan.net>
parents:
73
diff
changeset
|
103 |
|
158
74422bae017d
Always use canonical id and turn cache time generator a class method
nanaya <me@nanaya.pro>
parents:
157
diff
changeset
|
104 @client_config_id %= @client_config_count |
| 15 | 105 end |
| 106 end |