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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
# frozen_string_literal: true
class Combustion::Databases::MySQL < Combustion::Databases::Base
ACCESS_DENIED_ERROR = 10_145
def reset
establish_connection(configuration.merge(:database => nil))
super
end
private
def charset
configuration[:charset] || ENV["CHARSET"] || "utf8"
end
def charset_error
return "" unless configuration[:charset]
"(if you set the charset manually, make sure you have a matching collation)"
end
def collation
configuration[:collation] || ENV["COLLATION"] || "utf8_unicode_ci"
end
def create
connection.create_database configuration[:database], creation_options
establish_connection configuration
rescue error_class => error
rescue_create_from error
end
def create_as_root(error)
establish_connection configuration.merge(
:database => nil,
:username => "root",
:password => request_password(error)
)
connection.create_database configuration[:database], creation_options
connection.execute grant_statement
establish_connection configuration
end
def creation_options
{:charset => charset, :collation => collation}
end
def drop
connection.drop_database configuration[:database]
end
def error_class
if configuration[:adapter][/jdbc/]
# FIXME: After Jdbcmysql gives this class
require "active_record/railties/jdbcmysql_error"
ArJdbcMySQL::Error
elsif configuration[:adapter][/mysql2/] && defined?(Mysql2)
Mysql2::Error
elsif configuration[:adapter][/trilogy/] && defined?(Trilogy)
Trilogy::Error
else
Mysql::Error
end
end
def grant_statement
<<-SQL
GRANT ALL PRIVILEGES ON #{configuration["database"]}.*
TO '#{configuration[:username]}'@'localhost'
IDENTIFIED BY '#{configuration[:password]}' WITH GRANT OPTION;
SQL
end
def request_password(error)
print <<-TXT.strip
#{error.error}.
Please provide the root password for your mysql installation
>
TXT
$stdin.gets.strip
end
def rescue_create_from(error)
if error.errno == ACCESS_DENIED_ERROR
create_as_root(error)
return
end
warn <<-TXT
#{error.error}
Couldn't create database for #{configuration.inspect}, charset: #{charset}, collation: #{collation}
#{charset_error}
TXT
end
end
|
