t 0.1.0 → 0.2.0

data/lib/t/cli/list/add/all.rb

@@ -0,0 +1,169 @@
+require 'active_support/core_ext/array/grouping'
+require 't/rcfile'
+require 'thor'
+require 'twitter'
+
+module T
+  class CLI
+    class List
+      class Add
+        class All < Thor
+          DEFAULT_HOST = 'api.twitter.com'
+          DEFAULT_PROTOCOL = 'https'
+          MAX_USERS_PER_LIST = 500
+
+          check_unknown_options!
+
+          def initialize(*)
+            super
+            @rcfile = RCFile.instance
+          end
+
+          desc "friends LIST_NAME", "Add all friends to a list."
+          def friends(list_name)
+            list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            existing_list_members = list_member_ids.length
+            if existing_list_members >= MAX_USERS_PER_LIST
+              return say "The list \"#{list_name}\" are already contains the maximum of #{MAX_USERS_PER_LIST} members."
+            end
+            friend_ids = []
+            cursor = -1
+            until cursor == 0
+              friends = client.friend_ids(:cursor => cursor)
+              friend_ids += friends.ids
+              cursor = friends.next_cursor
+            end
+            list_member_ids_to_add = (friend_ids - list_member_ids)
+            number = list_member_ids_to_add.length
+            if number.zero?
+              return say "All of @#{@rcfile.default_profile[0]}'s friends are already members of the list \"#{list_name}\"."
+            elsif existing_list_members + number > MAX_USERS_PER_LIST
+              return unless yes? "Lists can't have more than #{MAX_USERS_PER_LIST} members. Do you want to add up to #{MAX_USERS_PER_LIST} friends to the list \"#{list_name}\"?"
+            else
+              return unless yes? "Are you sure you want to add #{number} #{number == 1 ? 'friend' : 'friends'} to the list \"#{list_name}\"?"
+            end
+            max_members_to_add = MAX_USERS_PER_LIST - existing_list_members
+            list_member_ids_to_add[0...max_members_to_add].in_groups_of(100, false) do |user_id_group|
+              client.list_add_members(list_name, user_id_group)
+            end
+            number_added = [number, max_members_to_add].min
+            say "@#{@rcfile.default_profile[0]} added #{number_added} #{number_added == 1 ? 'friend' : 'friends'} to the list \"#{list_name}\"."
+            say
+            say "Run `#{$0} list remove all friends #{list_name}` to undo."
+          end
+
+          desc "followers LIST_NAME", "Add all followers to a list."
+          def followers(list_name)
+            list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            existing_list_members = list_member_ids.length
+            if existing_list_members >= MAX_USERS_PER_LIST
+              return say "The list \"#{list_name}\" are already contains the maximum of #{MAX_USERS_PER_LIST} members."
+            end
+            follower_ids = []
+            cursor = -1
+            until cursor == 0
+              followers = client.follower_ids(:cursor => cursor)
+              follower_ids += followers.ids
+              cursor = followers.next_cursor
+            end
+            list_member_ids_to_add = (follower_ids - list_member_ids)
+            number = list_member_ids_to_add.length
+            if number.zero?
+              return say "All of @#{@rcfile.default_profile[0]}'s followers are already members of the list \"#{list_name}\"."
+            elsif existing_list_members + number > MAX_USERS_PER_LIST
+              return unless yes? "Lists can't have more than #{MAX_USERS_PER_LIST} members. Do you want to add up to #{MAX_USERS_PER_LIST} followers to the list \"#{list_name}\"?"
+            else
+              return unless yes? "Are you sure you want to add #{number} #{number == 1 ? 'follower' : 'followers'} to the list \"#{list_name}\"?"
+            end
+            max_members_to_add = MAX_USERS_PER_LIST - existing_list_members
+            list_member_ids_to_add[0...max_members_to_add].in_groups_of(100, false) do |user_id_group|
+              client.list_add_members(list_name, user_id_group)
+            end
+            number_added = [number, max_members_to_add].min
+            say "@#{@rcfile.default_profile[0]} added #{number_added} #{number_added == 1 ? 'follower' : 'followers'} to the list \"#{list_name}\"."
+            say
+            say "Run `#{$0} list remove all followers #{list_name}` to undo."
+          end
+
+          desc "listed FROM_LIST_NAME TO_LIST_NAME", "Add all list memebers to a list."
+          def listed(from_list_name, to_list_name)
+            to_list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(to_list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              to_list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            existing_list_members = to_list_member_ids.length
+            if existing_list_members >= MAX_USERS_PER_LIST
+              return say "The list \"#{to_list_name}\" are already contains the maximum of #{MAX_USERS_PER_LIST} members."
+            end
+            from_list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(from_list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              from_list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            list_member_ids_to_add = (from_list_member_ids - to_list_member_ids)
+            number = list_member_ids_to_add.length
+            if number.zero?
+              return say "All of the members of the list \"#{from_list_name}\" are already members of the list \"#{to_list_name}\"."
+            elsif existing_list_members + number > MAX_USERS_PER_LIST
+              return unless yes? "Lists can't have more than #{MAX_USERS_PER_LIST} members. Do you want to add up to #{MAX_USERS_PER_LIST} members to the list \"#{to_list_name}\"?"
+            else
+              return unless yes? "Are you sure you want to add #{number} #{number == 1 ? 'member' : 'members'} to the list \"#{to_list_name}\"?"
+            end
+            max_members_to_add = MAX_USERS_PER_LIST - existing_list_members
+            list_member_ids_to_add[0...max_members_to_add].in_groups_of(100, false) do |user_id_group|
+              client.list_add_members(to_list_name, user_id_group)
+            end
+            number_added = [number, max_members_to_add].min
+            say "@#{@rcfile.default_profile[0]} added #{number_added} #{number_added == 1 ? 'member' : 'members'} to the list \"#{to_list_name}\"."
+            say
+            say "Run `#{$0} list remove all listed #{from_list_name} #{to_list_name}` 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
+          end
+
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,67 @@
+require 't/core_ext/string'
+require 't/rcfile'
+require 'thor'
+require 'twitter'
+
+module T
+  class CLI
+    class List
+      class Remove < Thor
+        DEFAULT_HOST = 'api.twitter.com'
+        DEFAULT_PROTOCOL = 'https'
+
+        check_unknown_options!
+
+        def initialize(*)
+          super
+          @rcfile = RCFile.instance
+        end
+
+        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.each do |screen_name|
+            screen_name = screen_name.strip_at
+            client.list_remove_member(list_name, screen_name)
+            say "@#{@rcfile.default_profile[0]} removed @#{screen_name} from the list \"#{list_name}\"."
+          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
+
+        desc "all SUBCOMMAND ...ARGS", "Remove all users to a list."
+        require 't/cli/list/remove/all'
+        subcommand 'all', CLI::List::Remove::All
+
+      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
+end

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

@@ -0,0 +1,162 @@
+require 't/rcfile'
+require 'thor'
+require 'twitter'
+
+module T
+  class CLI
+    class List
+      class Remove
+        class All < Thor
+          DEFAULT_HOST = 'api.twitter.com'
+          DEFAULT_PROTOCOL = 'https'
+
+          check_unknown_options!
+
+          def initialize(*)
+            super
+            @rcfile = RCFile.instance
+          end
+
+          desc "friends LIST_NAME", "Remove all friends from a list."
+          def friends(list_name)
+            list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            friend_ids = []
+            cursor = -1
+            until cursor == 0
+              friends = client.friend_ids(:cursor => cursor)
+              friend_ids += friends.ids
+              cursor = friends.next_cursor
+            end
+            list_member_ids_to_remove = (friend_ids - list_member_ids)
+            number = list_member_ids_to_remove.length
+            if number.zero?
+              return say "None of @#{@rcfile.default_profile[0]}'s friends are members of the list \"#{list_name}\"."
+            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.each do |list_member_id|
+              client.list_remove_member(list_name, list_member_id)
+            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."
+          end
+
+          desc "followers LIST_NAME", "Remove all followers from a list."
+          def followers(list_name)
+            list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            follower_ids = []
+            cursor = -1
+            until cursor == 0
+              followers = client.follower_ids(:cursor => cursor)
+              follower_ids += followers.ids
+              cursor = followers.next_cursor
+            end
+            list_member_ids_to_remove = (follower_ids - list_member_ids)
+            number = list_member_ids_to_remove.length
+            if number.zero?
+              return say "None of @#{@rcfile.default_profile[0]}'s followers are members of the list \"#{list_name}\"."
+            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.each do |list_member_id|
+              client.list_remove_member(list_name, list_member_id)
+            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."
+          end
+
+          desc "listed FROM_LIST_NAME TO_LIST_NAME", "Remove all list members from a list."
+          def listed(from_list_name, to_list_name)
+            to_list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(to_list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              to_list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            from_list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(from_list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              from_list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            list_member_ids_to_remove = (from_list_member_ids - to_list_member_ids)
+            number = list_member_ids_to_remove.length
+            if number.zero?
+              return say "None of the members of the list \"#{from_list_name}\" are members of the list \"#{to_list_name}\"."
+            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.each do |list_member_id|
+              client.list_remove_member(to_list_name, list_member_id)
+            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."
+          end
+
+          desc "members LIST_NAME", "Remove all members from a list."
+          def members(list_name)
+            list_member_ids = []
+            cursor = -1
+            until cursor == 0
+              list_members = client.list_members(list_name, :cursor => cursor, :skip_status => true, :include_entities => false)
+              list_member_ids += list_members.users.collect{|user| user.id}
+              cursor = list_members.next_cursor
+            end
+            number = list_member_ids.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_member_ids.each do |list_member_id|
+              client.list_remove_member(list_name, list_member_id)
+            end
+            say "@#{@rcfile.default_profile[0]} removed #{number} #{number == 1 ? 'member' : 'members'} from the list \"#{list_name}\"."
+          end
+          map %w(users) => :members
+
+        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
+  end
+end

data/lib/t/cli/set.rb

@@ -0,0 +1,86 @@
+require 't/core_ext/string'
+require 't/rcfile'
+require 'thor'
+require 'twitter'
+
+module T
+  class CLI
+    class Set < Thor
+      DEFAULT_HOST = 'api.twitter.com'
+      DEFAULT_PROTOCOL = 'https'
+
+      check_unknown_options!
+
+      def initialize(*)
+        super
+        @rcfile = RCFile.instance
+      end
+
+      desc "bio DESCRIPTION", "Edits your Bio information on your Twitter profile."
+      def bio(description)
+        client.update_profile(:description => description, :include_entities => false)
+        say "@#{@rcfile.default_profile[0]}'s bio has been updated."
+      end
+
+      desc "default SCREEN_NAME [CONSUMER_KEY]", "Set your default account."
+      def default(screen_name, consumer_key=nil)
+        screen_name = screen_name.strip_at
+        @rcfile.path = parent_options['profile'] if parent_options['profile']
+        consumer_key = rcfile[screen_name].keys.last if consumer_key.nil?
+        @rcfile.default_profile = {'username' => screen_name, 'consumer_key' => consumer_key}
+        say "Default account has been updated."
+      end
+
+      desc "language LANGUAGE_NAME", "Selects the language you'd like to receive notifications in."
+      def language(language_name)
+        client.settings(:lang => language_name)
+        say "@#{@rcfile.default_profile[0]}'s language has been updated."
+      end
+
+      desc "location PLACE_NAME", "Updates the location field in your profile."
+      def location(place_name)
+        client.update_profile(:location => place_name, :include_entities => false)
+        say "@#{@rcfile.default_profile[0]}'s location has been updated."
+      end
+
+      desc "name NAME", "Sets the name field on your Twitter profile."
+      def name(name)
+        client.update_profile(:name => name, :include_entities => false)
+        say "@#{@rcfile.default_profile[0]}'s name has been updated."
+      end
+
+      desc "url URL", "Sets the URL field on your profile."
+      def url(url)
+        client.update_profile(:url => url, :include_entities => false)
+        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