termtter 1.7.2 → 1.8.0

data/lib/plugins/growl.rb

@@ -8,6 +8,53 @@ require 'cgi'
 
 begin
   require 'ruby-growl'
+  if RUBY_VERSION >= "1.9" # Fix ruby-growl for multibyte chars if Ruby version is over 1.9
+    class Growl
+      private
+
+      def notification_packet(name, title, description, priority, sticky)
+        flags = 0
+        data = []
+
+        packet = [
+          GROWL_PROTOCOL_VERSION,
+          GROWL_TYPE_NOTIFICATION,
+        ]
+
+        flags = 0
+        flags |= ((0x7 & priority) << 1) # 3 bits for priority
+        flags |= 1 if sticky # 1 bit for sticky
+
+        packet << flags
+        packet << name.bytesize
+        packet << title.bytesize
+        packet << description.bytesize
+        packet << @app_name.bytesize
+
+        data << name
+        data << title
+        data << description
+        data << @app_name
+
+        packet << data.join
+        packet = packet.pack GNN_FORMAT
+
+        checksum = MD5.new packet
+        checksum.update @password unless @password.nil?
+
+        packet << checksum.digest
+
+        return packet.force_encoding('utf-8')
+      end
+
+      def send(packet)
+        set_sndbuf packet.bytesize
+        @socket.send packet, 0
+        @socket.flush
+      end
+    end
+  end
+
   growl = Growl.new "localhost", "termtter", ["update_friends_timeline"]
 rescue LoadError
   growl = nil

data/lib/plugins/gyazo.rb

@@ -3,26 +3,26 @@ require 'net/http'
 def post_gyazo
   browser_cmd = 'firefox'
   gyazo_url = ""
-  
+
   idfile = ENV['HOME'] + "/.gyazo.id"
-  
-  id = ''
-  if File.exist?(idfile) then
+
+  id = nil
+  if File.exist?(idfile)
     id = File.read(idfile).chomp
   else
     id = Time.new.strftime("%Y%m%d%H%M%S")
     File.open(idfile,"w").print(id+"\n")
   end
-  
+
   tmpfile = "/tmp/image_upload#{$$}.png"
-  
-  system "import #{tmpfile}"
-  
+
+  system import, tmpfile
+
   imagedata = File.read(tmpfile)
   File.delete(tmpfile)
-  
+
   boundary = '----BOUNDARYBOUNDARY----'
-  
+
   data = <<EOF
 --#{boundary}\r
 content-disposition: form-data; name="id"\r
@@ -36,19 +36,19 @@ content-disposition: form-data; name="imagedata"\r
 --#{boundary}--\r
 EOF
 
