termtter 1.3.1 → 1.4.0

data/lib/plugins/defaults/stdout.rb

@@ -10,12 +10,14 @@ config.plugins.stdout.set_default(
   '<90><%=time%> [<%=status_id%>]</90> <<%=color%>><%=s.user.screen_name%>: <%=text%></<%=color%>> ' +
   '<90><%=reply_to_status_id ? " (reply_to [#{reply_to_status_id}]) " : ""%><%=source%><%=s.user.protected ? "[P]" : ""%></90>'
 )
+config.plugins.stdout.set_default(:time_format_today, '%H:%M:%S')
+config.plugins.stdout.set_default(:time_format_not_today, '%y/%m/%d %H:%M')
 config.plugins.stdout.set_default(:enable_pager, true)
 config.plugins.stdout.set_default(:pager, 'less -R -f +G')
 config.plugins.stdout.set_default(:window_height, 50)
 config.plugins.stdout.set_default(:typable_ids, ('aa'..'zz').to_a)
 config.plugins.stdout.set_default(:typable_id_prefix, '$')
-config.plugins.stdout.set_default(:show_as_thread, false)
+config.plugins.stdout.set_default(:show_as_thread, false) # db plugin is required
 
 module Termtter
   class TypableIdGenerator
@@ -27,30 +29,49 @@ module Termtter
       end
       @ids = ids
       @table = {}
+      @rtable = {}
     end
 
     def next(data)
       id = @ids.shift
       @ids.push id
+      @rtable.delete(@table[id])
       @table[id] = data
+      @rtable[data] = id
       id
     end
 
     def get(id)
       @table[id]
     end
+
+    def get_id(data)
+      @rtable[data] || self.next(data)
+    end
   end
 
   module Client
     @typable_id_generator = TypableIdGenerator.new(config.plugins.stdout.typable_ids)
 
     def self.data_to_typable_id(data)
-      id = config.plugins.stdout.typable_id_prefix + @typable_id_generator.next(data)
+      id = config.plugins.stdout.typable_id_prefix + @typable_id_generator.get_id(data)
     end
 
     def self.typable_id_to_data(id)
       @typable_id_generator.get(id)
     end
+
+    def self.time_format_for(statuses)
+      t0 = Time.now
+      t1 = Time.parse(statuses.first[:created_at])
+      t2 = Time.parse(statuses.last[:created_at])
+      if [t0.year, t0.month, t0.day] == [t1.year, t1.month, t1.day] \
+        and [t1.year, t1.month, t1.day] == [t2.year, t2.month, t2.day]
+        config.plugins.stdout.time_format_today
+      else
+        config.plugins.stdout.time_format_not_today
+      end
+    end
   end
 
   class StdOut < Hook
@@ -59,27 +80,16 @@ module Termtter
     end
 
     def call(statuses, event)
-      print_statuses(statuses)
+      print_statuses(statuses, event)
     end
 
-    def print_statuses(statuses, sort = true, time_format = nil)
+    def print_statuses(statuses, event, sort = true, time_format = nil)
       return unless statuses and statuses.first
-      unless time_format
-        t0 = Time.now
-        t1 = Time.parse(statuses.first[:created_at])
-        t2 = Time.parse(statuses.last[:created_at])
-        time_format = 
-          if [t0.year, t0.month, t0.day] == [t1.year, t1.month, t1.day] \
-            and [t1.year, t1.month, t1.day] == [t2.year, t2.month, t2.day]
-            '%H:%M:%S'
-          else
-            '%y/%m/%d %H:%M'
-          end
-      end
+      time_format ||= Termtter::Client.time_format_for statuses
 
       output_text = ''
       statuses.each do |s|
-        output_text << status_line(s, time_format)
+        output_text << status_line(s, time_format, event)
       end
 
       if config.plugins.stdout.enable_pager && ENV['LINES'] && statuses.size > ENV['LINES'].to_i
@@ -93,7 +103,7 @@ module Termtter
       end
     end
 
