Merge pull request #52 from cowboyrushforth/twitter

adding twitter support

lib/lolcommits.rb

@@ -22,6 +22,7 @@
 require 'lolcommits/plugins/dot_com'
 require 'lolcommits/plugins/tranzlate'
 require 'lolcommits/plugins/statsd'
+require 'lolcommits/plugins/lol_twitter'
 
 # require runner after all the plugins have been required
 require 'lolcommits/runner'

lib/lolcommits/configuration.rb

@@ -71,7 +71,7 @@ def puts_plugins
       puts "Available plugins: #{names.join(', ')}"
     end
 
-    def do_configure!(plugin)
+    def do_configure!(plugin, forced_options=nil)
       if plugin.nil? || plugin.strip == ''
         puts_plugins
         print "Name of plugin to configure: "
@@ -90,14 +90,19 @@ def do_configure!(plugin)
         return
       end
 
-      options = plugin_object.options.inject(Hash.new) do |acc, option|
-        print "#{option}: "
-        val = STDIN.gets.strip
-        val = true  if val == 'true'
-        val = false if val == 'false'
+      if forced_options.nil?
+        options = plugin_object.options.inject(Hash.new) do |acc, option|
+          print "#{option}: "
+          val = STDIN.gets.strip
+          val = true  if val == 'true'
+          val = false if val == 'false'
 
-        acc.merge(option => val)
+          acc.merge(option => val)
+        end
+      else
+        options = forced_options
       end
+
       config = self.user_configuration || Hash.new
       config[plugin] = options
       File.open(self.user_configuration_file, 'w') do |f| 

lib/lolcommits/plugins/lol_twitter.rb

@@ -0,0 +1,99 @@
+require 'yaml'
+require 'twitter'
+require 'oauth'
+
+TWITTER_CONSUMER_KEY = 'qc096dJJCxIiqDNUqEsqQ'
+TWITTER_CONSUMER_SECRET = 'rvjNdtwSr1H0TvBvjpk6c4bvrNydHmmbvv7gXZQI'
+
+module Lolcommits
+
+  class LolTwitter < Plugin
+
+    def initialize(runner)
+      super
+      self.name    = 'twitter'
+      self.default = false
+    end
+
+    def initial_twitter_auth
+      puts "\n--------------------------------------------"
+      puts "Need to grab twitter tokens (first time only)"
+      puts "---------------------------------------------"
+
+      consumer = OAuth::Consumer.new(TWITTER_CONSUMER_KEY, 
+                                     TWITTER_CONSUMER_SECRET,
+                                     :site => 'http://api.twitter.com',
+                                     :request_endpoint => 'http://api.twitter.com',
+                                     :sign_in => true)
+
+      request_token = consumer.get_request_token
+      rtoken  = request_token.token
+      rsecret = request_token.secret
+
+      puts "\n1.) Open the following url in your browser, get the PIN:\n\n"
+      puts request_token.authorize_url
+      puts "\n2.) Enter PIN, then press enter:"
+
+      begin
+        STDOUT.flush
+        twitter_pin = STDIN.gets.chomp
+      rescue
+      end
+
+      if (twitter_pin.nil?) || (twitter_pin.length == 0)
+        puts "\n\tERROR: Could not read PIN, auth fail"
+        return
+      end
+
+      begin
+        OAuth::RequestToken.new(consumer, rtoken, rsecret)
+        access_token = request_token.get_access_token(:oauth_verifier => twitter_pin)
+      rescue Twitter::Unauthorized
+        puts "> FAIL!"
+        return
+      end
+
+      # saves the config back to yaml file.
+      self.runner.config.do_configure!('twitter', { 'enabled'      => true,
+                                                    'access_token' => access_token.token,
+                                                    'secret'       => access_token.secret })
+    end
+
+    def run
+      commit_msg = self.runner.message
+      available_commit_msg_size = 128 
+      tweet_msg = commit_msg.length > available_commit_msg_size ? "#{commit_msg[0..(available_commit_msg_size-3)]}..." : commit_msg
+      tweet_text = "#{tweet_msg} #lolcommits"
+      puts "Tweeting: #{tweet_text}"
+
+      if configuration['access_token'].nil? || configuration['secret'].nil?
+        initial_twitter_auth()
+      end
+
+      if configuration['access_token'].nil? || configuration['secret'].nil?
+        puts "Missing Twitter Credentials - Skipping The Tweet"
+        return
+      end
+
+      Twitter.configure do |config|
+        config.consumer_key = TWITTER_CONSUMER_KEY
+        config.consumer_secret = TWITTER_CONSUMER_SECRET
+      end
+
+      client = Twitter::Client.new(
+        :oauth_token => configuration['access_token'],
+        :oauth_token_secret => configuration['secret']
+      )
+      retries = 2
+      begin
+        if client.update_with_media(tweet_text, File.open(self.runner.main_image, 'r'))
+          puts "\t--> Tweet Sent!"
+        end
+      rescue Twitter::Error::InternalServerError
+        retries -= 1
+        retry if retries > 0
+        puts "\t ! --> Tweet 500 Error - Tweet Not Posted"
+      end
+    end
+  end
+end

lib/lolcommits/runner.rb

@@ -12,6 +12,7 @@ class Runner
 
     # Executed Last
     set_callback :run, :after,  :cleanup!
+    set_callback :run, :after,  :execute_lolcommits_lol_twitter
     set_callback :run, :after,  :execute_lolcommits_stats_d
     set_callback :run, :after,  :execute_lolcommits_dot_com
     set_callback :run, :after,  :execute_lolcommits_loltext

lolcommits.gemspec

@@ -22,6 +22,8 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency('git', '~> 1.2.5')
   s.add_runtime_dependency('choice', '~> 0.1.6')
   s.add_runtime_dependency('launchy', '~> 2.1.1')
+  s.add_runtime_dependency("twitter")
+  s.add_runtime_dependency("oauth")
 
   s.add_development_dependency('rdoc')
   s.add_development_dependency('aruba')