chirpstream 0.0.2 → 0.0.3

data/Rakefile

@@ -6,10 +6,10 @@ begin
     s.email = "joshbuddy@gmail.com"
     s.homepage = "http://github.com/joshbuddy/chirpstream"
     s.authors = ["Joshua Hull"]
-    s.files = FileList["[A-Z]*", "{lib}/**/*"]
+    s.files = FileList["[A-Z]*", "{lib}/**/*", "{bin}/**/*"]
     s.add_dependency 'eventmachine', ">= 0.12.10"
     s.add_dependency 'em-http-request', ">= 0.2.7"
-    s.add_dependency 'json', ">= 1.2.4"
+    s.add_dependency 'yajl-ruby', ">= 0.7.5"
     s.add_dependency 'load_path_find', ">= 0.0.5"
   end
   Jeweler::GemcutterTasks.new

data/Readme.rdoc

@@ -12,8 +12,9 @@ This is an experimental EventMachine-based client for http://chirpstream.twitter
   
   chirp = Chirpstream.new('joshbuddy', 'xxxxxxxx')
   chirp.reconnect{ puts "reconnecting..." }
-  chirp.tweet {|t| puts "#{t['text']} (from #{t['user']['name'].foreground(:red)} (#{('@' + t['user']['screen_name']).foreground(:green)}))" }
-  chirp.follow {|t| puts "#{t.data['source'].data['screen_name'].foreground(:green)} following #{t.data['target'].data['screen_name'].foreground(:green)}" }
-  chirp.favorite {|t| puts "#{t.data['source'].data['screen_name'].foreground(:green)} <3 -> #{t.data['target_object'].data['text']}" }
-  chirp.retweet {|t| puts "#{t.data['source'].data['screen_name'].foreground(:green)} RT -> #{t.data['target_object'].data['text']}" }
+  chirp.tweet {|t| puts "#{t.text} (from #{t.user.name.foreground(:red)} (#{('@' + t.user.screen_name).foreground(:green)}))" }
+  chirp.follow {|t| puts "#{t.source.screen_name.foreground(:green)} following #{t.target.screen_name.foreground(:green)}" }
+  chirp.favorite {|t| puts "#{t.source.screen_name.foreground(:green)} <3 -> #{t.target_object.text}" }
+  chirp.retweet {|t| puts "#{t.source.screen_name.foreground(:green)} RT -> #{t.target_object.text}" }
   chirp.connect
+

data/VERSION

@@ -1 +1 @@
-0.0.2
+0.0.3

data/bin/chirp_growl

@@ -0,0 +1,59 @@
+#!/usr/bin/env ruby -rubygems
+
+begin
+  require 'growl'
+rescue LoadError
+  puts "You'll need to install growl\nsudo gem install growl"
+  exit(1)
+end
+
+begin
+  require 'json'
+rescue LoadError
+  puts "You'll need to install json\nsudo gem install json"
+  exit(1)
+end
+
+require File.join(File.dirname(__FILE__), '..', 'lib', 'chirpstream')
+require 'etc'
+require 'fileutils'
+
+FileUtils.mkdir_p(File.join(Etc.getpwuid.dir, '.chirpstream'))
+FileUtils.mkdir_p(File.join(Etc.getpwuid.dir, '.chirpstream', 'cache'))
+
+json_file = File.join(Etc.getpwuid.dir, '.chirpstream', 'user.json')
+cache_dir = File.join(Etc.getpwuid.dir, '.chirpstream', 'cache')
+
+if File.exist?(json_file)
+  data = JSON.parse(File.read(json_file))
+  @username = data['username']
+  @password = data['password']
+else
+  puts "Enter your username:"
+  @username = gets.strip
+  puts "Enter your password:"
+  @password = gets.strip
+  File.open(json_file, 'w') {|f| f << {:username => @username, :password => @password}.to_json}
+end
+
+chirp = Chirpstream.new(@username, @password)
+
+Growl.notify {
+  self.message = 'Starting chirp_growl...'
+  self.title = 'ChirpStreeeeem!'
+  self.image = File.join(File.dirname(__FILE__), '..', 'image', 'bird.png')
+}
+
+chirp.tweet { |t| 
+  t.user.with_profile_image(cache_dir) do |image_path|
+    Growl.notify {
+      self.message = t.text
+      self.title = "@#{t.user.screen_name}"
+      self.image = image_path if image_path
+    }
+  end
+}
+
+chirp.connect
+
+

