snapgit 0.6.4

data/lib/lolcommits/plugins/lol_slack.rb

@@ -0,0 +1,68 @@
+# -*- encoding : utf-8 -*-
+require 'rest_client'
+
+module Lolcommits
+  class LolSlack < Plugin
+    ENDPOINT_URL = 'https://slack.com/api/files.upload'.freeze
+    RETRY_COUNT  = 2
+
+    def self.name
+      'slack'
+    end
+
+    def self.runner_order
+      :postcapture
+    end
+
+    def configured?
+      !configuration['access_token'].nil?
+    end
+
+    def configure
+      print "Open the URL below and issue a token for your user:\n"
+      print "https://api.slack.com/web\n"
+      print "Enter the generated token below, then press enter: (e.g. xxxx-xxxxxxxxx-xxxx) \n"
+      code = STDIN.gets.to_s.strip
+      print "Enter a comma-seperated list of channel IDs to post images in, then press enter: (e.g. C1234567890,C1234567890)\n"
+      print "NOTE: you must use channel IDs (not channel names). Grab them from here; https://api.slack.com/methods/channels.list/test\n"
+      channels = STDIN.gets.to_s.strip
+
+      { 'access_token' => code,
+        'channels' => channels }
+    end
+
+    def configure_options!
+      options = super
+      if options['enabled']
+        config = configure
+        options.merge!(config)
+      end
+      options
+    end
+
+    def run_postcapture
+      return unless valid_configuration?
+
+      retries = RETRY_COUNT
+      begin
+
+        response = RestClient.post(
+          ENDPOINT_URL,
+          :file     => File.new(runner.main_image),
+          :token    => configuration['access_token'],
+          :filetype => 'jpg',
+          :filename => runner.sha,
+          :title    => runner.message + "[#{runner.git_info.repo}]",
+          :channels => configuration['channels'])
+
+        debug response
+      rescue => e
+        retries -= 1
+        retry if retries > 0
+        puts "Posting to slack failed - #{e.message}"
+        puts 'Try running config again:'
+        puts "\tlolcommits --config --plugin slack"
+      end
+    end
+  end
+end

data/lib/lolcommits/plugins/lol_tumblr.rb

