lib/generators/doorkeeper/migration_generator.rb in doorkeeper-4.2.6 vs lib/generators/doorkeeper/migration_generator.rb in doorkeeper-4.3.0

@@ -4,12 +4,24 @@
   include Rails::Generators::Migration
   source_root File.expand_path('../templates', __FILE__)
   desc 'Installs Doorkeeper migration file.'
 
   def install
-    migration_template 'migration.rb', 'db/migrate/create_doorkeeper_tables.rb'
+    migration_template(
+      'migration.rb.erb',
+      'db/migrate/create_doorkeeper_tables.rb',
+      migration_version: migration_version
+    )
   end
 
   def self.next_migration_number(dirname)
     ActiveRecord::Generators::Base.next_migration_number(dirname)
+  end
+
+  private
+
+  def migration_version
+    if ActiveRecord::VERSION::MAJOR >= 5
+      "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
+    end
   end
 end