-    def status_line(s, time_format, indent = 0)
+    def status_line(s, time_format, event, indent = 0)
       return '' unless s
       text = TermColor.escape(s.text)
       color = config.plugins.stdout.colors[s.user.id.to_i % config.plugins.stdout.colors.size]
@@ -114,10 +124,13 @@ module Termtter
 
       erbed_text = ERB.new(config.plugins.stdout.timeline_format).result(binding)
       indent_text = indent > 0 ? "#{'    ' * (indent - 1)} ┗ " : ''
+
+      erbed_text = Client.get_hooks(:pre_coloring).inject(erbed_text){|result, hook| hook.call(result, event)}
+
       text = TermColor.parse(indent_text + erbed_text) + "\n"
       text = TermColor.unescape(text)
       if config.plugins.stdout.show_as_thread && s.in_reply_to_status_id
-        text << status_line(Status[s.in_reply_to_status_id], time_format, indent + 1)
+        text << status_line(Status[s.in_reply_to_status_id], time_format, event, indent + 1)
       end
       text
     end

data/lib/plugins/defaults.rb

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
 module Termtter::Client
+  # TODO: ↓不要な気もする
   config.set_default(:stdout, true)
   config.set_default(:standard_commands, true)
   config.set_default(:standard_completion, true)

data/lib/plugins/erb.rb

@@ -1,14 +1,10 @@
 # -*- coding: utf-8 -*-
-
 require 'erb'
-
-Termtter::Client.register_hook(
-  :name => :erb,
-  :point => :modify_arg_for_update,
-  :exec_proc => lambda {|cmd, arg|
+module Termtter::Client
+  register_hook(:erb, :point => :modify_arg_for_update) do |cmd, arg|
     ERB.new(arg).result(binding)
-  }
-)
+  end
+end
 
 # erb.rb
 #   enable to <%= %> in the command update

data/lib/plugins/expand-tinyurl.rb

@@ -5,6 +5,8 @@ URL_SHORTTERS = [
   { :host => "is.gd", :pattern => %r'(http://is\.gd(/[\w/]+))' },
   { :host => "bit.ly", :pattern => %r'(http://bit\.ly(/[\w/]+))' },
   { :host => "ff.im", :pattern => %r'(http://ff\.im(/[-\w/]+))'},
+  { :host => "to.ly", :pattern => %r'(http://to\.ly(/[-\w/]+))'},
+  { :host => "j.mp", :pattern => %r'(http://j\.mp(/[\w/]+))' },
 ]
 
 config.plugins.expand_tinyurl.set_default(:shortters, [])
@@ -52,4 +54,6 @@ def expand_url(host, path)
   res = http_class.new(host).head(path)
   return nil unless res.code == "301" or res.code == "302"
   res['Location'].force_encoding(Encoding::UTF_8)
+rescue
+  nil
 end

data/lib/plugins/fibyou.rb

@@ -0,0 +1,12 @@
+# It depends on defaults/fib.rb
+
+module Termtter::Client
+  register_command(:fibyou) do |arg|
+    /(\w+)\s(\d+)/ =~ arg
+    name = normalize_as_user_name($1)
+    n = $2.to_i
+    text = "@#{name} fib(#{n}) = #{fib n}"
+    Termtter::API.twitter.update(text)
+    puts "=> " << text
+  end
+end

data/lib/plugins/group.rb

@@ -1,17 +1,5 @@
 # -*- coding: utf-8 -*-
 
-module Termtter
-  class Status
-    def is_member?(group = nil)
-      if group
-        config.plugins.group.groups[group].include? self.user.screen_name
-      else
-        config.plugins.group.groups.values.flatten.include? self.user.screen_name
-      end
-    end
-  end
-end
-
 module Termtter::Client
   config.plugins.group.
     set_default(:groups, {})
