class Rails::Console

def available_environments
  Dir['config/environments/*.rb'].map { |fname| File.basename(fname, '.*') }
end

def debugger?
  options[:debugger]
end

def environment
  options[:environment] ||= ENV['RAILS_ENV'] || ENV['RACK_ENV'] || 'development'
end

def environment?
  environment
end

def initialize(app, options={})
  @app     = app
  @options = options
  app.sandbox = sandbox?
  app.load_console
  @console = app.config.console || IRB
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

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

def sandbox?
  options[:sandbox]
end

def set_environment!
  Rails.env = environment
end

def start(*args)
  new(*args).start
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