data/lib/chirpstream/event/delete.rb

@@ -1,4 +1,7 @@
 class Chirpstream
   class Delete < Event
+    ATTRS = [:id, :user_id]
+    attr_accessor *ATTRS
+    user_writer :user_id
   end
 end

data/lib/chirpstream/event/favorite.rb

@@ -1,4 +1,8 @@
 class Chirpstream
   class Favorite < Event
+    ATTRS = [ :target, :source, :target_object ]
+    attr_accessor *ATTRS
+    user_writer :target, :source
+    tweet_writer :target_object
   end
 end

data/lib/chirpstream/event/follow.rb

@@ -1,4 +1,7 @@
 class Chirpstream
   class Follow < Event
+    ATTRS = [ :target, :source ]
+    attr_accessor *ATTRS
+    user_writer :target, :source
   end
 end

data/lib/chirpstream/event/retweet.rb

@@ -1,4 +1,8 @@
 class Chirpstream
   class Retweet < Event
+    ATTRS = [ :target, :source, :target_object ]
+    attr_accessor *ATTRS
+    user_writer :target, :source
+    tweet_writer :target_object
   end
 end

data/lib/chirpstream/event.rb

@@ -1,19 +1,5 @@
 class Chirpstream
   
-  class Event
-    
-    attr_reader :data
-    
-    def initialize(data)
-      @data = data
-    end
-    
-    def load_all(&block)
-      if unloaded_data = @data.values.find{|d| d.respond_to?(:loaded?) && !d.loaded? }
-        unloaded_data.load { load_all(&block) }
-      else
-        block.call(self)
-      end
-    end
+  class Event < TwitterObject
   end
 end

data/lib/chirpstream/tweet.rb

@@ -1,29 +1,17 @@
 class Chirpstream
-  class Tweet
+  class Tweet < TwitterObject
     
-    URL = "http://api.twitter.com/1/statuses/show/%s.json"
-
-    attr_reader :id, :data
-    
-    def initialize(base, id)
-      @base = base
-      @id = id
-    end
-    
-    def load
-      unless loaded?
-        http = EM::HttpRequest.new(URL % id).get(:head => {'authorization' => [@base.username, @base.password]})
-        http.callback {
-          if http.response_header.status == 200
-            @data = JSON.parse(http.response)
-            yield self
-          end
-        }
-      end
+    ATTRS = [:coordinates, :favorited, :created_at, :truncated, :contributors, :text, :id, :geo, :in_reply_to_user_id, :place, :source, :in_reply_to_screen_name, :in_reply_to_status_id, :user]
+  
+    attr_accessor *ATTRS
+    user_writer :user
+  
+    def tweet_loadable_id
+      id
     end
     
     def loaded?
-      @data
+      !text.nil?
     end
   end
 end

data/lib/chirpstream/twitter_object.rb

