Extend Rake with a more elegant and reliable way

sentry-ruby/lib/sentry/rake.rb

@@ -1,30 +1,31 @@
 require "rake"
 require "rake/task"
 
-module Rake
- class Application
+module Sentry
+ module Rake
+ module Application
+ def display_error_message(*)
+ Sentry.capture_exception(ex, hint: { background: false }) do |scope|
+ task_name = top_level_tasks.join(' ')
+ scope.set_transaction_name(task_name)
+ scope.set_tag("rake_task", task_name)
+ end if Sentry.initialized? && !Sentry.configuration.skip_rake_integration
 
- alias orig_display_error_messsage display_error_message
- def display_error_message(ex)
- Sentry.capture_exception(ex, hint: { background: false }) do |scope|
- task_name = top_level_tasks.join(' ')
- scope.set_transaction_name(task_name)
- scope.set_tag("rake_task", task_name)
- end if Sentry.initialized? && !Sentry.configuration.skip_rake_integration
-
- orig_display_error_messsage(ex)
+ super
+ end
  end
- end
 
- class Task
- alias orig_execute execute
+ module Task
+ def execute(*)
+ return super unless Sentry.initialized? && Sentry.get_current_hub
 
- def execute(args=nil)
- return orig_execute(args) unless Sentry.initialized? && Sentry.get_current_hub
-
- Sentry.get_current_hub.with_background_worker_disabled do
- orig_execute(args)
+ Sentry.get_current_hub.with_background_worker_disabled do
+ super
+ end
  end
  end
  end
 end
+
+Rake::Application.prepend(Sentry::Rake::Application)
+Rake::Task.prepend(Sentry::Rake::Task)