Class: Rails::DBConsole

Inherits:
Object
  • Object
show all
Defined in:
lib/railties/rails/commands/dbconsole.rb

Instance Method Summary collapse

Instance Method Details

#startObject



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
77
78
79
80
81
# File 'lib/railties/rails/commands/dbconsole.rb', line 3

def start
  ENV["RAILS_ENV"] ||= @options[:environment] || environment
  config = db_config.configuration_hash

  case db_config.adapter
  when /^(jdbc)?mysql/
    args = {
      host: "--host",
      port: "--port",
      socket: "--socket",
      username: "--user",
      encoding: "--default-character-set",
      sslca: "--ssl-ca",
      sslcert: "--ssl-cert",
      sslcapath: "--ssl-capath",
      sslcipher: "--ssl-cipher",
      sslkey: "--ssl-key"
    }.filter_map { |opt, arg| "#{arg}=#{config[opt]}" if config[opt] }

    if config[:password] && @options[:include_password]
      args << "--password=#{config[:password]}"
    elsif config[:password] && !config[:password].to_s.empty?
      args << "-p"
    end

    args << db_config.database

    ActiveRecord::Base.connection.class.find_cmd_and_exec(["mysql", "mysql5"], *args)

  when /^postgres|^postgis/
    ENV["PGUSER"] = config[:username] if config[:username]
    ENV["PGHOST"] = config[:host] if config[:host]
    ENV["PGPORT"] = config[:port].to_s if config[:port]
    ENV["PGPASSWORD"] = config[:password].to_s if config[:password] && @options[:include_password]
    ENV["PGSSLMODE"] = config[:sslmode].to_s if config[:sslmode]
    ENV["PGSSLCERT"] = config[:sslcert].to_s if config[:sslcert]
    ENV["PGSSLKEY"] = config[:sslkey].to_s if config[:sslkey]
    ENV["PGSSLROOTCERT"] = config[:sslrootcert].to_s if config[:sslrootcert]
    ActiveRecord::Base.connection.class.find_cmd_and_exec("psql", db_config.database)

  when "sqlite3", "litedb"
    args = []

    args << "-#{@options[:mode]}" if @options[:mode]
    args << "-header" if @options[:header]
    args << File.expand_path(db_config.database, Rails.respond_to?(:root) ? Rails.root : nil)

    ActiveRecord::Base.connection.class.find_cmd_and_exec("sqlite3", *args)

  when "oracle", "oracle_enhanced"
    logon = ""

    if config[:username]
      logon = config[:username].dup
      logon << "/#{config[:password]}" if config[:password] && @options[:include_password]
      logon << "@#{db_config.database}" if db_config.database
    end

    ActiveRecord::Base.connection.class.find_cmd_and_exec("sqlplus", logon)

  when "sqlserver"
    args = []

    args += ["-d", db_config.database.to_s] if db_config.database
    args += ["-U", config[:username].to_s] if config[:username]
    args += ["-P", config[:password].to_s] if config[:password]

    if config[:host]
      host_arg = +"tcp:#{config[:host]}"
      host_arg << ",#{config[:port]}" if config[:port]
      args += ["-S", host_arg]
    end

    ActiveRecord::Base.connection.class.find_cmd_and_exec("sqlcmd", *args)

  else
    abort "Unknown command-line client for #{db_config.database}."
  end
end