termtter 1.5.0 → 1.6.0

data/lib/plugins/friends.rb

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+module Termtter::Client
+
+  class << self
+    def get_friends(user_name, max)
+      friends = []
+      page = 0
+      begin
+        friends += tmp = Termtter::API::twitter.friends(user_name,
+            :page => page += 1)
+        puts "#{friends.length}/#{max}"
+      rescue
+      end until (tmp.empty? or friends.length > max)
+      friends.take(max)
+    end
+  end
+
+  register_command(
+    :name => :friends, :aliases => [:following],
+    :exec_proc => lambda {|arg|
+      user_name = arg.empty? ? config.user_name : arg
+      public_storage[:friends] = friends = get_friends(user_name, 2000)
+      puts friends.map(&:screen_name).join(' ')
+    },
+    :help => ["friends,following [USERNAME]", "Show user's friends."]
+  )
+
+  register_command(
+    :name => :diff_follow, :aliases => [:diff],
+    :exec_proc => lambda {|arg|
+      user_name = arg.empty? ? config.user_name : arg
+      friends = public_storage[:friends]
+      followers = public_storage[:followers]
+      if friends.nil? || followers.nil?
+        puts 'Do followers and friends first.'
+        return
+      end
+      friends = friends.map(&:screen_name)
+      followers = followers.map(&:screen_name)
+      puts "friends - followers:"
+      puts (friends - followers).map{|s|"http://#{config.host}/#{s}"}.join("\n")
+      puts
+      puts "followers - friends:"
+      puts (followers - friends).map{|s|"http://#{config.host}/#{s}"}.join("\n")
+    },
+    :help => ["diff_follow,diff",
+        "Show difference between frineds and followers."]
+  )
+
+end

data/lib/plugins/g.rb

@@ -0,0 +1,16 @@
+module Termtter::Client
+  begin
+    require 'g'
+    register_command(
+      :name      => :g,
+      :help      => ['g obj', "Do you know 'g'? It's like 'p'."],
+      :exec_proc => lambda {|arg|
+        # we get arg as String, so without eval() it is not very useful
+        # but we shouldn't blindly eval user-supplied string like this, either
+        g eval(arg)
+      }
+    )
+  rescue e
+    handle_error(e)
+  end
+end

data/lib/plugins/gsub.rb

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+
+config.plugins.gsub.set_default(:table, [
+  [/([★☆△▽…□♪♬])(?=\S)/, '\1 '],
+  [/(?=\S)(https?:\/\/)/, ' \1'],
+])
+
+Termtter::Client.register_hook(
+  :name => :gsub,
+  :point => :filter_for_output,
+  :exec => lambda {|statuses, event|
+    statuses.each do |s|
+      t = s.text
+      config.plugins.gsub.table.each {|a, b| t.gsub!(a, b || '') }
+    end
+  }
+)

data/lib/plugins/haml.rb

@@ -0,0 +1,55 @@
+# coding: utf-8
+
+require 'haml'
+require 'tempfile'
+
+module Termtter::Plugins
+  class Haml
+    def initialize(config = Termtter::Config.instance, logger = Termtter::Client.logger)
+      @config, @logger = config, logger
+
+      plugin_config.set_default :options, {}
+    end
+
+    def plugin_config
+      @config.plugins.haml
+    end
+
+    def run(arg)
+      begin
+        hamlified = haml(arg)
+      rescue => e
+        @logger.error e
+        return
+      end
+
+      return if hamlified.nil? || hamlified.empty?
+
+      Termtter::API.twitter.update(hamlified)
+      puts "=> #{hamlified}"
+    end
+
+    def haml(format)
+      return unless input = editor(:haml)
+
+      opts = plugin_config.options.merge(format.empty? ? {} : {:format => format.to_sym})
+      ::Haml::Engine.new(input, opts).render.chomp
+    end
+
+    def editor(extname)
+      unless cmd = ENV['VISUAL'] || ENV['EDITOR']
+        raise 'Please set VISUAL or EDITOR variable.'
+      end
+
+      # XXX: works only in Ruby 1.8.7 or later
+      Tempfile.open(['tmp', ".#{extname}"]) do |f|
+        system cmd, f.path
+        return f.read
+      end
+    end
+  end
+end
+
+Termtter::Client.register_command(:haml) do |arg|
+  Termtter::Plugins::Haml.new.run(arg)
+end

