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