module ORS
  module Helpers

    include Config

    def setup_repo server
      info "[#{server}] installing codebase..."

      execute_command server, %(cd #{base_path}),
                              %(rm -rf #{deploy_directory}),
                              %(git clone #{repo}:#{name} #{deploy_directory}),
                              %(mkdir -p #{deploy_directory}/tmp/pids),
                              %(mkdir -p #{deploy_directory}/log)
    end

    def setup_ruby server
      info "[#{server}] installing ruby and gems..."

      execute_command server, prepare_initial_environment,
                              %(gem install rubygems-update),
                              %(gem update --system),
                              %(gem install bundler),
                              %(bundle install --without development test osx_development > bundler.log)
    end

    def update_code server
      info "[#{server}] updating codebase..."

      execute_command server, prepare_environment,
                              %(git fetch),
                              %(git checkout -q -f origin/#{environment}),
                              %(git reset --hard),
                              %(git submodule update --init)
    end

    def bundle_install server
      info "[#{server}] installing bundle..."

      execute_command server, prepare_environment,
                              %(bundle install --without development test osx_development > bundler.log)
    end

    def start_server server
      info "[#{server}] starting unicorn..."

      execute_command server, prepare_environment,
                              %(if [ -f config.ru ]; then RAILS_ENV=#{environment} bundle exec unicorn -c config/unicorn.rb -D -E #{environment}; else RAILS_ENV=#{environment} bundle exec unicorn_rails -c config/unicorn.rb -D -E #{environment}; fi)
    end

    def stop_server server
      info "[#{server}] stopping unicorn..."

      execute_command server, prepare_environment,
                              %(kill \\`cat tmp/pids/unicorn.pid\\`)
    end

    def restart_server server
      info "[#{server}] restarting unicorn..."

      execute_command server, prepare_environment,
                              %(kill -USR2 \\`cat tmp/pids/unicorn.pid\\`)
    end

    def run_migrations server
      info "[#{server}] running migrations..."

      execute_command server, prepare_environment,
                              %(RAILS_ENV=#{environment} bundle exec rake db:migrate db:seed)
    end

    def execute_in_parallel servers
      servers.map do |server|
        Thread.new(server) do |server|
          yield server
        end
      end.map {|thread| thread.join }
    end

    # options = {:exec => ?, :capture => ?, :quiet_ssh => ?}
    def execute_command server, *command_array
      options = {:exec => false, :capture => false, :quiet_ssh => false}
      options.merge!(command_array.pop) if command_array.last.is_a?(Hash)
      options[:local] = true if server.to_s == "localhost"

      command = build_command(server, command_array, options)

      if pretending
        info("[#{server}] #{command}")
      else
        if options[:exec]
          exec command
        else
          results = `#{command}`
          if options[:capture]
            return results
          else
            results.split("\n").each do |result|
              info "[#{server}] #{result.chomp}\n"
            end
          end
        end
      end
    end

    def build_command server, *commands_and_maybe_options
      return "" if commands_and_maybe_options.empty?

      if commands_and_maybe_options.last.is_a?(Hash)
        options = commands_and_maybe_options.pop
        command_array = commands_and_maybe_options
      else
        command_array = commands_and_maybe_options
        options = {}
      end

      commands = command_array.join " && "
      psuedo_tty = options[:exec] ? '-t ' : ''
      quiet_ssh = options[:quiet_ssh] ? '-q ' : ''

      if options[:local]
        commands
      else
        if use_gateway
          %(ssh #{quiet_ssh}#{psuedo_tty}#{gateway} 'ssh #{quiet_ssh}#{psuedo_tty}#{deploy_user}@#{server} "#{commands}"')
        else
          %(ssh #{quiet_ssh}#{psuedo_tty}#{deploy_user}@#{server} "#{commands}")
        end
      end
    end

    def prepare_initial_environment
      # We do 2 cd's and a git checkout here because the master
      # branch may not always contain the proper rvmrc/Gemfile
      # we need when setting up the rest of the deploy
      prepare_environment + [
                             %(git checkout -q -f origin/#{environment}),
                             %(git reset --hard),
                             %(cd ../),
                             %(cd #{deploy_directory})
                            ]
    end

    def prepare_environment
      [%(source ~/.rvm/scripts/rvm),
       %({ cd #{deploy_directory} > /dev/null; })] # Silence RVM's "Using... gemset..."
    end

    def info message
      STDOUT.puts message
    end

    def fatal message
      info message
      exit 1
    end

  end
end