Attributes for devise generator

heartcombo · Jul 17, 2010 · aecc014 · aecc014
1 parent 55fd7e3
commit aecc014
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/lib/generators/active_record/devise_generator.rb b/lib/generators/active_record/devise_generator.rb
@@ -4,6 +4,8 @@
 module ActiveRecord
   module Generators
     class DeviseGenerator < ActiveRecord::Generators::Base
+      argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"
+
       include Devise::Generators::OrmHelpers
       source_root File.expand_path("../templates", __FILE__)
 
@@ -23,4 +25,4 @@ def inject_devise_content
       end
     end
   end
-end
+end
diff --git a/lib/generators/active_record/templates/migration.rb b/lib/generators/active_record/templates/migration.rb
@@ -10,6 +10,10 @@ def self.up
       # t.lockable :lock_strategy => :<%= Devise.lock_strategy %>, :unlock_strategy => :<%= Devise.unlock_strategy %>
       # t.token_authenticatable
 
+<% for attribute in attributes -%>
+      t.<%= attribute.type %> :<%= attribute.name %>
+<% end -%>
+
       t.timestamps
     end