diff options
Diffstat (limited to 'railties/lib/initializer.rb')
| -rw-r--r-- | railties/lib/initializer.rb | 158 |
1 files changed, 158 insertions, 0 deletions
diff --git a/railties/lib/initializer.rb b/railties/lib/initializer.rb new file mode 100644 index 0000000000..2a172cc53d --- /dev/null +++ b/railties/lib/initializer.rb @@ -0,0 +1,158 @@ +require 'logger' + +RAILS_ENV = ENV['RAILS_ENV'] || 'development' + +module Rails + class Initializer + attr_reader :configuration + + def self.run(command = :process, configuration = Configuration.new) + yield configuration if block_given? + new(configuration).send(command) + end + + def initialize(configuration) + @configuration = configuration + end + + def process + set_load_path + + require_frameworks + require_environment + + initialize_database + initialize_logger + initialize_framework_logging + initialize_routing + end + + def set_load_path + configuration.load_paths.reverse.each { |dir| $LOAD_PATH.unshift(dir) if File.directory?(dir) } + $LOAD_PATH.uniq! + end + + def require_frameworks + configuration.frameworks.each { |framework| require(framework.to_s) } + end + + def require_environment + require_dependency(configuration.environment_file) + end + + def initialize_database + return unless configuration.frameworks.include?(:active_record) + ActiveRecord::Base.configurations = configuration.database_configuration + ActiveRecord::Base.establish_connection + end + + def initialize_logger + begin + logger = Logger.new(configuration.log_path) + logger.level = configuration.log_level + rescue StandardError + logger = Logger.new(STDERR) + logger.level = Logger::WARN + logger.warn( + "Rails Error: Unable to access log file. Please ensure that #{configuration.log_path} exists and is chmod 0666. " + + "The log level has been raised to WARN and the output directed to STDERR until the problem is fixed." + ) + end + + Object.const_set "RAILS_DEFAULT_LOGGER", logger + end + + def initialize_framework_logging + [ActiveRecord, ActionController, ActionMailer].each { |mod| mod::Base.logger ||= RAILS_DEFAULT_LOGGER } + end + + def initialize_framework_views + [ActionController, ActionMailer].each { |mod| mod::Base.template_root ||= configuration.view_path } + end + + def initialize_routing + return unless configuration.frameworks.include?(:action_controller) + ActionController::Routing::Routes.reload + Object.const_set "Controllers", Dependencies::LoadingModule.root(*configuration.controller_paths) + end + end + + class Configuration + attr_accessor :frameworks, :load_paths, :log_level, :log_path, :database_configuration_file, :view_path, :controller_paths + + def initialize + self.frameworks = default_frameworks + self.load_paths = default_load_paths + self.log_path = default_log_path + self.log_level = default_log_level + self.view_path = default_view_path + self.controller_paths = default_controller_paths + self.database_configuration_file = default_database_configuration_file + end + + def database_configuration + YAML::load(ERB.new((IO.read(database_configuration_file))).result) + end + + def environment_file + "environments/#{environment}" + end + + def environment + ::RAILS_ENV + end + + private + def default_frameworks + [ :active_support, :active_record, :action_controller, :action_mailer, :action_web_service ] + end + + def default_load_paths + paths = ["#{environment}/test/mocks/#{environment}"] + + # Then model subdirectories. + paths.concat(Dir["#{RAILS_ROOT}/app/models/[_a-z]*"]) + paths.concat(Dir["#{RAILS_ROOT}/components/[_a-z]*"]) + + # Followed by the standard includes. + paths.concat %w( + app + app/models + app/controllers + app/helpers + app/apis + components + config + lib + vendor + vendor/rails/railties + vendor/rails/railties/lib + vendor/rails/actionpack/lib + vendor/rails/activesupport/lib + vendor/rails/activerecord/lib + vendor/rails/actionmailer/lib + vendor/rails/actionwebservice/lib + ).map { |dir| "#{RAILS_ROOT}/#{dir}" }.select { |dir| File.directory?(dir) } + end + + def default_log_path + File.join(RAILS_ROOT, 'log', "#{environment}.log") + end + + def default_log_level + environment == 'production' ? Logger::INFO : Logger::DEBUG + end + + def default_database_configuration_file + File.join(RAILS_ROOT, 'config', 'database.yml') + end + + def default_view_path + File.join(RAILS_ROOT, 'app', 'views') + end + + def default_controller_paths + [ File.join(RAILS_ROOT, 'app', 'controllers'), File.join(RAILS_ROOT, 'components') ] + end + end +end
\ No newline at end of file |