module Airbrake::Capistrano

def self.load_into(config)
  config.load do
    after 'deploy',            'airbrake:deploy'
    after 'deploy:migrations', 'airbrake:deploy'
    after 'deploy:cold',       'airbrake:deploy'
    namespace :airbrake do
      desc "Notify Airbrake of the deploy"
      task :deploy, except: { no_release: true }, on_error: :continue do
        run(
          <<-CMD, once: true
            cd #{config.release_path} && \
            RACK_ENV=#{fetch(:rack_env, nil)} \
            RAILS_ENV=#{fetch(:rails_env, nil)} \
            bundle exec rake airbrake:deploy \
              USERNAME=#{Shellwords.shellescape(ENV.fetch('USER', nil) || ENV.fetch('USERNAME', nil))} \
              ENVIRONMENT=#{fetch(:airbrake_env, fetch(:rails_env, 'production'))} \
              REVISION=#{current_revision.strip} \
              REPOSITORY=#{repository} \
              VERSION=#{fetch(:app_version, nil)}
          CMD
        )
        logger.info 'Notified Airbrake of the deploy'
      end
    end
  end
end