lib/generators/mailkick/install_generator.rb in mailkick-0.1.6 vs lib/generators/mailkick/install_generator.rb in mailkick-0.2.0

- old
+ new

@@ -20,10 +20,16 @@ "%.3d" % next_migration_number end end def copy_migration - migration_template "install.rb", "db/migrate/install_mailkick.rb" + migration_template "install.rb", "db/migrate/install_mailkick.rb", migration_version: migration_version + end + + def migration_version + if ActiveRecord::VERSION::MAJOR >= 5 + "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]" + end end end end end