Update rolify_generator.rb

mido18 · Apr 12, 2014 · c7b4bc3 · c7b4bc3
1 parent 38336a4
commit c7b4bc3
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/generators/active_record/rolify_generator.rb b/lib/generators/active_record/rolify_generator.rb
@@ -13,6 +13,15 @@ def generate_model
       end
 
       def inject_role_class
+        if args[1]=="engine"
+          if args[2]=="devise"
+            require 'devise'
+            require "#{ENGINE_ROOT}/config/initializers/devise.rb"
+            require "#{ENGINE_ROOT}/app/models/#{user_cname.downcase}.rb"
+          else
+            require "#{ENGINE_ROOT}/app/models/#{user_cname.downcase}.rb"
+          end
+        end
         inject_into_class(model_path, class_name, model_content)
       end
 
@@ -47,4 +56,4 @@ def model_content
       end
     end
   end
-end
+end