# These tasks are added to the project if you install annotate as a Rails plugin. # (They are not used to build annotate itself.) # Append annotations to Rake tasks for ActiveRecord, so annotate automatically gets # run after doing db:migrate. migration_tasks = %w(db:migrate db:migrate:up db:migrate:down db:migrate:reset db:migrate:redo db:rollback) if defined?(Rails::Application) && Rails.version.split('.').first.to_i >= 6 require 'active_record' databases = ActiveRecord::Tasks::DatabaseTasks.setup_initial_database_yaml ActiveRecord::Tasks::DatabaseTasks.for_each(databases) do |spec_name| migration_tasks.concat(%w(db:migrate db:migrate:up db:migrate:down).map { |task| "#{task}:#{spec_name}" }) end end migration_tasks.each do |task| next unless Rake::Task.task_defined?(task) Rake::Task[task].enhance do Rake::Task[Rake.application.top_level_tasks.last].enhance do annotation_options_task = if Rake::Task.task_defined?('app:set_annotation_options') 'app:set_annotation_options' else 'set_annotation_options' end Rake::Task[annotation_options_task].invoke Annotate::Migration.update_annotations end end end module Annotate class Migration @@working = false def self.update_annotations unless @@working || Annotate::Helpers.skip_on_migration? @@working = true self.update_models if Annotate::Helpers.include_models? self.update_routes if Annotate::Helpers.include_routes? end end def self.update_models if Rake::Task.task_defined?("annotate_models") Rake::Task["annotate_models"].invoke elsif Rake::Task.task_defined?("app:annotate_models") Rake::Task["app:annotate_models"].invoke end end def self.update_routes if Rake::Task.task_defined?("annotate_routes") Rake::Task["annotate_routes"].invoke elsif Rake::Task.task_defined?("app:annotate_routes") Rake::Task["app:annotate_routes"].invoke end end end end