t 0.3.1 → 0.4.0

data/lib/t/cli/list/remove.rb

@@ -1,19 +1,20 @@
+require 'active_support/core_ext/array/grouping'
 require 'retryable'
 require 't/core_ext/enumerable'
 require 't/core_ext/string'
 require 't/collectable'
 require 't/rcfile'
+require 't/requestable'
 require 'thor'
-require 'twitter'
 
 module T
   class CLI
     class List
       class Remove < Thor
         include T::Collectable
+        include T::Requestable
 
-        DEFAULT_HOST = 'api.twitter.com'
-        DEFAULT_PROTOCOL = 'https'
+        MAX_USERS_PER_REQUEST = 100
 
         check_unknown_options!
 
@@ -37,14 +38,14 @@ module T
           else
             return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'friend' : 'friends'} from the list \"#{list_name}\"?"
           end
-          list_member_ids_to_remove.threaded_map do |list_member_id|
+          list_member_ids_to_remove.in_groups_of(MAX_USERS_PER_REQUEST, false).threaded_each do |user_id_group|
             retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
-              client.list_remove_member(list_name, list_member_id)
+              client.list_remove_members(list_name, user_id_group)
             end
           end
           say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'friend' : 'friends'} from the list \"#{list_name}\"."
           say
-          say "Run `#{$0} list add all friends #{list_name}` to undo."
+          say "Run `#{File.basename($0)} list add all friends #{list_name}` to undo."
         end
 
         desc "followers LIST_NAME", "Remove all followers from a list."
@@ -62,14 +63,14 @@ module T
           else
             return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'follower' : 'followers'} from the list \"#{list_name}\"?"
           end
-          list_member_ids_to_remove.threaded_map do |list_member_id|
+          list_member_ids_to_remove.in_groups_of(MAX_USERS_PER_REQUEST, false).threaded_each do |user_id_group|
             retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
-              client.list_remove_member(list_name, list_member_id)
+              client.list_remove_members(list_name, user_id_group)
             end
           end
           say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'follower' : 'followers'} from the list \"#{list_name}\"."
           say
-          say "Run `#{$0} list add all followers #{list_name}` to undo."
+          say "Run `#{File.basename($0)} list add all followers #{list_name}` to undo."
         end
 
         desc "listed FROM_LIST_NAME TO_LIST_NAME", "Remove all list members from a list."
@@ -87,14 +88,14 @@ module T
           else
             return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{to_list_name}\"?"
           end
-          list_member_ids_to_remove.threaded_map do |list_member_id|
+          list_member_ids_to_remove.in_groups_of(MAX_USERS_PER_REQUEST, false).threaded_each do |user_id_group|
             retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
-              client.list_remove_member(to_list_name, list_member_id)
+              client.list_remove_members(to_list_name, user_id_group)
             end
           end
           say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{to_list_name}\"."
           say
-          say "Run `#{$0} list add all listed #{from_list_name} #{to_list_name}` to undo."
+          say "Run `#{File.basename($0)} list add all listed #{from_list_name} #{to_list_name}` to undo."
         end
 
         desc "members LIST_NAME", "Remove all members from a list."
@@ -105,9 +106,9 @@ module T
           number = list_members.length
           return say "The list \"#{list_name}\" doesn't have any members." if number.zero?
           return unless yes? "Are you sure you want to remove #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{list_name}\"?"
-          list_members.collect(&:id).threaded_map do |list_member_id|
+          list_members.collect(&:id).in_groups_of(MAX_USERS_PER_REQUEST, false).threaded_each do |user_id_group|
             retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
-              client.list_remove_member(list_name, list_member_id)
+              client.list_remove_members(list_name, user_id_group)
             end
           end
           say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{list_name}\"."
@@ -117,41 +118,16 @@ module T
         desc "users LIST_NAME SCREEN_NAME [SCREEN_NAME...]", "Remove users from a list."
         def users(list_name, screen_name, *screen_names)
           screen_names.unshift(screen_name)
