earthquake 0.0.2 → 0.1.0

data/Gemfile

@@ -6,10 +6,12 @@ source "http://rubygems.org"
 gem 'twitter-stream', :require => 'twitter/json_stream'
 gem 'notify'
 gem 'i18n'
-gem 'activesupport', :require => 'active_support/core_ext'
+gem 'activesupport', :require => ['active_support/core_ext', 'active_support/dependencies']
 gem 'awesome_print', :require => 'ap'
 gem 'launchy'
 gem 'oauth'
+gem 'twitter_oauth'
+gem 'termcolor'
 
 # Add dependencies to develop your gem here.
 # Include everything needed to run rake, tests, features, etc.

data/Gemfile.lock

@@ -7,14 +7,17 @@ GEM
     diff-lcs (1.1.2)
     eventmachine (0.12.10)
     git (1.2.5)
+    highline (1.6.1)
     i18n (0.5.0)
     jeweler (1.5.2)
       bundler (~> 1.0.0)
       git (>= 1.2.5)
       rake
+    json (1.5.1)
     launchy (0.4.0)
       configuration (>= 0.0.5)
       rake (>= 0.8.1)
+    mime-types (1.16)
     notify (0.3.0)
     oauth (0.4.4)
     rake (0.8.7)
@@ -28,9 +31,15 @@ GEM
     rspec-expectations (2.3.0)
       diff-lcs (~> 1.1.2)
     rspec-mocks (2.3.0)
+    termcolor (1.2.1)
+      highline (>= 1.5.0)
     twitter-stream (0.1.12)
       eventmachine (>= 0.12.8)
       roauth (>= 0.0.2)
+    twitter_oauth (0.4.3)
+      json (>= 1.1.9)
+      mime-types (>= 1.16)
+      oauth (>= 0.4.1)
 
 PLATFORMS
   ruby
@@ -46,4 +55,6 @@ DEPENDENCIES
   oauth
   rcov
   rspec (~> 2.3.0)
+  termcolor
   twitter-stream
+  twitter_oauth

data/README.md

@@ -3,6 +3,8 @@ Earthquake
 
 Twitter Client on Terminal with Streaming API.
 
+It supports ruby 1.9 only.
+
 Install
 ----
 
@@ -19,9 +21,7 @@ TODO
 ----
 
 * plugin system
- * command
  * filter
-* colorize
 * reconnect
 
 Copyright

data/VERSION

@@ -1 +1 @@
-0.0.2
+0.1.0

data/earthquake.gemspec

