module Twitter
  module CLI
    module Helpers
      class NoActiveAccount < StandardError; end
      class NoAccounts < StandardError; end
      
      def output_tweets(collection, options={})
        options = {
          :cache        => false,
          :since_prefix => '',
          :empty_msg    => 'Nothing new since your last check.'
        }.merge(options)
        
        if collection.size > 0
          justify   = collection.collect { |s| s.user.screen_name }.max { |a,b| a.length <=> b.length }.length rescue 0
          indention = ' ' * (justify + 3)
          say("\n#{indention}#{collection.size} new tweet(s) found.\n\n")
          collection.each do |s|
            Tweet.create_from_tweet(current_account, s) if options[:cache]
            
            occurred_at    = Time.parse(s.created_at).strftime('On %b %d at %l:%M%P')
            formatted_time = '-' * occurred_at.length + "\n#{indention}#{occurred_at}"
            formatted_name = s.user.screen_name.rjust(justify + 1)
            formatted_msg  = ''
            
            s.text.split(' ').each_with_index do |word, idx|
              formatted_msg += "#{word} "
              
              sixth_word = idx != 0 && idx % 6 == 0
              formatted_msg += "\n#{indention}" if sixth_word
            end
            
            say "#{CGI::unescapeHTML(formatted_name)}: #{CGI::unescapeHTML(formatted_msg)}\n#{indention}#{formatted_time}\n\n"
          end
          
          Configuration["#{options[:since_prefix]}_since_id"] = collection.first.id
        else
          say(options[:empty_msg])
        end
      end
      
      def base(username=current_account.username, password=current_account.password)
        @base ||= Twitter::Base.new(username, password)
      end
      
      def current_account
        @current_account ||= Account.active        
        raise Account.count == 0 ? NoAccounts : NoActiveAccount if @current_account.nil?
        @current_account
      end
      
      def attempt_import(&block)
        tweet_file = File.join(ENV['HOME'], '.twitter')
        if File.exists?(tweet_file)
          say '.twitter file found, attempting import...'
          config = YAML::load(File.read(tweet_file))
          if !config['email'].nil? && !config['password'].nil?
            Account.add(:username => config['email'], :password => config['password'])
            say 'Account imported'
            block.call if block_given?
            true
          else
            say "Either your username or password were blank in your .twitter file so I could not import. Use 'twitter add' to add an account."
            false
          end
        end
      end
                  
      def do_work(&block)
        connect
        begin
          block.call
        rescue Twitter::RateExceeded
          say("Twitter says you've been making too many requests. Wait for a bit and try again.")
        rescue Twitter::Unavailable
          say("Twitter is unavailable right now. Try again later.")
        rescue Twitter::CantConnect => msg
          say("Can't connect to twitter because: #{msg}")
        rescue Twitter::CLI::Helpers::NoActiveAccount
          say("You have not set an active account. Use 'twitter change' to set one now.")
        rescue Twitter::CLI::Helpers::NoAccounts
          unless attempt_import { block.call }
            say("You have not created any accounts. Use 'twitter add' to create one now.")
          end
        end
      end
      
      def connect
        ActiveRecord::Base.logger = Logger.new('/tmp/twitter_ar_logger.log')
        ActiveRecord::Base.establish_connection(Twitter::CLI::Config)
        ActiveRecord::Base.connection
      end

      def migrate
        connect
        ActiveRecord::Migrator.migrate("#{CLI_ROOT}/migrations/")
      end

      def connect_and_migrate
        say('Attempting to establish connection...')
        connect
        say('Connection established...migrating database...')
        migrate
      end
    end
  end
end