# this file can either be required manually by a user, in which case 'bugsnag'
# needs to be required, or it can be required automatically in the railtie,
# in which case 'bugsnag' has already been required
require 'bugsnag' unless defined?(Bugsnag)

Rake::TaskManager.record_task_metadata = true

if Gem::Version.new(RUBY_VERSION.dup) >= Gem::Version.new('2.0')
  module Bugsnag
    module RakeTask
      FRAMEWORK_ATTRIBUTES = {
        framework: 'Rake'
      }

      # Executes the rake task with Bugsnag setup with contextual data.
      def execute(args = nil)
        Bugsnag.configuration.detected_app_type = "rake"

        old_task = Bugsnag.configuration.request_data[:bugsnag_running_task]
        Bugsnag.configuration.set_request_data :bugsnag_running_task, self
        Bugsnag.configuration.runtime_versions["rake"] = ::Rake::VERSION

        super
      rescue Exception => ex
        Bugsnag.notify(ex, true) do |report|
          report.severity = "error"
          report.severity_reason = {
            type: Bugsnag::Report::UNHANDLED_EXCEPTION_MIDDLEWARE,
            attributes: FRAMEWORK_ATTRIBUTES
          }
        end
        raise
      ensure
        Bugsnag.configuration.set_request_data :bugsnag_running_task, old_task
      end
    end
  end

  Rake::Task.send(:prepend, Bugsnag::RakeTask)
else
  class Rake::Task
    FRAMEWORK_ATTRIBUTES = {
      framework: 'Rake'
    }

    ##
    # Executes the rake task with Bugsnag setup with contextual data.
    def execute_with_bugsnag(args=nil)
      Bugsnag.configuration.detected_app_type = "rake"

      old_task = Bugsnag.configuration.request_data[:bugsnag_running_task]
      Bugsnag.configuration.set_request_data :bugsnag_running_task, self
      Bugsnag.configuration.runtime_versions["rake"] = ::Rake::VERSION

      execute_without_bugsnag(args)
    rescue Exception => ex
      Bugsnag.notify(ex, true) do |report|
        report.severity = "error"
        report.severity_reason = {
          type: Bugsnag::Report::UNHANDLED_EXCEPTION_MIDDLEWARE,
          attributes: FRAMEWORK_ATTRIBUTES
        }
      end
      raise
    ensure
      Bugsnag.configuration.set_request_data :bugsnag_running_task, old_task
    end

    alias_method :execute_without_bugsnag, :execute
    alias_method :execute, :execute_with_bugsnag
  end
end

Bugsnag.configuration.internal_middleware.use(Bugsnag::Middleware::Rake)