From c7b4bc3d94140bf3b5d1cf92c945c1530026d872 Mon Sep 17 00:00:00 2001 From: mido18 Date: Sun, 13 Apr 2014 00:07:34 +0200 Subject: [PATCH] Update rolify_generator.rb --- lib/generators/active_record/rolify_generator.rb | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/lib/generators/active_record/rolify_generator.rb b/lib/generators/active_record/rolify_generator.rb index 8dfd6ea3..2d7d3137 100644 --- 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 \ No newline at end of file +end