twat 0.6.3 → 0.9.2

data/lib/twat/endpoint.rb

@@ -1,16 +1,22 @@
 module Twat
 
-  %w[twitter identica].each do |filename|
+  ENDPOINTS = ["identi.ca", "twitter"]
+
+  %w[base twitter identica].each do |filename|
     require File.join(File.expand_path("../endpoints", __FILE__), filename)
   end
 
   # Proxies to the actual endpoint classes
   class Endpoint
     def self.new(endpoint)
-      @implementation = {
-        :twitter => Endpoints::Twitter,
-        :"identi.ca" => Endpoints::Identica
-      }[endpoint].new
+      begin
+        @implementation = {
+          :twitter => Endpoints::Twitter,
+          :"identi.ca" => Endpoints::Identica
+        }[endpoint].new
+      rescue NoMethodError
+        raise Exceptions::NoSuchEndpoint
+      end
     end
   end
 end

data/lib/twat/endpoints/base.rb

@@ -0,0 +1,39 @@
+module Twat::Endpoints
+  class Base
+
+    @@endpoint_set=nil
+
+    def authorize_account(name)
+      oauth_conf = oauth_options.merge ({ :site => url })
+
+      oauth = OAuth::Consumer.new(consumer_info[:consumer_key], consumer_info[:consumer_secret], oauth_conf)
+      token_request = oauth.get_request_token()
+      puts "Please authenticate the application at #{token_request.authorize_url}, then enter pin"
+      pin = STDIN.gets.chomp
+      begin
+        access_token = token_request.get_access_token(oauth_verifier: pin)
+        account_settings = {
+          oauth_token: access_token.token,
+          oauth_token_secret: access_token.secret,
+          endpoint: endpoint_name
+        }
+        config.accounts[name.to_sym] = account_settings
+        config.save!
+      rescue OAuth::Unauthorized
+        puts "Couldn't authenticate you, did you enter the pin correctly?"
+      end
+    end
+
+    def config
+      return @config if @config
+      @config = ::Twat::Config.new
+      @config.create! unless @config.exists?
+      return @config
+    end
+
+    def endpoint_name
+      raise "Endpoint not set! Naughty programmer"
+    end
+
+  end
+end

data/lib/twat/endpoints/identica.rb

@@ -1,5 +1,5 @@
 module Twat::Endpoints
-  class Identica
+  class Identica < Base
 
     def initialize
       unless @@endpoint_set
@@ -38,5 +38,9 @@ module Twat::Endpoints
       }
     end
 
+    def endpoint_name
+      :"identi.ca"
+    end
+
   end
 end

data/lib/twat/endpoints/twitter.rb

@@ -1,5 +1,5 @@
 module Twat::Endpoints
-  class Twitter
+  class Twitter < Base
 
     def url
       "https://api.twitter.com/"
@@ -16,6 +16,10 @@ module Twat::Endpoints
       {}
     end
 
+    def endpoint_name
+      :twitter
+    end
+
   end
 end
 

data/lib/twat/exceptions.rb

@@ -1,53 +1,100 @@
 module Twat
-  class NoSuchAccount < Exception; end
-  class NoDefaultAccount < Exception; end
-  class NoSuchCommand < Exception; end
-  class NoConfigFile < Exception; end
-  class RequiresOptVal < Exception; end
-  class Usage < Exception; end
-  class InvalidCredentials < Exception; end
-  class ConfigVersionIncorrect < Exception; end
-  class InvalidSetOpt < Exception; end
-  class InvalidBool < Exception; end
-  class InvalidInt < Exception; end
-  class TweetTooLong < Exception; end
+  module Exceptions
 