-  header ={
+  header = {
     'Content-Length' => data.length.to_s,
     'Content-type' => "multipart/form-data; boundary=#{boundary}"
   }
 
-  Net::HTTP.start("gyazo.com",80){|http|
-    res = http.post("/upload.cgi",data,header)
+  Net::HTTP.start("gyazo.com", 80){|http|
+    res = http.post("/upload.cgi", data, header)
     url = res.response.to_ary[1]
     puts url
     system "#{browser_cmd} #{url}"
     gyazo_url = url
   }
-  return gyazo_url
+  gyazo_url
 end
 
 Termtter::Client.register_command(
@@ -63,10 +63,9 @@ Termtter::Client.register_command(
 )
 
 # gyazo:
-#   Capture an arbitary desktop area and upload the image to gyazo.com, 
+#   Capture an arbitary desktop area and upload the image to gyazo.com,
 #   open it up with the browser you specified. Then, tweet the link with
-#   a message. 
-# 
+#   a message.
 #   You need ImageMagick tools and a web browser(default:Firefox, as you
 #   can see.)
 #
@@ -75,4 +74,3 @@ Termtter::Client.register_command(
 #
 # example:
 #   gyazo What a lame dialogue message! (capture process starts...)
-#

data/lib/plugins/hi.rb

@@ -1,15 +1,36 @@
 # -*- 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
+  {
+    :english => ['hi', 'hey', 'hello', 'How are you?', "How's going?"],
+    :spanish => ['¡Hola!', '¿Cómo estás?'],
+    :german => ['Guten Tag!'],
+  }.each do |language, greetings|
+    greetings.each do |greeting|
+      # '¿Cómo estás?' -> 'como_estas'
+      # MEMO:
+      #   command_name = greeting.tr('áó', 'ao').scan(/\w+/).join('_').downcase
+      # works only on ruby 1.9
+      command_name = greeting.
+        gsub('á', 'a').
+        gsub('ó', 'o').
+        scan(/[a-zA-Z]+/).
+        join('_').
+        downcase
+      register_command(
+        :name => command_name,
+        :author => 'ujihisa',
+        :help => ["#{command_name} [(Optinal) USER]", "Post a greeting message in #{language.to_s.capitalize}"],
+        :exec_proc => lambda {|arg|
+        result =
+          if arg.empty?
+            Termtter::API.twitter.update(greeting)
+          else
+            name = normalize_as_user_name(arg)
+            Termtter::API.twitter.update("@#{name} #{greeting}")
+          end
+        puts "=> " << result.text
+        })
     end
   end
 end

data/lib/plugins/http_server.rb

@@ -47,7 +47,7 @@ module Termtter::Client
 
   @http_server.mount_proc('/reload.html') do |req, res|
     # MEMO: ブラウザで画面を二つ開いてるとデータの取り合いになっておかしな感じになる。。。
-    res['Content-Type'] = 'text/javascript; charset=utf-8';
+    res['Content-Type'] = 'text/javascript; charset=utf-8'
     res.body = @http_server_statuses_store.to_json
     @http_server_statuses_store.clear
   end
@@ -67,7 +67,7 @@ module Termtter::Client
       begin
         command = req.path.sub(/^\//, '')
         execute(command)
-        res['Content-Type'] = 'text/javascript; charset=utf-8';
+        res['Content-Type'] = 'text/javascript; charset=utf-8'
         res.body = @http_server_output
       rescue Termtter::CommandNotFound => e
         res.status = 404
@@ -80,3 +80,4 @@ module Termtter::Client
     @http_server.start
   end
 end
+# vim: fileencoding=utf8

data/lib/plugins/irc_gw.rb

@@ -1,10 +1,14 @@
 # -*- coding: utf-8 -*-
 
 require 'net/irc'
+require 'set'
 
 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)
+config.plugins.irc_gw.set_default(:command_regexps, [/^(.+): *(.*)/])
+
+Termtter::Client.plug 'multi_output'
 
 module Termtter::Client
   class << self
@@ -15,12 +19,12 @@ module Termtter::Client
       begin
         puts "collecting friends (#{frinends.length})"
         last = Termtter::API::twitter.friends(user_name, :cursor => last ? last.next_cursor : -1)
-	frinends += last.users
+        frinends += last.users
       rescue Timeout::Error, StandardError # XXX
         break
       end until last.next_cursor == 0
       puts "You have #{frinends.length} friends."
-      frinends.map(&:screen_name)
+      Set.new(frinends.map(&:screen_name))
     end
   end
 end
@@ -58,17 +62,40 @@ class TermtterIrcGateway < Net::IRC::Server::Session
   def initialize(*args)
     super
     @@listners << self
+    @friends = Set.new
+    @commands = []
+
+    Termtter::Client.register_hook(:collect_user_names_for_irc_gw, :point => :pre_filter) do |statuses, event|
+      new_users = []
+      statuses.each do |s|
+        screen_name = s.user.screen_name
+        next if screen_name == @user # XXX
+        next if @friends.include? screen_name
+        @friends << screen_name
+        new_users << screen_name
+      end
+      join_members(new_users)
+    end
+
+    Termtter::Client.register_command(
+      :name => :collect_friends,
+      :help => 'Collect friends for IRC.',
+      :exec => lambda {|arg|
+        sync_friends
+      })
+
+    Termtter::Client.register_hook(:collect_commands_for_irc_gw, :point => :post_command) do |text|
+      sync_commands if text =~ /plug/
+    end
   end
 
   def call(statuses, event)
-    msg_type =
-      case event
-      when :update_friends_timeline
-        PRIVMSG
-      else
-        time_format = Termtter::Client.time_format_for statuses
-        NOTICE
-      end
+    if event == :update_friends_timeline
+      msg_type = PRIVMSG
+    else
+      time_format = Termtter::Client.time_format_for statuses
+      msg_type = NOTICE
+    end
     statuses.each do |s|
       typable_id = Termtter::Client.data_to_typable_id(s.id)
       time = Time.parse(s.created_at).strftime(time_format) if time_format
@@ -88,9 +115,10 @@ class TermtterIrcGateway < Net::IRC::Server::Session
 
   def on_user(m)
     super
+    @user = m.params.first
     post @prefix, JOIN, main_channel
     post server_name, MODE, main_channel, "+o", @prefix.nick
-    check_friends
+    sync_commands
     self.call(@@last_statuses || [], :update_friends_timeline)
   end
 
@@ -101,10 +129,20 @@ class TermtterIrcGateway < Net::IRC::Server::Session
       return unless Termtter::Client.find_command(termtter_command)
       post '#termtter', NOTICE, main_channel, '> ' + termtter_command
       Termtter::Client.execute(termtter_command)
-    else
-      Termtter::Client.execute('update ' + message)
-      post @prefix, TOPIC, main_channel, message
+      return
+    end
+    config.plugins.irc_gw.command_regexps and
+    config.plugins.irc_gw.command_regexps.each do |rule|
+      if message =~ rule
+        command = message.scan(rule).first.join(' ')
+        next unless Termtter::Client.find_command(command)
+        post '#termtter', NOTICE, main_channel, '> ' + command
+        Termtter::Client.execute(command)
+        return
+      end
     end
+    Termtter::Client.execute('update ' + message)
+    post @prefix, TOPIC, main_channel, message
   rescue Exception => e
     post '#termtter', NOTICE, main_channel, "#{e.class.to_s}: #{e.message}"
     Termtter::Client.handle_error(e)
@@ -116,11 +154,26 @@ class TermtterIrcGateway < Net::IRC::Server::Session
     end
   end
 
-  def check_friends
+  def sync_friends
+    previous_friends = @friends
+    new_friends = Termtter::Client.following_friends
+    diff = new_friends - previous_friends
+    join_members(diff)
+    @friends += diff
+  end
+
+  def sync_commands
+    previous_commands = @commands
+    new_commands = Termtter::Client.commands.keys.map{|s| s.to_s.split(' ')}.flatten.uniq
+    join_members(new_commands - previous_commands)
+    @commands = new_commands
+  end
+
+  def join_members(members)
     params = []
     max_params_count = 3
-    Termtter::Client.following_friends.each do |name|
-      prefix = Prefix.new("#{name}!#{name}@localhost")
+    members.each do |member|
+      prefix = Prefix.new("#{member}!#{member}@localhost")
       post prefix, JOIN, main_channel
       params << prefix.nick
       next if params.size < max_params_count
@@ -130,6 +183,7 @@ class TermtterIrcGateway < Net::IRC::Server::Session
     end
     post server_name, MODE, main_channel, "+#{"v" * params.size}", *params unless params.empty?
   end
+
 end
 
 unless defined? IRC_SERVER

data/lib/plugins/line.rb

@@ -0,0 +1,10 @@
+Termtter::Client.register_command(
+  :name => :line,
+  :help => 'Draws a line without any side effect on Twitter.',
+  #:aliases => [:l],
+  :exec_proc => lambda {|arg|
+    window_width = (ENV['COLUMNS'] || `stty size`[/\d+\s+(\d+)/, 1]).to_i
+    text = '<dark>' + (' ' * window_width) + '</dark>'
+    puts TermColor.parse(text)
+  }
+)

data/lib/plugins/linefeed.rb

@@ -17,9 +17,14 @@ Termtter::Client.register_hook(
         cs = line.unpack 'U*'
         while cs.size > 0
           sz = 0
-          l = cs.take_while{|c| sz += c < 0x100 ? 1 : 2; sz < width}
+          l = cs.take_while do |c|
+            sz2 = c < 0x100 ? 1 : 2
+            sz += sz2
+            sz < width || (sz2 == 1 && c != 32)
+          end
           ocs += l
           cs = cs.drop(l.size)
+          cs = cs.drop(1) if cs[0] == 32
           ocs += [0x0a]
         end
       end

data/lib/plugins/list_switch.rb

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+
+module Termtter
+  module Client
+    register_command('list switch',
+      :help => ["list switch LISTNAME", "Switch to the list"]
+    ) do |arg|
+      @since_id = nil
+      Termtter::Plugins::ListSwitch.list_switch(arg)
+    end
+
+    # TODO: コマンド名が微妙...
+    register_command('list switch restore',
+      :help => ["list switch restore LISTNAME", "Restore to switch list "]
+    ) do |arg|
+      Termtter::Plugins::ListSwitch.restore
+    end
+  end
+
+  class RubytterProxy
+    alias_method :call_rubytter_without_list_switch, :call_rubytter
+    def call_rubytter_with_list_switch(method, *args, &block)
+      Termtter::Plugins::ListSwitch.call_rubytter(self, method, *args, &block)
+    end
+    alias_method :call_rubytter, :call_rubytter_with_list_switch
+  end
+
+  module Plugins
+    module ListSwitch
+      extend self
+
+      attr_reader :active, :list_name, :list_user_name, :list_user_id
+
+      def call_rubytter(rubytter_proxy, method, *args, &block)
+        if active
+          case method
+          when :home_timeline
+            # => list_statuses(user_name, slug, options)
+            method, args = :list_statuses, [list_user_name, list_name, *args]
+          when :follow
+            # => add_member_to_list(slug, user.id)
+            method, args = :add_member_to_list, [list_name, *args]
+          when :leave
+            # => remove_member_from_list(slug, user.id)
+            method, args = :remove_member_from_list, [list_name, *args]
+          end
+        end
+        rubytter_proxy.call_rubytter_without_list_switch(method, *args, &block)
+      end
+
+      def list_switch(full_name)
+        @active = true
+        @list_user_name, @list_name = split_list_name(full_name)
+        user = Termtter::API.twitter.cached_user(list_user_name) ||
+                  Termtter::API.twitter.user(list_user_name)
+        @list_user_id = user.id
+        # TODO: やっつけなのでちゃんとやる
+        config.prompt = full_name + '> '
+      end
+
+      def restore
+        @active = false
+        # TODO: やっつけなのでちゃんとやる
+        config.prompt = '> '
+      end
+
+      def split_list_name(list_name)
+        if /([^\/]+)\/([^\/]+)/ =~ list_name
+          [Termtter::Client.normalize_as_user_name($1), $2]
+        else
+          [config.user_name, $2]
+        end
+      end
+    end
+  end
+end