require 'optparse' require 'irb' require 'irb/completion' module Rails class Console class << self def start(*args) new(*args).start end def parse_arguments(arguments) options = {} OptionParser.new do |opt| opt.banner = "Usage: rails console [environment] [options]" opt.on('-s', '--sandbox', 'Rollback database modifications on exit.') { |v| options[:sandbox] = v } opt.on("-e", "--environment=name", String, "Specifies the environment to run this console under (test/development/production).", "Default: development") { |v| options[:environment] = v.strip } opt.parse!(arguments) end if arguments.first && arguments.first[0] != '-' env = arguments.first if available_environments.include? env options[:environment] = env else options[:environment] = %w(production development test).detect {|e| e =~ /^#{env}/} || env end end options end private def available_environments Dir['config/environments/*.rb'].map { |fname| File.basename(fname, '.*') } end end attr_reader :options, :app, :console def initialize(app, options={}) @app = app @options = options app.sandbox = sandbox? app.load_console @console = app.config.console || IRB end def sandbox? options[:sandbox] end def environment options[:environment] ||= ENV['RAILS_ENV'] || ENV['RACK_ENV'] || 'development' end def environment? environment end def set_environment! Rails.env = environment end def start set_environment! if environment? if sandbox? puts "Loading #{Rails.env} environment in sandbox (Rails #{Rails.version})" puts "Any modifications you make will be rolled back on exit" else puts "Loading #{Rails.env} environment (Rails #{Rails.version})" end if defined?(console::ExtendCommandBundle) console::ExtendCommandBundle.send :include, Rails::ConsoleMethods end console.start end end end