53 lines
1.7 KiB
Ruby
53 lines
1.7 KiB
Ruby
require 'optparse'
|
|
require 'irb'
|
|
require 'irb/completion'
|
|
|
|
module Rails
|
|
class Console
|
|
def self.start(app)
|
|
new(app).start
|
|
end
|
|
|
|
def initialize(app)
|
|
@app = app
|
|
end
|
|
|
|
def start
|
|
options = {}
|
|
|
|
OptionParser.new do |opt|
|
|
opt.banner = "Usage: console [environment] [options]"
|
|
opt.on('-s', '--sandbox', 'Rollback database modifications on exit.') { |v| options[:sandbox] = v }
|
|
opt.on("--debugger", 'Enable ruby-debugging for the console.') { |v| options[:debugger] = v }
|
|
opt.on('--irb', "DEPRECATED: Invoke `/your/choice/of/ruby script/rails console` instead") { |v| abort '--irb option is no longer supported. Invoke `/your/choice/of/ruby script/rails console` instead' }
|
|
opt.parse!(ARGV)
|
|
end
|
|
|
|
@app.load_console(options[:sandbox])
|
|
|
|
if options[:debugger]
|
|
begin
|
|
require 'ruby-debug'
|
|
puts "=> Debugger enabled"
|
|
rescue Exception
|
|
puts "You need to install ruby-debug to run the console in debugging mode. With gems, use 'gem install ruby-debug'"
|
|
exit
|
|
end
|
|
end
|
|
|
|
if options[: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
|
|
IRB.start
|
|
end
|
|
end
|
|
end
|
|
|
|
# Has to set the RAILS_ENV before config/application is required
|
|
if ARGV.first && !ARGV.first.index("-") && env = ARGV.pop # has to pop the env ARGV so IRB doesn't freak
|
|
ENV['RAILS_ENV'] = %w(production development test).detect {|e| e =~ /^#{env}/} || env
|
|
end
|