-          screen_names.threaded_map do |screen_name|
+          screen_names.map!(&:strip_at)
+          screen_names.in_groups_of(MAX_USERS_PER_REQUEST, false).threaded_each do |user_id_group|
             retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
-              client.list_remove_member(list_name, screen_name)
+              client.list_remove_members(list_name, user_id_group)
             end
           end
           number = screen_names.length
           say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'user' : 'users'} from the list \"#{list_name}\"."
           say
-          say "Run `#{$0} list add users #{list_name} #{screen_names.join(' ')}` to undo."
-        end
-
-      private
-
-        def base_url
-          "#{protocol}://#{host}"
-        end
-
-        def client
-          return @client if @client
-          @rcfile.path = parent_options['profile'] if parent_options['profile']
-          @client = Twitter::Client.new(
-            :endpoint => base_url,
-            :consumer_key => @rcfile.default_consumer_key,
-            :consumer_secret => @rcfile.default_consumer_secret,
-            :oauth_token => @rcfile.default_token,
-            :oauth_token_secret  => @rcfile.default_secret
-          )
-        end
-
-        def host
-          parent_options['host'] || DEFAULT_HOST
-        end
-
-        def protocol
-          parent_options['no_ssl'] ? 'http' : DEFAULT_PROTOCOL
+          say "Run `#{File.basename($0)} list add users #{list_name} #{screen_names.join(' ')}` to undo."
         end
 
       end

data/lib/t/cli/search.rb

@@ -3,17 +3,16 @@ require 'pager'
 require 'retryable'
 require 't/core_ext/enumerable'
 require 't/rcfile'
+require 't/requestable'
 require 'thor'
-require 'twitter'
 
 module T
   class CLI
     class Search < Thor
       include ActionView::Helpers::DateHelper
       include Pager
+      include T::Requestable
 
-      DEFAULT_HOST = 'api.twitter.com'
-      DEFAULT_PROTOCOL = 'https'
       DEFAULT_NUM_RESULTS = 20
       MAX_PAGES = 16
       MAX_NUM_RESULTS = 200
@@ -40,12 +39,12 @@ module T
         end
       end
 
-      desc "timeline QUERY", "Returns Tweets in your timeline that match a specified query."
-      def timeline(query)
+      desc "favorites QUERY", "Returns Tweets you've favorited that mach a specified query."
+      def favorites(query)
         timeline = 1.upto(MAX_PAGES).threaded_map do |page|
           retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
-            client.home_timeline(:page => page, :count => MAX_NUM_RESULTS).map do |status|
-              status if /#{query}/i.match(status.text)
+            client.favorites(:page => page, :count => MAX_NUM_RESULTS).select do |status|
+              /#{query}/i.match(status.text)
             end
           end
         end
@@ -54,15 +53,14 @@ module T
           say "#{status.user.screen_name.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
         end
       end
-      map %w(tl) => :timeline
+      map %w(faves) => :favorites
 
-      desc "user SCREEN_NAME QUERY", "Returns Tweets in a user's timeline that match a specified query."
-      def user(screen_name, query)
-        screen_name = screen_name.strip_at
+      desc "mentions QUERY", "Returns Tweets mentioning you that mach a specified query."
+      def mentions(query)
         timeline = 1.upto(MAX_PAGES).threaded_map do |page|
           retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
-            client.user_timeline(screen_name, :page => page, :count => MAX_NUM_RESULTS).map do |status|
-              status if /#{query}/i.match(status.text)
+            client.mentions(:page => page, :count => MAX_NUM_RESULTS).select do |status|
+              /#{query}/i.match(status.text)
             end
           end
         end
@@ -71,31 +69,54 @@ module T
           say "#{status.user.screen_name.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
         end
       end
+      map %w(replies) => :mentions
 