@@ -0,0 +1,135 @@
+class TwitterObject
+
+  SINGLE_USER_URL = "http://api.twitter.com/1/users/show/%s.json"
+  SINGLE_TWEET_URL = "http://api.twitter.com/1/statuses/show/%s.json"
+
+  attr_reader :base
+
+  def initialize(base, data = nil)
+    @base = base
+    from_json(data) if data
+    yield self if block_given?
+  end
+  
+  def self.user_writer(*attrs)
+    attrs.each do |attr|
+      module_eval "
+      def #{attr}=(#{attr})
+        @#{attr} = if #{attr}.is_a?(Hash)
+          Chirpstream::User.new(base, #{attr})
+        else
+          Chirpstream::User.new(base) {|u| u.id = #{attr}}
+        end
+      end
+      "
+    end
+  end
+  
+  def self.tweet_writer(*attrs)
+    attrs.each do |attr|
+      module_eval "
+      def #{attr}=(#{attr})
+        @#{attr} = if #{attr}.is_a?(Hash)
+          Chirpstream::Tweet.new(base, #{attr})
+        else
+          Chirpstream::Tweet.new(base) {|t| t.id = #{attr}}
+        end
+      end
+      "
+    end
+  end
+  
+  def from_json(data)
+    self.class.attrs.each { |a| self.send(:"#{a}=", data[a.to_s]) }
+  end
+  
+  def self.attrs
+    const_get(:ATTRS)
+  end
+  
+  def load_all(&block)
+    attrs = self.class.attrs
+    if respond_to?(:loaded?) && !loaded?
+      if respond_to?(:user_loadable_id)
+        from_json(get_user_data(user_loadable_id)[user_loadable_id])
+        load_all(&block)
+      elsif respond_to?(:tweet_loadable_id)
+        from_json(get_tweet_data(tweet_loadable_id)[tweet_loadable_id])
+        load_all(&block)
+      end
+    else
+      tweet_ids = {}
+      user_ids = {}
+      attrs.each do |a|
+        obj = send(a)
+        if obj.respond_to?(:loaded?) && !obj.loaded?
+          if obj.respond_to?(:user_loadable_id)
+            user_ids[a] = obj.user_loadable_id
+          elsif obj.respond_to?(:tweet_loadable_id)
+            tweet_ids[a] = obj.tweet_loadable_id
+          end
+        end
+      end
+      get_tweet_data(tweet_ids.values) { |tweet_data|
+        tweet_ids.each{|k,v| self.send(:"#{k}=", tweet_data[v])}
+        user_data = get_user_data(user_ids.values) { |user_data|
+          user_ids.each{|k,v| self.send(:"#{k}=", user_data[v])}
+          yield self
+        }
+      }
+    end
+    
+  end
+  
+  def get_tweet_data(ids)
+    ids = Array(ids).uniq
+    data = {}
+    if ids.empty?
+      yield data
+    else
+      load_tweet_data(ids, data) { yield data }
+    end
+  end
+  
+  def load_tweet_data(ids, data, &block)
+    id = ids.shift
+    if (id)
+      parser = Yajl::Parser.new
+      parser.on_parse_complete = proc { |parsed|
+        data[id] = parsed
+        load_tweet_data(ids, data, &block)
+      }
+      http = EM::HttpRequest.new("http://api.twitter.com/1/statuses/show/%s.json" % id).get :head => {'authorization' => [base.username, base.password]}
+      http.stream { |chunk|
+        parser << chunk
+      }
+    else
+      yield
+    end
+  end
+  
+  def get_user_data(ids)
+    ids = Array(ids).uniq
+    data = {}
+    if ids.empty?
+      yield data
+    else
+      load_user_data(ids, data) { yield data }
+    end
+  end
+  
+  def load_user_data(ids, data)
+    parser = Yajl::Parser.new
+    parser.on_parse_complete = proc { |parsed|
+      parsed.each do |user|
+        data[user["id"]] = user
+      end
+      yield
+    }
+    http = EM::HttpRequest.new("http://api.twitter.com/1/users/lookup.json").post :head => {'authorization' => [base.username, base.password]}, :body => {'user_id' => ids.join(',')}
+    http.stream { |chunk|
+      parser << chunk
+    }
+  end
+  
+end

data/lib/chirpstream/user.rb

@@ -1,29 +1,46 @@
+require 'em-http'
+require 'digest/md5'
+
 class Chirpstream
-  class User
-  
-    URL = "http://api.twitter.com/1/users/show/%s.json"
-  
-    attr_reader :id, :data
+  class User < TwitterObject
 
-    def initialize(base, id)
-      @base = base
-      @id = id
+    ATTRS = [
+      :profile_background_tile, :name, :profile_sidebar_border_color, :profile_sidebar_fill_color, :profile_image_url, :location, :created_at,
+      :profile_link_color, :url, :contributors_enabled, :favourites_count, :utc_offset, :id, :followers_count, :protected, :lang,
+      :profile_text_color, :geo_enabled, :profile_background_color, :time_zone, :notifications, :description, :verified, :profile_background_image_url,
+      :statuses_count, :friends_count, :screen_name, :following
+    ]
+   
+    attr_accessor *ATTRS
+    
+    def loaded?
+      name
+    end
+    
+    def user_loadable_id
+      id
     end
-  
-    def load
-      unless loaded?
-        http = EM::HttpRequest.new(URL % id).get(:head => {'authorization' => [@base.username, @base.password]})
-        http.callback {
+    
+    def with_profile_image(cache_dir)
+      raise unless loaded?
+      
+      cached_file = File.join(cache_dir, "#{Digest::MD5.hexdigest(profile_image_url)}#{File.extname(profile_image_url)}")
+      
+      if File.exist?(cached_file)
+        yield cached_file
+      else
+        http = EM::HttpRequest.new(profile_image_url).get :head => {'authorization' => [base.username, base.password]}
+        http.callback do
           if http.response_header.status == 200
-            @data = JSON.parse(http.response)
-            yield self
+            File.open(cached_file, 'w') {|f| f << http.response}
+            yield cached_file
+          else
+            yield nil
           end
-        }
+        end
       end
+      
     end
-  
-    def loaded?
-      @data
-    end
+    
   end
 end

data/lib/chirpstream.rb

@@ -1,11 +1,12 @@
 require 'eventmachine'
 require 'em-http'
-require 'json'
+require 'yajl'
 require 'pp'
 require 'load_path_find'
 
 $LOAD_PATH.add_current
 
+require 'chirpstream/twitter_object'
 require 'chirpstream/event'
 require 'chirpstream/event/follow'
 require 'chirpstream/event/retweet'
@@ -61,63 +62,63 @@ class Chirpstream
   def dispatch_friend(data)
     unless @handlers.friend.empty?
       data['friends'].each_slice(100) do |friend_ids|
-        friend_http = EM::HttpRequest.new("http://api.twitter.com/1/users/lookup.json").post(:body => {'user_id' => friend_ids.join(',')}, :head => {'authorization' => [@username, @password]})
-        friend_http.callback do
-          friends = JSON.parse(friend_http.response)
+        parser = Yajl::Parser.new
+        parser.on_parse_complete = proc { |friends|
           friends.each do |friend|
             @handlers.friend.each{|h| h.call(friend)}
           end
-        end
+        }
+        friend_http = EM::HttpRequest.new("http://api.twitter.com/1/users/lookup.json").post(:body => {'user_id' => friend_ids.join(',')}, :head => {'authorization' => [@username, @password]})
+        http.stream { |chunk|
+          parser << chunk
+        }
       end
     end
   end
   
   def dispatch_tweet(data)
     unless @handlers.tweet.empty?
-      @handlers.tweet.each{|h| h.call(data)}
+      tweet = Tweet.new(self, data)
+      tweet.load_all { |t|
+        @handlers.tweet.each{|h| h.call(tweet)}
+      }
     end
   end
   
   def dispatch_follow(data)
     unless @handlers.follow.empty?
-      data['target'] = User.new(self, data['target']['id'])
-      data['source'] = User.new(self, data['source']['id'])
-      e = Follow.new(data)
-      e.load_all {
-        @handlers.follow.each{|h| h.call(e)}
+      follow = Follow.new(self, data)
+      follow.load_all { |f|
+        @handlers.follow.each{|h| h.call(f)}
       }
     end
   end
   
   def dispatch_favorite(data)
-    return if @handlers.follow.empty?
-    data['target'] = User.new(self, data['target']['id'])
-    data['source'] = User.new(self, data['source']['id'])
-    data['target_object'] = Tweet.new(self, data['target_object']['id'])
-    e = Favorite.new(data)
-    e.load_all {
-      @handlers.favorite.each{|h| h.call(e)}
-    }
+    unless @handlers.favorite.empty?
+      favorite = Favorite.new(self, data)
+      favorite.load_all { |f|
+        @handlers.favorite.each{|h| h.call(f)}
+      }
+    end
   end
   
   def dispatch_retweet(data)
-    return if @handlers.retweet.empty?
-    data['target'] = User.new(self, data['target']['id'])
-    data['source'] = User.new(self, data['source']['id'])
-    data['target_object'] = Tweet.new(self, data['target_object']['id'])
-    e = Retweet.new(data)
-    e.load_all {
-      @handlers.retweet.each{|h| h.call(e)}
-    }
+    unless @handlers.retweet.empty?
+      retweet = Retweet.new(self, data)
+      retweet.load_all { |f|
+        @handlers.retweet.each{|h| h.call(f)}
+      }
+    end
   end
     
   def dispatch_delete(data)
-    return if @handlers.delete.empty?
-    data['delete']['user_id'] = User.new(self, data['delete']['user_id'])
-    e = Delete.new(data)
-    e.load_all {
-      @handlers.delete.each{|h| h.call(e)}
-    }
+    unless @handlers.delete.empty?
+      delete = Delete.new(self, data)
+      delete.load_all { |f|
+        @handlers.delete.each{|h| h.call(f)}
+      }
+    end
   end
   
   def dispatch_reconnect
@@ -125,45 +126,44 @@ class Chirpstream
     @handlers.reconnect.each{|h| h.call}
   end
   
+  def handle_tweet(parsed_data)
+    if parsed_data['friends']
+      dispatch_friend(parsed_data)
+    elsif parsed_data['text']
+      dispatch_tweet(parsed_data)
+    elsif parsed_data['event']
+      case parsed_data['event']
+      when 'follow'
+        dispatch_follow(parsed_data)
+      when 'favorite'
+        dispatch_favorite(parsed_data)
+      when 'retweet'
+        dispatch_retweet(parsed_data)
+      else
+        puts "weird event"
+        pp parsed_data
+      end
+    elsif parsed_data['delete']
+      dispatch_delete(parsed_data)
+    else
+      puts "i didn't know what to do with this!"
+      pp parsed_data
+    end
+  end
+  
   def connect
     unless EM.reactor_running?
       EM.run { connect }
     else
+      parser = Yajl::Parser.new
+      parser.on_parse_complete = method(:handle_tweet)
       http = EM::HttpRequest.new(@connect_url).get :head => {'authorization' => [@username, @password]}
       http.errback { |e, err|
         dispatch_reconnect
         connect
       }
-      http.stream { |chunk| 
-        @data << chunk
-        begin
-          parsed_data = JSON.parse(@data)
-          @data = ''
-          if parsed_data['friends']
-            dispatch_friend(parsed_data)
-          elsif parsed_data['text']
-            dispatch_tweet(parsed_data)
-          elsif parsed_data['event']
-            case parsed_data['event']
-            when 'follow'
-              dispatch_follow(parsed_data)
-            when 'favorite'
-              dispatch_favorite(parsed_data)
-            when 'retweet'
-              dispatch_retweet(parsed_data)
-            else
-              puts "weird event"
-              pp parsed_data
-            end
-          elsif parsed_data['delete']
-            dispatch_delete(parsed_data)
-          else
-            puts "i didn't know what to do with this!"
-            pp parsed_data
-          end
-        rescue JSON::ParserError
-          #puts "need more"
-        end
+      http.stream { |chunk|
+        parser << chunk
       }
     end
   end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Joshua Hull
@@ -14,8 +14,8 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-04-22 00:00:00 -04:00
-default_executable: 
+date: 2010-05-01 00:00:00 -04:00
+default_executable: chirp_growl
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: eventmachine
@@ -46,17 +46,17 @@ dependencies:
   type: :runtime
   version_requirements: *id002
 - !ruby/object:Gem::Dependency 
-  name: json
+  name: yajl-ruby
   prerelease: false
   requirement: &id003 !ruby/object:Gem::Requirement 
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
         segments: 
-        - 1
-        - 2
-        - 4
-        version: 1.2.4
+        - 0
+        - 7
+        - 5
+        version: 0.7.5
   type: :runtime
   version_requirements: *id003
 - !ruby/object:Gem::Dependency 
@@ -75,8 +75,8 @@ dependencies:
   version_requirements: *id004
 description: Eventmachine-based Chirpstream client
 email: joshbuddy@gmail.com
-executables: []
-
+executables: 
+- chirp_growl
 extensions: []
 
 extra_rdoc_files: []
@@ -86,14 +86,15 @@ files:
 - Rakefile
 - Readme.rdoc
 - VERSION
+- bin/chirp_growl
 - lib/chirpstream.rb
 - lib/chirpstream/event.rb
 - lib/chirpstream/event/delete.rb
 - lib/chirpstream/event/favorite.rb
 - lib/chirpstream/event/follow.rb
 - lib/chirpstream/event/retweet.rb
-- lib/chirpstream/loadable.rb
 - lib/chirpstream/tweet.rb
+- lib/chirpstream/twitter_object.rb
 - lib/chirpstream/user.rb
 has_rdoc: true
 homepage: http://github.com/joshbuddy/chirpstream