@@ -0,0 +1,127 @@
+# -*- encoding : utf-8 -*-
+require 'yaml'
+require 'oauth'
+require 'webrick'
+require 'cgi'
+require 'tumblr_client'
+
+module Lolcommits
+  class LolTumblr < Plugin
+    TUMBLR_API_ENDPOINT    = 'https://www.tumblr.com'.freeze
+    TUMBLR_CONSUMER_KEY    = '2FtMEDpEPkxjoUdkpHh42h9wqTu9IVS7Ra0QyNZGixdCvhllN2'.freeze
+    TUMBLR_CONSUMER_SECRET = 'qWuvxgFUR2YyWKtbWOkDTMAiBEbj7ZGaNLaNQPba0PI1N4JpBs'.freeze
+
+    def run_postcapture
+      return unless valid_configuration?
+      puts 'Posting to Tumblr'
+      r = client.photo(configuration['tumblr_name'], :data => runner.main_image)
+      if r.key?('id')
+        puts "\t--> Post successful!"
+      else
+        puts "Tumblr post FAILED! #{r}"
+      end
+    rescue Faraday::Error => e
+      puts "Tumblr post FAILED! #{e.message}"
+    end
+
+    def configure_options!
+      options = super
+      # ask user to configure tokens if enabling
+      if options['enabled']
+        auth_config = configure_auth!
+        return unless auth_config
+        options = options.merge(auth_config).merge(configure_tumblr_name)
+      end
+      options
+    end
+
+    def configure_auth!
+      puts '---------------------------'
+      puts 'Need to grab tumblr tokens'
+      puts '---------------------------'
+
+      request_token = oauth_consumer.get_request_token(:exclude_callback => true)
+      print "\n1) Please open this url in your browser to authorize lolcommits:\n\n"
+      puts request_token.authorize_url
+      print "\n2) Launching a local server to complete the OAuth authentication process:\n\n"
+      begin
+        server = WEBrick::HTTPServer.new :Port => 3000
+        server.mount_proc '/', server_callback(server)
+        server.start
+        debug "Requesting Tumblr OAuth Token with verifier: #{@verifier}"
+        access_token = request_token.get_access_token(:oauth_verifier => @verifier)
+      rescue Errno::EADDRINUSE
+        puts "\nERROR You have something running on port 3000. Please turn it off to complete the authorization process"
+        return
+      rescue OAuth::Unauthorized
+        puts "\nERROR: Tumblr OAuth verification faile!"
+        return
+      end
+      return unless access_token.token && access_token.secret
+      puts ''
+      puts '------------------------------'
+      puts 'Thanks! Tumblr Auth Succeeded'
+      puts '------------------------------'
+
+      {
+        'access_token' => access_token.token,
+        'secret'       => access_token.secret
+      }
+    end
+
+    def configure_tumblr_name
+      print "\n3) What's your tumblr name? (i.e. 'http://[THIS PART HERE].tumblr.com'): "
+      { 'tumblr_name' => STDIN.gets.strip }
+    end
+
+    def configured?
+      !configuration['enabled'].nil? &&
+        configuration['access_token'] &&
+        configuration['secret']
+    end
+
+    def client
+      @client ||= Tumblr.new(:consumer_key       => TUMBLR_CONSUMER_KEY,
+                             :consumer_secret    => TUMBLR_CONSUMER_SECRET,
+                             :oauth_token        => configuration['access_token'],
+                             :oauth_token_secret => configuration['secret']
+                            )
+    end
+
+    def oauth_consumer
+      @oauth_consumer ||= OAuth::Consumer.new(TUMBLR_CONSUMER_KEY,
+                                              TUMBLR_CONSUMER_SECRET,
+                                              :site             => TUMBLR_API_ENDPOINT,
+                                              :request_endpoint => TUMBLR_API_ENDPOINT,
+                                              :http_methdo => :get
+                                             )
+    end
+
+    def config_with_default(key, default = nil)
+      if configuration[key]
+        configuration[key].strip.empty? ? default : configuration[key]
+      else
+        default
+      end
+    end
+
+    def self.name
+      'tumblr'
+    end
+
+    def self.runner_order
+      :postcapture
+    end
+
+    protected
+
+    def server_callback(server)
+      proc do |req, res|
+        q = CGI.parse req.request_uri.query
+        @verifier = q['oauth_verifier'][0]
+        server.stop
+        res.body = 'Lolcommits authorization complete!'
+      end
+    end
+  end
+end

data/lib/lolcommits/plugins/lol_twitter.rb

