@@ -69,26 +69,26 @@ module Twibot processed[:tweet] = tweets.first.id if tweets.length > 0 processed[:reply] = tweets.first.id if tweets.length > 0 - + # for searches, use latest tweet on public timeline # if handle_tweets && config[:timeline_for].to_s != "public" sandbox { tweets = twitter.timeline_for(:public, { :count => 1 }) } end if tweets.length > 0 - handlers[:search].each_key {|q| processed[:search][q] = tweets.first.id } + handlers_for_type(:search).each_key {|q| processed[:search][q] = tweets.first.id } end - + load_followers - + when Numeric, /\d+/ # a tweet ID to start from processed[:tweet] = processed[:reply] = processed[:message] = config[:process] handlers[:search].each_key {|q| processed[:search][q] = config[:process] } else abort "Unknown process option #{config[:process]}, aborting..." end - + load_friends unless handlers_for_type(:follower).empty? - + poll end @@ -107,60 +107,60 @@ module Twibot message_count += receive_replies || 0 message_count += receive_tweets || 0 message_count += receive_searches || 0 - + receive_followers - + run_hook :after_all, message_count - + interval = message_count > 0 ? min_interval : [interval + step, max].min - + log.debug "#{config[:host]} sleeping for #{interval}s" sleep interval end end - - + + def friend_ids @friend_ids ||= {} end - + def add_friend!(user_or_id, only_local=false) id = id_for_user_or_id(user_or_id) sandbox(0) { twitter.friend(:add, id) } unless only_local friend_ids[id] = true end - + def remove_friend!(user_or_id, only_local=false) id = id_for_user_or_id(user_or_id) sandbox(0) { twitter.friend(:remove, id) } unless only_local friend_ids[id] = false end - + def is_friend?(user_or_id) !!friend_ids[id_for_user_or_id(user_or_id)] end - + def follower_ids @follower_ids ||= {} end - + def add_follower!(user_or_id) follower_ids[id_for_user_or_id(user_or_id)] = true end - + def remove_follower!(user_or_id) follower_ids[id_for_user_or_id(user_or_id)] = false end - + def is_follower?(user_or_id) !!follower_ids[id_for_user_or_id(user_or_id)] end - + def id_for_user_or_id(user_or_id) (user_or_id.respond_to?(:screen_name) ? user_or_id.id : user_or_id).to_i end - - + + # # retrieve a list of friend ids and store it as a Hash # @@ -169,7 +169,7 @@ module Twibot twitter.graph(:friends, config[:login]).each {|id| add_friend!(id, true) } end end - + # # retrieve a list of friend ids and store it as a Hash # @@ -178,22 +178,22 @@ module Twibot twitter.graph(:followers, config[:login]).each {|id| add_follower!(id) } end end - - + + # # returns a Hash of all registered hooks # def hooks @hooks ||= {} end - + # # registers a block to be called at the given +event+ # def add_hook(event, &blk) hooks[event.to_sym] = blk end - + # # calls the hook method for the +event+ if one has # been defined @@ -201,7 +201,7 @@ module Twibot def run_hook(event, *args) hooks[event.to_sym].call(*args) if hooks[event.to_sym].respond_to? :call end - + # # Receive direct messages # @@ -249,22 +249,22 @@ module Twibot # def receive_searches result_count = 0 - + handlers_for_type(:search).each_pair do |query, search_handlers| options = { :q => query, :rpp => 100 } [:lang, :geocode].each do |param| options[param] = search_handlers.first.options[param] if search_handlers.first.options[param] end options[:since_id] = processed[:search][query] if processed[:search][query] - + result_count += sandbox(0) do dispatch_messages([:search, query], twitter.search(options.merge(options)), %w{tweet tweets}.map {|l| "#{l} for \"#{query}\""}) end end - + result_count end - + # # Receive any new followers # @@ -279,7 +279,7 @@ module Twibot end log.info "#{config[:host]}: Received #{newbies.size} new #{newbies.size == 1 ? 'follower' : 'followers'}" end - + # # Dispatch a collection of messages # @@ -296,7 +296,7 @@ module Twibot log.info "#{config[:host]}: Received #{num} #{num == 1 ? labels[0] : labels[1]}" num end - + # # invokes the given block, running the before and # after hooks for the given type @@ -308,7 +308,7 @@ module Twibot run_hook :"after_#{event}" value end - + # # Return logger instance # lib/twibot/bot.rb @@ -7,15 +7,15 @@ module Twibot handlers_for_type(type) << handler handler end - + def handlers_for_type(type) if type.is_a? Array handlers[type.first][type.last] ||= [] else - handlers[type] + handlers[type] || {} end end - + def dispatch(type, message) handlers_for_type(type).each { |handler| handler.dispatch(message) } end lib/twibot/handlers.rb 2679b4a966823f6e4190e1dad115fbda3b3850af Christian Johansen christian@cjohansen.no http://github.com/cjohansen/twibot/commit/8cd96251d9de93e21795c7ea659cef39d202f4ae 8cd96251d9de93e21795c7ea659cef39d202f4ae 2009-08-14T13:59:34-07:00 2009-08-14T13:59:34-07:00 always return hash from handlers_for_type 68b2f06cdb1f21949d7b82ca1b40e9fbbb8d0644 Christian Johansen christian@cjohansen.no