summaryrefslogtreecommitdiff
path: root/data
diff options
context:
space:
mode:
Diffstat (limited to 'data')
-rw-r--r--data/rbot/plugins/twitter.rb154
1 files changed, 99 insertions, 55 deletions
diff --git a/data/rbot/plugins/twitter.rb b/data/rbot/plugins/twitter.rb
index ec5e3b92..df834b0c 100644
--- a/data/rbot/plugins/twitter.rb
+++ b/data/rbot/plugins/twitter.rb
@@ -5,6 +5,7 @@
#
# Author:: Carter Parks (carterparks) <carter@carterparks.com>
# Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
+# Author:: "NeoLobster" <neolobster@snugglenets.com>
#
# Copyright:: (C) 2007 Carter Parks
# Copyright:: (C) 2007 Giuseppe Bilotta
@@ -12,16 +13,26 @@
# Users can setup their twitter username and password and then begin updating
# twitter whenever
+require 'oauth'
+require 'yaml'
require 'rexml/rexml'
-require 'cgi'
class TwitterPlugin < Plugin
- Config.register Config::IntegerValue.new('twitter.status_count',
- :default => 1, :validate => Proc.new { |v| v > 0 && v <= 10},
- :desc => "Maximum number of status updates shown by 'twitter status'")
- Config.register Config::IntegerValue.new('twitter.friends_status_count',
- :default => 3, :validate => Proc.new { |v| v > 0 && v <= 10},
- :desc => "Maximum number of status updates shown by 'twitter friends status'")
+ Config.register Config::StringValue.new('twitter.key',
+ :default => "",
+ :desc => "Twitter OAuth Consumer Key")
+
+ Config.register Config::StringValue.new('twitter.secret',
+ :default => "",
+ :desc => "Twitter OAuth Consumer Secret")
+
+ Config.register Config::IntegerValue.new('twitter.status_count',
+ :default => 1, :validate => Proc.new { |v| v > 0 && v <= 10},
+ :desc => "Maximum number of status updates shown by 'twitter status'")
+
+ Config.register Config::IntegerValue.new('twitter.friends_status_count',
+ :default => 3, :validate => Proc.new { |v| v > 0 && v <= 10},
+ :desc => "Maximum number of status updates shown by 'twitter friends status'")
def initialize
super
@@ -34,47 +45,48 @@ class TwitterPlugin < Plugin
val
end
end
-
- @header = {
- 'X-Twitter-Client' => 'rbot twitter plugin'
- }
end
- # return a help string when the bot is asked for help on this plugin
def help(plugin, topic="")
- return "twitter status [nick] => show nick's (or your) status, use 'twitter friends status [nick]' to also show the friends' timeline | twitter update [status] => updates your status on twitter | twitter identify [username] [password] => ties your nick to your twitter username and password | twitter actions [on|off] => enable/disable twitting of actions (/me does ...)"
+ return "twitter status [nick] => show nick's (or your) status, use 'twitter friends status [nick]' to also show the friends' timeline | twitter update [status] => updates your status on twitter | twitter authorize => Generates an authorization URL which will give you a PIN to authorize the bot to use your twitter account. | twitter pin [pin] => Finishes bot authorization using the PIN provided by the URL from twitter authorize. | twitter deauthorize => Makes the bot forget your Twitter account. | twitter actions [on|off] => enable/disable twitting of actions (/me does ...)"
end
# update the status on twitter
def get_status(m, params)
-
- nick = params[:nick] || @registry[m.sourcenick + "_username"]
-
friends = params[:friends]
+ if @registry.has_key?(m.sourcenick + "_access_token")
+ @access_token = YAML::load(@registry[m.sourcenick + "_access_token"])
+ nick = params[:nick] || @access_token.params[:screen_name]
+ else
+ if friends
+ m.reply "You are not authorized with Twitter. Please use 'twitter authorize' first to use this feature."
+ return false
+ end
+ nick = params[:nick]
+ end
if not nick
- m.reply "you should specify the username of the twitter touse, or identify using 'twitter identify [username] [password]'"
+ m.reply "you should specify the username of the twitter to use, or identify using 'twitter authorize'"
return false
end
+ count = @bot.config['twitter.friends_status_count']
user = URI.escape(nick)
-
- count = @bot.config['twitter.status_count']
- unless friends
- uri = "http://twitter.com/statuses/user_timeline/#{user}.xml?count=#{count}"
+ if @registry.has_key?(m.sourcenick + "_access_token")
+ if friends
+ #no change to count variable
+ uri = "https://api.twitter.com/1/statuses/friends_timeline.xml?count=#{count}"
+ response = @access_token.get(uri).body
+ else
+ count = @bot.config['twitter.status_count']
+ uri = "https://api.twitter.com/1/statuses/user_timeline.xml?screen_name=#{user}&count=#{count}"
+ response = @access_token.get(uri).body
+ end
else
- count = @bot.config['twitter.friends_status_count']
- auth = ""
- if m.private?
- auth << URI.escape(@registry[m.sourcenick + "_username"])
- auth << ":"
- auth << URI.escape(@registry[m.sourcenick + "_password"])
- auth << "@"
- end
- uri = "http://#{auth}twitter.com/statuses/friends_timeline/#{user}.xml"
+ #unauthorized user, will try to get from public timeline the old way
+ uri = "http://twitter.com/statuses/user_timeline/#{user}.xml?count=#{count}"
+ response = @bot.httputil.get(uri, :cache => false)
end
-
- response = @bot.httputil.get(uri, :headers => @header, :cache => false)
debug response
texts = []
@@ -124,35 +136,66 @@ class TwitterPlugin < Plugin
end
end
- # update the status on twitter
- def update_status(m, params)
-
+ def deauthorize(m, params)
+ if @registry.has_key?(m.sourcenick + "_request_token")
+ @registry.delete(m.sourcenick + "_request_token")
+ end
+ if @registry.has_key?(m.sourcenick + "_access_token")
+ @registry.delete(m.sourcenick + "_access_token")
+ end
+ m.reply "Done! You can reauthorize this account in the future by using 'twitter authorize'"
+ end
- unless @registry.has_key?(m.sourcenick + "_password") && @registry.has_key?(m.sourcenick + "_username")
- m.reply "you must identify using 'twitter identify [username] [password]'"
- return false
+ def authorize(m, params)
+ #remove all old authorization data
+ if @registry.has_key?(m.sourcenick + "_request_token")
+ @registry.delete(m.sourcenick + "_request_token")
+ end
+ if @registry.has_key?(m.sourcenick + "_access_token")
+ @registry.delete(m.sourcenick + "_access_token")
end
- user = URI.escape(@registry[m.sourcenick + "_username"])
- pass = URI.escape(@registry[m.sourcenick + "_password"])
- uri = "http://#{user}:#{pass}@twitter.com/statuses/update.xml"
+ key = @bot.config['twitter.key']
+ secret = @bot.config['twitter.secret']
+ @consumer = OAuth::Consumer.new(key, secret, {
+ :site => "https://api.twitter.com",
+ :request_token_path => "/oauth/request_token",
+ :access_token_path => "/oauth/access_token",
+ :authorize_path => "/oauth/authorize"
+ } )
+ @request_token = @consumer.get_request_token
+ @registry[m.sourcenick + "_request_token"] = YAML::dump(@request_token)
+ m.reply "Go to this URL to get your authorization PIN, then use 'twitter pin <pin>' to finish authorization: " + @request_token.authorize_url
+ end
- msg = params[:status].to_s
+ def pin(m, params)
+ unless @registry.has_key?(m.sourcenick + "_request_token")
+ m.reply "You must first use twitter authorize to get the PIN"
+ return false
+ end
+ @request_token = YAML::load(@registry[m.sourcenick + "_request_token"])
+ @access_token = @request_token.get_access_token( { :oauth_verifier => params[:pin] } )
+ @registry[m.sourcenick + "_access_token"] = YAML::dump(@access_token)
+ m.reply "Okay, you're all set"
+ end
- if msg.length > 160
- m.reply "your status message update is too long, please keep it under 140 characters if possible, 160 characters maximum"
- return
+ # update the status on twitter
+ def update_status(m, params)
+ unless @registry.has_key?(m.sourcenick + "_access_token")
+ m.reply "You must first authorize your Twitter account before tweeting."
+ return false;
end
+ @access_token = YAML::load(@registry[m.sourcenick + "_access_token"])
+
+ uri = "https://api.twitter.com/statuses/update.json"
+ msg = params[:status].to_s
if msg.length > 140
- m.reply "your status message is longer than 140 characters, which is not optimal, but I'm going to update anyway"
+ m.reply "your status message update is too long, please keep it under 140 characters"
+ return
end
- source = "source=rbot"
- msg = "status=#{CGI.escape(msg)}"
- body = [source,msg].join("&")
-
- response = @bot.httputil.post(uri, body, :headers => @header)
+ response = @access_token.post(uri, { :status => msg })
debug response
reply_method = params[:notify] ? :notify : :reply
@@ -167,7 +210,7 @@ class TwitterPlugin < Plugin
def identify(m, params)
@registry[m.sourcenick + "_username"] = params[:username].to_s
@registry[m.sourcenick + "_password"] = params[:password].to_s
- m.reply "you're all setup!"
+ m.reply "you're all set up!"
end
# update on ACTION if the user has enabled the option
@@ -198,9 +241,10 @@ end
# create an instance of our plugin class and register for the "length" command
plugin = TwitterPlugin.new
-plugin.map 'twitter identify :username :password', :action => "identify", :public => false
plugin.map 'twitter update *status', :action => "update_status", :threaded => true
-plugin.map 'twitter status [:nick]', :action => "get_status", :threaded => true
+plugin.map 'twitter authorize', :action => "authorize", :public => false
+plugin.map 'twitter deauthorize', :action => "deauthorize", :public => false
+plugin.map 'twitter pin :pin', :action => "pin", :public => false
plugin.map 'twitter actions [:toggle]', :action => "actions", :requirements => { :toggle => /^on|off$/ }
+plugin.map 'twitter status [:nick]', :action => "get_status", :threaded => true
plugin.map 'twitter :friends [status] [:nick]', :action => "get_status", :requirements => { :friends => /^friends?$/ }, :threaded => true
-
generated by cgit v1.2.3 (git 2.39.1) at 2025-01-09 23:53:55 +0000