mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
be3e10cd26
I've also added a proper acceptance test which reproduced the issue. Closes #9513, #9515.
101 lines
2.6 KiB
Ruby
101 lines
2.6 KiB
Ruby
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.on("--debugger", 'Enable the debugger.') { |v| options[:debugger] = v }
|
|
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 debugger?
|
|
options[:debugger]
|
|
end
|
|
|
|
def start
|
|
require_debugger if debugger?
|
|
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
|
|
|
|
def require_debugger
|
|
require 'debugger'
|
|
puts "=> Debugger enabled"
|
|
rescue LoadError
|
|
puts "You're missing the 'debugger' gem. Add it to your Gemfile, bundle, and try again."
|
|
exit
|
|
end
|
|
end
|
|
end
|