From fcbe87c857b79583c3521ba8ac5ed7860ae262b8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Harold=20Gim=C3=A9nez?= Date: Fri, 28 Oct 2011 09:54:02 -0400 Subject: [PATCH] Set up attribute whitelisting if you're using active record --- lib/suspenders/app_builder.rb | 4 ++++ lib/suspenders/generators/app_generator.rb | 8 ++++++++ 2 files changed, 12 insertions(+) diff --git a/lib/suspenders/app_builder.rb b/lib/suspenders/app_builder.rb index 16a31953d..25bfaeb9b 100644 --- a/lib/suspenders/app_builder.rb +++ b/lib/suspenders/app_builder.rb @@ -133,6 +133,10 @@ def setup_root_route route "root :to => 'Clearance::Sessions#new'" end + def set_active_record_whitelist_attributes + inject_into_class "config/application.rb", "Application", " config.active_record.whitelist_attributes = true\n" + end + def migrate_database rake "db:migrate" end diff --git a/lib/suspenders/generators/app_generator.rb b/lib/suspenders/generators/app_generator.rb index 985837f6a..70612d248 100644 --- a/lib/suspenders/generators/app_generator.rb +++ b/lib/suspenders/generators/app_generator.rb @@ -28,6 +28,7 @@ def suspenders_customization invoke :setup_gitignore invoke :copy_miscellaneous_files invoke :setup_root_route + invoke :set_active_record_whitelist_attributes invoke :outro end @@ -101,6 +102,13 @@ def setup_root_route build(:setup_root_route) end + def set_active_record_whitelist_attributes + if 'active_record' == options[:orm] + say "Setting up active_record.whitelist_attributes" + build(:set_active_record_whitelist_attributes) + end + end + def outro say "Congratulations! You just pulled our suspenders." say "Remember to run 'rails generate airbrake' with your API key."