-    private
-
-      def base_url
-        "#{protocol}://#{host}"
-      end
-
-      def client
-        return @client if @client
-        @rcfile.path = parent_options['profile'] if parent_options['profile']
-        @client = Twitter::Client.new(
-          :endpoint => base_url,
-          :consumer_key => @rcfile.default_consumer_key,
-          :consumer_secret => @rcfile.default_consumer_secret,
-          :oauth_token => @rcfile.default_token,
-          :oauth_token_secret  => @rcfile.default_secret
-        )
+      desc "retweets QUERY", "Returns Tweets you've retweeted that mach a specified query."
+      def retweets(query)
+        timeline = 1.upto(MAX_PAGES).threaded_map do |page|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.retweeted_by(:page => page, :count => MAX_NUM_RESULTS).select do |status|
+              /#{query}/i.match(status.text)
+            end
+          end
+        end
+        page unless T.env.test?
+        timeline.flatten.compact.each do |status|
+          say "#{status.user.screen_name.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
+        end
       end
+      map %w(rts) => :retweets
 
-      def host
-        parent_options['host'] || DEFAULT_HOST
+      desc "timeline QUERY", "Returns Tweets in your timeline that match a specified query."
+      def timeline(query)
+        timeline = 1.upto(MAX_PAGES).threaded_map do |page|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.home_timeline(:page => page, :count => MAX_NUM_RESULTS).select do |status|
+              /#{query}/i.match(status.text)
+            end
+          end
+        end
+        page unless T.env.test?
+        timeline.flatten.compact.each do |status|
+          say "#{status.user.screen_name.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
+        end
       end
+      map %w(tl) => :timeline
 
-      def protocol
-        parent_options['no_ssl'] ? 'http' : DEFAULT_PROTOCOL
+      desc "user SCREEN_NAME QUERY", "Returns Tweets in a user's timeline that match a specified query."
+      def user(screen_name, query)
+        screen_name = screen_name.strip_at
+        timeline = 1.upto(MAX_PAGES).threaded_map do |page|
+          retryable(:tries => 3, :on => Twitter::Error::ServerError, :sleep => 0) do
+            client.user_timeline(screen_name, :page => page, :count => MAX_NUM_RESULTS).select do |status|
+              /#{query}/i.match(status.text)
+            end
+          end
+        end
+        page unless T.env.test?
+        timeline.flatten.compact.each do |status|
+          say "#{status.user.screen_name.rjust(MAX_SCREEN_NAME_SIZE)}: #{status.text} (#{time_ago_in_words(status.created_at)} ago)"
+        end
       end
 
     end

data/lib/t/cli/set.rb

@@ -1,13 +1,12 @@
 require 't/core_ext/string'
 require 't/rcfile'
+require 't/requestable'
 require 'thor'
-require 'twitter'
 
 module T
   class CLI
     class Set < Thor
-      DEFAULT_HOST = 'api.twitter.com'
-      DEFAULT_PROTOCOL = 'https'
+      include T::Requestable
 
       check_unknown_options!
 
@@ -55,32 +54,6 @@ module T
         say "@#{@rcfile.default_profile[0]}'s URL has been updated."
       end
 
-    private
-
-      def base_url
-        "#{protocol}://#{host}"
-      end
-
-      def client
-        return @client if @client
-        @rcfile.path = parent_options['profile'] if parent_options['profile']
-        @client = Twitter::Client.new(
-          :endpoint => base_url,
-          :consumer_key => @rcfile.default_consumer_key,
-          :consumer_secret => @rcfile.default_consumer_secret,
-          :oauth_token => @rcfile.default_token,
-          :oauth_token_secret  => @rcfile.default_secret
-        )
-      end
-
-      def host
-        parent_options['host'] || DEFAULT_HOST
-      end
-
-      def protocol
-        parent_options['no_ssl'] ? 'http' : DEFAULT_PROTOCOL
-      end
-
     end
   end
 end

data/lib/t/cli/unfollow.rb

@@ -3,16 +3,14 @@ require 't/core_ext/enumerable'
 require 't/core_ext/string'
 require 't/collectable'
 require 't/rcfile'
+require 't/requestable'
 require 'thor'
-require 'twitter'
 
 module T
   class CLI
     class Unfollow < Thor
       include T::Collectable
-
-      DEFAULT_HOST = 'api.twitter.com'
-      DEFAULT_PROTOCOL = 'https'
+      include T::Requestable
 
       check_unknown_options!
 