-  module HandledExceptions
-    def with_handled_exceptions(opts)
+    class TwatException < Exception; end
+
+    class Usage < TwatException; end
+
+    class AlreadyConfigured < TwatException
+      def msg
+        "Already configured, use add to add more accounts"
+      end
+    end
+
+    class ArgumentRequired < TwatException
+      def msg
+        "This command requires an argument"
+      end
+    end
+    class NoSuchAccount < TwatException
+      def msg
+        "No such account"
+      end
+    end
+    class NoDefaultAccount < TwatException
+      def msg
+        "No default account configured (try twat set default [account])."
+      end
+    end
+    class NoSuchCommand < TwatException
+      def msg
+        "No such command"
+      end
+    end
+    class NoSuchEndpoint < TwatException
+      def msg
+        "Available endpoints are #{::Twat::ENDPOINTS.join(", ")}"
+      end
+    end
+    class NoConfigFile < TwatException
+      def msg
+        "No config file, create one with twat -a [user|nick]"
+      end
+    end
+    class RequiresOptVal < TwatException
+      def msg
+        "--set must take an option=value pair as arguments"
+      end
+    end
+    class InvalidCredentials < TwatException
+      def msg
+        ["Invalid credentials, try reauthenticating with",
+          "twat -a ACCOUNT"]
+      end
+    end
+    class ConfigVersionIncorrect < TwatException
+      def msg
+        "Your config file is out of date. Run with --update-config to rememdy"
+      end
+    end
+    class InvalidSetOpt < TwatException
+      def msg
+        "There is no such configurable option"
+      end
+    end
+    class InvalidBool < TwatException
+      def msg
+        "Invalid value, valid values are #{Options::BOOL_VALS.join("|")}"
+      end
+    end
+    class InvalidInt < TwatException
+      def msg
+        "Invalid value, must be an integer"
+      end
+    end
+    class TweetTooLong < TwatException
+      def msg
+        "Twitter enforces a maximum status length of 140 characters"
+      end
+    end
+    class NoSuchTweet < TwatException
+      def msg
+        "Specified tweet was not found"
+      end
+    end
+    class NoInternetConnection < TwatException
+      def msg
+        "Couldn't connect to the endpoint"
+      end
+    end
+
+    def with_handled_exceptions(opts=nil)
       begin
-        # FIXME
-        yield opts
+        yield
       rescue Usage
-        opts.usage
-      rescue NoSuchAccount
-        puts "No such account"
-        opts.usage
-      rescue NoDefaultAccount
-        puts "No default account configured."
-      rescue NoSuchCommand
-        puts "No such command"
-        opts.usage
-      rescue NoConfigFile
-        puts "No config file, create one with twat -a [user|nick]"
-        opts.usage
-      rescue InvalidSetOpt
-        puts "There is no such configurable option"
-        opts.usage
-      rescue RequiresOptVal
-        puts "--set must take an option=value pair as arguments"
-      rescue InvalidCredentials
-        puts "Invalid credentials, try reauthenticating with"
-        puts "twat -a #{opts[:account]}"
-      rescue ConfigVersionIncorrect
-        puts "Your config file is out of date. Run with --update-config to rememdy"
-      rescue InvalidBool
-        puts "Invalid value, valid values are #{Options::BOOL_VALID.join("|")}"
-      rescue InvalidInt
-        puts "Invalid value, must be an integer"
-      rescue Errno::ECONNRESET
-        puts "Connection was reset by third party."
-      rescue TweetTooLong
-        puts "Twitter enforces a maximum status length of 140 characters"
+        opts.usage if opts
+      rescue TwatException => e
+        puts e.msg
       end
     end
   end

data/lib/twat/follow_mixin.rb

