lita-tweet 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2cae612dab3855ddc2a3b67400a8be73545a5708
-  data.tar.gz: 6f813d3fcf0a68f4b530e72091403e4da718d047
+  metadata.gz: 2075bb6c559760c56f41340005b53999d1dfa134
+  data.tar.gz: dd51909e34b310662c31ae91b983a2937f7a2762
 SHA512:
-  metadata.gz: 7af5e3d30400abfe6a4caa4da0072cbcbf7498d57526ee18acae71f339cc810e51c9ef31c24bae6443712bfe4cec75f594d494bd7d8e8691c82c8ac00c9e3bdf
-  data.tar.gz: ba01ba95692904af032fcba3faaea7517aa1e2b24056c179ef4e06c747d3da7139ad2cfa21df1452470ec793480f1a776f1a20c218bbc40ee58424af85fa7063
+  metadata.gz: 838bd08b7db776304826b8211349c8d983c318fd95a5eb2e776b1fa096669c3481a0a6dd97cc6cb6da4ed37814c342b1e1802705c95365156b8e4aee432e227b
+  data.tar.gz: 87353ed0740d0b6673e24c2535b36e32608a16a4a6e5c084c24e4db00eb7e889e3c7a7e9f2a861a277f288d78506342ca8c6fa537f4a01cd4a07c662b3c98c80

data/lib/keyword_struct.rb

@@ -0,0 +1,5 @@
+class KeywordStruct < Struct
+  def initialize(**kwargs)
+    super(*members.map{|k| kwargs[k] })
+  end
+end

data/lib/lita/handlers/tweet.rb

@@ -1,7 +1,115 @@
+require_relative "./tweet/data"
+
 module Lita
   module Handlers
     class Tweet < Handler
-      
+      config :consumer_key, type: String
+      config :consumer_secret, type: String
+      config :http_url
+
+      route %r{^tweet\s(.+)}, :tweet, command: true, restrict_to: :tweeters,
+        help: {"tweet MESSAGE" => "Post a tweet."}
+      route %r{^untweet}, :untweet, command: true, restrict_to: :tweeters,
+        help: {"untweet" => "Delete the last tweet."}
+      route %r{^twitter accounts}, :accounts, command: true,
+        restrict_to: :tweeters, help: {
+          "twitter accounts" => "List accounts that can be tweeted from.",
+          "twitter accounts add" => "Authorize a new account for tweeting.",
+          "twitter accounts remove NAME" => "Remove the twitter account NAME"
+        }
+      # route %r{^twitter channels\s(.+?)\s(.+)}, :channels, command: true, help: {
+      #   "twitter channels" => "List account to channel mappings.",
+      #   "twitter channels NAME CHANNEL" => "Tweet as twitter account NAME when told to tweet in channel CHANNEL."
+      # }
+
+      TWITTER_AUTH_URL = "/twitter/auth"
+      TWITTER_AUTH_CALLBACK_URL = "/twitter/callback"
+      http.get TWITTER_AUTH_URL, :twitter_auth
+      http.get TWITTER_AUTH_CALLBACK_URL, :twitter_auth_callback
+
+      def tweet(response)
+        text = response.match_data[1]
+        if text.nil? || text.empty?
+          return response.reply("I need something to tweet!")
+        end
+
+        account = account_for(response)
+        return response.relpy(no_accounts) if account.nil?
+
+        tweet = account.tweet(text)
+        twitter_data.set_last_tweet(account.username, tweet.id)
+        response.reply(tweet.url)
+      end
+
+      def untweet(response)
+        account = account_for(response)
+        return response.relpy(no_accounts) if account.nil?
+
+        if account.untweet
+          response.reply("Removed last tweet.")
+        else
+          response.reply("Couldn't find a last tweet to remove!")
+        end
+      end
+
+      def accounts(response)
+        case response.args[1]
+        when "add"
+          response.reply(add_account)
+        when "remove"
+          response.reply(remove_account(response.args[2]))
+        else
+          response.reply(list_accounts)
+        end
+      end
+
+      # def channels(response)
+      #   # do channel stuff here
+      # end
+
+      def twitter_auth(request, response)
+        callback_url = TWITTER_AUTH_CALLBACK_URL
+        request_token = twitter_data.create_request_token(callback_url)
+        response.status = 302
+        response.headers["Location"] = request_token.authorize_url
+      end
+
+      def twitter_auth_callback(request, response)
+        token = request.params["oauth_token"]
+        verifier = request.params["oauth_verifier"]
+        account = twitter_data.authorize_account(token, verifier)
+        response.body << "Done! You can now tweet from @#{account.username}."
+      end
+
+    private
+
+      def list_accounts
+        names = twitter_data.usernames
+
+        if names.empty?
+          "No authorized accounts. Use `twitter accounts add` to add one."
+        else
+          usernames = names.map{|n| " - @#{n}" }.join("\n")
+          "Authorized Twitter accounts:\n" << usernames
+        end
+      end
+
+      def add_account
+        auth_uri = twitter_data.bot_uri(TWITTER_AUTH_URL)
+        "Authorize your account for tweeting here:\n#{auth_uri}"
+      end
+
+      def remove_account(name)
+        twitter_data.remove_account(name)
+        "Removed @#{name}."
+      end
+      def account_for(response)
+        twitter_data.account(twitter_data.usernames.first)
+      end
+
+      def twitter_data
+        @twitter_data ||= Data.new(redis, config, robot)
+      end
 
       Lita.register_handler(self)
     end

