module Hanzo
  class Deploy < Base

  protected

    def initialize_variables
      @env = extract_argument(1)

      if @env.nil? and Hanzo::Installers::Remotes.environments.keys.length == 1
        @env = Hanzo::Installers::Remotes.environments.keys.first
      end
    end

    def initialize_cli
      initialize_help and return if @env.nil?

      deploy
      run_migrations
    end

    def initialize_help
      @options.banner = "Usage: hanzo deploy ENVIRONMENT"
    end

    def deploy
      branch = Hanzo.ask("Branch to deploy in #{@env}:") { |q| q.default = "HEAD" }

      Hanzo.run "git push -f #{@env} #{branch}:master"
    end

    def run_migrations
      if Dir.exists?('db/migrate')
        migration = Hanzo.agree('Run migrations?')
        Hanzo.run "heroku run rake db:migrate --remote #{@env}" if migration
      end
    end
  end
end