data/lib/plugins/hatebu_and_update.rb

@@ -32,12 +32,12 @@ module Termtter::Client
         auth.authorize(req)
         title = nil
         tinyurl = nil
-        Termtter::API.connection.start('b.hatena.ne.jp', 80) do |http|
+        Net::HTTP.start('b.hatena.ne.jp', 80) do |http|
           res = http.request(req)
           title = Nokogiri::XML(res.body).search('link[title]').first['title'] rescue nil
         end
         if title
-          Termtter::API.connection.start('tinyurl.com', 80) do |http|
+          Net::HTTP.start('tinyurl.com', 80) do |http|
             tinyurl = http.get('/api-create.php?url=' + URI.escape(url)).body
           end
 

data/lib/plugins/history.rb

@@ -21,6 +21,15 @@ module Termtter::Client
     if File.exist?(filename)
       begin
         history = Marshal.load Zlib::Inflate.inflate(File.read(filename))
+      rescue Zlib::BufError => e
+        ui = create_highline
+        delete = ui.ask("Unable to read #{filename}. Do you wish to remove it?")
+        if delete =~ /^y/i
+          if File.delete(filename) > 1
+            puts "Removed #{filename}"
+          end
+        end
+        history = nil
       end
       if history
         keys.each do |key|

data/lib/plugins/irc_gw.rb

@@ -8,7 +8,7 @@ config.plugins.irc_gw.set_default(:logger_level, Logger::ERROR)
 
 module Termtter::Client
   class << self
-    def friends
+    def following_friends
       user_name = config.user_name
 
       frinends = []
@@ -92,8 +92,15 @@ class TermtterIrcGateway < Net::IRC::Server::Session
 
   def on_privmsg(m)
     target, message = *m.params
