From 34a64070ca385ad743024f23f6411540dd0cb6e6 Mon Sep 17 00:00:00 2001 From: Rodrigo Flores Date: Wed, 2 May 2012 14:49:33 -0300 Subject: [PATCH] Cached the class_path and changed a variable name (kudos to @britto!) --- lib/generators/active_record/devise_generator.rb | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/lib/generators/active_record/devise_generator.rb b/lib/generators/active_record/devise_generator.rb index 791f16e954..8aa912ed22 100644 --- a/lib/generators/active_record/devise_generator.rb +++ b/lib/generators/active_record/devise_generator.rb @@ -27,10 +27,12 @@ def inject_devise_content attr_accessible :email, :password, :password_confirmation, :remember_me CONTENT - indent_size = class_name.to_s.split("::").size - 1 - content = content.split("\n").map { |line| " " * indent_size + line}.join("\n") + class_path = class_name.to_s.split("::") - inject_into_class(model_path, class_name.to_s.split("::").last, content) if model_exists? + indent_depth = class_path.size - 1 + content = content.split("\n").map { |line| " " * indent_depth + line } .join("\n") + + inject_into_class(model_path, class_path.last, content) if model_exists? end def migration_data