@@ -0,0 +1,134 @@
+module FollowMixin
+  POLLING_RESOLUTION = 20 # Readline scan time in hz
+  # Shim to bail out if we're in test
+  def untested
+    true
+  end
+
+  class PrintNothing < Exception; end
+
+  def run
+    auth!
+    enable_readline! if untested
+    @tweetstack = ::Twat::TweetStack.new
+
+    @failcount = 0
+
+    # Get 5 tweets
+    tweets = retrieve_tweets(:count => 5)
+    while untested do
+      begin
+        last_id = process_followed(tweets) if tweets.any?
+        (config.polling_interval * POLLING_RESOLUTION).times do
+          reader.tick
+          lines = 0
+          reader.each_line do |i|
+            lines += 1
+            handle_input(i)
+          end
+          sleep 1.0/POLLING_RESOLUTION if lines == 0
+        end
+        tweets = retrieve_tweets(:since_id => last_id)
+        @failcount = 0
+      rescue Interrupt
+        break
+      rescue Twitter::Error::ServiceUnavailable
+        break unless fail_or_bail
+        sleeptime = 60 * (@failcount + 1)
+        reader.puts_above "#{"(__-){".red}: the fail whale has been rolled out, sleeping for #{sleeptime} seconds"
+        reader.wait sleeptime
+      rescue Errno::ECONNRESET, Errno::ETIMEDOUT, SocketError
+        break unless fail_or_bail
+      end
+    end
+  end
+
+  private
+
+  def retrieve_tweets(opts)
+    new_tweets(opts)
+  rescue SocketError
+    raise ::Twat::Exceptions::NoInternetConnection
+  end
+
+  def handle_input(inp)
+    ret = process_input(inp)
+    return if ret.nil?
+
+    if ret === true
+      reader.puts_above(inp.green)
+    elsif ret === false
+      reader.puts_above(inp.red)
+    else
+      reader.puts_above(ret)
+    end
+  rescue PrintNothing
+    nil
+  end
+
+  def process_input(inp)
+    case inp
+    when /^[rR][tT] ([0-9]{1,2})/
+      begin
+        retweet($1.to_i)
+        return true
+      rescue ::Twat::Exceptions::NoSuchTweet
+        return "#{inp.red} #{":: No such tweet".bold.red}"
+      end
+    when /^follow (.*)/
+      Twitter.follow($1)
+      return true
+    when /^reply ([0-9]{1,2}) (.*)$/
+      # TODO This should be a method of the tweetstack
+      in_reply_to = get_tweet($1.to_i)
+      msg = "@#{in_reply_to.as_user} #{$2}"
+
+      if msg.length > 140
+        return "#{msg} #{":: Too long".bold.red}"
+      else
+        Twitter.update(msg, in_reply_to_status_id: in_reply_to.id)
+        return msg.green
+      end
+    else
+      # Assume they want to tweet something
+      if inp.length > 140
+        return "#{inp.red} #{":: Too long".bold.red}"
+      else
+        Twitter.update(inp)
+        return true
+      end
+    end
+  end
+
+  def get_tweet(idx)
+    raise ::Twat::Exceptions::NoSuchTweet unless @tweetstack.include?(idx)
+    @tweetstack[idx]
+  end
+
+  # Wrapper methods from the tweetstack implementation
+  def retweet(idx)
+    Twitter.retweet(get_tweet(idx).id)
+  end
+
+  def fail_or_bail
+    if @failcount > 2
+      puts "3 consecutive failures, giving up"
+    else
+      @failcount += 1
+      return true
+    end
+  end
+
+  def process_followed(tweets)
+    last_id = nil
+    tweets.reverse.each do |tweet|
+      id = @tweetstack << tweet
+      beep if config.beep? && tweet.text.mentions?(config.account_name)
+      reader.puts_above format(tweet, @tweetstack.last)
+      last_id = tweet.id
+    end
+
+    return last_id
+  end
+
+end

data/lib/twat/options.rb

@@ -1,5 +1,8 @@
 module Twat
   class Options
+
+    include ::Twat::Exceptions
+
     BOOL_TRUE=["yes", "true", "1", "on"]
     BOOL_FALSE=["no", "false", "0", "off"]
     BOOL_VALS = BOOL_TRUE + BOOL_FALSE
@@ -23,41 +26,54 @@ module Twat
       !!/^[0-9]+$/.match(val)
     end
 
+    def self.to_bool(value)
+      case
+      when bool_true?(value)
+        return true
+      when bool_false?(value)
+        return false
+      else
+        raise InvalidBool
+      end
+    end
+
     # A set of wrappers around the global config object to set given attributes
     # Catching failures is convenient because of the method_missing? hook
     def method_missing(sym, *args, &block)
-      puts sym
-      raise InvalidSetOpt
+      raise InvalidSetOpt, "Options doesn't know about #{sym}"
     end
 
     def config
-      @config ||= Config.new
+      @config ||= Config.new.tap do |n|
+        raise NoConfigFile unless n.exists?
+      end
     end
 
     # This is deliberately not abstracted (it could be easily accessed from
     # withing the method_missing method, but that will just lead to nastiness
     # later when I implement colors, for example.
     def default=(value)
-      val = value.to_sym
-      unless config.accounts.include?(val)
+      value = value.to_sym
+      unless config.accounts.include?(value)
         raise NoSuchAccount
       end
 
-      config[:default] = val
+      config[:default] = value
       config.save!
     end
 
     def colors=(value)
-      val = value.to_sym
-      raise InvalidBool unless Options.bool_valid?(val)
-      config[:colors] = val
+      config[:colors] = Options.to_bool(value)
       config.save!
     end
 
     def beep=(value)
-      val = value.to_sym
-      raise InvalidBool unless Options.bool_valid?(val)
-      config[:beep] = val
+      config[:beep] = Options.to_bool(value)
+      config.save!
+    end
+
+    def show_mentions=(value)
+      config[:show_mentions] = Options.to_bool(value)
       config.save!
     end