@@ -36,7 +34,7 @@ module T
         end
         say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
         say
-        say "Run `#{$0} follow all listed #{list_name}` to follow again."
+        say "Run `#{File.basename($0)} follow all listed #{list_name}` to follow again."
       end
 
       desc "followers", "Unfollow all followers."
@@ -58,7 +56,7 @@ module T
         end
         say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
         say
-        say "Run `#{$0} follow all followers` to stop."
+        say "Run `#{File.basename($0)} follow all followers` to stop."
       end
 
       desc "friends", "Unfollow all friends."
@@ -77,7 +75,7 @@ module T
         end
         say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
         say
-        say "Run `#{$0} follow users #{screen_names.join(' ')}` to follow again."
+        say "Run `#{File.basename($0)} follow users #{screen_names.join(' ')}` to follow again."
       end
       map %w(everyone everybody) => :friends
 
@@ -101,7 +99,7 @@ module T
         end
         say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
         say
-        say "Run `#{$0} follow users #{screen_names.join(' ')}` to follow again."
+        say "Run `#{File.basename($0)} follow users #{screen_names.join(' ')}` to follow again."
       end
 
       desc "users SCREEN_NAME [SCREEN_NAME...]", "Allows you to stop following users."
@@ -115,33 +113,7 @@ module T
         number = screen_names.length
         say "@#{@rcfile.default_profile[0]} is no longer following #{number} #{number == 1 ? 'user' : 'users'}."
         say
-        say "Run `#{$0} follow users #{screen_names.join(' ')}` to follow again."
-      end
-
-    private
-
-      def base_url
-        "#{protocol}://#{host}"
-      end
-
-      def client
-        return @client if @client
-        @rcfile.path = parent_options['profile'] if parent_options['profile']
-        @client = Twitter::Client.new(
-          :endpoint => base_url,
-          :consumer_key => @rcfile.default_consumer_key,
-          :consumer_secret => @rcfile.default_consumer_secret,
-          :oauth_token => @rcfile.default_token,
-          :oauth_token_secret  => @rcfile.default_secret
-        )
-      end
-
-      def host
-        parent_options['host'] || DEFAULT_HOST
-      end
-
-      def protocol
-        parent_options['no_ssl'] ? 'http' : DEFAULT_PROTOCOL
+        say "Run `#{File.basename($0)} follow users #{screen_names.join(' ')}` to follow again."
       end
 
     end

data/lib/t/core_ext/enumerable.rb

@@ -1,5 +1,14 @@
 module Enumerable
 
+  def threaded_each
+    threads = []
+    result = each do |object|
+      threads << Thread.new{yield object}
+    end
+    threads.each(&:join)
+    result
+  end
+
   def threaded_map
     results = map{nil}
     threads = []

data/lib/t/rcfile.rb

@@ -1,7 +1,3 @@
-# rcfile = RCFile.instance (should load current data from)
-# rcfile[profile] = {}
-# rcfile.write
-
 require 'singleton'
 require 'yaml'
 

data/lib/t/requestable.rb

@@ -0,0 +1,37 @@
+require 'twitter'
+
+module T
+  module Requestable
+    DEFAULT_HOST = 'api.twitter.com'
+    DEFAULT_PROTOCOL = 'https'
+
+    def self.included(base)
+
+    private
+
+      def base_url
+        "#{protocol}://#{host}"
+      end
+
+      def client
+        return @client if @client
+        @rcfile.path = parent_options['profile'] if parent_options['profile']
+        @client = Twitter::Client.new(
+          :endpoint => base_url,
+          :consumer_key => @rcfile.default_consumer_key,
+          :consumer_secret => @rcfile.default_consumer_secret,
+          :oauth_token => @rcfile.default_token,
+          :oauth_token_secret  => @rcfile.default_secret
+        )
+      end
+
+      def host
+        parent_options['host'] || DEFAULT_HOST
+      end
+
+      def protocol
+        parent_options['no_ssl'] ? 'http' : DEFAULT_PROTOCOL
+      end
+    end
+  end
+end