jugyo-termtter 0.7.0 → 0.7.5

data/lib/plugin/keyword.rb

@@ -12,5 +12,5 @@ end
 # keyword.rb
 #   provides a keyword watching method
 # example config
-#   configatron.timeline_format = '<%= color(time, 90) %> <%= color(status, s.has_keyword ? 4 : status_color) %> <%= color(id, 90) %>'
+#   configatron.plugins.stdout.timeline_format = '<%= color(time, 90) %> <%= color(status, s.has_keyword ? 4 : status_color) %> <%= color(id, 90) %>'
 #   configatron.plugins.keyword.keywords = [ /motemen/ ]

data/lib/plugin/log.rb

@@ -1,16 +1,22 @@
 module Termtter::Client
   public_storage[:log] = []
+  configatron.plugins.log.set_default('max_size', 1/0.0)
 
   add_help '/WORD', 'Search log for WORD'
 
-  add_hook do |statuses,event|
+  add_hook do |statuses, event|
     case event
     when :update_friends_timeline
       public_storage[:log] += statuses
+      max_size = configatron.plugins.log.max_size
+      if public_storage[:log].size > max_size
+        public_storage[:log] = public_storage[:log][-max_size..-1]
+      end
+      public_storage[:log] = public_storage[:log].uniq.sort_by{|a| a.created_at} if statuses.first
     end
   end
 
-  add_command %r'^/(.+)' do |m,t|
+  add_command %r'^/(.+)' do |m, t|
     pat = Regexp.new(m[1])
     statuses = public_storage[:log].select { |s| s.text =~ pat }
     call_hooks(statuses, :list_friends_timeline, t)

data/lib/plugin/plugin.rb

@@ -16,21 +16,21 @@ module Termtter::Client
 
   add_help 'plugins', 'Show list of plugins'
   add_command /^plugins$/ do |m, t|
-    puts public_storage[:plugins].join("\n")
+    puts public_storage[:plugins].sort.join("\n")
   end
 
   def self.find_plugin_candidates(a, b)
