Contents

# 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
      Annotated::Migration.update_annotations
    end
  end
end

module Annotated
  class Migration
    @@working = false

    def self.update_annotations
      unless @@working || Annotated::Helpers.skip_on_migration?
        @@working = true

        update_models if Annotated::Helpers.include_models?
        update_routes if Annotated::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

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
annotated-0.1.1	lib/tasks/annotate_models_migrate.rake
annotated-0.1.0	lib/tasks/annotate_models_migrate.rake
annotated-0.0.3	lib/tasks/annotate_models_migrate.rake
annotated-0.0.2	lib/tasks/annotate_models_migrate.rake
annotated-0.0.1	lib/tasks/annotate_models_migrate.rake