From a518b9d475ca8223c89cb036ee3841bab35e916b Mon Sep 17 00:00:00 2001 From: Matthias Hecker Date: Thu, 14 Nov 2013 15:51:26 +0000 Subject: [PATCH] twitter: filter ret.latest status if non specified --- data/rbot/plugins/twitter.rb | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/data/rbot/plugins/twitter.rb b/data/rbot/plugins/twitter.rb index 991b5bd3..3ed8e62c 100644 --- a/data/rbot/plugins/twitter.rb +++ b/data/rbot/plugins/twitter.rb @@ -38,14 +38,23 @@ class TwitterPlugin < Plugin :default => 3, :validate => Proc.new { |v| v > 0 && v <= 10}, :desc => "Maximum number of status updates shown by 'twitter [home|mentions|retweets] status'") + URL_PATTERN = %r{twitter\.com/([^/]+)(?:/status/(\d+))?} + def twitter_filter(s) - loc = Utils.check_location(s, Regexp.new('twitter\.com/(#!/)?.*/status/\d+')) + loc = Utils.check_location(s, URL_PATTERN) return nil unless loc - id = loc.first.match(/\/status\/(\d+)/)[1] - - response = @app_access_token.get('/1.1/statuses/show/'+id+'.json').body + matches = loc.first.match URL_PATTERN + if matches[2] # status id matched + id = matches[2] + url = '/1.1/statuses/show/%s.json' % id + else # no status id, get the latest status of that user + user = matches[1] + url = '/1.1/statuses/user_timeline.json?screen_name=%s&count=1&include_rts=true' % user + end + response = @app_access_token.get(url).body begin tweet = JSON.parse(response) + tweet = tweet.first if tweet.instance_of? Array status = { :date => (Time.parse(tweet["created_at"]) rescue ""), :id => (tweet["id_str"] rescue ""), -- 2.39.2