draftcode-termtter 1.0.8

data/lib/plugins/reverse.rb

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+  add_filter do |statuses, _|
+    statuses.map do |s|
+      s.text = s.text.split(//).reverse.to_s
+      s
+    end
+  end
+end
+
+# filter-reverse.rb
+#   reverse texts

data/lib/plugins/say.rb

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+
+raise 'say.rb runs only in OSX Leopard' if /darwin9/ !~ RUBY_PLATFORM
+
+# say :: String -> String -> IO ()
+def say(who, what)
+  voices = %w(Alex Alex Bruce Fred Ralph Agnes Kathy Vicki)
+  voice = voices[who.hash % voices.size]
+  system 'say', '-v', voice, what
+end
+
+module Termtter::Client
+  register_hook(
+    :name => :say,
+    :points => [:output],
+    :exec_proc => lambda {|statuses, event|
+      statuses.reverse.each do |s|
+        text_without_uri = s[:text].gsub(%r|https?://[^\s]+|, 'U.R.I.')
+        say s[:screen_name], text_without_uri
+      end
+    }
+  )
+end
+
+# KNOWN BUG:
+# * exit or <C-c> doen't work quickly.

data/lib/plugins/scrape.rb

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+
+  def self.scrape_members(members)
+    statuses = []
+    members.each_with_index do |member, index|
+      puts "member #{index+1}/#{members.size} #{member}"
+      statuses += Termtter::API.twitter.user_timeline(member)
+    end
+    statuses
+  end    
+  
+  def self.scrape_group(group)
+    members = config.plugins.group.groups[group] || []
+    scrape_members(members)
+  end
+
+  register_command(
+                   :name => :scrape_group,
+                   :exec_proc => lambda{ |args|
+                     groups = args.split(' ').map{|g| g.to_sym}
+                     if groups.include? :all
+                       groups = config.plugins.group.groups.keys
+                       puts "get all groups..."
+                     end
+                     members = []
+                     groups.each do |group|
+                       members += config.plugins.group.groups[group]
+                     end
+                     statuses = scrape_members(members.uniq.compact.sort)
+                     output(statuses_to_hash(statuses), :pre_filter)
+                   },
+                   :completion_proc => lambda {|cmd, args|
+                     arg = args.split(' ').last
+                     prefix = args.split(' ')[0..-2].join(' ')
+                     find_group_candidates arg, "#{cmd} #{prefix} %s"
+                   },
+                   :help => ['scrape_group GROUPNAME(S)', 'Get the group timeline']
+                   )
+end

data/lib/plugins/screen-notify.rb

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+
+config.screen_notify.set_default(:format, "[termtter] %s")
+
+module Termtter::Client
+  register_hook(
+    :name => :screen_notify,
+    :points => [:output],
+    :exec_proc => lambda{|statuses, event|
+      return unless event = :update_friends_timeline
+      Thread.new(statuses) do |ss|
+        ss.each do |s|
+          msg = config.screen_notify.format % s.user.screen_name
+          system 'screen', '-X', 'eval', "bell_msg '#{msg}'", 'bell'
+          sleep 1
+        end
+      end
+    }
+  )
+end

data/lib/plugins/screen.rb

@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+
+module Termtter
+  module Plugin
+    module Screen
+      def self.set_title(title)
+        print "\033k#{title}\033\\\n"
+      end
+    end
+  end
+end
+
+# Add below to your ~/.termtter
+#
+# require 'plugins/yonda'
+# require 'plugins/screen'
+# module Termtter::Client
+#   register_hook(:name => :screen,
+#                 :points => [:post_exec__update_timeline, :plugin_yonda_yonda, :post_exec_yonda],
+#                 :exec_proc => lambda { |cmd, arg, result|
+#                   Termtter::Plugin::Screen::set_title("termtter(#{public_storage[:unread_count]})")
+#                 }
+#   )
+# end

data/lib/plugins/shell.rb

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+  register_command :name => :shell, :aliases => [:sh],
+    :help => ['shell,sh', 'Start your shell'],
+    :exec_proc => lambda {|args|
+      begin
+        pause
+        system ENV['SHELL'] || ENV['COMSPEC']
+      ensure
+        resume
+      end
+    }
+end

data/lib/plugins/sl.rb

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+
+module Termtter
+
+  module Client
+
+    public_storage[:current] = ''
+    public_storage[:orig_prompt] = config.prompt
+    config.prompt = "~/ #{public_storage[:orig_prompt]}"
+
+    register_command(
+      :name      => :sl, :aliases => [],
+      :exec_proc => lambda {|arg| system("sl") },
+      :help      => ['sl', 'The train pass in front of your screen']
+    )
+
+    register_command(
+      :name      => :pwd, :aliases => [],
+      :exec_proc => lambda {|arg| public_storage[:current] },
+      :help      => ['pwd', 'Show current direcroty']
+    )
+
+    register_command(
+      :name => :ls, :aliases => [],
+      :exec_proc => lambda {|arg|
+        call_commands("list #{arg.empty? ? public_storage[:current] : arg}")
+      },
+      :completion_proc => lambda {|cmd, args|
+        find_user_candidates args, "#{cmd} %s"
+      },
+      :help => ['ls', 'Show list in current directory']
+    )
+
+    register_command(
+      :name => :cd, :aliases => [],
+      :exec_proc => lambda {|arg|
+        public_storage[:current] =
+          (arg.nil? || /\~/ =~ arg) ? '' : arg
+        config.prompt = "~/#{public_storage[:current]} #{public_storage[:orig_prompt]}"
+      },
+      :completion_proc => lambda {|cmd, args|
+        find_user_candidates args, "#{cmd} %s"
+      },
+      :help => ['cd USER', 'Change current directory']
+    )
+  end
+end
+

data/lib/plugins/spam.rb

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+
+message = '*super spam time*'
+Termtter::API.twitter.update(message)
+puts "=> #{message}"
+
+Termtter::Client.register_hook(
+  :name => :span,
+  :point => /^pre_exec/,
+  :exec => lambda{|*arg|
+    text = arg.join(' ')
+    Termtter::API.twitter.update(text)
+    puts "=> #{text}"
+    raise Termtter::CommandCanceled
+  }
+)

data/lib/plugins/standard_plugins.rb

@@ -0,0 +1,525 @@
+# -*- coding: utf-8 -*-
+
+require 'erb'
+require 'set'
+
+config.plugins.standard.set_default(
+ :limit_format,
+ '<<%=remaining_color%>><%=limit.remaining_hits%></<%=remaining_color%>>/<%=limit.hourly_limit%> until <%=limit.reset_time%> (<%=remaining_time%> remaining)')
+
+module Termtter::Client
+
+  register_command(
+    :name => :update, :aliases => [:u],
+    :exec_proc => lambda {|arg|
+      unless arg.empty?
+        Termtter::API.twitter.update(arg)
+        puts "=> #{arg}"
+      end
+    },
+    :completion_proc => lambda {|cmd, args|
+      if /(.*)@([^\s]*)$/ =~ args
+        find_user_candidates $2, "#{cmd} #{$1}@%s"
+      end
+    },
+    :help => ["update,u TEXT", "Post a new message"]
+  )
+
+  direct_message_struct = Struct.new(:id, :text, :user, :created_at)
+  direct_message_struct.class_eval do
+    def method_missing(*args, &block)
+      nil
+    end
+  end
+  register_command(
+    :name => :direct, :aliases => [:d],
+    :exec_proc => lambda {|arg|
+      case arg
+      when /^([^\s]+)\s+(.*)\s*$/
+        user, text = $1, $2
+        Termtter::API.twitter.direct_message(user, text)
+        puts "=> to:#{user} message:#{text}"
+      when 'list'
+        output(
+          Termtter::API.twitter.direct_messages.map { |d|
+            direct_message_struct.new(d.id, "#{d.text} => #{d.recipient_screen_name}", d.sender, d.created_at)
+          },
+          :direct_messages
+        )
+      when 'sent_list'
+        output(
+          Termtter::API.twitter.sent_direct_messages.map { |d|
+            direct_message_struct.new(d.id, "#{d.text} => #{d.recipient_screen_name}", d.sender, d.created_at)
+          },
+          :direct_messages
+        )
+      end
+    },
+    :completion_proc => lambda {|cmd, arg|
+      find_user_candidates(arg, "#{cmd} %s") + 
+        ["list", "sent_list"].grep(/^#{Regexp.quote(arg)}/).map { |i| "#{cmd} #{i}" }
+    },
+    :help => [
+      ["direct,d USERNAME TEXT", "Send direct message"],
+      ["direct,d list", 'List direct messages'],
+      ["direct,d sent_list", 'List sent direct messages']
+    ]
+  )
+
+  register_command(
+    :name => :profile, :aliases => [:p],
+    :exec_proc => lambda {|arg|
+      user = Termtter::API.twitter.user(arg.strip)
+      attrs = %w[ name screen_name url description profile_image_url location protected following
+        friends_count followers_count statuses_count favourites_count
+        id time_zone created_at utc_offset notifications
+      ]
+      label_width = attrs.map(&:size).max
+      attrs.each do |attr|
+        value = user.__send__(attr.to_sym)
+        puts "#{attr.gsub('_', ' ').rjust(label_width)}: #{value}"
+      end
+    },
+    :completion_proc => lambda {|cmd, arg|
+      find_user_candidates arg, "#{cmd} %s"
+    },
+    :help => ["profile,p USERNAME", "Show user's profile"]
+  )
+
+  register_command(
+    :name => :followers,
+    :exec_proc => lambda {|arg|
+      user_name = arg.strip
+      user_name = config.user_name if user_name.empty?
+
+      followers = []
+      page = 0
+      begin
+        followers += tmp = Termtter::API.twitter.followers(user_name, :page => page+=1)
+      end until tmp.empty?
+      Termtter::Client.public_storage[:followers] = followers
+      public_storage[:users] += followers.map(&:screen_name)
+      puts followers.map(&:screen_name).join(' ')
+    },
+    :completion_proc => lambda {|cmd, arg|
+      find_user_candidates arg, "#{cmd} %s"
+    },
+    :help => ["followers", "Show followers"]
+  )
+
+  register_command(
+    :name => :list, :aliases => [:l],
+    :exec_proc => lambda {|arg|
+      if arg.empty?
+        event = :list_friends_timeline
+        statuses = Termtter::API.twitter.friends_timeline
+      else
+        event = :list_user_timeline
+        statuses = Termtter::API.twitter.user_timeline(arg)
+      end
+      output(statuses, event)
+    },
+    :completion_proc => lambda {|cmd, arg|
+      find_user_candidates arg, "#{cmd} %s"
+    },
+    :help => ["list,l [USERNAME]", "List the posts"]
+  )
+
+  public_storage[:search_keywords] = Set.new
+  register_command(
+    :name => :search, :aliases => [:s],
+    :exec_proc => lambda {|arg|
+      statuses = Termtter::API.twitter.search(arg)
+      public_storage[:search_keywords] << arg
+      output(statuses, :search)
+    },
+    :completion_proc => lambda {|cmd, arg|
+      public_storage[:search_keywords].grep(/#{Regexp.quote(arg)}/).map { |i| "#{cmd} #{i}" }
+    },
+    :help => ["search,s TEXT", "Search for Twitter"]
+  )
+
+  register_command(
+    :name => :replies, :aliases => [:r],
+    :exec_proc => lambda {|arg|
+      output(Termtter::API.twitter.replies, :replies)
+    },
+    :help => ["replies,r", "List the most recent @replies for the authenticating user"]
+  )
+
+  register_command(
+    :name => :show,
+    :exec_proc => lambda {|arg|
+      id = arg.gsub(/.*:\s*/, '')
+      output([Termtter::API.twitter.show(id)], :show)
+    },
+    :completion_proc => lambda {|cmd, arg|
+      case arg
+      when /(\w+):\s*(\d+)\s*$/
+        find_status_ids($2).map{|id| "#{cmd} #{$1}: #{id}"}
+      else
+        users = find_users(arg)
+        unless users.empty?
+          users.map {|user| "#{cmd} #{user}:"}
+        else
+          find_status_ids(arg).map {|id| "#{cmd} #{id}"}
+        end
+      end
+    },
+    :help => ["show ID", "Show a single status"]
+  )
+
+  register_command(
+    :name => :shows,
+    :exec_proc => lambda {|arg|
+      id = arg.gsub(/.*:\s*/, '')
+      # TODO: Implement
+      #output([Termtter::API.twitter.show(id)], :show)
+      puts "Not implemented yet."
+    },
+    :completion_proc => get_command(:show).completion_proc
+  )
+
+  register_command(
+    :name => :follow, :aliases => [],
+    :exec_proc => lambda {|args|
+      args.split(' ').each do |arg|
+        if /^(\w+)/ =~ arg
+          res = Termtter::API::twitter.follow($1.strip)
+        end
+      end
+    },
+    :completion_proc => lambda {|cmd, args|
+      find_user_candidates args, "#{cmd} %s"
+    },
+    :help => ['follow USER', 'Follow user']
+  )
+
+  register_command(
+    :name => :leave, :aliases => [],
+    :exec_proc => lambda {|args|
+      args.split(' ').each do |arg|
+        if /^(\w+)/ =~ arg
+          res = Termtter::API::twitter.leave($1.strip)
+        end
+      end
+    },
+    :completion_proc => lambda {|cmd, args|
+      find_user_candidates args, "#{cmd} %s"
+    },
+    :help => ['leave USER', 'Leave user']
+  )
+
+  register_command(
+    :name => :favorite, :aliases => [:fav],
+    :exec_proc => lambda {|arg|
+      id = 0
+      case arg
+      when /^\d+/
+        id = arg.to_i
+      when /^@([A-Za-z0-9_]+)/
+        user = $1
+        statuses = Termtter::API.twitter.user_timeline(user)
+        return if statuses.empty?
+        id = statuses[0].id
+      when /^\/(.*)$/
+        word = $1
+        raise "Not implemented yet."
+      else
+        return
+      end
+
+      r = Termtter::API.twitter.favorite id
+      puts "Favorited status ##{r.id} on user @#{r.user.screen_name} #{r.text}"
+    },
+    :completion_proc => lambda {|cmd, arg|
+      case arg
+      when /@(.*)/
+        find_user_candidates $1, "#{cmd} @%s"
+      when /(\d+)/
+        find_status_ids(arg).map{|id| "#{cmd} #{id}"}
+      else
+        %w(favorite).grep(/^#{Regexp.quote arg}/)
+      end
+    },
+    :help => ['favorite,fav (ID|@USER|/WORD)', 'Favorite a status']
+  )
+
+  def self.show_settings(conf, level = 0)
+    conf.__values__.each do |k, v|
+      if v.instance_of? Termtter::Config
+        puts "#{k}:"
+        show_settings v, level + 1
+      else
+        print '  ' * level
+        puts "#{k} = #{v.nil? ? 'nil' : v.inspect}"
+      end
+    end
+  end
+
+  register_command(
+    :name => :settings, :aliases => [:set],
+    :exec_proc => lambda {|_|
+      show_settings config
+    },
+    :help => ['settings,set', 'Show your settings']
+  )
+
+  # TODO: Change colors when remaining_hits is low.
+  # TODO: Simmulate remaining_hits.
+  register_command(
+    :name => :limit, :aliases => [:lm],
+    :exec_proc => lambda {|arg|
+      limit = Termtter::API.twitter.limit_status
+      remaining_time = "%dmin %dsec" % (Time.parse(limit.reset_time) - Time.now).divmod(60)
+      remaining_color =
+        case limit.remaining_hits / limit.hourly_limit.to_f
+        when 0.2..0.4 then :yellow
+        when 0..0.2   then :red
+        else               :green
+        end
+      erbed_text = ERB.new(config.plugins.standard.limit_format).result(binding)
+      puts TermColor.parse(erbed_text)
+    },
+    :help => ["limit,lm", "Show the API limit status"]
+  )
+
+  register_command(
+    :name => :pause,
+    :exec_proc => lambda {|arg| pause},
+    :help => ["pause", "Pause updating"]
+  )
+
+  register_command(
+    :name => :resume,
+    :exec_proc => lambda {|arg| resume},
+    :help => ["resume", "Resume updating"]
+  )
+
+  register_command(
+    :name => :exit, :aliases => [:e, :q, :quit],
+    :exec_proc => lambda {|arg| exit},
+    :help => ['exit,e', 'Exit']
+  )
+
+  register_command(
+    :name => :help, :aliases => [:h],
+    :exec_proc => lambda {|arg|
+      helps = []
+      @commands.map do |name, command|
+        next unless command.help
+        case command.help[0]
+        when String
+          helps << command.help
+        when Array
+          helps += command.help
+        end
+      end
+      helps.compact!
+      puts formatted_help(helps)
+    },
+    :help => ["help,h", "Print this help message"]
+  )
+
+  def self.formatted_help(helps)
+    helps = helps.sort_by {|help| help[0] }
+    width = helps.map {|n, _| n.size }.max
+    space = 3
+    helps.map {|name, desc|
+      name.to_s.ljust(width + space) + desc.to_s
+    }.join("\n")
+  end
+
+  # completion for standard commands
+
+  public_storage[:users] ||= Set.new
+  public_storage[:status_ids] ||= Set.new
+  public_storage[:last_status_id] ||= Hash.new
+
+  register_hook(
+    :name => :for_completion,
+    :points => [:pre_filter],
+    :exec_proc => lambda {|statuses, event|
+      statuses.each do |s|
+        public_storage[:users].add(s.user.screen_name)
+        public_storage[:users] += s.text.scan(/@([a-zA-Z_0-9]*)/).flatten
+        public_storage[:status_ids].add(s.id)
+        public_storage[:status_ids].add(s.in_reply_to_status_id) if s.in_reply_to_status_id
+        public_storage[:last_status_id].store(s.user.screen_name, s.id)
+      end
+    }
+  )
+
+  public_storage[:plugins] = (Dir["#{File.dirname(__FILE__)}/*.rb"] + Dir["#{Termtter::CONF_DIR}/plugins/*.rb"]).map do |f|
+    f.match(%r|([^/]+).rb$|)[1]
+  end
+
+  register_command(
+    :name      => :plugin,
+    :exec_proc => lambda {|arg|
+      if arg.empty?
+        puts 'Should specify plugin name.'
+        return
+      end
+      begin
+        result = plugin arg
+      rescue LoadError
+      ensure
+        puts "=> #{result.inspect}"
+      end
+    },
+    :completion_proc => lambda {|cmd, args|
+      find_user_candidates args, "#{cmd} %s"
+      unless args.empty?
+        find_plugin_candidates args, "#{cmd} %s"
+      else
+        public_storage[:plugins].sort
+      end
+    },
+    :help      => ['plugin FILE', 'Load a plugin']
+  )
+
+  register_command(
+    :name      => :plugins,
+    :exec_proc => lambda {|arg|
+      puts public_storage[:plugins].sort.join("\n")
+    },
+    :help      => ['plugins', 'Show list of plugins']
+  )
+
+  register_command(
+    :name => :reply,
+    :aliases => [:re],
+    :exec_proc => lambda {|arg|
+      case arg
+      when /^\s*(?:list|ls)\s*(?:\s+(\w+))?\s*$/
+        public_storage[:log4re] = if $1
+                                    public_storage[:log].
+                                      select{|l| l.user.screen_name == $1}.
+                                      sort {|a,b| a.id <=> b.id}
+                                  else
+                                    public_storage[:log].sort {|a,b| a.id <=> b.id}
+                                  end
+        public_storage[:log4re].each_with_index do |s, i|
+          puts "#{i}: #{s.user.screen_name}: #{s.text}"
+        end
+      when /^\s*(?:up(?:date)?)\s+(\d+)\s+(.+)$/
+        id   = public_storage[:log4re][$1.to_i].id
+        text = $2
+        user = public_storage[:log4re][$1.to_i].user
+        update_with_user_and_id(text, user.screen_name, id) if user
+        public_storage.delete :log4re
+      when /^\s*(\d+)\s+(.+)$/
+        id, text = $1, $2
+        user = public_storage[:log].select {|l| l.id == id.to_i }.first.user
+        update_with_user_and_id(text, user.screen_name, id) if user
+      when /^\s*@(\w+)/
+        in_reply_to_status_id = Termtter::API.twitter.user($1).status.id rescue nil
+        params = in_reply_to_status_id ? {:in_reply_to_status_id => in_reply_to_status_id} : {}
+        Termtter::API.twitter.update(arg, params)
+        puts "=> #{arg}"
+      end
+    },
+    :completion_proc => lambda {|cmd, arg|
+      case arg
+      when /(.*)@([^\s]*)$/
+        find_user_candidates $2, "#{cmd} #{$1}@%s"
+      when /(\d+)/
+        find_status_ids(arg).map{|id| "#{cmd} #{id}"}
+      end
+    },
+    :help => ["reply,re @USERNAME or STATUS_ID", "Send a reply"]
+  )
+
+  register_command(
+    :name => :redo,
+    :aliases => [:"."],
+    :exec_proc => lambda {|arg|
+      break if Readline::HISTORY.length < 2
+      i = Readline::HISTORY.length - 2
+      input = ""
+      begin
+        input = Readline::HISTORY[i]
+        i -= 1
+        return if i <= 0
+      end while input == "redo" or input == "."
+      begin
+        Termtter::Client.call_commands(input)
+      rescue CommandNotFound => e
+        warn "Unknown command \"#{e}\""
+        warn 'Enter "help" for instructions'
+      rescue => e
+        handle_error e
+      end
+    },
+    :help => ["redo,.", "Execute previous command"]
+  )
+
+  def self.update_with_user_and_id(text, username, id)
+    text = ERB.new("@#{username} #{text}").result(binding).gsub(/\n/, ' ')
+    result = Termtter::API.twitter.update(text, {'in_reply_to_status_id' => id})
+    puts "=> #{text}"
+    result
+  end
+
+=begin
+  = Termtter reply command
+  == Usage
+  === list
+  * ステータスリストを連番と一緒に出す。
+  > reply [list|ls]
+  0: foo: foo's message
+  1: bar: bar's message
+  ..
+
+  * ユーザ指定してリスト作成。
+  > reply [list|ls] foo
+  0: foo: foo's message0
+  1: foo: foo's message1
+
+  === reply
+  メッセージ送信の際、@usernameが自動的に付与される。
+
+  * status_idを自分で入力してreply送信
+  > reply 1234567890 message to status_id
+  => @foo message to status_id (reply to 1234567890)
+
+  * reply listコマンドで出したステータス番号に対してreply送信
+  > reply up 0 message to status no
+  => @foo message to status_no
+
+  * 対象ユーザの最後の発言に対してreply
+  > reply @foo message to foo
+  => @foo message to foo
+
+  == Todo
+  * 英語で説明 => ヘルプを設定する
+  * リファクタ
+  * 補完
+  * 確認画面を出したい
+=end
+
+  def self.find_plugin_candidates(a, b)
+    public_storage[:plugins].
+      grep(/^#{Regexp.quote a}/i).
+      map {|u| b % u }
+  end
+
+  def self.find_status_ids(text)
+    public_storage[:status_ids].select {|id| /#{Regexp.quote(text)}/ =~ id.to_s}
+  end
+
+  def self.find_users(text)
+    public_storage[:users].select {|user| /^#{Regexp.quote(text)}/ =~ user}
+  end
+
+  def self.find_user_candidates(a, b)
+    users = 
+      if a.nil? || a.empty?
+        public_storage[:users].to_a
+      else
+        public_storage[:users].grep(/^#{Regexp.quote a}/i)
+      end
+    users.map {|u| b % u }
+  end
+end