airbrake/airbrake

# frozen_string_literal: true

# This is not bulletproof, but if this file is executed before a task
# definition, we can grab tasks descriptions and locations.
# See: https://goo.gl/ksn6PE
Rake::TaskManager.record_task_metadata = true

module Airbrake
  module Rake
    # Redefine +Rake::Task#execute+, so it can report errors to Airbrake.
    module Task
      # A wrapper around the original +#execute+, that catches all errors and
      # notifies Airbrake.
      #
      # rubocop:disable Lint/RescueException
      def execute(args = nil)
        super(args)
      rescue Exception => ex
        notify_airbrake(ex, args)
        raise ex
      end
      # rubocop:enable Lint/RescueException

      private

      def notify_airbrake(exception, args)
        notice = Airbrake.build_notice(exception)
        notice[:context][:component] = 'rake'
        notice[:context][:action] = name
        notice[:params].merge!(
          rake_task: task_info,
          execute_args: args,
          argv: ARGV.join(' '),
        )

        Airbrake.notify_sync(notice)
      end

      # rubocop:disable Metrics/CyclomaticComplexity, Metrics/AbcSize
      def task_info
        info = {}

        info[:name] = name
        info[:timestamp] = timestamp.to_s
        info[:investigation] = investigation

        info[:full_comment] = full_comment if full_comment
        info[:arg_names] = arg_names if arg_names.any?
        info[:arg_description] = arg_description if arg_description
        info[:locations] = locations if locations.any?
        info[:sources] = sources if sources.any?

        if prerequisite_tasks.any?
          info[:prerequisite_tasks] = prerequisite_tasks.map do |p|
            p.__send__(:task_info)
          end
        end

        info
      end
      # rubocop:enable Metrics/CyclomaticComplexity, Metrics/AbcSize
    end
  end
end

Rake::Task.prepend(Airbrake::Rake::Task)