forked from jmccartie/retweetbot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
retweetbot.rb
110 lines (87 loc) · 3.11 KB
/
retweetbot.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
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
#!/usr/bin/env ruby
require 'rubygems'
require 'bundler/setup'
Bundler.require(:default)
require 'logger'
module RetweetBot
LOGGER = Logger.new(STDERR)
LOGGER.level = Logger::DEBUG
JSON_PARSER = Yajl::Parser.new(:symbolize_keys => true)
class App
CONFIG = {
twitter_username: ENV['TWITTER_USERNAME'],
twitter_password: ENV['TWITTER_PASSWORD'],
consumer_key: ENV['TWITTER_CONSUMER_KEY'],
consumer_secret: ENV['TWITTER_CONSUMER_SECRET'],
access_token: ENV['TWITTER_ACCESS_TOKEN'],
access_token_secret: ENV['TWITTER_ACCESS_TOKEN_SECRET'],
filter: ENV['TWITTER_FILTER'] ? ENV['TWITTER_FILTER'].split(",") : "",
follow: ENV['TWITTER_FOLLOW'] ? ENV['TWITTER_FOLLOW'].split(",") : ""
}
def initialize
raise "Twitter oAuth not authorized." unless twitter_client.authorized?
end
def start
welcome = "Listening to Twitter stream for #{CONFIG[:filter].join(', ')}."
welcome << " (but only from #{CONFIG[:follow].count} users)" unless CONFIG[:follow].nil?
LOGGER.info welcome
twitter_stream.each_item do |item|
JSON_PARSER.parse(item) do |status|
if status.has_key?(:text) and status[:user][:screen_name] != CONFIG[:twitter_username] and CONFIG[:follow].include?(status[:user][:id].to_s)
LOGGER.info "@#{status[:user][:name]}: #{status[:text]}"
result = twitter_client.retweet(status[:id])
LOGGER.warn "ERROR: #{result[:errors]}" if result.has_key?(:errors)
end
end
end
twitter_stream.on_error do |message|
LOGGER.fatal "Error: #{message}"
EventMachine.stop
end
twitter_stream.on_reconnect do |timeout, retries|
LOGGER.warn "Reconnecting in: #{timeout} seconds."
end
twitter_stream.on_max_reconnects do |timeout, retries|
LOGGER.fatal "Failed after #{retries} failed reconnects."
EventMachine.stop
end
end
def stop
twitter_stream.stop
end
private
def twitter_stream
@twitter_stream ||= Twitter::JSONStream.connect(
:method => 'POST',
# :ssl => true, # Would be needed by OAuth
# :oauth => twitter_stream_oauth, # Sadly, OAuth does not work on streaming API
:auth => CONFIG[:twitter_username] + ":" + CONFIG[:twitter_password],
:filters => CONFIG[:filter]
)
end
def twitter_client
@twitter_client ||= TwitterOAuth::Client.new(
:consumer_key => CONFIG[:consumer_key],
:consumer_secret => CONFIG[:consumer_secret],
:token => CONFIG[:access_token],
:secret => CONFIG[:access_token_secret]
)
end
def twitter_stream_oauth
{
:consumer_key => CONFIG[:consumer_key],
:consumer_secret => CONFIG[:consumer_secret],
:access_key => CONFIG[:access_token],
:access_secret => CONFIG[:access_token_secret]
}
end
end # class RetweetBot
end # module RetweetBot
retweetbot = RetweetBot::App.new
EventMachine::run do
retweetbot.start
trap('TERM') do
retweetbot.stop
EventMachine.stop if EventMachine.reactor_running?
end
end