@@ -34,15 +22,25 @@ module Termtter::Client
    :exec_proc => lambda {|arg|
      unless arg.empty?
        group_name = arg.to_sym
+       groups = config.plugins.group.groups
        if group_name == :all
-         targets = config.plugins.group.groups.values.flatten.uniq
+         targets = groups.values.flatten.uniq
+       elsif groups.keys.include? group_name
+         targets = groups[group_name]
        else
-         targets = config.plugins.group.groups[group_name]
+         ignore_cased = groups.keys.map(&:to_s).grep(/^#{group_name}$/i).map(&:to_sym)
+         if ignore_cased.length == 1
+           targets = groups[ignore_cased.first]
+         elsif ignore_cased.length > 1 
+           puts "which? #{ignore_cased.inspect.gsub(':','')}"
+         end
+       end
+       if targets
+         statuses = targets ? targets.map { |target|
+           public_storage[:tweet][target]
+         }.flatten.uniq.compact.sort_by{ |s| s[:id]} : []
+         output(statuses, :search)
        end
-       statuses = targets ? targets.map { |target|
-          public_storage[:tweet][target]
-        }.flatten.uniq.compact.sort_by{ |s| s[:id]} : []
-       output(statuses, :search)
      else
        config.plugins.group.groups.each_pair do |key, value|
          puts "#{key}: #{value.join(',')}"
@@ -69,10 +67,10 @@ module Termtter::Client
     :exec_proc => lambda do |statuses, event|
       return statuses unless event == :update_friends_timeline
       return statuses unless config.plugins.group.default_filter
-      skip_group = config.plugins.group.default_filter
+      filter_group = config.plugins.group.default_filter
       r = []
       statuses.each do |s|
-        unless self.is_member?(s, :dqn)
+        unless self.is_member?(s, filter_group)
           r << s
         end
       end
@@ -86,5 +84,6 @@ end
 #   config.plugins.group.groups = {
 #     :rits => %w(hakobe isano hitode909)
 #   }
+#   config.plugins.group.default_filter = :rits
 # NOTE: group.rb needs plugin/log
 

data/lib/plugins/hi.rb

@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+module Termtter::Client
+  [:hi, :hola].each do |hi|
+    register_command(hi, :help => ["#{hi} [(Optinal) USER]", "Post a #{hi}"]) do |arg|
+      result =
+        if arg.empty?
+          Termtter::API.twitter.update(hi.to_s)
+        else
+          name = normalize_as_user_name(arg)
+          Termtter::API.twitter.update("#{hi} @#{name}")
+        end
+      puts "=> " << result.text
+    end
+  end
+end

data/lib/plugins/hugeurl.rb

@@ -26,7 +26,7 @@ Termtter::Client::register_hook(
         config.plugins.hugeurl.skip_users.include?(s.user.screen_name) and next
         s.text.gsub!(HUGEURL_TARGET_PATTERN) do |m|
           res = http.get('/hugeurl?url=' + m)
-          res.code == '200' ? res.body : m
+          res.code == '200' && res.body !~ /^</ ? res.body : m
         end
       end
     end

data/lib/plugins/irc_gw.rb

@@ -2,10 +2,24 @@
 
 require 'net/irc'
 
-# TODO: post text of stdout too
-
 config.plugins.irc_gw.set_default(:port, 16669)
 config.plugins.irc_gw.set_default(:last_statuses_count, 100)
+config.plugins.irc_gw.set_default(:logger_level, Logger::ERROR)
+
+module Termtter::Client
+  class << self
+    def friends
+      user_name = config.user_name
+
+      frinends = []
+      page = 0
+      begin
+        frinends += tmp = Termtter::API::twitter.friends(user_name, :page => page+=1)
+      end until tmp.empty?
+      frinends.map(&:screen_name)
+    end
+  end
+end
 
 class TermtterIrcGateway < Net::IRC::Server::Session
   @@listners = []
@@ -25,6 +39,13 @@ class TermtterIrcGateway < Net::IRC::Server::Session
       end
     }
   )
+  if Termtter::Client.respond_to? :register_output
+    Termtter::Client.register_output(:irc) do |message|
+      @@listners.each do |listener|
+        listener.log(message.gsub(/\e\[\d+m/, '')) # remove escape sequence
+      end
+    end
+  end
 
   def server_name; 'termtter' end
   def server_version; '0.0.0' end
@@ -41,11 +62,13 @@ class TermtterIrcGateway < Net::IRC::Server::Session
       when :update_friends_timeline
         PRIVMSG
       else
+        time_format = Termtter::Client.time_format_for statuses
         NOTICE
       end
-
     statuses.each do |s|
-      post s.user.screen_name, msg_type, main_channel, [s.text, s.id].join(' ')
+      typable_id = Termtter::Client.data_to_typable_id(s.id)
+      time = Time.parse(s.created_at).strftime(time_format) if time_format
+      post s.user.screen_name, msg_type, main_channel, [time, s.text, typable_id].compact.join(' ')
     end
   end
 
@@ -61,21 +84,46 @@ class TermtterIrcGateway < Net::IRC::Server::Session
     super
     post @prefix, JOIN, main_channel
     post server_name, MODE, main_channel, "+o", @prefix.nick
+    check_friends
     self.call(@@last_statuses || [], :update_friends_timeline)
   end
 
   def on_privmsg(m)
     target, message = *m.params
     Termtter::Client.call_commands('update ' + message)
+    post @prefix, TOPIC, main_channel, message
+  end
+
+  def log(str)
+    str.each_line do |line|
+      post server_name, NOTICE, main_channel, line
+    end
+  end
+
+  def check_friends
+    params = []
+    max_params_count = 3
+    Termtter::Client.friends.each do |name|
+      prefix = Prefix.new("#{name}!#{name}@localhost")
+      post prefix, JOIN, main_channel
+      params << prefix.nick
+      next if params.size < max_params_count
+
+      post server_name, MODE, main_channel, "+#{"v" * params.size}", *params
+      params = []
+    end
+    post server_name, MODE, main_channel, "+#{"v" * params.size}", *params unless params.empty?
   end
 end
 
 unless defined? IRC_SERVER
+  logger = Logger.new($stdout)
+  logger.level = config.plugins.irc_gw.logger_level
   IRC_SERVER = Net::IRC::Server.new(
     'localhost',
     config.plugins.irc_gw.port,
     TermtterIrcGateway,
-    :logger => Termtter::Client.logger
+    :logger => logger
   )
   Thread.start do
     IRC_SERVER.start

data/lib/plugins/itunes.rb

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+require 'appscript' or raise 'itunes plugin cannot run'
+
+config.plugins.itunes.set_default(:prefix, 'Listening now:')
+config.plugins.itunes.set_default(:suffix, '#iTunes #listening')
+config.plugins.itunes.set_default(
+  :format,
+  '<%=prefix%> <%=track_name%> (<%=time%>) <%=artist%> <%=album%> <%=suffix%>')
+
+module Termtter::Client
+  register_command :name => :listening_now, :aliases => [:ln],
+    :help => ['listening_now,ln', "Post the information of listening now."],
+    :exec_proc => lambda {|args|
+      begin
+        prefix     = config.plugins.itunes.prefix
+        track_name = Appscript.app('iTunes').current_track.name.get
+        artist     = Appscript.app('iTunes').current_track.artist.get
+        genre      = Appscript.app('iTunes').current_track.genre.get
+        time       = Appscript.app('iTunes').current_track.time.get
+        album      = Appscript.app('iTunes').current_track.album.get
+        suffix     = config.plugins.itunes.suffix
+        erbed_text = ERB.new(config.plugins.itunes.format).result(binding)
+        erbed_text.gsub!(/\s{2,}/, ' ')
+        if args.length > 0
+          erbed_text = args + ' ' + erbed_text
+        end
+        Termtter::API.twitter.update(erbed_text)
+        puts "=> " << erbed_text
+      rescue => e
+        p e
+      end
+    }
+end

data/lib/plugins/jakigan.rb

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+
+if RUBY_VERSION >= '1.9'
+  # nop
+elsif RUBY_VERSION >= '1.8.7'
+  class Array
+    alias sample choice
+  end
+else
+  class Array
+    def sample
+      at(rand(size))
+    end
+  end
+end
+
+module Termtter::Client
+  module Jakigan
+    TEMPLATES = [
+      # http://twitter.com/yugui/status/3086394151
+      '%s、か。ククク、その真の意図、邪気眼を持たぬ者には分かるまい。',
+      # http://twitter.com/Sixeight/statuses/3705973751
+      '僕の邪気眼は%sです。'
+    ]
+  end
+  register_macro(:jkg, "update #{Jakigan::TEMPLATES.sample}",
+    :help => ['jkg {MESSAGE}', 'update "{MESSAGE}+something jkg."'],
+    :alias => :jk
+  )
+end

data/lib/plugins/list_with_opts.rb

@@ -3,7 +3,7 @@ module Termtter::Client
   register_command(
     :name => :list, :aliases => [:l],
     :exec_proc => lambda {|arg|
-      _, options, user = */((?:\-[a-z][= ]\S+\s*)+)?(\w+)?/.match(arg)
+      _, options, user = */((?:\-[a-z][= ]\S+\s*)+)?(?:@?(\w+))?/.match(arg)
       params = {}
       options.scan(/(\-[a-z])[= ](\S+)/).each do |k,v|
         v = v.sub(/^['"]/,'').sub(/['"]$/,'')

data/lib/plugins/multi_output.rb

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+  @outputs = { }
+  class << self
+    def register_output(as, &block)
+      @outputs[as] = block
+    end
+
+    def delete_output(name)
+      @outputs.delete(name)
+    end
+
+    def puts message
+      @outputs.each_value do |block|
+        block.call(message)
+      end
+    end
+  end
+end
+
+module Termtter::Client
+  register_command(
+    :name => :outputs,
+    :exec_proc => lambda {|arg|
+      puts @outputs.keys.inspect
+    },
+    :help => ['outputs', 'Show outputs']
+    )
+
+  register_output(:stdout) do |msg|
+    STDOUT.puts(msg)
+  end
+end

data/lib/plugins/multi_post.rb

@@ -3,30 +3,45 @@
 module Termtter::Client
   class << self
     def wassr_update(text)
+      if text.match(/^(\d+)\s+(.+)$/) and
+          (s = Termtter::API.twitter.show($1) rescue nil)
+        tmp_text = "@#{s.user.screen_name} #{$2}"
+      else
+        tmp_text = text
+      end
+
       Net::HTTP.version_1_2
       req = Net::HTTP::Post.new("/statuses/update.json?")
       req.basic_auth config.plugins.wassr.username, config.plugins.wassr.password
       Net::HTTP.start('api.wassr.jp', 80) do |http|
-        res = http.request(req, "status=#{URI.escape(text)}&source=Termtter")
+        res = http.request(req, "status=#{URI.escape(tmp_text)}&source=Termtter")
       end
     end
   end
-end
 
-Termtter::Client.register_hook(
-  :name => :multi_post,
-  :points => [:modify_arg_for_update, :modify_arg_for_reply],
-  :exec_proc => lambda {|cmd, arg|
-    begin
-      wassr_arg = arg.gsub(/\d{10,}/, '')
-      Termtter::Client.wassr_update(wassr_arg.strip)
-    rescue
-      puts "RuntimeError: #{$!}"
-    end
+  register_hook(
+    :name => :multi_post,
+    :points => [:post_exec_update, :post_exec_reply, :post_exec_retweet],
+    :exec_proc => lambda {|cmd, arg, result|
+      prefix = config.plugins.stdout.typable_id_prefix
+      if result
+        wassr_arg = result
+      elsif arg.match(/(\$([a-z]{2}))/) and
+          s = Termtter::API.twitter.show(typable_id_to_data($2))
+        wassr_arg = arg.sub($1, "@" + s.user.screen_name)
+      else
+        wassr_arg = arg
+      end
+
+      begin
+        Termtter::Client.wassr_update(wassr_arg.strip)
+      rescue
+        puts "RuntimeError: #{$!}"
+      end
+    }
+  )
+end
 
-    return arg
-  }
-)
 
 # multi_post.rb
 # One post, multi update.

data/lib/plugins/open.rb

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+
+module Termtter
+  module Client
+    plug 'uri-open'
+    register_command(
+      :name => :open, :aliases => [:o],
+      :exec_proc => proc {|arg|
+        sid, n = arg.split(/ +/)
+        if sid.nil?
+          call_commands "uri-open"
+        else
+          target = (n || 1).to_i
+          target_uri = nil
+          ids = find_status_ids(sid) || []
+          ids.each do |id|
+            text = Termtter::API.twitter.show(id).text
+            URI.extract(text, %w[http https]).each_with_index do |uri, i|
+              print "#{i + 1}) #{uri}"
+              if i + 1 == target
+                target_uri = uri
+                print " <-"
+              end
+              puts ""
+            end
+          end
+          if target_uri
+            open_uri target_uri
+          end
+        end
+      },
+      :help => ['open,o', 'Open n-th URI in the message']
+    )
+  end
+end
+
+# usage
+# > open $dq
+# open 1st URI in $dq
+# > open $dq 2
+# open 2nd URI in $dq
+#
+# see also
+#   http://twitter.com/takiuchi

data/lib/plugins/paranoid.rb

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+module Termtter::Client
+  register_command(
+    :paranoid,
+    :help => ['paranoid message', 'Something like `update`'],
+    :alias => :pd) do |arg|
+      str = arg.gsub(/\w+/, '@\0').gsub(/\#@/, '#')
+      result = Termtter::API.twitter.update(str)
+      puts "paranoid'ed=> " << result.text
+    end
+end

data/lib/plugins/quick_exit.rb

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+#
+#config.plugins.alias.aliases = { :e => 'exit', :q => 'exit'}
+#
+ 
+module Termtter::Client
+  register_alias(:e, 'exit')
+  register_alias(:q, 'exit')
+end

data/lib/plugins/replace.rb

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+  def self.delete_and_replace(recent, pattern_reg, replace, global)
+    new_text =
+      if global
+        recent.text.gsub(pattern_reg, replace)
+      else
+        recent.text.sub(pattern_reg, replace)
+      end
+
+    param =
+      if recent.in_reply_to_status_id
+        {:in_reply_to_status_id => recent.in_reply_to_status_id}
+      else
+        {}
+      end
+
+    if new_text == recent.text
+      puts "It was not replaced."
+      raise Termtter::CommandCanceled
+    end
+
+    if /^y?$/i !~ Readline.readline("\"replace #{new_text}\" [Y/n] ", false)
+      puts 'canceled.'
+      raise Termtter::CommandCanceled
+    else
+      result = Termtter::API.twitter.remove_status(recent.id)
+      puts "deleted => #{result.text}"
+      result = Termtter::API.twitter.update(new_text, param)
+      puts "updated => #{result.text}"
+    end
+  end
+
+  register_command(:name => :replace,
+                   :aliases => [:typo],
+                   :help => ['replace,typo /PATTERN/REPLACE/',
+                             'Delete and replace most recent tweet.'],
+                   :exec_proc => lambda {|arg|
+                     recent =
+                     Termtter::API.twitter.user_timeline(config.user_name)[0]
+                     pattern, replace, mode =
+                     /^s?\/(.*?(?:(?!\\).))\/(.*)\/$/.match(arg).to_a.values_at(1, 2, 3)
+
+                     if pattern == ""
+                       puts "PATTERN is empty."
+                       raise Termtter::CommandCanceled
+                     end
+
+                     delete_and_replace(recent, Regexp.new(pattern, /i/.match(mode)),
+                                        replace, /g/ =~ mode)
+                   }
+                   )
+end

data/lib/plugins/saykanji.rb

@@ -40,7 +40,7 @@ def saykanji(text, say_speed)
     end
   }
   text_to_say = `echo #{text_wakati.join}|mecab -O yomi`
-  system "SayKana", "-s", "#{say_speed}", "#{text_to_say}"
+  system "SayKana -s #{say_speed} \"#{text_to_say}\" 2>/dev/null"
 end
 
 def say(who, text)