index_generator.rb in gindex-0.1.2

- old
+ new

@@ -1,10 +1,12 @@
 require "rails/generators/migration"
 
 class IndexGenerator < Rails::Generators::Base
   include Rails::Generators::Migration
+
   source_root File.expand_path("../templates", __FILE__)
+
   argument :table, type: :string
   argument :columns, type: :array
 
   # Implement the required interface for Rails::Generators::Migration.
   def self.next_migration_number(dirname) #:nodoc:
@@ -16,7 +18,13 @@
     end
   end
 
   def copy_migration
     migration_template "index_migration.rb", "db/migrate/add_index_on_#{columns.join('_and_')}_to_#{table}.rb"
+  end
+
+  def migration_version
+    if ActiveRecord::VERSION::MAJOR >= 5
+      "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
+    end
   end
 end