@@ -0,0 +1,95 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{earthquake}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["jugyo"]
+  s.date = %q{2011-03-16}
+  s.default_executable = %q{earthquake}
+  s.description = %q{Twitter Client on Terminal with Twitter Streaming API.}
+  s.email = %q{jugyo.org@gmail.com}
+  s.executables = ["earthquake"]
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.md"
+  ]
+  s.files = [
+    ".document",
+    ".rspec",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE.txt",
+    "README.md",
+    "Rakefile",
+    "VERSION",
+    "bin/earthquake",
+    "earthquake.gemspec",
+    "lib/earthquake.rb",
+    "lib/earthquake/commands.rb",
+    "lib/earthquake/core.rb",
+    "lib/earthquake/ext.rb",
+    "lib/earthquake/get_access_token.rb",
+    "lib/earthquake/input.rb",
+    "lib/earthquake/output.rb",
+    "lib/earthquake/twitter.rb",
+    "spec/earthquake_spec.rb",
+    "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/jugyo/earthquake}
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.6.0}
+  s.summary = %q{Twitter Client on Terminal.}
+  s.test_files = [
+    "spec/earthquake_spec.rb",
+    "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<twitter-stream>, [">= 0"])
+      s.add_runtime_dependency(%q<notify>, [">= 0"])
+      s.add_runtime_dependency(%q<i18n>, [">= 0"])
+      s.add_runtime_dependency(%q<activesupport>, [">= 0"])
+      s.add_runtime_dependency(%q<awesome_print>, [">= 0"])
+      s.add_runtime_dependency(%q<launchy>, [">= 0"])
+      s.add_runtime_dependency(%q<oauth>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, ["~> 2.3.0"])
+      s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.5.2"])
+      s.add_development_dependency(%q<rcov>, [">= 0"])
+    else
+      s.add_dependency(%q<twitter-stream>, [">= 0"])
+      s.add_dependency(%q<notify>, [">= 0"])
+      s.add_dependency(%q<i18n>, [">= 0"])
+      s.add_dependency(%q<activesupport>, [">= 0"])
+      s.add_dependency(%q<awesome_print>, [">= 0"])
+      s.add_dependency(%q<launchy>, [">= 0"])
+      s.add_dependency(%q<oauth>, [">= 0"])
+      s.add_dependency(%q<rspec>, ["~> 2.3.0"])
+      s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.5.2"])
+      s.add_dependency(%q<rcov>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<twitter-stream>, [">= 0"])
+    s.add_dependency(%q<notify>, [">= 0"])
+    s.add_dependency(%q<i18n>, [">= 0"])
+    s.add_dependency(%q<activesupport>, [">= 0"])
+    s.add_dependency(%q<awesome_print>, [">= 0"])
+    s.add_dependency(%q<launchy>, [">= 0"])
+    s.add_dependency(%q<oauth>, [">= 0"])
+    s.add_dependency(%q<rspec>, ["~> 2.3.0"])
+    s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.5.2"])
+    s.add_dependency(%q<rcov>, [">= 0"])
+  end
+end
+

data/lib/earthquake.rb

@@ -4,10 +4,14 @@ require 'readline'
 require 'bundler/setup'
 Bundler.require :default
 
-$:.unshift File.expand_path('..', __FILE__)
+Thread.abort_on_exception = true
 
-require 'earthquake/core'
-
-module Earthquake
-  extend Earthquake::Core
-end
+%w(
+  ext
+  core
+  output
+  input
+  get_access_token
+  twitter
+  commands
+).each { |name| require_dependency File.expand_path("../earthquake/#{name}", __FILE__) }

data/lib/earthquake/commands.rb

@@ -0,0 +1,51 @@
+# encoding: UTF-8
+module Earthquake
+  init do
+    command :exit do |m|
+      stop
+    end
+
+    command :help do |m|
+      puts "TODO..."
+    end
+
+    command :restart do |m|
+      puts 'restarting...'
+      exec File.expand_path('../../..//bin/earthquake', __FILE__)
+    end
+
+    command :eval do |m|
+      ap eval(m[1])
+    end
+
+    # update
+    command %r|^[^/]+| do |m|
+      twitter.update(m[0]) if confirm(m[0])
+    end
+
+    command %r|^/reply (\d+)\s+(.*)|, :as => :reply do |m|
+      # TODO
+      ap m
+    end
+
+    command :status do |m|
+      puts_item twitter.status(m[1])
+    end
+
+    command :delete do |m|
+      twitter.status_destroy(m[1])
+    end
+
+    command :retweet do |m|
+      twitter.retweet(m[1])
+    end
+
+    command :favorite do |m|
+      twitter.favorite(m[1])
+    end
+
+    command :unfavorite do |m|
+      twitter.unfavorite(m[1])
+    end
+  end
+end

data/lib/earthquake/core.rb

@@ -1,9 +1,33 @@
+# encoding: UTF-8
 module Earthquake
   module Core
     attr_accessor :config
-    attr_reader :item_queue
 
-    def init(*argv)
+    def item_queue
+      @item_queue ||= []
+    end
+
+    def inits
+      @inits ||= []
+    end
+
+    def init(&block)
+      inits << block
+    end
+
+    def init_all
+      inits.each { |block| block.call }
+      inits.clear
+    end
+
+    def reload
+      loaded = ActiveSupport::Dependencies.loaded.dup
+      ActiveSupport::Dependencies.clear
+      loaded.each { |lib| require_dependency lib }
+      init_all
+    end
+
+    def load_config(*argv)
       # TODO: parse argv
       self.config = {
         :dir             => File.expand_path('~/.earthquake'),
@@ -13,84 +37,37 @@ module Earthquake
       config[:file] ||= File.join(config[:dir], 'config')
       load config[:file]
 
-      get_access_token unless self.config[:access_key] && self.config[:access_secret]
-    end
-
-    def get_access_token
-      consumer = OAuth::Consumer.new(
-        self.config[:consumer_key],
-        self.config[:consumer_secret],
-        :site => 'http://api.twitter.com'
-      )
-      request_token = consumer.get_request_token
-
-      puts "1) open: #{request_token.authorize_url}"
-      Launchy::Browser.run(request_token.authorize_url)
-
-      print "2) Enter the PIN: "
-      pin = gets.strip
-
-      access_token = request_token.get_access_token(:oauth_verifier => pin)
-      config[:access_key] = access_token.token
-      config[:access_secret] = access_token.secret
-
-      puts "Saving 'access_key' and 'access_secret' to '#{config[:file]}'"
-      File.open(config[:file], 'a') do |f|
-        f << "\n"
-        f << "Earthquake.config[:access_key] = '#{config[:access_key]}'"
-        f << "\n"
-        f << "Earthquake.config[:access_secret] = '#{config[:access_secret]}'"
-      end
+      get_access_token unless self.config[:token] && self.config[:secret]
     end
 
     def start(*argv)
-      init(*argv)
-
-      @item_queue = []
+      load_config(*argv)
 
-      Thread.abort_on_exception = true
+      init_all
 
       Thread.start do
-        while buf = Readline.readline("[earthquake] ", true)
-          begin
-            buf = buf.strip
-
-            if buf == '/exit'
-              # TODO: refactor
-              @stream.stop if @stream
-              EventMachine.stop if EventMachine.reactor_running? 
-            end
-          rescue => e
-            notify "[ERROR] #{e}"
-          end
+        while buf = Readline.readline("⚡ ", true)
+          input(buf.strip)
         end
       end
 
       Thread.start do
         loop do
-          # TODO: handle the response that include friends
-          if Readline.line_buffer.empty?
-            while item = item_queue.shift
-              begin
-                insert "#{item["user"]["screen_name"]}: #{item["text"]}"
-              rescue => e
-                insert do
-                  ap item
-                end
-                # notify "[ERROR] #{e}"
-              end
-            end
+          if Readline.line_buffer.nil? || Readline.line_buffer.empty?
+            output
+            sleep 1
+          else
+            sleep 2
           end
-          sleep 1
         end
       end
 
       EventMachine::run {
-        @stream = Twitter::JSONStream.connect(
+        @stream = ::Twitter::JSONStream.connect(
           :ssl   => true,
           :host  => 'userstream.twitter.com',
           :path  => '/2/user.json',
-          :oauth => config.slice(:consumer_key, :consumer_secret, :access_key, :access_secret)
+          :oauth => config.slice(:consumer_key, :consumer_secret).merge(:access_key => config[:token], :access_secret => config[:secret])
         )
 
         @stream.each_item do |item|
@@ -112,27 +89,19 @@ module Earthquake
           $stdout.flush
         end
 
-        trap('TERM') {
-          @stream.stop
-          EventMachine.stop if EventMachine.reactor_running? 
-        }
+        trap('TERM') { stop }
       }
     end
 
-    def notify(message)
-      Notify.notify 'earthquake', message
+    def stop
+      @stream.stop
+      EventMachine.stop if EventMachine.reactor_running? 
     end
 
-    def insert(*messages)
-      clear_line
-      puts messages unless messages.empty?
-      yield if block_given?
-    ensure
-      Readline.refresh_line
-    end
-
-    def clear_line
-      print "\e[0G" + "\e[K"
+    def notify(message)
+      Notify.notify 'earthquake', message
     end
   end
+
+  extend Core
 end

data/lib/earthquake/ext.rb

@@ -0,0 +1,7 @@
+class String
+  alias_method :t, :termcolor
+
+  def e
+    TermColor.escape(self)
+  end
+end

data/lib/earthquake/get_access_token.rb

@@ -0,0 +1,32 @@
+module Earthquake
+  module GetAccessToken
+    def get_access_token
+      consumer = OAuth::Consumer.new(
+        self.config[:consumer_key],
+        self.config[:consumer_secret],
+        :site => 'http://api.twitter.com'
+      )
+      request_token = consumer.get_request_token
+
+      puts "1) open: #{request_token.authorize_url}"
+      Launchy::Browser.run(request_token.authorize_url)
+
+      print "2) Enter the PIN: "
+      pin = gets.strip
+
+      access_token = request_token.get_access_token(:oauth_verifier => pin)
+      config[:token] = access_token.token
+      config[:secret] = access_token.secret
+
+      puts "Saving 'token' and 'secret' to '#{config[:file]}'"
+      File.open(config[:file], 'a') do |f|
+        f << "\n"
+        f << "Earthquake.config[:token] = '#{config[:token]}'"
+        f << "\n"
+        f << "Earthquake.config[:secret] = '#{config[:secret]}'"
+      end
+    end
+  end
+
+  extend GetAccessToken
+end

data/lib/earthquake/input.rb

@@ -0,0 +1,64 @@
+# encoding: UTF-8
+module Earthquake
+  module Input
+    def commands
+      @commands ||= []
+    end
+
+    def command_names
+      @command_names ||= []
+    end
+
+    def input(text)
+      begin
+        reload if config[:debug]
+        if command = commands.detect { |c| c[:pattern] =~ text }
+          command[:block].call($~)
+        elsif !text.empty?
+          puts "<yellow>Command not found</yellow>".t
+        end
+      rescue Exception => e
+        puts "<on_red>[ERROR] #{e.message.e}\n#{e.backtrace.join("\n").e}</on_red>".t
+      end
+    end
+
+    def command(pattern, options = {}, &block)
+      if block
+        if pattern.is_a?(String) || pattern.is_a?(Symbol)
+          command_name = "#{config[:command_prefix]}#{pattern}"
+          command_names << command_name
+          pattern = /^#{Regexp.quote(command_name)}\s*(.*)$/
+        end
+        command_names << "#{config[:command_prefix]}#{options[:as]}" if options[:as]
+        commands << {:pattern => pattern, :block => block}
+      else
+        commands.detect { |c| c[:name] == name }
+      end
+    end
+
+    def confirm(message, type = :y)
+      case type
+      when :y
+        print "#{message} [Yn] "
+        return !(gets.strip =~ /^n$/i)
+      when :n
+        print "#{message} [yN] "
+        return !!(gets.strip =~ /^y$/i)
+      else
+        raise "type must be :y or :n"
+      end
+    end
+  end
+
+  init do
+    Readline.completion_proc = lambda { |text|
+      command_names.grep /^#{Regexp.quote(text)}/
+    }
+
+    config[:command_prefix] ||= '/'
+
+    commands.clear
+  end
+
+  extend Input
+end

data/lib/earthquake/output.rb

@@ -0,0 +1,71 @@
+module Earthquake
+  module Output
+    def output
+      return if item_queue.empty?
+      insert do
+        while item = item_queue.shift
+          puts_item(item)
+        end
+      end
+    end
+
+    def puts_item(item)
+      output_handers.each { |p| p.call(item) }
+    rescue => e
+      puts e, e.backtrace
+    end
+
+    def output_handers
+      @output_handers ||= []
+    end
+
+    def output_hander(&block)
+      output_handers << block
+    end
+
+    def insert(*messages)
+      clear_line
+      puts messages unless messages.empty?
+      yield if block_given?
+    ensure
+      Readline.refresh_line
+    end
+
+    def clear_line
+      print "\e[0G" + "\e[K"
+    end
+
+    def color_of(screen_name)
+      config[:colors][screen_name.to_i(36) % config[:colors].size]
+    end
+  end
+
+  init do
+    output_handers.clear
+
+    config[:colors] = (31..36).to_a + (91..96).to_a
+
+    output_hander do |item|
+      if item["text"]
+        misc = (item["in_reply_to_status_id"] ? " (reply to #{item["in_reply_to_status_id"]})" : "")
+        user_color = color_of(item["user"]["screen_name"])
+        text = item["text"].e.gsub(/@([0-9A-Za-z_]+)/) do |i|
+          c = color_of($1)
+          "<#{c}>#{i}</#{c}>"
+        end
+        status = "<90>[#{item["id"].to_s.e}]</90> " +
+                 "<#{user_color}>#{item["user"]["screen_name"].e}</#{user_color}>: " +
+                 "#{text}<90>#{misc.e}</90>"
+        puts status.t
+      end
+    end
+
+    output_hander do |item|
+      if item["delete"]
+        puts "[deleted] #{item["delete"]["status"]["id"]}"
+      end
+    end
+  end
+
+  extend Output
+end

data/lib/earthquake/twitter.rb

@@ -0,0 +1,11 @@
+module Earthquake
+  module Twitter
+    attr_reader :twitter
+  end
+
+  init do
+    @twitter = TwitterOAuth::Client.new(config.slice(:consumer_key, :consumer_secret, :token, :secret))
+  end
+
+  extend Twitter
+end

metadata

@@ -2,7 +2,7 @@
 name: earthquake
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors: 
 - jugyo
@@ -153,8 +153,15 @@ files:
 - Rakefile
 - VERSION
 - bin/earthquake
+- earthquake.gemspec
 - lib/earthquake.rb
+- lib/earthquake/commands.rb
 - lib/earthquake/core.rb
+- lib/earthquake/ext.rb
+- lib/earthquake/get_access_token.rb
+- lib/earthquake/input.rb
+- lib/earthquake/output.rb
+- lib/earthquake/twitter.rb
 - spec/earthquake_spec.rb
 - spec/spec_helper.rb
 has_rdoc: true