1 | #! /usr/bin/perl -w |
---|
2 | |
---|
3 | use strict; |
---|
4 | use warnings; |
---|
5 | use utf8; |
---|
6 | |
---|
7 | ## IMPORTANT ## |
---|
8 | # When Net::Twitter::Lite encounters a Twitter API error or a network error, |
---|
9 | # it throws a Net::Twitter::Lite::Error object. |
---|
10 | # You can catch and process these exceptions by using eval blocks and testing $@ |
---|
11 | ## from http://search.cpan.org/perldoc?Net::Twitter::Lite#ERROR_HANDLING |
---|
12 | use Net::Twitter::Lite; |
---|
13 | use FindBin qw($Bin); |
---|
14 | use YAML::Tiny; |
---|
15 | |
---|
16 | sub VERBOSE () { $ARGV[0] eq 'verbose' }; |
---|
17 | sub DEBUG () { VERBOSE or $ARGV[0] eq 'debug' }; |
---|
18 | use Data::Dumper; |
---|
19 | |
---|
20 | DEBUG and warn "$0: debug mode"; |
---|
21 | |
---|
22 | my $conf = loadconf("$Bin/config.yml"); |
---|
23 | if (! defined $conf) { |
---|
24 | die "$0: cannot parse config file.\n"; |
---|
25 | } |
---|
26 | my $stat = loadconf("$Bin/status.yml"); |
---|
27 | if (! defined $stat) { |
---|
28 | $stat = {}; |
---|
29 | } |
---|
30 | |
---|
31 | my $bot = login($conf); |
---|
32 | if (! $bot->authorized) { |
---|
33 | die "$0: this client is not yet authorized.\n"; |
---|
34 | } |
---|
35 | |
---|
36 | my $tweets = {}; |
---|
37 | my $tweet; |
---|
38 | |
---|
39 | $tweet = or_search($bot, $conf->{hashtag}, $stat->{search}); |
---|
40 | if ($tweet) { |
---|
41 | %$tweets = (%$tweets, %$tweet); |
---|
42 | } |
---|
43 | |
---|
44 | $tweet = mentions_ids($bot, $stat->{mention}); |
---|
45 | if ($tweet) { |
---|
46 | %$tweets = (%$tweets, %$tweet); |
---|
47 | } |
---|
48 | |
---|
49 | foreach my $id (sort keys %$tweets) { |
---|
50 | # $tweets->{$id}{type} eq 'search' => found by search API |
---|
51 | # eq 'mention' => found by mention API |
---|
52 | if ($tweets->{$id}{type} eq 'retweet') { |
---|
53 | next; |
---|
54 | } |
---|
55 | DEBUG or sleep($conf->{sleep}); |
---|
56 | |
---|
57 | # do retweet found tweets |
---|
58 | my $res; |
---|
59 | eval { |
---|
60 | DEBUG or $res = $bot->retweet($id); |
---|
61 | DEBUG and warn "retweet($id) => ", Dumper($tweets->{$id}); |
---|
62 | }; |
---|
63 | if ($@) { |
---|
64 | evalrescue($@); |
---|
65 | warn "status_id => $id\n"; |
---|
66 | next; |
---|
67 | } |
---|
68 | |
---|
69 | $stat->{$tweets->{$id}{type}} = $id; |
---|
70 | } |
---|
71 | |
---|
72 | if ($tweets) { |
---|
73 | # save last status to yaml file |
---|
74 | DEBUG or YAML::Tiny::DumpFile("$Bin/status.yml", $stat); |
---|
75 | DEBUG and warn "status.yml => ", Dumper($stat); |
---|
76 | } |
---|
77 | |
---|
78 | |
---|
79 | sub loadconf { |
---|
80 | # load configration data from yaml formatted file |
---|
81 | # param => scalar string of filename |
---|
82 | # ret => hash object of yaml data |
---|
83 | |
---|
84 | my $file = shift @_; |
---|
85 | |
---|
86 | my $yaml = YAML::Tiny->read($file); |
---|
87 | |
---|
88 | if ($!) { |
---|
89 | warn "$0: '$file' $!\n"; |
---|
90 | } |
---|
91 | |
---|
92 | return $yaml->[0]; |
---|
93 | } |
---|
94 | |
---|
95 | sub login { |
---|
96 | # make Net::Twitter::Lite object and login |
---|
97 | # param => hash object of configration |
---|
98 | # ret => Net::Twitter::Lite object |
---|
99 | |
---|
100 | my $conf = shift @_; |
---|
101 | |
---|
102 | my $bot = Net::Twitter::Lite->new( |
---|
103 | consumer_key => $conf->{consumer_key}, |
---|
104 | consumer_secret => $conf->{consumer_secret}, |
---|
105 | ); |
---|
106 | |
---|
107 | $bot->access_token($conf->{access_token}); |
---|
108 | $bot->access_token_secret($conf->{access_token_secret}); |
---|
109 | |
---|
110 | return $bot; |
---|
111 | } |
---|
112 | |
---|
113 | sub or_search { |
---|
114 | # search tweets containing keywords |
---|
115 | # param => Net::Twitter::Lite object, ArrayRef of keywords, since_id |
---|
116 | # ret => HashRef of status_id (timeline order is destroyed) |
---|
117 | # or undef (none is found) |
---|
118 | |
---|
119 | my $bot = shift @_; |
---|
120 | my $keywords = shift @_; |
---|
121 | my $since_id = shift @_ || 1; |
---|
122 | |
---|
123 | my $key = ""; |
---|
124 | foreach my $word (@$keywords) { |
---|
125 | if ($key) { |
---|
126 | $key .= " OR $word"; |
---|
127 | } |
---|
128 | else { |
---|
129 | $key = $word; |
---|
130 | } |
---|
131 | } |
---|
132 | DEBUG and warn "searching '$key'"; |
---|
133 | |
---|
134 | my $res; |
---|
135 | my $ids = {}; |
---|
136 | eval { |
---|
137 | if ($key) { |
---|
138 | $res = $bot->search( |
---|
139 | { |
---|
140 | q => $key, |
---|
141 | since_id => $since_id, |
---|
142 | } |
---|
143 | ); |
---|
144 | } |
---|
145 | if ($res->{results}) { |
---|
146 | VERBOSE and warn Dumper($res); |
---|
147 | foreach my $tweet (@{$res->{results}}) { |
---|
148 | my $res = $bot->show_status($tweet->{id}); |
---|
149 | if ($res->{retweeted_status}) { |
---|
150 | $ids->{$tweet->{id}}{type} = 'retweet'; |
---|
151 | } |
---|
152 | else { |
---|
153 | $ids->{$tweet->{id}}{type} = 'search'; |
---|
154 | } |
---|
155 | VERBOSE and warn Dumper($res); |
---|
156 | } |
---|
157 | } |
---|
158 | }; |
---|
159 | if ($@) { |
---|
160 | evalrescue($@); |
---|
161 | } |
---|
162 | |
---|
163 | DEBUG and warn "search result => ", Dumper($ids); |
---|
164 | return $ids; |
---|
165 | } |
---|
166 | |
---|
167 | sub mentions_ids { |
---|
168 | # return status_ids mentioned to me |
---|
169 | # param => Net::Twitter::Lite object, since_id |
---|
170 | # ret => HashRef of status_id (timeline order is destroyed) |
---|
171 | # or undef (none is found) |
---|
172 | |
---|
173 | my $bot = shift @_; |
---|
174 | my $since_id = shift @_ || 1; |
---|
175 | |
---|
176 | my $res; |
---|
177 | eval { |
---|
178 | $res = $bot->mentions( |
---|
179 | { |
---|
180 | since_id => $since_id, |
---|
181 | } |
---|
182 | ); |
---|
183 | VERBOSE and warn Dumper($res); |
---|
184 | }; |
---|
185 | if ($@) { |
---|
186 | evalrescue($@); |
---|
187 | } |
---|
188 | |
---|
189 | my $ids = {}; |
---|
190 | if ($res && @{$res}) { |
---|
191 | $ids = { |
---|
192 | map { |
---|
193 | $_->{id} => { |
---|
194 | type => 'mention', |
---|
195 | } |
---|
196 | } @{$res} |
---|
197 | }; |
---|
198 | } |
---|
199 | |
---|
200 | DEBUG and warn "mentions result => ", Dumper($ids); |
---|
201 | return $ids; |
---|
202 | } |
---|
203 | |
---|
204 | sub evalrescue { |
---|
205 | # output error message at eval error |
---|
206 | |
---|
207 | use Scalar::Util qw(blessed); |
---|
208 | |
---|
209 | if (blessed $@ && $@->isa('Net::Twitter::Lite::Error')) { |
---|
210 | warn $@->error; |
---|
211 | if ($@->twitter_error) { |
---|
212 | my %twitter_error = %{$@->twitter_error}; |
---|
213 | map { |
---|
214 | $twitter_error{"$_ => "} = $twitter_error{$_} . "\n"; |
---|
215 | delete $twitter_error{$_} |
---|
216 | } keys %twitter_error; |
---|
217 | warn join("", %twitter_error); |
---|
218 | } |
---|
219 | } |
---|
220 | else { |
---|
221 | warn $@; |
---|
222 | } |
---|
223 | } |
---|