diff --git a/lib/bootstrappers/app_builder.rb b/lib/bootstrappers/app_builder.rb index cbf7457..9b026b6 100644 --- a/lib/bootstrappers/app_builder.rb +++ b/lib/bootstrappers/app_builder.rb @@ -97,6 +97,31 @@ def use_mysql_config_template end + def use_postgresql_config_template + + template 'postgresql_database.yml.erb', 'config/database.yml',:force => true + template 'postgresql_database.yml.erb', 'config/database.yml.example', :force => true + + db_user_name = ask("What is your local database user name? [postgres]") + db_password = ask("What is your local database password? ['']") + + replace_in_file 'config/database.yml', 'username: root', "username: #{db_user_name}" if db_user_name.present? + replace_in_file 'config/database.yml', 'password: ""', "password: '#{db_password}'" if db_password.present? + + end + + def use_sqlite_config_template + + template 'sqlite_database.yml.erb', 'config/database.yml',:force => true + template 'sqlite_database.yml.erb', 'config/database.yml.example', :force => true + + db_user_name = ask("What is your local database user name? [root]") + db_password = ask("What is your local database password? ['']") + + replace_in_file 'config/database.yml', 'username: root', "username: #{db_user_name}" if db_user_name.present? + replace_in_file 'config/database.yml', 'password: ""', "password: '#{db_password}'" if db_password.present? + + end end end diff --git a/lib/bootstrappers/generators/app_generator.rb b/lib/bootstrappers/generators/app_generator.rb index cbeeb3f..ab5fce9 100644 --- a/lib/bootstrappers/generators/app_generator.rb +++ b/lib/bootstrappers/generators/app_generator.rb @@ -87,8 +87,13 @@ def setup_capistrano def setup_database say 'Setting up database' - if 'mysql' == options[:database] + case options[:database] + when 'mysql' build :use_mysql_config_template + when 'sqlite' + build :use_sqlite_config_template + when 'postgresql' + build :use_postgresql_config_template end build :create_database