Fix rubocop warnings
[chef.git] / cookbooks / mysql / libraries / mysql.rb
1 require "chef/mixin/shell_out"
2 require "rexml/document"
3
4 class Chef
5   class MySQL
6     include Chef::Mixin::ShellOut
7
8     USER_PRIVILEGES = [
9       :select, :insert, :update, :delete, :create, :drop, :reload,
10       :shutdown, :process, :file, :grant, :references, :index, :alter,
11       :show_db, :super, :create_tmp_table, :lock_tables, :execute,
12       :repl_slave, :repl_client, :create_view, :show_view, :create_routine,
13       :alter_routine, :create_user, :event, :trigger, :create_tablespace
14     ].freeze
15
16     DATABASE_PRIVILEGES = [
17       :select, :insert, :update, :delete, :create, :drop, :grant,
18       :references, :index, :alter, :create_tmp_table, :lock_tables,
19       :create_view, :show_view, :create_routine, :alter_routine,
20       :execute, :event, :trigger
21     ].freeze
22
23     def execute(options)
24       # Create argument array
25       args = []
26
27       # Work out how to authenticate
28       if options[:user]
29         args.push("--username")
30         args.push(options[:user])
31
32         if options[:password]
33           args.push("--password")
34           args.push(options[:password])
35         end
36       else
37         args.push("--defaults-file=/etc/mysql/debian.cnf")
38       end
39
40       # Set output format
41       args.push("--xml") if options[:xml]
42
43       # Add any SQL command to execute
44       if options[:command]
45         args.push("--execute")
46         args.push(options[:command])
47       end
48
49       # Add the database name
50       args.push(options[:database] || "mysql")
51
52       # Run the command
53       shell_out!("/usr/bin/mysql", *args, :user => "root", :group => "root")
54     end
55
56     def query(sql, options = {})
57       # Run the query
58       result = execute(options.merge(:command => sql, :xml => true))
59
60       # Parse the output
61       document = REXML::Document.new(result.stdout)
62
63       # Create
64       records = []
65
66       # Loop over the rows in the result set
67       document.root.each_element("/resultset/row") do |row|
68         # Create a record
69         record = {}
70
71         # Loop over the fields, adding them to the record
72         row.each_element("field") do |field|
73           name = field.attributes["name"].downcase
74           value = field.text
75
76           record[name.to_sym] = value
77         end
78
79         # Add the record to the record list
80         records << record
81       end
82
83       # Return the record list
84       records
85     end
86
87     def users
88       @users ||= query("SELECT * FROM user").each_with_object({}) do |user, users|
89         name = "'#{user[:user]}'@'#{user[:host]}'"
90
91         users[name] = USER_PRIVILEGES.each_with_object({}) do |privilege, privileges|
92           privileges[privilege] = user["#{privilege}_priv".to_sym] == "Y"
93         end
94       end
95     end
96
97     def databases
98       @databases ||= query("SHOW databases").each_with_object({}) do |database, databases|
99         databases[database[:database]] = {
100           :permissions => {}
101         }
102       end
103
104       query("SELECT * FROM db").each do |record|
105         database = @databases[record[:db]]
106
107         next unless database
108
109         user = "'#{record[:user]}'@'#{record[:host]}'"
110
111         database[:permissions][user] = DATABASE_PRIVILEGES.each_with_object([]) do |privilege, privileges|
112           privileges << privilege if record["#{privilege}_priv".to_sym] == "Y"
113         end
114       end
115
116       @databases
117     end
118
119     def canonicalise_user(user)
120       local, host = user.split("@")
121
122       host ||= "%"
123
124       local = "'#{local}'" unless local =~ /^'.*'$/
125       host = "'#{host}'" unless host =~ /^'.*'$/
126
127       "#{local}@#{host}"
128     end
129
130     def privilege_name(privilege)
131       case privilege
132       when :grant
133         "GRANT OPTION"
134       when :create_tmp_table
135         "CREATE TEMPORARY TABLES"
136       else
137         privilege.to_s.upcase.tr("_", " ")
138       end
139     end
140   end
141 end