@@ -0,0 +1,155 @@
+# -*- encoding : utf-8 -*-
+require 'yaml'
+require 'oauth'
+require 'twitter'
+
+module Lolcommits
+  class LolTwitter < Plugin
+    TWITTER_API_ENDPOINT    = 'https://api.twitter.com'.freeze
+    TWITTER_CONSUMER_KEY    = 'qc096dJJCxIiqDNUqEsqQ'.freeze
+    TWITTER_CONSUMER_SECRET = 'rvjNdtwSr1H0TvBvjpk6c4bvrNydHmmbvv7gXZQI'.freeze
+    TWITTER_RETRIES         = 2
+    TWITTER_PIN_REGEX       = /^\d{4,}$/ # 4 or more digits
+    DEFAULT_SUFFIX          = '#lolcommits'.freeze
+
+    def run_postcapture
+      return unless valid_configuration?
+      tweet = build_tweet(runner.message)
+
+      attempts = 0
+      begin
+        attempts += 1
+        puts "Tweeting: #{tweet}"
+        debug "--> Tweeting! (attempt: #{attempts}, tweet length: #{tweet.length} chars)"
+        if client.update_with_media(tweet, File.open(runner.main_image, 'r'))
+          puts "\t--> Tweet Sent!"
+        end
+      rescue Twitter::Error::ServerError,
+             Twitter::Error::ClientError => e
+        debug "Tweet FAILED! #{e.class} - #{e.message}"
+        retry if attempts < TWITTER_RETRIES
+        puts "ERROR: Tweet FAILED! (after #{attempts} attempts) - #{e.message}"
+      end
+    end
+
+    def build_tweet(commit_message)
+      prefix = config_with_default('prefix', '')
+      suffix = " #{config_with_default('suffix', DEFAULT_SUFFIX)}"
+      prefix = "#{prefix} " unless prefix.empty?
+
+      available_commit_msg_size = max_tweet_size - (prefix.length + suffix.length)
+      if commit_message.length > available_commit_msg_size
+        commit_message = "#{commit_message[0..(available_commit_msg_size - 3)]}..."
+      end
+
+      "#{prefix}#{commit_message}#{suffix}"
+    end
+
+    def configure_options!
+      options = super
+      # ask user to configure tokens if enabling
+      if options['enabled']
+        auth_config = configure_auth!
+        return unless auth_config
+        options = options.merge(auth_config).merge(configure_prefix_suffix)
+      end
+      options
+    end
+
+    def configure_auth!
+      puts '---------------------------'
+      puts 'Need to grab twitter tokens'
+      puts '---------------------------'
+
+      request_token = oauth_consumer.get_request_token
+      rtoken        = request_token.token
+      rsecret       = request_token.secret
+
+      print "\n1) Please open this url in your browser to get a PIN for lolcommits:\n\n"
+      puts request_token.authorize_url
+      print "\n2) Enter PIN, then press enter: "
+      twitter_pin = STDIN.gets.strip.downcase.to_s
+
+      unless twitter_pin =~ TWITTER_PIN_REGEX
+        puts "\nERROR: '#{twitter_pin}' is not a valid Twitter Auth PIN"
+        return
+      end
+
+      begin
+        debug "Requesting Twitter OAuth Token with PIN: #{twitter_pin}"
+        OAuth::RequestToken.new(oauth_consumer, rtoken, rsecret)
+        access_token = request_token.get_access_token(:oauth_verifier => twitter_pin)
+      rescue OAuth::Unauthorized
+        puts "\nERROR: Twitter PIN Auth FAILED!"
+        return
+      end
+
+      return unless access_token.token && access_token.secret
+      puts ''
+      puts '------------------------------'
+      puts 'Thanks! Twitter Auth Succeeded'
+      puts '------------------------------'
+      {
+        'access_token' => access_token.token,
+        'secret'       => access_token.secret
+      }
+    end
+
+    def configure_prefix_suffix
+      print "\n3) Prefix all tweets with something? e.g. @user (leave blank for no prefix): "
+      prefix = STDIN.gets.strip
+      print "\n4) End all tweets with something? e.g. #hashtag (leave blank for default suffix #{DEFAULT_SUFFIX}): "
+      suffix = STDIN.gets.strip
+
+      config = {}
+      config['prefix'] = prefix unless prefix.empty?
+      config['suffix'] = suffix unless suffix.empty?
+      config
+    end
+
+    def configured?
+      !configuration['enabled'].nil? &&
+        configuration['access_token'] &&
+        configuration['secret']
+    end
+
+    def client
+      @client ||= Twitter::REST::Client.new do |config|
+        config.consumer_key        = TWITTER_CONSUMER_KEY
+        config.consumer_secret     = TWITTER_CONSUMER_SECRET
+        config.access_token        = configuration['access_token']
+        config.access_token_secret = configuration['secret']
+      end
+    end
+
+    def oauth_consumer
+      @oauth_consumer ||= OAuth::Consumer.new(
+        TWITTER_CONSUMER_KEY,
+        TWITTER_CONSUMER_SECRET,
+        :site             => TWITTER_API_ENDPOINT,
+        :request_endpoint => TWITTER_API_ENDPOINT,
+        :sign_in          => true
+      )
+    end
+
+    def config_with_default(key, default = nil)
+      if configuration[key]
+        configuration[key].strip.empty? ? default : configuration[key]
+      else
+        default
+      end
+    end
+
+    def max_tweet_size
+      139 - client.configuration.characters_reserved_per_media
+    end
+
+    def self.name
+      'twitter'
+    end
+
+    def self.runner_order
+      :postcapture
+    end
+  end
+end

data/lib/lolcommits/plugins/lol_yammer.rb

