termtter 0.8.14 → 1.0.0

data/lib/plugins/multi_reply.rb

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+  register_command(
+    :name => :multi_reply, :aliases => [:mr],
+    :exec_proc => lambda {|arg|
+      text = ERB.new(arg).result(binding).gsub(/\n/, ' ')
+      unless text.empty?
+        /(@(.+))*\s+(.+)/ =~ text
+        if $1
+          msg = $3
+          text = $1.split(/\s+/).map {|u| "#{u} #{msg}" }
+        end
+        Array(text).each do |post|
+          Termtter::API.twitter.update(post)
+          puts "=> #{post}"
+        end
+      end
+    },
+    :completion_proc => lambda {|cmd, args|
+      if /(.*)@([^\s]*)$/ =~ args
+        find_user_candidates $2, "#{cmd} #{$1}@%s"
+      end
+    },
+    :help => ["multi_reply,mp TEXT", "Reply to multi user"]
+  )
+end

data/lib/{plugin → plugins}/notify-send.rb

@@ -7,7 +7,7 @@ Termtter::Client.add_hook do |statuses, event|
     text = statuses.take(max).map {|s|
       status_text = CGI.escapeHTML(s.text)
       status_text.gsub!(%r{https?://[-_.!~*\'()a-zA-Z0-9;\/?:\@&=+\$,%#]+},'<a href="\0">\0</a>')
-      "<b>#{s.user_screen_name}:</b> <span font=\"9.0\">#{status_text}</span>"
+      "<b>#{s.user.screen_name}:</b> <span font=\"9.0\">#{status_text}</span>"
     }.join("\n")
 
     text << %Q|\n<a href="http://twitter.com/">more...</a>| if statuses.size > max

data/lib/plugins/outputz.rb

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+  config.plugins.outputz.set_default(:uri, 'termtter://twitter.com/status/update')
+
+  key = config.plugins.outputz.secret_key
+  if key.empty?
+    puts 'Need your secret key'
+    puts 'please set config.plugins.outputz.secret_key'
+  else
+    register_hook(
+      :name => :outputz,
+      :points => [:pre_exec_update],
+      :exec_proc => lambda {|cmd, arg|
+        Thread.new do
+          Termtter::API.connection.start('outputz.com', 80) do |http|
+              key  = CGI.escape key
+              uri  = CGI.escape config.plugins.outputz.uri
+              size = arg.split(//).size
+              http.post('/api/post', "key=#{key}&uri=#{uri}&size=#{size}")
+            end
+        end
+      }
+    )
+  end
+end
+
+# outputz.rb
+#   a plugin that report to outputz your post
+#
+# settings (note: must this order)
+#   config.plugins.outputz.secret_key = 'your secret key'
+#   plugin 'outputz'

data/lib/{plugin → plugins}/primes.rb

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
 def primes(n)
+  return "" if n < 3
   table = []
   (2 .. n).each do |i|
     table << i
@@ -18,6 +19,12 @@ def primes(n)
 end
 
 module Termtter::Client
-  add_command /^primes\s(\d+)/ do|m,t|t.update_status x="primes(#{n=m[1].to_i}) = {#{primes n}}"
-  puts "=> #{x}" end
+  register_command(
+    :name => :primes,
+    :exec_proc => lambda {|arg|
+      n = arg.to_i
+      text = "primes(#{n}) = {#{primes n}}}"
+      puts "=> " << text
+    }
+  )
 end

data/lib/plugins/quicklook.rb

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+
+require 'uri'
+require 'open-uri'
+require 'pathname'
+require 'tmpdir'
+
+config.plugins.quicklook.set_default(:quicklook_tmpdir, "#{Dir.tmpdir}/termtter-quicklook-tmpdir")
+tmpdir = Pathname.new(config.plugins.quicklook.quicklook_tmpdir)
+tmpdir.mkdir unless tmpdir.exist?
+
+def quicklook(url)
+  tmpdir = Pathname.new(config.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
+  register_command(
+    :name => :quicklook, :aliases => [:ql],
+    :exec_proc => proc{|arg|
+      status = Termtter::API.twitter.show(arg)
+      if (status)
+        uris = URI.regexp.match(status.text).to_a
+        quicklook(uris.first) unless uris.empty?
+      end
+    }
+  )
+end
+
+# quicklook.rb
+# REQUIREMENTS:
+#   plugin 'expand-tinyurl'
+# TODO:
+#   Close quicklook window automatically.

data/lib/{plugin → plugins}/reblog.rb

@@ -17,7 +17,7 @@ module Termtter::Client
           status = t.show(id).first
         end
     
-        Tumblr::API.write(configatron.plugins.reblog.email, configatron.plugins.reblog.password) do
+        Tumblr::API.write(config.plugins.reblog.email, config.plugins.reblog.password) do
           quote("#{status.text}", "<a href=\"http://twitter.com/#{status.user_screen_name}/status/#{status.id}\">Twitter / #{status.user_name}</a>")
         end
       end
@@ -34,7 +34,7 @@ end
 # reblog.rb
 # tumblr reblog it!
 #
-# configatron.plugins.reblog.email = 'your-email-on-tumblr'
-# configatron.plugins.reblog.password = 'your-password-on-tumblr'
+# config.plugins.reblog.email = 'your-email-on-tumblr'
+# config.plugins.reblog.password = 'your-password-on-tumblr'
 #
 #   reblog 1114860346

data/lib/{filter → plugins}/reverse.rb

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

data/lib/{plugin → plugins}/say.rb

@@ -14,7 +14,7 @@ module Termtter::Client
     if !statuses.empty? && event == :update_friends_timeline
       statuses.reverse.each do |s|
         text_without_uri = s.text.gsub(%r|https?://[^\s]+|, 'U.R.I.')
-        say s.user_screen_name, text_without_uri
+        say s.user.screen_name, text_without_uri
       end
     end
   end

data/lib/{plugin → plugins}/scrape.rb

@@ -6,13 +6,13 @@ module Termtter::Client
     statuses = []
     members.each_with_index do |member, index|
       puts "member #{index+1}/#{members.size} #{member}"
-      statuses += Termtter::API.twitter.get_user_timeline(member)
+      statuses += Termtter::API.twitter.user_timeline(member)
     end
     statuses
   end    
   
   def self.scrape_group(group)
-    members = configatron.plugins.group.groups[group] || []
+    members = config.plugins.group.groups[group] || []
     scrape_members(members)
   end
 
@@ -21,12 +21,12 @@ module Termtter::Client
                    :exec_proc => lambda{ |args|
                      groups = args.split(' ').map{|g| g.to_sym}
                      if groups.include? :all
-                       groups = configatron.plugins.group.groups.keys
+                       groups = config.plugins.group.groups.keys
                        puts "get all groups..."
                      end
                      members = []
                      groups.each do |group|
-                       members += configatron.plugins.group.groups[group]
+                       members += config.plugins.group.groups[group]
                      end
                      statuses = scrape_members(members.uniq.compact.sort)
                      call_hooks(statuses, :pre_filter)

data/lib/plugins/screen-notify.rb

@@ -0,0 +1,13 @@
+
+if config.screen_notify.format.nil? or config.screen_notify.format.empty?
+  config.screen_notify.format = "[termtter] @%s"
+end
+
+Termtter::Client.add_hook do |statuses, event|
+  if !statuses.empty? && event == :update_friends_timeline
+    statuses.reverse.each do |s|
+      msg = config.screen_notify.format % [s.user.screen_name, s.text]
+      system 'screen', '-X', 'eval', "bell_msg '#{msg}'", 'bell'
+    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/{plugin → plugins}/sl.rb

@@ -5,8 +5,8 @@ module Termtter
   module Client
 
     public_storage[:current] = ''
-    public_storage[:orig_prompt] = configatron.prompt
-    configatron.prompt = "~/ #{public_storage[:orig_prompt]}"
+    public_storage[:orig_prompt] = config.prompt
+    config.prompt = "~/ #{public_storage[:orig_prompt]}"
 
     register_command(
       :name      => :sl, :aliases => [],
@@ -23,7 +23,7 @@ module Termtter
     register_command(
       :name => :ls, :aliases => [],
       :exec_proc => lambda {|arg|
-        call_commands("list #{arg.empty? ? public_storage[:current] : arg}", API.twitter)
+        call_commands("list #{arg.empty? ? public_storage[:current] : arg}")
       },
       :completion_proc => lambda {|cmd, args|
         find_user_candidates args, "#{cmd} %s"
@@ -36,7 +36,7 @@ module Termtter
       :exec_proc => lambda {|arg|
         public_storage[:current] =
           (arg.nil? || /\~/ =~ arg) ? '' : arg
-        configatron.prompt = "~/#{public_storage[:current]} #{public_storage[:orig_prompt]}"
+        config.prompt = "~/#{public_storage[:current]} #{public_storage[:orig_prompt]}"
       },
       :completion_proc => lambda {|cmd, args|
         find_user_candidates args, "#{cmd} %s"

data/lib/plugins/spam.rb

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+
+Termtter::API.twitter.update('*super spam time*')
+Termtter::Client.register_hook(
+  :name => :span,
+  :points => [/^pre_exec/],
+  :exec_proc => lambda{|*arg|
+    text = arg.join(' ')
+    Termtter::API.twitter.update(text)
+    puts "=> #{text}"
+    false
+  }
+)

data/lib/{plugin → plugins}/standard_plugins.rb

@@ -9,9 +9,12 @@ module Termtter::Client
   register_command(
     :name => :update, :aliases => [:u],
     :exec_proc => lambda {|arg|
-      text = ERB.new(arg).result(binding).gsub(/\n/, ' ')
-      Termtter::API.twitter.update_status(text)
-      puts "=> #{text}"
+      unless arg =~ /^\s*$/
+        text = ERB.new(arg).result(binding).gsub(/\n/, ' ')
+        result = Termtter::API.twitter.update(text, :source=>Termtter::APP_NAME)
+        puts "=> #{text}"
+        result
+      end
     },
     :completion_proc => lambda {|cmd, args|
       if /(.*)@([^\s]*)$/ =~ args
@@ -39,7 +42,7 @@ module Termtter::Client
   register_command(
     :name => :profile, :aliases => [:p],
     :exec_proc => lambda {|arg|
-      user = Termtter::API.twitter.get_user_profile(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
@@ -58,19 +61,27 @@ module Termtter::Client
   register_command(
     :name => :followers,
     :exec_proc => lambda {|arg|
-      followers = Termtter::API.twitter.followers
+      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
       puts followers.map{|f|f.screen_name}.join(' ')
     }
+    # TODO :completion_proc
   )
 
   register_command(
     :name => :list, :aliases => [:l],
     :exec_proc => lambda {|arg|
       unless arg.empty?
-        call_hooks(Termtter::API.twitter.get_user_timeline(arg), :list_user_timeline)
+        call_hooks(Termtter::API.twitter.user_timeline(arg), :list_user_timeline)
       else
-        call_hooks(Termtter::API.twitter.get_friends_timeline(), :list_friends_timeline)
+        call_hooks(Termtter::API.twitter.friends_timeline(), :list_friends_timeline)
       end
     },
     :completion_proc => lambda {|cmd, arg|
@@ -122,13 +133,43 @@ module Termtter::Client
     :completion_proc => get_command(:show).completion_proc
   )
 
+  register_command(
+    :name => :follow, :aliases => [],
+    :exec_proc => lambda {|args|
+      args.split(' ').each do |arg|
+        if arg =~ /^(\w+)/
+          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 arg =~ /^(\w+)/
+          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']
+  )
+
   # 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.get_rate_limit_status
-      remaining_time = "%dmin %dsec" % (limit.reset_time - Time.now).divmod(60)
+      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
@@ -162,10 +203,20 @@ module Termtter::Client
     :name => :default_error_handler,
     :points => [:on_error],
     :exec_proc => lambda {|e|
-      puts "Error: #{e}"
-      if configatron.devel == true
-        puts e.backtrace.join("\n")
+      case e
+      when Rubytter::APIError
+        case e.response.code
+        when /401/
+          warn '[ERROR] Unauthorized: maybe you tried to show protected user status'
+        when /403/
+          warn '[ERROR] Access denied: maybe that user is protected'
+        when /404/
+          warn '[ERROR] Not found: maybe there is no such user'
+        end
+      else
+        warn "[ERROR] Something wrong: #{e.message}"
       end
+      raise e if config.system.devel == true
     }
   )
 
@@ -197,7 +248,7 @@ module Termtter::Client
     if arg
       `#{arg}`.each_line do |line|
         unless line.strip.empty?
-          Termtter::API.twitter.update_status(line)
+          Termtter::API.twitter.update(line)
           puts "=> #{line}"
         end
       end
@@ -222,11 +273,14 @@ module Termtter::Client
   public_storage[:status_ids] ||= Set.new
 
   add_hook do |statuses, event, t|
-    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.to_s)
-      public_storage[:status_ids].add(s.in_reply_to_status_id.to_s) if s.in_reply_to_status_id
+    case event
+    when :update_friends_timeline, :list_friends_timeline, :list_user_timeline, :replies
+      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.to_s)
+        public_storage[:status_ids].add(s.in_reply_to_status_id.to_s) if s.in_reply_to_status_id
+      end
     end
   end
 

data/lib/plugins/stdout.rb

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+
+require 'termcolor'
+require 'erb'
+
+config.plugins.stdout.set_default(
+  :colors,
+  [:none, :red, :green, :yellow, :blue, :magenta, :cyan])
+config.plugins.stdout.set_default(
+  :timeline_format,
+  '<90><%=time%></90> <<%=status_color%>><%=status%></<%=status_color%>> <90><%=id%></90>')
+config.plugins.stdout.set_default(:search_highlihgt_format, '<on_magenta><white>\1</white></on_magenta>')
+
+$highline = HighLine.new
+
+def color(str, value)
+  return str if value == :none
+  case value
+  when String, Symbol
+    $highline.color(str, value)
+  else
+    "\e[#{value}m#{str}\e[0m"
+  end
+end
+
+module Termtter::Client
+
+  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 = config.plugins.stdout.colors[s.user.screen_name.hash % config.plugins.stdout.colors.size]
+      status = "#{s.user.screen_name}: #{text}"
+      if s.in_reply_to_status_id
+        status += " (repl. to #{s.in_reply_to_status_id})"
+      end
+
+      time = "(#{Time.parse(s.created_at).strftime(time_format)})"
+      id = s.id
+      erbed_text = ERB.new(config.plugins.stdout.timeline_format).result(binding)
+      puts TermColor.parse(erbed_text)
+    end
+  end
+
+  def self.print_statuses_with_date(statuses, sort = true)
+    print_statuses(statuses, sort, '%m-%d %H:%M')
+  end
+
+  def self.print_search_results(result, time_format = '%H:%M:%S')
+    result.results.sort_by{|r| r.created_at}.each do |r|
+      text = r.text.
+                gsub(/(\n|\r)/, '').
+                gsub(/(#{Regexp.escape(result.query)})/i, config.plugins.stdout.search_highlihgt_format)
+      status_color = config.plugins.stdout.colors[r.from_user_id.to_i.hash % config.plugins.stdout.colors.size]
+      status = "#{r.from_user}: #{text}"
+      time = "(#{Time.parse(r.created_at).strftime(time_format)})"
+      id = r.id
+      erbed_text = ERB.new(config.plugins.stdout.timeline_format).result(binding)
+      puts TermColor.parse(erbed_text)
+    end
+  end
+
+  add_hook do |result, event|
+    case event
+    when :update_friends_timeline, :list_friends_timeline
+      print_statuses(result) unless result.empty?
+    when :list_user_timeline, :replies
+      print_statuses_with_date(result) unless result.empty?
+    when :show
+      print_statuses_with_date([result])
+    when :search
+      print_search_results(result)
+    end
+  end
+
+end
+# stdout.rb
+#   output statuses to stdout
+# example config
+#   config.plugins.stdout.colors = [:none, :red, :green, :yellow, :blue, :magenta, :cyan]
+#   config.plugins.stdout.timeline_format = '<90><%=time%></90> <<%=status_color%>><%=status%></<%=status_color%>> <90><%=id%></90>'

data/lib/plugins/storage/DB.rb

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+
+require 'sqlite3'
+require 'singleton'
+
+module Termtter::Storage
+  class DB
+    include Singleton
+    attr_reader :db
+
+    def initialize
+      @db = SQLite3::Database.new(Termtter::CONF_DIR + '/storage.db')
+      @db.type_translation = true
+      create_table
+    end
+
+    def create_table
+      sql =<<-SQL
+CREATE TABLE IF NOT EXISTS user (
+    id          int NOT NULL,
+    screen_name text,
+    PRIMARY KEY (id)
+);
+CREATE TABLE IF NOT EXISTS post (
+    post_id          int NOT NULL,  -- twitter側のpostのid
+    created_at	     int,    	    -- 日付(RubyでUNIX時間に変換)
+    in_reply_to_status_id int, 	    -- あったほうがよいらしい
+    in_reply_to_user_id int,  	    -- あったほうがよいらしい
+    post_text text,
+    user_id int NOT NULL,
+    PRIMARY KEY (post_id)
+);
+      SQL
+      @db.execute_batch(sql)
+    end
+  end
+end

data/lib/plugins/storage/status.rb

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+
+require File.dirname(__FILE__) + '/DB'
+require 'sqlite3'
+
+module Termtter::Storage
+  class Status
+    KEYS = %w[post_id created_at in_reply_to_status_id in_reply_to_user_id post_text user_id screen_name]
+
+    def size
+      DB.instance.db.get_first_value("select count(*) from post").to_i
+    end
+
+    def self.search(query)
+      raise "query must be Hash(#{query}, #{query.class})" unless query.kind_of? Hash
+
+      DB.instance.db.execute("select created_at, screen_name, post_text, in_reply_to_status_id, post_id from post inner join user on post.user_id = user.id where post_text like '%' || ? || '%' ", query[:post_text]) do |created_at, screen_name, post_text, in_reply_to_status_id, post_id|
+
+#      DB.instance.db.execute("select created_at, screen_name, post_text, in_reply_to_status_id, post_id from post inner join user on post.user_id = user.id where post_text like '%of%' ", query[:post_text]) do |created_at, screen_name, post_text, in_reply_to_status_id, post_id|
+        p [created_at, screen_name, post_text, in_reply_to_status_id, post_id]
+      end
+    end
+
+    def self.insert(data)
+      raise "data must be Hash(#{data}, #{data.class})" unless data.kind_of? Hash
+      # 条件しぼりたいけどやりかたがうまくわからない
+      #      raise "unko" unless data.keys.all?{|c| KEYS.include? c}
+      begin
+      DB.instance.db.execute(
+                             "insert into post values(?,?,?,?,?,?)",
+                             data[:post_id],
+                             data[:created_at],
+                             data[:in_reply_to_status_id],
+                             data[:in_reply_to_user_id],
+                             data[:post_text],
+                             data[:user_id])
+        begin
+          DB.instance.db.execute(
+                                 "insert into user values(?,?)",
+                                 data[:user_id],
+                                 data[:screen_name])
+        rescue # FIXME: specify exceptions here
+        end
+        rescue # FIXME: specify exceptions here
+      end
+    end
+  end
+end

data/lib/plugins/storage/status_mook.rb

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+
+require 'sqlite3'
+
+module Termtter::Storage
+  class Status
+    def initialize
+    end
+
+    def self.create
+    end
+
+    def self.all
+      []
+    end
+
+    def self.search
+    end
+
+    private
+      def db
+        @db ||= connect
+      end
+
+      def connect
+        @db = SQLite3::Database.new(File.expand_path('~/test.db'))
+        @db.type_translation = true
+      end
+  end
+end