Contents

require 'rails/generators'
require 'rails/generators/migration'
require 'rails/generators/active_record'

class VanityGenerator < Rails::Generators::Base
  include Rails::Generators::Migration
  source_root File.expand_path('../../templates', __FILE__)

  def self.next_migration_number(path)
    ::ActiveRecord::Generators::Base.next_migration_number(path)
  end

  def create_migration_file
    migration_template "vanity_migration.rb.erb", destination("vanity_migration.rb"), :migration_version=>migration_version
    migration_template "add_unique_indexes_migration.rb.erb", destination("add_vanity_unique_indexes.rb"), :migration_version=>migration_version
    migration_template "add_participants_unique_index_migration.rb.erb", destination("add_participants_unique_index_migration.rb"), :migration_version=>migration_version
  end

  private

  def destination(name)
    File.join(Rails.root, 'db', 'migrate', name)
  end

  def versioned?
    ActiveRecord::VERSION::MAJOR >= 5
  end

  def migration_version
    if versioned?
      "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
vanity-3.1.0	lib/generators/vanity/migration_generator.rb
vanity-3.0.2	lib/generators/vanity/migration_generator.rb
vanity-3.0.1	lib/generators/vanity/migration_generator.rb
vanity-3.0.0	lib/generators/vanity/migration_generator.rb