data/lib/lita/handlers/tweet/account.rb

@@ -0,0 +1,34 @@
+require "twitter"
+require "keyword_struct"
+
+module Lita
+  module Handlers
+    class Tweet < Handler
+
+      Account = KeywordStruct.new(:username, :token, :secret, :config, :last_tweet) do
+        def username
+          @username ||= client.user.screen_name
+        end
+
+        def tweet(text)
+          client.update(text)
+        end
+
+        def untweet
+          last_tweet ? client.destroy_status(last_tweet) : false
+        end
+
+      private
+
+        def client
+          @client ||= ::Twitter::REST::Client.new do |c|
+            c.consumer_key        = config.consumer_key
+            c.consumer_secret     = config.consumer_secret
+            c.access_token        = token
+            c.access_token_secret = secret
+          end
+        end
+      end
+    end
+  end
+end

data/lib/lita/handlers/tweet/data.rb

@@ -0,0 +1,94 @@
+require "oauth"
+require_relative "./account"
+
+module Lita
+  module Handlers
+    class Tweet < Handler
+      Data = Struct.new(:redis, :config, :robot) do
+        def usernames
+          redis.smembers("twitter_accounts")
+        end
+
+        def account(username)
+          hash = redis.hgetall("twitter_accounts:#{username}")
+          data = hash.each_with_object({}){|(k,v),h| h[k.to_sym] = v }
+          Account.new(**data.merge(config: config))
+        end
+
+        def add_account(token, secret)
+          account = Account.new(token: token, secret: secret, config: config)
+          username = account.username
+
+          redis.sadd("twitter_accounts", username)
+          redis.hmset("twitter_accounts:#{username}",
+            "username", username,
+            "token", token,
+            "secret", secret
+          )
+
+          account
+        end
+
+        def remove_account(username)
+          redis.del("twitter_accounts:#{username}")
+          redis.srem("twitter_accounts", username)
+        end
+
+        def set_last_tweet(username, tweet_id)
+          redis.hset("twitter_accounts:#{username}", "last_tweet", tweet_id)
+        end
+
+        def get_last_tweet(username)
+          redis.hget("twitter_accounts:#{username}", "last_tweet")
+        end
+
+        def create_request_token(callback_path)
+          request_token = consumer.get_request_token(
+            oauth_callback: bot_uri(callback_path).to_s)
+          request_hash = request_token.params
+
+          key = "request_token:#{request_hash[:oauth_token]}"
+          redis.hmset(key, *request_hash.to_a.flatten)
+          redis.expire(key, 120)
+
+          request_token
+        end
+
+        def authorize_account(token, verifier)
+          request_token = find_request_token(token)
+          access_token = request_token.get_access_token(oauth_verifier: verifier)
+          add_account(access_token.token, access_token.secret)
+        end
+
+        def bot_uri(path = "")
+          base_uri + path
+        end
+
+      private
+
+        def find_request_token(oauth_token)
+          params = redis.hgetall("request_token:#{oauth_token}")
+          params.keys.each{|k| params[k.to_sym] = params[k] }
+          OAuth::RequestToken.from_hash(consumer, params)
+        end
+
+        def base_uri
+          @base_url ||= begin
+            http = robot.config.http
+            URI(config.http_url || "http://#{http.host}:#{http.port}")
+          end
+        end
+
+        def consumer
+          @consumer ||= OAuth::Consumer.new(
+            config.consumer_key,
+            config.consumer_secret,
+            :site => 'https://api.twitter.com',
+            :authorize_path => '/oauth/authenticate',
+            :sign_in => true
+          )
+        end
+      end
+    end
+  end
+end

data/lita-tweet.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "lita-tweet"
-  spec.version       = "0.1.0"
+  spec.version       = "0.3.0"
   spec.authors       = ["Andre Arko"]
   spec.email         = ["andre@arko.net"]
   spec.description   = "Tweeting for Lita"
@@ -15,8 +15,10 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_runtime_dependency "lita", ">= 4.7"
+  spec.add_runtime_dependency "twitter", "~> 5.16"
+  spec.add_runtime_dependency "oauth", "~> 0.5.1"
 
-  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "bundler", "~> 1.12"
   spec.add_development_dependency "pry-byebug"
   spec.add_development_dependency "rake"
   spec.add_development_dependency "rack-test"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lita-tweet
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Andre Arko
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-08-03 00:00:00.000000000 Z
+date: 2016-08-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: lita
@@ -24,20 +24,48 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '4.7'
+- !ruby/object:Gem::Dependency
+  name: twitter
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.16'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.16'
+- !ruby/object:Gem::Dependency
+  name: oauth
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.5.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.5.1
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: '1.12'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: '1.12'
 - !ruby/object:Gem::Dependency
   name: pry-byebug
   requirement: !ruby/object:Gem::Requirement
@@ -135,8 +163,11 @@ files:
 - LICENSE
 - README.md
 - Rakefile
+- lib/keyword_struct.rb
 - lib/lita-tweet.rb
 - lib/lita/handlers/tweet.rb
+- lib/lita/handlers/tweet/account.rb
+- lib/lita/handlers/tweet/data.rb
 - lita-tweet.gemspec
 - locales/en.yml
 - spec/lita/handlers/tweet_spec.rb