-    Termtter::Client.call_commands('update ' + message)
-    post @prefix, TOPIC, main_channel, message
+    if message =~ / +\//
+      termtter_command = message.gsub(/ +\//, '')
+      return if Termtter::Client.find_commands(termtter_command).empty?
+      post '#termtter', NOTICE, main_channel, '> ' + termtter_command
+      Termtter::Client.call_commands(termtter_command)
+    else
+      Termtter::Client.call_commands('update ' + message)
+      post @prefix, TOPIC, main_channel, message
+    end
   rescue Exception => e
     post '#termtter', NOTICE, main_channel, "#{e.class.to_s}: #{e.message}"
     Termtter::Client.handle_error(e)
@@ -108,7 +115,7 @@ class TermtterIrcGateway < Net::IRC::Server::Session
   def check_friends
     params = []
     max_params_count = 3
-    Termtter::Client.friends.each do |name|
+    Termtter::Client.following_friends.each do |name|
       prefix = Prefix.new("#{name}!#{name}@localhost")
       post prefix, JOIN, main_channel
       params << prefix.nick

data/lib/plugins/linefeed.rb

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+config.plugins.linefeed.set_default(:width, 50)
+config.plugins.linefeed.set_default(:right, 10)
+
+Termtter::Client.register_hook(
+  :name => :linefeed,
+  :point => :filter_for_output,
+  :exec => lambda {|statuses, event|
+    width = config.plugins.linefeed.width
+    right = config.plugins.linefeed.right
+    statuses.each do |s|
+      t = s.text
+      ocs = []
+      sz = nil
+      t.gsub!(%r{(https?://)}, "\n\\1")
+      t.split("\n").each do |line|
+        cs = line.unpack 'U*'
+        while cs.size > 0
+          sz = 0
+          l = cs.take_while{|c| sz += c < 0x100 ? 1 : 2; sz < width}
+          ocs += l
+          cs = cs.drop(l.size)
+          ocs += [0x0a]
+        end
+      end
+      t2 = ocs.pack('U*')
+      t2.chop! if sz < (width - right)
+      s.text[0, t.size] = t2
+    end
+  }
+)

data/lib/plugins/md5pass.rb

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+require 'digest/md5'
+require 'base64'
+
+config.plugins.md5pass.set_default(:salt, 'salt')
+config.plugins.md5pass.set_default(:len, 12)
+config.plugins.md5pass.set_default(:times, 10)
+
+module Termtter::Client
+
+  def self.gen_pass(master_pass)
+      salt = config.plugins.md5pass.salt
+      len = config.plugins.md5pass.len
+      times = config.plugins.md5pass.times
+      url = "http://#{config.host}/"
+      user = config.user_name
+      str = (url + salt + user + master_pass) * (2 ** times);
+      Base64.encode64(Digest::MD5.digest(str))[0, len]
+  end
+
+  hl = create_highline
+  mp = hl.ask('your master password for md5pass: ') { |q| q.echo = false }
+  config.password = gen_pass(mp)
+
+  register_command(
+    :name => :show_md5pass,
+    :exec_proc => lambda {|arg| puts "=> #{gen_pass(arg)}" },
+    :help => ["show_md5pass MASTER_PASSWORD", "Show your md5 password."]
+  )
+
+end
+
+# config example:
+#   config.system.load_plugins = 'md5pass'
+#   config.plugins.md5pass.salt = 'set_random_string'
+#
+# 1. Decide your master password and salt.
+# 2. Configure. (see above setting)
+# 3. Start termtter.
+# 4. Check your generated password using 'show_md5pass MASTER_PASSWORD'
+# 5. Change your Twitter password to it.
+# 6. Restart termtter.

data/lib/plugins/outputz.rb

@@ -13,7 +13,7 @@ module Termtter::Client
       :points => [:pre_exec_update],
       :exec_proc => lambda {|cmd, arg|
         Thread.new do
-          Termtter::API.connection.start('outputz.com', 80) do |http|
+          Net::HTTP.start('outputz.com', 80) do |http|
               key  = CGI.escape key
               uri  = CGI.escape config.plugins.outputz.uri
               size = arg.split(//).size

data/lib/plugins/primes.rb

@@ -23,7 +23,7 @@ module Termtter::Client
     :name => :primes,
     :exec_proc => lambda {|arg|
       n = arg.to_i
-      text = "primes(#{n}) = {#{primes n}}}"
+      text = "primes(#{n}) = {#{primes n}}"
       puts "=> " << text
     }
   )

data/lib/plugins/quote.rb

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+
+config.plugins.quote.set_default(:format, '<%= comment %>QT @<%=s.user.screen_name%>: <%=s.text%>')
+config.plugins.quote.set_default(:confirm_protected, true)
+
+module Termtter::Client
+  def self.post_quote(s, comment = nil)
+    if s.user.protected && config.plugins.quote.confirm_protected &&
+        !confirm("#{s.user.screen_name} is protected! Are you sure?", false)
+      return
+    end
+
+    comment += ' ' unless comment.nil?
+    text = ERB.new(config.plugins.quote.format).result(binding)
+    Termtter::API.twitter.update(text)
+    puts "=> #{text}"
+
+    return text
+  end
+
+  register_command(
+    :name      => :quote,
+    :aliases   => [:qt],
+    :help      => ['quote,qt (ID|@USER)', 'Post a quote message'],
+    :exec_proc => lambda {|arg|
+      arg, comment = arg.split(/\s/, 2)
+
+      if public_storage[:typable_id] && s = typable_id_status(arg)
+        post_quote(s, comment)
+      else
+        case arg
+        when /(\d+)/
+          post_quote(Termtter::API.twitter.show(arg), comment)
+        when /@([A-Za-z0-9_]+)/
+          user = $1
+          statuses = Termtter::API.twitter.user_timeline(user)
+          return if statuses.empty?
+          post_quote(statuses[0], comment)
+        end
+      end
+    }
+  )
+end

data/lib/plugins/reduce_text.rb

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+module Termtter::Client
+  @last_screen_name = nil
+  @last_source = nil
+  @last_time = nil
+
+  register_hook(
+    :name => :reduce_screenname,
+    :point => :prepare_screenname,
+    :exec => lambda {|n, event|
+      @last_screen_name == n ? '' : @last_screen_name = n
+    }
+  )
+
+  register_hook(
+    :name => :reduce_source,
+    :point => :prepare_source,
+    :exec => lambda {|n, event| @last_source == n ? '' : @last_source = n }
+  )
+
+  register_hook(
+    :name => :reduce_time,
+    :point => :prepare_time,
+    :exec => lambda {|n, event| @last_time == n ? '' : @last_time = n }
+  )
+end

data/lib/plugins/reverse.rb

@@ -1,13 +1,14 @@
 # -*- coding: utf-8 -*-
 
-module Termtter::Client
-  add_filter do |statuses, _|
-    statuses.map do |s|
+Termtter::Client.register_hook(
+  :name => :reverse,
+  :point => :filter_for_output,
+  :exec => lambda {|statuses, event|
+    statuses.each do |s|
       s.text = s.text.split(//).reverse.to_s
-      s
     end
-  end
-end
+  }
+)
 
 # filter-reverse.rb
 #   reverse texts

data/lib/plugins/source.rb

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+def __dir__
+  File.dirname(File.expand_path(__FILE__))
+end
+
+def let(o)
+  yield o
+end
+
+def source(pluginname)
+  File.read(
+    let(
+      Pathname(__dir__) + "#{pluginname}.rb") {|a|
+        File.exist?(a) ? a : Pathname(__dir__) + "defaults/#{pluginname}.rb" })
+end
+
+def truncate(s)
+  s.each_char.take(140).join
+end
+
+module Termtter::Client
+  register_command(:source, :help => ["source {plugin-name}", "shows the source code of the plugin"]) do |arg|
+    puts source(arg)
+  end
+
+  register_command(:sourceyou, :help => ["sourceyou @username {plugin-name}", "gives the source code of the plugin"]) do |arg|
+    /(\w+)\s(\w+)/ =~ arg
+    text = truncate("@#{normalize_as_user_name($1)} #{source($2)}")
+    Termtter::API.twitter.update(text)
+  end
+end

data/lib/plugins/storage/status.rb

@@ -37,7 +37,7 @@ module Termtter::Storage
           }
         }
       end
-      Rubytter.json_to_struct(result)
+      Rubytter.structize(result)
     end
 
     def self.search_user(query)
@@ -60,7 +60,7 @@ module Termtter::Storage
           }
         }
       end
-      Rubytter.json_to_struct(result)
+      Rubytter.structize(result)
     end
 
     def self.insert(data)

data/lib/plugins/storage.rb

@@ -44,7 +44,7 @@ module Termtter::Client
     :aliases => [:ssu],
     :exec_proc => lambda {|arg|
       unless arg.strip.empty?
-        key = arg.strip
+        key = arg.strip.gsub(/^@/, '')
         statuses = Termtter::Storage::Status.search_user({:user => key})
         output(statuses, :search)
       end

data/lib/plugins/stream.rb

@@ -0,0 +1,192 @@
+# -*- coding: utf-8 -*-
+
+require 'tweetstream'
+require File.dirname(__FILE__) + '/../termtter/active_rubytter'
+
+config.plugins.stream.set_default :max_following, 400
+config.plugins.stream.set_default :timeline_format, '<yellow>[S]</yellow> $orig'
+config.plugins.stream.set_default :retry_wait_base, 60
+config.plugins.stream.set_default :retry_wait_max, 3600
+
+module Termtter::Client
+
+  config.plugins.stream.keywords = []
+
+  class << self
+    if defined?(DB)
+      def friends(max)
+        Status.group(:user_id).
+          select(:user_id, :screen_name).
+          join(:users, :id => :user_id).
+          order(:COUNT.sql_function.desc).take(max)
+      end
+    else
+      def friends(max)
+        friends = []
+        page    = 0
+        begin
+          friends += tmp = Termtter::API::twitter.friends(config.user_name, :page => page+=1)
+          p friends.length
+        rescue
+        end until (tmp.empty? or friends.length > max)
+        friends.take(max)
+      end
+    end
+
+    def swap_timeline_format(format)
+      original = config.plugins.stdout.timeline_format
+      if /\$orig/ =~ format
+        format.gsub!(/\$orig/, original)
+      end
+      config.plugins.stdout.timeline_format = format
+      yield
+      config.plugins.stdout.timeline_format = original
+    end
+
+    def kill_thread(name)
+      config.plugins.stream.__send__(name).kill rescue nil
+      config.plugins.stream.__assign__(name, nil)
+    end
+    def alive_thread?(name)
+      config.plugins.stream.__send__(name).alive? rescue false
+    end
+    private :kill_thread
+    private :alive_thread?
+  end
+
+
+  help = ['keyword_stream [:stop|:show|:add|:delete] [KEYWORDS]',
+          'Tracking keyword using Stream API']
+  register_command(:keyword_stream, :help => help) do |arg|
+    catch(:exit) do
+      throw :exit if arg.empty?
+      args = arg.split /[, ]/
+      case args[0]
+      when ':stop'
+        kill_thread :keyword_stream if alive_thread? :keywor_stream
+        config.plugins.stream.keywords.clear
+        puts 'keyword_stream has stopped'
+        throw :exit
+      when ':show'
+        puts alive_thread?(:keyword_stream) ? 'streaming alive' : 'not alive'
+        unless config.plugins.stream.keywords.empty?
+          puts config.plugins.stream.keywords.join(', ')
+        end
+        throw :exit
+      when ':add'
+        args.shift
+        config.plugins.stream.keywords |= args
+      when ':delete'
+        args.shift
+        config.plugins.stream.keywords -= args
+        if config.plugins.stream.keywords.empty?
+          kill_thread :keyword_stream if alive_thread? :keywor_stream
+          puts 'keyword_stream has stopped'
+          throw :exit
+        end
+      when ':start'
+      when /^:.*/
+        puts "Unknown keyword_stream options"
+        throw :exit
+      else
+        config.plugins.stream.keywords = args
+      end
+
+      kill_thread :keyword_stream if alive_thread? :keywor_stream
+
+      keywords = config.plugins.stream.keywords
+
+      puts "streaming: #{keywords.join(', ')}"
+      config.plugins.stream.keyword_stream = Thread.new do
+        retry_wait = config.plugins.stream.retry_wait_base,
+        begin
+          TweetStream::Client.new(config.user_name, config.password).
+            filter(:track => keywords) do |status|
+            print "\e[0G" + "\e[K" unless win?
+            swap_timeline_format(config.plugins.stream.timeline_format) do
+              output [Termtter::ActiveRubytter.new(status)],
+                      :update_friends_timeline
+            end
+            Readline.refresh_line
+          end
+        rescue
+          puts "stream is down"
+          puts "wait #{config.plugins.stream.retry_wait}sec"
+          sleep retry_wait
+          retry_wait = retry_wait * 2;
+          retry_wait = config.plugin.stream.retry_max unless
+          retry_max > config.plugin.stream.retry_max
+          retry
+        end
+      end
+    end
+
+    at_exit do
+      kill_thread :keyword_stream
+    end
+  end
+
+  help = ['hash_stream HASHTAG', 'Tracking hashtag using Stream API']
+  register_command(:hash_stream, :help => help) do |arg|
+    arg = "##{arg}" unless /^#/ =~ arg
+    call_commands("keyword_stream #{arg}")
+  end
+
+  help = ['stream USERNAME', 'Tracking users using Stream API']
+  register_command(:stream, :help => help) do |arg|
+    catch(:exit) do
+      args = arg.split
+
+      case args[0]
+      when ':stop'
+        kill_thread :thread
+        puts 'stream is down'
+        throw :exit
+      end
+
+      if config.plugins.stream.thread.class == Thread
+        puts 'already streaming'
+        throw :exit
+      end
+
+      targets = args.map { |name|
+        Termtter::API.twitter.user(name).id rescue nil
+      }
+
+      max = config.plugins.stream.max_following
+      unless targets and targets.length > 0
+        keys = [:user_id, :"`user_id`", :id, :"`id`"]
+        targets = friends(max).map{ |u|
+          keys.map{ |k| u[k] rescue nil}.compact.first
+        }.compact
+      end
+
+      config.plugins.stream.thread = Thread.new do
+        begin
+          current_targets = targets.take(max)
+          targets = targets.take(max)
+          message = "streaming #{current_targets.length} friend"
+          message << (current_targets.size == 1 ? '.' : 's.')
+          puts message
+          TweetStream::Client.new(config.user_name, config.password).
+            filter(:follow => current_targets) do |status|
+            print "\e[0G" + "\e[K" unless win?
+            swap_timeline_format(config.plugins.stream.timeline_format) do
+              output [Termtter::ActiveRubytter.new(status)], :update_friends_timeline
+            end
+            Readline.refresh_line
+          end
+        rescue(NoMethodError) => e    # #<NoMethodError: private method `split' called for nil:NilClass>
+          puts "stream seems broken (#{e.inspect})."
+          max -= 10 if max > 10
+          retry
+        end
+      end
+
+      at_exit do
+        kill_thread :stream
+      end
+    end
+  end
+end
+

data/lib/plugins/switch_user.rb

@@ -1,22 +1 @@
-# -*- 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
+puts 'Plugin "switch_user" was obsoleted, and use plugin "switch" that is loaded as default.'