draftcode-termtter 1.0.8

data/lib/plugins/stdout.rb

@@ -0,0 +1,220 @@
+# -*- coding: utf-8 -*-
+
+require 'termcolor'
+require 'erb'
+require 'tempfile'
+require 'curses'
+require 'dl/import'
+
+module Readline
+  begin
+    module LIBREADLINE
+      if DL.const_defined? :Importable
+        extend DL::Importable
+      else
+        extend DL::Importer
+      end
+      pathes = Array(ENV['TERMTTER_EXT_LIB'] || [
+        '/opt/local/lib/libreadline.dylib',
+        '/usr/lib/libreadline.so',
+        '/usr/local/lib/libreadline.so',
+        File.join(Gem.bindir, 'readline.dll')
+      ])
+      dlload(pathes.find { |path| File.exist?(path)})
+      extern 'int rl_refresh_line(int, int)'
+    end
+    def self.refresh_line
+      LIBREADLINE.rl_refresh_line(0, 0)
+    end
+  rescue Exception
+    def self.refresh_line;end
+  end
+end
+
+if win?
+  require 'iconv'
+
+  module Readline
+    $iconv_sj_to_u8 = Iconv.new('UTF-8', "CP#{$wGetACP.call()}")
+    alias :old_readline :readline
+    def readline(*a)
+      str = old_readline(*a)
+      out = ''
+      loop do
+        begin
+          out << $iconv_sj_to_u8.iconv(str)
+          break
+        rescue Iconv::Failure
+          out << "#{$!.success}?"
+          str = $!.failed[1..-1]
+        end
+      end
+      return out
+    end
+    module_function :old_readline, :readline
+  end
+end
+
+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_highlight_format, '<on_magenta><white>\1</white></on_magenta>')
+
+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)
+
+class String
+  def truncate_column(col)
+    count = 0
+    ary = []
+    ret = []
+    self.split(//u).each {|c|
+      count += (c.size == 1 ? 1 : 2)
+      if count > col then
+          ret.push(ary.to_s)
+          ary = []
+          count = 0
+      end
+      ary.push(c)
+    }
+    ret.push(ary.to_s) unless ary.empty?
+    ret
+  end
+end
+
+module Termtter
+  class StdOut < Hook
+    def initialize
+      super(:name => :stdout, :points => [:output])
+      @input_thread = nil
+      Client.register_hook(
+        :name => :stdout_exit,
+        :points => [:exit],
+        :exec_proc => lambda { @input_thread.kill if @input_thread }
+      )
+      Client.register_hook(
+          :name => :stdout_readline_yield_thread,
+          :points => [:before_task_thread_run],
+          :exec_proc => lambda { start_input_thread() unless @input_thread }
+      )
+    end
+
+    def call(statuses, event)
+      print_statuses(statuses)
+    end
+
+    def trap_setting()
+      begin
+        stty_save = `stty -g`.chomp
+        trap("INT") do
+          begin
+            system "stty", stty_save
+          ensure
+            exit
+          end
+        end
+      rescue Errno::ENOENT
+      end
+    end
+
+    def setup_readline
+      if Readline.respond_to?(:basic_word_break_characters=)
+        Readline.basic_word_break_characters= "\t\n\"\\'`><=;|&{("
+      end
+      Readline.completion_proc = Client.get_command_completion_proc()
+      vi_or_emacs = config.editing_mod
+      unless vi_or_emacs.empty?
+        Readline.__send__("#{vi_or_emacs}_editing_mode")
+      end
+    end
+
+    def start_input_thread
+      setup_readline()
+      trap_setting()
+      @input_thread = Thread.new do
+        while buf = Readline.readline(ERB.new(config.prompt).result(API.twitter.__send__(:binding)), true)
+          Readline::HISTORY.pop if buf.empty?
+          begin
+            Client.call_commands(buf)
+          rescue CommandNotFound => e
+            warn "Unknown command \"#{e}\""
+            warn 'Enter "help" for instructions'
+          rescue => e
+            Client.handle_error e
+          end
+        end
+      end
+      @input_thread.join
+    end
+
+    def print_statuses(statuses, 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
+
+      output_text = ''
+      output_text += "\e[1K\e[0G" unless win?
+      Curses::init_screen;
+      cols = Curses.cols;
+      Curses::close_screen
+      statuses.each do |s|
+        text = s.text
+        status_color = config.plugins.stdout.colors[s.user.id.hash % config.plugins.stdout.colors.size]
+        status = "#{s.user.screen_name}: #{TermColor.escape(text)}"
+        if s.in_reply_to_status_id
+          status += " (reply to #{s.in_reply_to_status_id})"
+        end
+
+        time = "(#{Time.parse(s.created_at).strftime(time_format)})"
+        id = s.id
+        len =  id.to_s.length + 2
+        if cols > 0 then
+          status = status.truncate_column(cols-len-1).join( "\n" )
+          status = status.gsub( /\n/, "\n".ljust(len) )
+        end
+        source =
+          case s.source
+          when />(.*?)</ then $1
+          when 'web' then 'web'
+          end
+
+        erbed_text = ERB.new(config.plugins.stdout.timeline_format).result(binding)
+        output_text << TermColor.parse(erbed_text) + "\n"
+      end
+
+      if config.plugins.stdout.enable_pager && ENV['LINES'] && statuses.size > ENV['LINES'].to_i
+        file = Tempfile.new('termtter')
+        file.print output_text
+        file.close
+        system "#{config.plugins.stdout.pager} #{file.path}"
+        file.close(true)
+      else
+        output_text << TermColor.parse("<90>-----Fetched on " + Time.now.strftime(time_format) + "</90>\n")
+        print output_text
+      end
+      Readline.refresh_line
+    end
+  end
+
+  Client.register_hook(StdOut.new)
+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.rb

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+
+require 'pp'
+require 'time'
+
+require File.dirname(__FILE__) + '/storage/status'
+
+module Termtter::Client
+  public_storage[:log] = []
+
+  register_hook(
+    :name => :storage,
+    :points => [:pre_filter],
+    :exec_proc => lambda {|statuses, event|
+      statuses.each do |s|
+        Termtter::Storage::Status.insert(
+          :post_id => s.id,
+          :created_at => Time.parse(s.created_at).to_i,
+          :in_reply_to_status_id => s.in_reply_to_status_id,
+          :in_reply_to_user_id => s.in_reply_to_user_id,
+          :text => s.text,
+          :user_id => s.user.id,
+          :screen_name => s.user.screen_name
+        )
+      end
+    }
+  )
+
+  register_command(
+    :name => :search_storage,
+    :aliases => [:ss],
+    :exec_proc => lambda {|arg|
+      unless arg.strip.empty?
+        key = arg.strip
+        statuses = Termtter::Storage::Status.search({:text => key})
+        output(statuses, :search)
+      end
+    },
+    :help => [ 'search_storage WORD', 'Search storage for WORD' ]
+  )
+
+  register_command(
+    :name => :search_storage_user,
+    :aliases => [:ssu],
+    :exec_proc => lambda {|arg|
+      unless arg.strip.empty?
+        key = arg.strip
+        statuses = Termtter::Storage::Status.search_user({:user => key})
+        output(statuses, :search)
+      end
+    },
+    :help => [ 'search_storage_user SCREEN_NAME', 'Search storage for SCREE_NAME' ]
+  )
+
+end

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,83 @@
+# -*- 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 self.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
+      if query[:text] == nil then
+        query[:text] = '';
+      end
+      if query[:user] == nil then
+        query[:user] = '';
+      end
+      result = []
+      sql = "select created_at, screen_name, post_text, in_reply_to_status_id, post_id, user_id "
+      sql += "from post inner join user on post.user_id = user.id where post_text like '%' || ? || '%'"
+      DB.instance.db.execute(sql,
+                             query[:text]) do |created_at, screen_name, post_text, in_reply_to_status_id, post_id, user_id|
+        created_at = Time.at(created_at).to_s
+        result << {
+          :id => post_id,
+          :created_at => created_at,
+          :text => post_text,
+          :in_reply_to_status_id => in_reply_to_status_id,
+          :in_reply_to_user_id => nil,
+          :user => {
+            :id => user_id,
+            :screen_name => screen_name
+          }
+        }
+      end
+      Rubytter.json_to_struct(result)
+    end
+
+    def self.search_user(query)
+      raise "query must be Hash(#{query}, #{query.class})" unless query.kind_of? Hash
+      result = []
+      sql = "select created_at, screen_name, post_text, in_reply_to_status_id, post_id, user_id "
+      sql += "from post inner join user on post.user_id = user.id where "
+      sql += query[:user].split(' ').map!{|que| que.gsub(/(\w+)/, 'screen_name like \'%\1%\'')}.join(' or ')
+      DB.instance.db.execute(sql) do |created_at, screen_name, post_text, in_reply_to_status_id, post_id, user_id|
+        created_at = Time.at(created_at).to_s
+        result << {
+          :id => post_id,
+          :created_at => created_at,
+          :text => post_text,
+          :in_reply_to_status_id => in_reply_to_status_id,
+          :in_reply_to_user_id => nil,
+          :user => {
+            :id => user_id,
+            :screen_name => screen_name
+          }
+        }
+      end
+      Rubytter.json_to_struct(result)
+    end
+
+    def self.insert(data)
+      return unless data[:text]
+      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[:text],
+        data[:user_id])
+      DB.instance.db.execute(
+        "insert into user values(?,?)",
+        data[:user_id],
+        data[:screen_name])
+    rescue SQLite3::SQLException
+    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

data/lib/plugins/switch_user.rb

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+
+module Termtter::Client
+  register_command(
+    :name => :switch_user,
+    :exec_proc => lambda {|arg|
+      Termtter::API.switch_user(arg)
+    },
+    :completion_proc => lambda {|cmd, arg|
+      # TODO
+    },
+    :help => ["switch_user USERNAME", "Switch twitter account."]
+  )
+
+  register_command(
+    :name => :restore_user,
+    :exec_proc => lambda {|arg|
+      Termtter::API.restore_user
+    },
+    :help => ["restore_user", "Restore default twitter account."]
+  )
+end