jugyo-termtter 0.7.6 → 0.7.7

data/lib/plugin/scrape.rb

@@ -0,0 +1,43 @@
+module Termtter::Client
+  def self.scrape_group(group, t)
+    members = configatron.plugins.group.groups[group] || []
+    members.map {|member|
+      t.get_user_timeline(member)
+    }.flatten
+  end
+
+  def self.scrape_groups(t)
+    groups = configatron.plugins.group.groups
+    groups.map {|group|
+      scrape_group(group, t)
+    }.flatten
+  end
+
+  add_help 'scrape_group GROUPNAME', 'Get the group timeline'
+  add_help 'scrape_groups', 'Get all groups timeline'
+
+  add_command /^(?:scrape_group)\s+(.+)/ do |m, t|
+    group_name = m[1].to_sym
+    statuses = scrape_group(group_name, t)
+    call_hooks(statuses, :pre_filter, t)
+    puts "done"
+  end
+
+  add_command /^(?:scrape_groups)\s*$/ do |m, t|
+    statuses = scrape_groups(t)
+    call_hooks(statuses, :pre_filter, t)
+    puts "done"
+  end
+
+  add_completion do |input|
+    case input
+    when /^(scrape_group)?\s+(.+)/
+      find_group_candidates($2, "#{$1} %s") if defined? find_group_candidates
+    when /^(scrape_group)\s+$/
+      configatron.plugins.group.groups.keys
+    else
+      %w(scrape_group scrape_groups).grep(/^#{Regexp.quote input}/)
+    end
+  end
+
+end

data/lib/plugin/screen.rb

@@ -0,0 +1,22 @@
+module Termtter
+  module Plugin
+    module Screen
+      def self.set_title(title)
+        print "\033k#{title}\033\\"
+      end
+    end
+  end
+end
+
+# Add below to your ~/.termtter
+#
+# require 'plugin/yonda'
+# require 'plugin/screen'
+# module Termtter::Client
+#   add_hook do |statuses, event|
+#     case event
+#     when :update_friends_timeline, :plugin_yonda_yonda
+#       Termtter::Plugin::Screen::set_title("termtter(#{public_storage[:unread_count]})")
+#     end
+#   end
+# end

data/lib/plugin/shell.rb

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

data/lib/plugin/sl.rb

@@ -12,20 +12,20 @@ module Termtter::Client
   end
 
   add_help 'cd USER', 'Change current directory'
-  add_command /^cd\s+(.*)/ do |m, t|
+  add_command /^(?:cd\s+|\.\/)(.*)/ do |m, t|
     directory = m[1].strip
-    direcroty = '' if /~/ =~ direcroty
-    public_storage[:current] = direcroty
-    puts "=> #{direcroty}"
+    directory = '' if /\~/ =~ directory
+    public_storage[:current] = directory
+    puts "=> #{directory}"
   end
   add_macro /^cd$/, 'eval public_storage[:current] = ""'
 
   add_completion do |input|
     case input
-    when /^(cd)\s+(.*)/
-      find_user_candidates $2, "#{$1} %s"
+    when /^(cd\s+|\.\/)(.*)/
+      find_user_candidates $2, "#{$1.gsub(/\s+/, ' ')}%s"
     else
-      %w[ sl ls cd pwd ].grep(/^#{Regexp.quote input}/)
+      %w[ sl ls cd pwd ./ ].grep(/^#{Regexp.quote input}/)
     end
   end
 end

data/lib/plugin/standard_plugins.rb

@@ -1,15 +1,35 @@
+require 'erb'
+
 module Termtter::Client
 
   # standard commands
 
   add_command /^(update|u)\s+(.*)/ do |m, t|
-    text = m[2]
+    text = ERB.new(m[2]).result(binding).gsub(/\n/, ' ')
     unless text.empty?
       t.update_status(text)
       puts "=> #{text}"
     end
   end
 
+  add_command /^(direct|d)\s+([^\s]+)\s+(.*)\s*$/ do |m, t|
+    user = m[2]
+    text = ERB.new(m[3]).result(binding).gsub(/\n/, ' ')
+    unless text.empty?
+      t.direct_message(user, text)
+      puts "=> to:#{user} message:#{text}"
+    end
+  end
+
+  add_command /^(profile|p)\s+([^\s]+)/ do |m, t|
+    profile = t.get_user_profile(m[2])
+    if profile
+      %w[ name favourites_count url id description protected utc_offset time_zone screen_name notifications statuses_count followers_count friends_count profile_image_url location following created_at ].each do |key|
+        puts "#{key} : #{profile[key]}" if profile.key?(key)
+      end
+    end
+  end
+
   add_command /^(list|l)\s*$/ do |m, t|
     statuses = t.get_friends_timeline()
     call_hooks(statuses, :list_friends_timeline, t)
@@ -28,45 +48,63 @@ module Termtter::Client
     call_hooks(t.replies(), :replies, t)
   end
 
-  add_command /^show\s+([^\s]+)/ do |m, t|
-    call_hooks(t.show(m[1]), :show, t)
+  add_command /^show(s)?\s+(?:[\w\d]+:)?(\d+)/ do |m, t|
+    call_hooks(t.show(m[2], 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
-
-  add_command /^resume\s*$/ do |m, t|
-    resume
-  end
-
-  add_command /^(exit|e)\s*$/ do |m, t|
-    exit
-  end
-
-  add_command /^help\s*$/ do |m, t|
-    puts <<-EOS
-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
-replies,r         List the most recent @replies for the authenticating user
-search,s TEXT     Search for Twitter
-show ID           Show a single status
-    EOS
-    puts formatted_help unless @@helps.empty?
+  register_command(
+    :name => :limit, :aliases => ['lm'],
+    :exec_proc => proc {|arg|
+      limit = Termtter::API.twitter.get_rate_limit_status
+      remaining_time = "%dmin %dsec" % (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
+      puts "=> #{color(limit.remaining_hits, remaining_color)}/#{limit.hourly_limit} until #{limit.reset_time} (#{remaining_time} remaining)"
+    },
+    :help => ["limit,lm", "Show the API limit status"]
+  )
+
+  register_command( 
+    :name => :pause,
+    :exec_proc => proc {|arg| pause},
+    :help => ["pause", "Pause updating"]
+  )
+
+  register_command(
+    :name => :resume,
+    :exec_proc => proc {|arg| resume},
+    :help => ["resume", "Resume updating"]
+  )
+
+  register_command(
+    :name => :exit, :aliases => ['e'],
+    :exec_proc => proc {|arg| exit},
+    :help => ['exit,e', 'Exit']
+  )
+
+  add_command /^(help|h)\s*$/ do |m, t|
+    # TODO: migrate to use Termtter::Command#help
+    helps = [
+      ["help,h", "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"],
+      ["update,u TEXT", "Post a new message"],
+      ["direct,d @USERNAME TEXT", "Send direct message"],
+      ["profile,p USERNAME", "Show user's profile"],
+      ["replies,r", "List the most recent @replies for the authenticating user"],
+      ["search,s TEXT", "Search for Twitter"],
+      ["show ID", "Show a single status"]
+    ]
+    helps += @@helps
+    helps += @@new_commands.map {|name, command| command.help}
+    helps.compact!
+    puts formatted_help(helps)
   end
 
   add_command /^eval\s+(.*)$/ do |m, t|
@@ -90,13 +128,13 @@ show ID           Show a single status
     end
   end
 
-  def self.formatted_help
-    width = @@helps.map {|n, d| n.size }.max
+  def self.formatted_help(helps)
+    helps = helps.sort_by{|help| help[0]}
+    width = helps.map {|n, d| n.size }.max
     space = 3
-    "\nuser commands:\n" +
-      @@helps.map {|name, desc|
-        name.to_s.ljust(width + space) + desc.to_s
-      }.join("\n")
+    helps.map {|name, desc|
+      name.to_s.ljust(width + space) + desc.to_s
+    }.join("\n")
   end
 
   # completion for standard commands
@@ -114,11 +152,15 @@ show ID           Show a single status
     end
   end
 
-  def self.find_status_id_candidates(a, b)
+  def self.find_status_id_candidates(a, b, u = nil)
+    candidates = public_storage[:status_ids].to_a
+    if u && c = public_storage[:log].select {|s| s.user_screen_name == u }.map {|s| s.id.to_s }
+      candidates = c unless c.empty?
+    end
     if a.empty?
-      public_storage[:status_ids].to_a
+      candidates
     else
-      public_storage[:status_ids].grep(/#{Regexp.quote a}/)
+      candidates.grep(/#{Regexp.quote a}/)
     end.
     map {|u| b % u }
   end
@@ -133,14 +175,24 @@ show ID           Show a single status
   end
 
   add_completion do |input|
-    standard_commands = %w[exit help list pause update resume replies search show limit]
+    standard_commands = %w[exit help list pause profile update direct resume replies search show limit]
     case input
     when /^(list|l)?\s+(.*)/
       find_user_candidates $2, "#{$1} %s"
     when /^(update|u)\s+(.*)@([^\s]*)$/
       find_user_candidates $3, "#{$1} #{$2}@%s"
-    when /^show\s+(.*)/
-      find_status_id_candidates $1, "show %s"
+    when /^(direct|d)\s+(.*)/
+      find_user_candidates $2, "#{$1} %s"
+    when /^(profile|p)\s+(.*)/
+      find_user_candidates $2, "#{$1} %s"
+    when /^show(s)?\s+(([\w\d]+):)?\s*(.*)/
+      if $2
+        find_status_id_candidates $4, "show#{$1} #{$2}%s", $3
+      else
+        result = find_user_candidates $4, "show#{$1} %s:"
+        result = find_status_id_candidates $4, "show#{$1} %s"  if result.empty?
+        result
+      end
     else
       standard_commands.grep(/^#{Regexp.quote input}/)
     end

data/lib/plugin/stdout.rb

@@ -3,53 +3,15 @@ require 'erb'
 
 configatron.plugins.stdout.set_default(
   :colors,
-  [:white, :red, :green, :yellow, :blue, :magenta, :cyan])
+  [:none, :red, :green, :yellow, :blue, :magenta, :cyan])
 configatron.plugins.stdout.set_default(
   :timeline_format,
   '<%= color(time, 90) %> <%= color(status, status_color) %> <%= color(id, 90) %>')
 
 $highline = HighLine.new
 
-if win?
-  require 'kconv'
-  require 'Win32API'
-  STD_OUTPUT_HANDLE = 0xFFFFFFF5
-  $wSetConsoleTextAttribute = Win32API.new('kernel32','SetConsoleTextAttribute','II','I')
-  $wGetConsoleScreenBufferInfo = Win32API.new("kernel32", "GetConsoleScreenBufferInfo", ['l', 'p'], 'i')
-  $wGetStdHandle = Win32API.new('kernel32','GetStdHandle','I','I')
-
-  $hStdOut = $wGetStdHandle.call(STD_OUTPUT_HANDLE)
-  lpBuffer = ' ' * 22
-  $wGetConsoleScreenBufferInfo.call($hStdOut, lpBuffer)
-  $oldColor = lpBuffer.unpack('SSSSSssssSS')[4]
-
-  $colorMap = {
-       0 => 7,     # black/white
-      37 => 8,     # white/intensity
-      31 => 4 + 8, # red/red
-      32 => 2 + 8, # green/green
-      33 => 6 + 8, # yellow/yellow
-      34 => 1 + 8, # blue/blue
-      35 => 5 + 8, # magenta/purple
-      36 => 3 + 8, # cyan/aqua
-      90 => 7,     # erase/white
-  }
-  def puts(str)
-    str = str.tosjis
-    tokens = str.split(/(\e\[\d+m)/)
-    tokens.each do |token|
-      if token =~ /\e\[(\d+)m/
-        $wSetConsoleTextAttribute.call $hStdOut, $colorMap[$1.to_i].to_i
-      else
-        STDOUT.print token
-      end
-    end
-    $wSetConsoleTextAttribute.call $hStdOut, $oldColor
-    STDOUT.puts
-  end
-end
-
 def color(str, value)
+  return str if value == :none
   case value
   when String, Symbol
     $highline.color(str, value)
@@ -58,47 +20,41 @@ def color(str, value)
   end
 end
 
-Termtter::Client.add_hook do |statuses, event|
-  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 = 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})"
-        end
-
-        time_format = case event
-          when :update_friends_timeline, :list_friends_timeline
-            '%H:%M:%S'
-          else
-            '%m-%d %H:%M'
-          end
-        time = "(#{s.created_at.strftime(time_format)})"
-
-        id = s.id
+module Termtter::Client
 
-        puts ERB.new(configatron.plugins.stdout.timeline_format).result(binding)
-      end
-    end
-  when :search
-    statuses.each do |s|
+  def self.print_statuses(statuses, sort = true, time_format = '%H:%M:%S')
+    (sort ? statuses.sort_by{ |s| s.id} : statuses).each do |s|
       text = s.text
       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')})"
+      if s.in_reply_to_status_id
+        status += " (reply to #{s.in_reply_to_status_id})"
+      end
+      
+      time = "(#{s.created_at.strftime(time_format)})"
       id = s.id
       puts ERB.new(configatron.plugins.stdout.timeline_format).result(binding)
     end
   end
-end
 
+  def self.print_statuses_with_date(statuses, sort = true)
+    print_statuses(statuses, sort, '%m-%d %H:%M')
+  end
+  
+  add_hook do |statuses, event|
+    next if statuses.empty?
+    
+    case event
+    when :update_friends_timeline, :list_friends_timeline
+      print_statuses(statuses)
+    when :search, :list_user_timeline, :show, :replies
+      print_statuses_with_date(statuses)
+    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.colors = [:none, :red, :green, :yellow, :blue, :magenta, :cyan]
 #   configatron.plugins.stdout.timeline_format = '<%= color(time, 90) %> <%= color(status, status_color) %> <%= color(id, 90) %>'

data/lib/plugin/system_status.rb

@@ -0,0 +1,41 @@
+require 'erb'
+
+configatron.plugins.system_status.set_default(:default_status_proc, proc { Time.now.strftime("%x %X") })
+configatron.plugins.system_status.set_default(:interval, 1)
+configatron.plugins.system_status.set_default(:default_color, :on_blue)
+configatron.plugins.system_status.set_default(:format, '<%= status %>')
+
+def out_put_status(status, color)
+  formatted_status = ERB.new(configatron.plugins.system_status.format).result(binding)
+  colored_status = color(formatted_status, color)
+  print "\e[s\e[1000G\e[#{status.size - 1}D#{colored_status}\e[u"
+  $stdout.flush
+end
+
+module Termtter::Client
+  Thread.new do
+    loop do
+      begin
+        status = public_storage[:system_status] ||
+                    configatron.plugins.system_status.default_status_proc.call
+        color = public_storage[:system_status_color] ||
+                        configatron.plugins.system_status.default_color
+      rescue => e
+        status = e.message
+        color = :on_red
+      end
+      out_put_status(status, color)
+      sleep configatron.plugins.system_status.interval
+    end
+  end
+end
+
+# system_status.rb
+#   show system status on left side.
+#   output public_storage[:system_status] or Time.now.strftime("%x %X") if nil
+# example config
+#   configatron.plugins.system_status.default_status_proc = proc { Time.now.strftime("%x %X") }
+#   configatron.plugins.system_status.interval = 1
+#   configatron.plugins.system_status.default_color = :on_blue
+#   configatron.plugins.system_status.format = '<%= status %>'
+

data/lib/plugin/update_editor.rb

@@ -0,0 +1,53 @@
+require 'tempfile'
+
+module Termtter::Client
+  if ENV['EDITOR']
+    configatron.plugins.update_editor.set_default('editor', ENV['EDITOR'])
+  else
+    configatron.plugins.update_editor.set_default('editor', 'vi')
+  end
+  configatron.plugins.update_editor.set_default('add_completion', false)
+
+  
+  def self.input_editor
+    file = Tempfile.new('termtter')
+    editor = configatron.plugins.update_editor.editor
+    if configatron.plugins.update_editor.add_completion
+      file.puts "\n"*100 + "__END__\n" + public_storage[:users].to_a.join(' ')
+    end
+    file.close
+    system("#{editor} #{file.path}")
+    result = file.open.read
+    file.close(false)
+    result
+  end
+
+  add_command /^(update_editor|ue)\s*$/ do |m, t|
+    pause
+    text = input_editor
+    unless text.empty?
+      text = ERB.new(text).result(binding)
+      text.split("\n").each do |post|
+        break if post =~ /^__END__$/
+        unless post.empty?
+          t.update_status(post)
+          puts "=> #{post}"
+        end
+      end
+    end
+    resume
+  end
+
+  add_help 'update_editor,ue', 'Update status from editor.'
+
+  add_completion do |input|
+    %w[ update_editor ].grep(/^#{Regexp.quote input}/)
+  end
+end
+
+# update_editor.rb
+#   update status from editor.
+# example:
+#   > update_editor
+#   (type your status, save, close the editor)
+#   => (your status)

data/lib/plugin/wassr_post.rb

@@ -0,0 +1,21 @@
+require 'uri'
+require 'net/http'
+
+module Termtter::Client
+  # NOTE: overwrite original update command
+  add_command /^(update|u)\s+(.*)/ do |m, t|
+    text = ERB.new(m[2]).result(binding).gsub(/\n/, ' ')
+    t.update_status(text)
+    puts "=> #{text}"
+    begin
+      Net::HTTP.version_1_2
+      req = Net::HTTP::Post.new("/statuses/update.json?")
+      req.basic_auth configatron.plugins.wassr_post.username, configatron.plugins.wassr_post.password
+      Net::HTTP.start('api.wassr.jp', 80) do |http|
+        res = http.request(req, "status=#{URI.escape(text)}&source=Termtter")
+      end
+    rescue
+      puts "RuntimeError: #{$!}"
+    end
+  end
+end

data/lib/plugin/yonda.rb

@@ -0,0 +1,18 @@
+module Termtter::Client
+  public_storage[:unread_count] = 0
+
+  add_help 'yonda,y', 'Mark as read'
+
+  add_command %r'^y(?:onda)?\s*$' do |m, t|
+    public_storage[:unread_count] = 0
+    print "\033[2J\033[H" # FIXME
+    call_hooks [], :plugin_yonda_yonda, t
+  end
+
+  add_hook do |statuses, event|
+    case event
+    when :update_friends_timeline
+      public_storage[:unread_count] += statuses.size
+    end
+  end
+end

data/lib/termtter/api.rb

@@ -0,0 +1,12 @@
+module Termtter
+  module API
+    class << self
+      attr_reader :connection, :twitter
+      def setup
+        @connection = Connection.new
+        @twitter = Termtter::Twitter.new(configatron.user_name, configatron.password, @connection)
+      end
+    end
+  end
+end
+# Termtter::API.connection, Termtter::API.twitter can be accessed.