require 'optparse'
require 'irb'
require "irb/completion"

module Rails
  class Console
    def self.start
      new.start
    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.parse!(ARGV)
      end

      require "#{Rails.root}/config/environment"
      require "rails/console_app"
      require "rails/console_sandbox" if options[:sandbox]
      require "rails/console_with_helpers"

      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

      ENV['RAILS_ENV'] =
        case ARGV.first
        when "p" then "production"
        when "d" then "development"
        when "t" then "test"
        else
          ARGV.first || ENV['RAILS_ENV'] || 'development'
        end

      if options[:sandbox]
        puts "Loading #{ENV['RAILS_ENV']} environment in sandbox (Rails #{Rails.version})"
        puts "Any modifications you make will be rolled back on exit"
      else
        puts "Loading #{ENV['RAILS_ENV']} environment (Rails #{Rails.version})"
      end
      IRB.start
    end
  end
end
require 'optparse'
require 'irb'
require "irb/completion"

module Rails
  class Console
    ENVIRONMENTS = %w(production development test)

    def self.start
      new.start
    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.parse!(ARGV)
      end

      if env = ARGV.first
        ENV['RAILS_ENV'] = ENVIRONMENTS.find { |e| e.index(env) } || env
      end

      require "#{Rails.root}/config/environment"
      require "rails/console_app"
      require "rails/console_sandbox" if options[:sandbox]
      require "rails/console_with_helpers"

      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 #{ENV['RAILS_ENV']} environment in sandbox (Rails #{Rails.version})"
        puts "Any modifications you make will be rolled back on exit"
      else
        puts "Loading #{ENV['RAILS_ENV']} environment (Rails #{Rails.version})"
      end
      IRB.start
    end
  end
end

Set RAILS_ENV before loading config/environment
Set RAILS_ENV before loading config/environment
