lib/generators/doeskeyvalue/doeskeyvalue_generator.rb in doeskeyvalue-0.2.2 vs lib/generators/doeskeyvalue/doeskeyvalue_generator.rb in doeskeyvalue-0.9.0

@@ -1,25 +1,29 @@
 # AWEXOME LABS
 # DoesKeyValue
 #
-# IndexTableGenerator -- generator for the Index database table
+# Generator -- Migration generator for the column-backed index table and/or
+#  the table-backed storage tables
 
-require 'rails/generators'
-require 'rails/generators/migration'
+require "rails/generators"
+require "rails/generators/migration"
 
 class DoeskeyvalueGenerator < Rails::Generators::Base
   include Rails::Generators::Migration
-  source_root File.expand_path('../templates', __FILE__)
+  source_root File.expand_path("../templates", __FILE__)
   
   def self.next_migration_number(path)
     if ActiveRecord::Base.timestamped_migrations
       Time.now.utc.strftime("%Y%m%d%H%M%S")
     else
       "%.3d" % (current_migration_number(path) + 1)
     end
   end
   
   def create_migration_file
-    migration_template 'create_key_value_index.rb', 'db/migrate/create_key_value_index.rb'
+    puts "Creating DoesKeyValue index table migration"
+    index_table_name = ARGV.first || "key_value_index"
+    puts "=> #{index_table_name} table to be created"
+    migration_template "create_key_value_index.rb", "db/migrate/create_#{index_table_name}.rb", table_name:index_table_name
   end
   
 end