@@ -0,0 +1,85 @@
+# -*- encoding : utf-8 -*-
+require 'yammer'
+require 'rest_client'
+
+# https://developer.yammer.com/oauth2-quickstart/
+YAMMER_CLIENT_ID        = 'bgORyeKtnjZJSMwp8oln9g'.freeze
+YAMMER_CLIENT_SECRET    = 'oer2WdGzh74a5QBbW3INUxblHK3yg9KvCZmiBa2r0'.freeze
+YAMMER_ACCESS_TOKEN_URL = 'https://www.yammer.com/oauth2/access_token.json'.freeze
+YAMMER_RETRY_COUNT      = 2
+
+module Lolcommits
+  class LolYammer < Plugin
+    def self.name
+      'yammer'
+    end
+
+    def self.runner_order
+      :postcapture
+    end
+
+    def configured?
+      !configuration['access_token'].nil?
+    end
+
+    def configure_access_token
+      print "Open the URL below and copy the `code` param from query after redirected, enter it as `access_token`:\n"
+      print "https://www.yammer.com/dialog/oauth?client_id=#{YAMMER_CLIENT_ID}&response_type=code\n"
+      print 'Enter code param from the redirected URL, then press enter: '
+      code = STDIN.gets.to_s.strip
+
+      url = YAMMER_ACCESS_TOKEN_URL
+      debug "access_token url: #{url}"
+      params = {
+        'client_id' => YAMMER_CLIENT_ID,
+        'client_secret' => YAMMER_CLIENT_SECRET,
+        'code' => code
+      }
+      debug "params : #{params.inspect}"
+      result = JSON.parse(RestClient.post(url, params))
+      debug "response : #{result.inspect}"
+      # no need for 'return', last line is always the return value
+      { 'access_token' => result['access_token']['token'] }
+    end
+
+    def configure_options!
+      options = super
+      # ask user to configure tokens if enabling
+      if options['enabled']
+        auth_config = configure_access_token
+        return unless auth_config
+        options.merge!(auth_config)
+      end
+      options
+    end
+
+    def run_postcapture
+      return unless valid_configuration?
+
+      commit_msg = runner.message
+      post = "#{commit_msg} #lolcommits"
+      puts "Yammer post: #{post}" unless runner.capture_stealth
+
+      Yammer.configure do |c|
+        c.client_id = YAMMER_CLIENT_ID
+        c.client_secret = YAMMER_CLIENT_SECRET
+      end
+
+      client = Yammer::Client.new(:access_token => configuration['access_token'])
+
+      retries = YAMMER_RETRY_COUNT
+      begin
+        lolimage = File.new(runner.main_image)
+        response = client.create_message(post, :attachment1 => lolimage)
+        debug response.body.inspect
+        puts "\t--> Status posted!" if response
+      rescue => e
+        retries -= 1
+        retry if retries > 0
+        puts "Status not posted - #{e.message}"
+        puts 'Try running config again:'
+        puts "\tlolcommits --config --plugin yammer"
+      end
+    end
+  end
+end

data/lib/lolcommits/plugins/lolsrv.rb

@@ -0,0 +1,59 @@
+# -*- encoding : utf-8 -*-
+require 'rest_client'
+require 'pp'
+require 'json'
+
+module Lolcommits
+  class Lolsrv < Plugin
+    def initialize(runner)
+      super
+      options << 'server'
+    end
+
+    def run_postcapture
+      return unless valid_configuration?
+      fork { sync }
+    end
+
+    def configured?
+      !configuration['enabled'].nil? && configuration['server']
+    end
+
+    def sync
+      existing = existing_lols
+      return unless existing.nil?
+      Dir[runner.config.loldir + '/*.{jpg,gif}'].each do |item|
+        sha = File.basename(item, '.*')
+        upload(item, sha) unless existing.include?(sha) || sha == 'tmp_snapshot'
+      end
+    end
+
+    def existing_lols
+      lols = JSON.parse(
+        RestClient.get(configuration['server'] + '/lols'))
+      lols.map { |lol| lol['sha'] }
+    rescue => e
+      log_error(e, "ERROR: existing lols could not be retrieved #{e.class} - #{e.message}")
+      return nil
+    end
+
+    def upload(file, sha)
+      RestClient.post(configuration['server'] + '/uplol',
+                      :lol  => File.new(file),
+                      :url  => runner.git_info.url + sha,
+                      :repo => runner.git_info.repo,
+                      :date => File.ctime(file),
+                      :sha  => sha)
+    rescue => e
+      log_error(e, "ERROR: Upload of lol #{sha} FAILED #{e.class} - #{e.message}")
+    end
+
+    def self.name
+      'lolsrv'
+    end
+
+    def self.runner_order
+      :postcapture
+    end
+  end
+end