aboutsummaryrefslogtreecommitdiffstats
path: root/railties/lib/rails/commands/console.rb
blob: 86376ac7e6ef04db652a5c992df855b66cfa8276 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
require 'optparse'
require 'irb'
require 'irb/completion'

module Rails
  class Console
    attr_reader :options, :app, :console, :arguments

    def self.start(*args)
      new(*args).start
    end

    def initialize(app, arguments = ARGV)
      @app       = app
      @arguments = arguments
      app.load_console
      @console   = app.config.console || IRB
    end

    def options
      @options ||= begin
        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!(arguments)
        end

        options
      end
    end

    def sandbox?
      options[:sandbox]
    end

    def debugger?
      options[:debugger]
    end

    def start
      app.sandbox = sandbox?

      require_debugger if debugger?

      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
      begin
        require 'ruby-debug'
        puts "=> Debugger enabled"
      rescue Exception
        puts "You need to install ruby-debug19 to run the console in debugging mode. With gems, use 'gem install ruby-debug19'"
        exit
      end
    end
  end
end

# Has to set the RAILS_ENV before config/application is required
if ARGV.first && !ARGV.first.index("-") && env = ARGV.shift # has to shift the env ARGV so IRB doesn't freak
  ENV['RAILS_ENV'] = %w(production development test).detect {|e| e =~ /^#{env}/} || env
end