class ElasticAPM::Spies::RakeSpy

def execute(*args)
  agent = ElasticAPM.start
  unless agent && agent.config.instrumented_rake_tasks.include?(name)
    return execute_without_apm(*args)
  end
  transaction =
    ElasticAPM.start_transaction("Rake::Task[#{name}]", 'Rake')
  begin
    result = execute_without_apm(*args)
    transaction.result = 'success' if transaction
  rescue StandardError => e
    transaction.result = 'error' if transaction
    ElasticAPM.report(e)
    raise
  ensure
    ElasticAPM.end_transaction
    ElasticAPM.stop
  end
  result
end

def install
  ::Rake::Task.class_eval do
    alias execute_without_apm execute
    def execute(*args)
      agent = ElasticAPM.start
      unless agent && agent.config.instrumented_rake_tasks.include?(name)
        return execute_without_apm(*args)
      end
      transaction =
        ElasticAPM.start_transaction("Rake::Task[#{name}]", 'Rake')
      begin
        result = execute_without_apm(*args)
        transaction.result = 'success' if transaction
      rescue StandardError => e
        transaction.result = 'error' if transaction
        ElasticAPM.report(e)
        raise
      ensure
        ElasticAPM.end_transaction
        ElasticAPM.stop
      end
      result
    end
  end
end