-    if a.empty?
-      public_storage[:plugins].to_a
-    else
-      public_storage[:plugins].grep(/^#{Regexp.quote a}/i)
-    end.
-    map {|u| b % u }
+    public_storage[:plugins].
+      grep(/^#{Regexp.quote a}/i).
+      map {|u| b % u }
   end
 
   add_completion do |input|
-    if input =~ /^(plugin)\s+(.*)/
+    case input
+    when /^(plugin)\s+(.+)/
       find_plugin_candidates $2, "#{$1} %s"
+    when /^(plugin)\s+$/
+      public_storage[:plugins].sort
     else
       %w[ plugin plugins ].grep(/^#{Regexp.quote input}/)
     end

data/lib/plugin/quicklook.rb

@@ -0,0 +1,36 @@
+require 'uri'
+require 'open-uri'
+require 'pathname'
+require 'tmpdir'
+
+configatron.plugins.quicklook.set_default(:quicklook_tmpdir, "#{Dir.tmpdir}/termtter-quicklook-tmpdir")
+tmpdir = Pathname.new(configatron.plugins.quicklook.quicklook_tmpdir)
+tmpdir.mkdir unless tmpdir.exist?
+
+def quicklook(url)
+  tmpdir = Pathname.new(configatron.plugins.quicklook.quicklook_tmpdir)
+  path   = tmpdir + Pathname.new(url).basename
+
+  Thread.new do
+    open(path, 'w') do |f|
+      f.write(open(url).read)
+    end
+    system("qlmanage -p #{path} > /dev/null 2>&1")
+  end
+end
+
+module Termtter::Client
+  add_command %r'^(?:quicklook|ql)\s+(\w+)$' do |m,t|
+    id = m[1]
+    status = t.show(id).first
+
+    if (status)
+      uris = URI.regexp.match(status.text).to_a
+      quicklook(uris.first) unless uris.empty?
+    end
+  end
+end
+
+# quicklook.rb
+# TODO:
+#   Close quicklook window automatically.

data/lib/plugin/reload.rb

@@ -0,0 +1,3 @@
+module Termtter::Client
+  add_macro /^reload$/, 'eval exec $0'
+end

data/lib/plugin/shell.rb

@@ -1,5 +1,4 @@
 module Termtter::Client
-  add_command /^shell/ do |_, _|
-    system ENV['SHELL']
-  end
+  add_help 'shell,sh', 'Start your shell'
+  add_macro /^(?:shell|sh)/, "eval system ENV['SHELL'] || ENV['COMSPEC']"
 end

data/lib/plugin/sl.rb

@@ -0,0 +1,3 @@
+module Termtter::Client
+  add_macro /^sl$/, 'eval system "sl"'
+end

data/lib/plugin/spam.rb

@@ -0,0 +1,7 @@
+Termtter::Twitter.new(configatron.user_name, configatron.password).update_status('*super spam time*')
+module Termtter::Client
+  clear_commands
+  add_command /.+/ do |m, t|
+    Thread.new { t.update_status(m[0]) }
+  end
+end

data/lib/plugin/standard_plugins.rb

@@ -32,6 +32,14 @@ module Termtter::Client
     call_hooks(t.show(m[1]), :show, t)
   end
 
+  # TODO: Change colors when remaining_hits is low.
+  # TODO: Simmulate remaining_hits.
+  add_command /^(limit|lm)\s*$/ do |m, t|
+    limit = t.get_rate_limit_status
+    puts "=> #{limit.remaining_hits}/#{limit.hourly_limit}"
+  end
+
+
   add_command /^pause\s*$/ do |m, t|
     pause
   end
@@ -50,6 +58,7 @@ exit,e            Exit
 help              Print this help message
 list,l            List the posts in your friends timeline
 list,l USERNAME   List the posts in the the given user's timeline
+limit,lm          Show the API limit status
 pause             Pause updating
 update,u TEXT     Post a new message
 resume            Resume updating
@@ -124,7 +133,7 @@ show ID           Show a single status
   end
 
   add_completion do |input|
-    standard_commands = %w[exit help list pause update resume replies search show]
+    standard_commands = %w[exit help list pause update resume replies search show limit]
     case input
     when /^(list|l)?\s+(.*)/
       find_user_candidates $2, "#{$1} %s"

data/lib/plugin/stdout.rb

@@ -1,10 +1,16 @@
+require 'highline'
 require 'erb'
 
-configatron.set_default(
+configatron.plugins.stdout.set_default(
+  :colors,
+  [:white, :red, :green, :yellow, :blue, :magenta, :cyan])
+configatron.plugins.stdout.set_default(
   :timeline_format,
   '<%= color(time, 90) %> <%= color(status, status_color) %> <%= color(id, 90) %>')
 
-if RUBY_PLATFORM.downcase =~ /mswin(?!ce)|mingw|bccwin/
+$highline = HighLine.new
+
+if win?
   require 'kconv'
   def color(str, num)
     str.to_s.tosjis
@@ -13,21 +19,24 @@ if RUBY_PLATFORM.downcase =~ /mswin(?!ce)|mingw|bccwin/
     STDOUT.puts(str.tosjis)
   end
 else
-  def color(str, num)
-    "\e[#{num}m#{str}\e[0m"
+  def color(str, value)
+    case value
+    when String, Symbol
+      $highline.color(str, value)
+    else
+      "\e[#{value}m#{str}\e[0m"
+    end
   end
 end
 
 Termtter::Client.add_hook do |statuses, event|
-  colors = %w(0 31 32 33 34 35 36 91 92 93 94 95 96)
-
   case event
   when :update_friends_timeline, :list_friends_timeline, :list_user_timeline, :show, :replies
     unless statuses.empty?
       statuses.reverse! if event == :update_friends_timeline
       statuses.each do |s|
         text = s.text
-        status_color = colors[s.user_screen_name.hash % colors.size]
+        status_color = configatron.plugins.stdout.colors[s.user_screen_name.hash % configatron.plugins.stdout.colors.size]
         status = "#{s.user_screen_name}: #{text}"
         if s.in_reply_to_status_id
           status += " (reply to #{s.in_reply_to_status_id})"
@@ -43,18 +52,24 @@ Termtter::Client.add_hook do |statuses, event|
 
         id = s.id
 
-        puts ERB.new(configatron.timeline_format).result(binding)
+        puts ERB.new(configatron.plugins.stdout.timeline_format).result(binding)
       end
     end
   when :search
     statuses.each do |s|
       text = s.text
-      status_color = colors[s.user_screen_name.hash % colors.size]
+      status_color = configatron.plugins.stdout.colors[s.user_screen_name.hash % configatron.plugins.stdout.colors.size]
 
       status = "#{s.user_screen_name}: #{text}"
       time = "(#{s.created_at.strftime('%m-%d %H:%M')})"
       id = s.id
-      puts ERB.new(configatron.timeline_format).result(binding)
+      puts ERB.new(configatron.plugins.stdout.timeline_format).result(binding)
     end
   end
 end
+
+# stdout.rb
+#   output statuses to stdout
+# example config
+#   configatron.plugins.stdout.colors = [:white, :red, :green, :yellow, :blue, :magenta, :cyan]
+#   configatron.plugins.stdout.timeline_format = '<%= color(time, 90) %> <%= color(status, status_color) %> <%= color(id, 90) %>'

data/lib/plugin/uri-open.rb

@@ -10,11 +10,17 @@ module Termtter::Client
   end
 
   def self.open_uri(uri)
-    # FIXME: works only in OSX and other *NIXs
-    if /linux/ =~ RUBY_PLATFORM
-      system 'firefox', uri
+    unless configatron.plugins.uri_open.browser.nil?
+      system configatron.plugins.uri_open.browser, uri
     else
-      system 'open', uri
+      case RUBY_PLATFORM
+      when /linux/
+        system 'firefox', uri
+      when /mswin(?!ce)|mingw|bccwin/
+        system 'explorer', uri
+      else
+        system 'open', uri
+      end
     end
   end
 

data/lib/plugin/yhara.rb

@@ -0,0 +1,142 @@
+if RUBY_VERSION < "1.8.7"
+  class Array
+    def choice
+      at(rand(size))
+    end
+  end
+end
+
+# based on new-harizon.rb
+module Yharian
+
+  VOICES = 
+    %w(Agnes Albert Bad\ News Bahh Bells Boing Bruce Bubbles Cellos Deranged Fred Hysterical Junior Kathy Pipe\ Organ Princess Ralph Trinoids Vicki Victoria Whisper Zarvox)
+  
+
+  class Speaker
+    attr_reader :name
+    def initialize(name)
+      @name = name
+    end
+    
+    def talk(context)
+      n = 7
+      words = (0..rand(n)).map { %w[y hara].choice }.
+        inject {|r, e| r + (rand < 0.97 ? ' ' : ', ') + e }
+      eos = %w(? ? . . . . . . . . !).choice
+      [Remark.new(self,words, eos)]
+    end
+
+    def voice(context = nil)
+      @name
+    end
+
+  end
+
+
+  class Alex < Speaker
+    def initialize
+      super 'Alex'
+    end
+  end
+
+  class Vicki < Speaker
+    def initialize
+      super 'Vicki'
+    end
+  end
+  
+  class Yhara < Speaker
+    def initialize
+      super 'yhara'
+    end
+
+    def voice(context = nil)
+      VOICES.choise
+    end
+  end
+
+  class Jenifer < Speaker
+    ARABIAN = %w[ايران نيست]
+
+    def initialize
+      super 'jenifer'
+    end
+    
+    def talk(context)
+      words = (0..rand(3)). map { ARABIAN.choice }.join(' ')
+      [Remark.new(self,words, '')]
+    end
+
+    def voice(context = nil)
+      'Princess'
+    end
+  end
+  
+  class Remark
+    attr_reader :speaker, :words, :eos, :pronounciation
+    
+    def initialize(speaker, words, eos, options = {})
+      @speaker = speaker
+      @words = words
+      @eos = eos # end of text : "?" or "." or "!"
+      @pronounciation = options[:pronounciation] || text
+    end
+    
+    def text
+      @words + @eos
+    end
+    
+    def interrogative?
+      @eos == '?'
+    end
+    
+    def display
+      puts "#{@speaker.name}: #{text}"
+    end
+    
+    def say(context = nil)
+      Kernel.say pronounciation, :voice => @speaker.voice(context)
+    end
+    
+    def correct?(s)
+      s.gsub(/[^yhar]/,'') == @words.gsub(/[^yhar]/,'')
+    end
+  end
+
+  @@context = []
+  @@speakers = [Alex.new, Vicki.new]
+
+  def self.text
+    if ( @@context.last && Yhara === @@context.last.speaker && rand < 0.25 ) || rand < 0.01
+      speaker = Jenifer.new
+    elsif @@context.last && @@context.last.words =~ /y hara/ and @@context.last.interrogative? and rand < 0.25
+      speaker = Yhara.new
+    else
+      speaker = @@speakers[rand(2)]
+    end
+
+    remark = speaker.talk(@@context).first
+    @@context.push remark
+    remark.text
+  end
+end
+
+module Termtter::Client
+
+  add_help 'yhara', 'Post a new Yharian sentence'
+  
+  add_command /^(yhara|y)\s*/ do |m, t|
+    text = Yharian::text
+    unless text.empty?
+      t.update_status(text)
+      puts "=> #{text}"
+    end
+  end
+end
+
+# yhara.rb
+#   post a new yharian sentence
+# example:
+#   > yhara
+#   => hara y y hara.

data/lib/termtter.rb

@@ -3,21 +3,25 @@ $:.unshift(File.dirname(__FILE__)) unless
 
 require 'rubygems'
 require 'json'
+require 'net/https'
 require 'open-uri'
 require 'cgi'
 require 'readline'
 require 'enumerator'
 require 'parsedate'
 require 'configatron'
-require 'filter'
 
 if RUBY_VERSION < '1.8.7'
   class Array
-    def take(n) at(0...n) end
+    def take(n) self[0...n] end
   end
 end
 
-if RUBY_PLATFORM.downcase =~ /mswin(?!ce)|mingw|bccwin/
+def win?
+  RUBY_PLATFORM.downcase =~ /mswin(?!ce)|mingw|bccwin/
+end
+
+if win?
   require 'kconv'
   module Readline
     alias :old_readline :readline
@@ -30,6 +34,17 @@ end
 
 configatron.set_default(:update_interval, 300)
 configatron.set_default(:prompt, '> ')
+configatron.set_default(:enable_ssl, false)
+configatron.proxy.set_default(:port, '8080')
+
+# FIXME: we need public_storage all around the script
+module Termtter
+  module Client
+    def self.public_storage
+      @@public_storage ||= {}
+    end
+  end
+end
 
 def plugin(s)
   require "plugin/#{s}"
@@ -37,32 +52,76 @@ end
 
 def filter(s)
   load "filter/#{s}.rb"
+rescue LoadError
+  raise
+else
+  Termtter::Client.public_storage[:filters] = []
+  Termtter::Client.public_storage[:filters] << s
+  true
 end
 
 # FIXME: delete this method after the major version up
-alias original_require require
-def require(s)
-  if %r|^termtter/(.*)| =~ s
-    puts "[WARNING] use plugin '#{$1}' instead of require"
-    puts "  Such a legacy .termtter file will not be supported until version 1.0.0"
-    s = "plugin/#{$1}"
+unless defined? original_require
+  alias original_require require
+  def require(s)
+    if %r|^termtter/(.*)| =~ s
+      puts "[WARNING] use plugin '#{$1}' instead of require"
+      puts "  Such a legacy .termtter file will not be supported until version 1.0.0"
+      s = "plugin/#{$1}"
+    end
+    original_require s
   end
-  original_require s
 end
 
 module Termtter
-  VERSION = '0.7.0'
+  VERSION = '0.7.5'
   APP_NAME = 'termtter'
 
+  class Connection
+    attr_reader :protocol, :port, :proxy_uri
+
+    def initialize
+      @proxy_host = configatron.proxy.host
+      @proxy_port = configatron.proxy.port
+      @proxy_user = configatron.proxy.user_name
+      @proxy_password = configatron.proxy.password
+      @proxy_uri = nil
+      @enable_ssl = configatron.enable_ssl
+      @protocol = "http"
+      @port = 80
+
+      unless @proxy_host.empty?
+        @http_class = Net::HTTP::Proxy(@proxy_host, @proxy_port,
+                                       @proxy_user, @proxy_password)
+        @proxy_uri =  "http://" + @proxy_host + ":" + @proxy_port + "/"
+      else
+        @http_class = Net::HTTP
+      end
+
+      if @enable_ssl
+        @protocol = "https"
+        @port = 443
+      end
+    end
+
+    def start(host, port, &block)
+      http = @http_class.new(host, port)
+      http.use_ssl = @enable_ssl
+      http.verify_mode = OpenSSL::SSL::VERIFY_NONE if http.use_ssl
+      http.start(&block)
+    end
+  end
+
   class Twitter
 
     def initialize(user_name, password)
       @user_name = user_name
       @password = password
+      @connection = Connection.new
     end
 
     def update_status(status)
-      Net::HTTP.start("twitter.com", 80) do |http|
+      @connection.start("twitter.com", @connection.port) do |http|
         uri = '/statuses/update.xml'
         http.request(post_request(uri), "status=#{CGI.escape(status)}&source=#{APP_NAME}")
       end
@@ -70,13 +129,13 @@ module Termtter
     end
 
     def get_friends_timeline(since_id = nil)
-      uri = "http://twitter.com/statuses/friends_timeline.json"
+      uri =  "#{@connection.protocol}://twitter.com/statuses/friends_timeline.json"
       uri << "?since_id=#{since_id}" if since_id
       return get_timeline(uri)
     end
 
     def get_user_timeline(screen_name)
-      return get_timeline("http://twitter.com/statuses/user_timeline/#{screen_name}.json")
+      return get_timeline("#{@connection.protocol}://twitter.com/statuses/user_timeline/#{screen_name}.json")
     rescue OpenURI::HTTPError => e
       puts "No such user: #{screen_name}"
       nears = near_users(screen_name)
@@ -85,7 +144,7 @@ module Termtter
     end
 
     def search(query)
-      results = JSON.parse(open('http://search.twitter.com/search.json?q=' + CGI.escape(query)).read)['results']
+      results = JSON.parse(open("#{@connection.protocol}://search.twitter.com/search.json?q=" + CGI.escape(query)).read, :proxy => @connection.proxy_uri)['results']
       return results.map do |s|
         status = Status.new
         status.id = s['id']
@@ -97,15 +156,15 @@ module Termtter
     end
 
     def show(id)
-      return get_timeline("http://twitter.com/statuses/show/#{id}.json")
+      return get_timeline("#{@connection.protocol}://twitter.com/statuses/show/#{id}.json")
     end
 
     def replies
-      return get_timeline("http://twitter.com/statuses/replies.json")
+      return get_timeline("#{@connection.protocol}://twitter.com/statuses/replies.json")
     end
 
     def get_timeline(uri)
-      data = JSON.parse(open(uri, :http_basic_authentication => [@user_name, @password]).read)
+      data = JSON.parse(open(uri, :http_basic_authentication => [user_name, password], :proxy => @connection.proxy_uri).read)
       data = [data] unless data.instance_of? Array
       return data.map do |s|
         status = Status.new
@@ -121,16 +180,48 @@ module Termtter
       end
     end
 
+    # note: APILimit.reset_time_in_seconds == APILimit.reset_time.to_i
+    APILIMIT = Struct.new("APILimit", :reset_time, :reset_time_in_seconds, :remaining_hits, :hourly_limit)
+    def get_rate_limit_status
+      uri = 'http://twitter.com/account/rate_limit_status.json'
+      data = JSON.parse(open(uri, :http_basic_authentication => [user_name, password], :proxy => @connection.proxy_uri).read)
+
+      reset_time = Time.parse(data['reset_time'])
+      reset_time_in_seconds = data['reset_time_in_seconds'].to_i
+      
+      APILIMIT.new(reset_time, reset_time_in_seconds, data['remaining_hits'], data['hourly_limit'])
+    end
+
+    alias :api_limit :get_rate_limit_status
+
+    private
+
+    def user_name
+      unless @user_name.instance_of? String
+        @user_name = Readline.readline('user name: ', false)
+      end
+      @user_name
+    end
+
+    def password
+      unless @password.instance_of? String
+        system 'stty -echo'
+        @password = Readline.readline('password: ', false)
+        system 'stty echo'
+        puts
+      end
+      @password
+    end
+
     def near_users(screen_name)
       Client::public_storage[:users].select {|user|
         /#{user}/i =~ screen_name || /#{screen_name}/i =~ user
       }.join(', ')
     end
-    private :near_users
 
     def post_request(uri)
       req = Net::HTTP::Post.new(uri)
-      req.basic_auth(@user_name, @password)
+      req.basic_auth(user_name, password)
       req.add_field('User-Agent', 'Termtter http://github.com/jugyo/termtter')
       req.add_field('X-Twitter-Client', 'Termtter')
       req.add_field('X-Twitter-Client-URL', 'http://github.com/jugyo/termtter')
@@ -144,6 +235,7 @@ module Termtter
     @@hooks = []
     @@commands = {}
     @@completions = []
+    @@filters = []
     @@helps = []
 
     class << self
@@ -159,6 +251,12 @@ module Termtter
         @@commands[regex] = block
       end
 
+      def add_macro(r, s)
+        add_command(r) do |m, t|
+          call_commands(s % m, t)
+        end
+      end
+
       def clear_commands
         @@commands.clear
       end
@@ -179,6 +277,27 @@ module Termtter
         @@helps.clear
       end
 
+      def add_filter(&filter)
+        @@filters << filter
+      end
+
+      def clear_filters
+        @@filters.clear
+      end
+
+      # memo: each filter must return Array of Status
+      def apply_filters(statuses)
+        filtered = statuses
+        @@filters.each do |f|
+          filtered = f.call(filtered)
+        end
+        filtered
+      rescue => e
+        puts "Error: #{e}"
+        puts e.backtrace.join("\n")
+        statuses
+      end
+
       Readline.basic_word_break_characters= "\t\n\"\\'`><=;|&{("
       Readline.completion_proc = proc {|input|
         @@completions.map {|completion|
@@ -186,10 +305,6 @@ module Termtter
         }.flatten.compact
       }
 
-      def public_storage
-        @@public_storage ||= {}
-      end
-
       def call_hooks(statuses, event, tw)
         statuses = apply_filters(statuses)
         @@hooks.each do |h|
@@ -231,6 +346,8 @@ module Termtter
       end
 
       def exit
+        call_hooks([], :exit, nil)
+        @@main_thread.kill
         @@update_thread.kill
         @@input_thread.kill
       end
@@ -240,7 +357,9 @@ module Termtter
         initialized = false
         @@pause = false
         tw = Termtter::Twitter.new(configatron.user_name, configatron.password)
+        call_hooks([], :initialize, tw)
 
+        @@input_thread = nil
         @@update_thread = Thread.new do
           since_id = nil
           loop do
@@ -251,9 +370,16 @@ module Termtter
               unless statuses.empty?
                 since_id = statuses[0].id
               end
+              print "\e[1K\e[0G" if !statuses.empty? && !win?
               call_hooks(statuses, :update_friends_timeline, tw)
               initialized = true
-
+              @@input_thread.kill if @@input_thread && !statuses.empty?
+            rescue OpenURI::HTTPError => e
+              if e.message == '401 Unauthorized'
+                puts 'Could not login'
+                puts 'plese check your account settings'
+                exit!
+              end
             rescue => e
               puts "Error: #{e}"
               puts e.backtrace.join("\n")
@@ -270,8 +396,25 @@ module Termtter
           Readline.__send__("#{vi_or_emacs}_editing_mode")
         end
 
-        @@input_thread = Thread.new do
-          while buf = Readline.readline(configatron.prompt, true)
+        begin
+          stty_save = `stty -g`.chomp
+          trap("INT") { system "stty", stty_save; exit }
+        rescue Errno::ENOENT
+        end
+
+        @@main_thread = Thread.new do
+          loop do
+            @@input_thread = create_input_thread(tw)
+            @@input_thread.join
+          end
+        end
+        @@main_thread.join
+      end
+
+      def create_input_thread(tw)
+        Thread.new do
+          erb = ERB.new(configatron.prompt)
+          while buf = Readline.readline(erb.result(tw.__send__(:binding)), true)
             begin
               call_commands(buf, tw)
             rescue CommandNotFound => e
@@ -283,18 +426,8 @@ module Termtter
             end
           end
         end
-
-        begin
-          stty_save = `stty -g`.chomp
-          trap("INT") { system "stty", stty_save; exit }
-        rescue Errno::ENOENT
-        end
-
-        @@input_thread.join
       end
-
     end
-
   end
 
   class CommandNotFound < StandardError; end
@@ -307,6 +440,9 @@ module Termtter
       attr_accessor attr.to_sym
     end
 
+    def eql?(other); self.id == other.id end
+    def hash; self.id end
+
     def english?
       self.class.english?(self.text)
     end