cookbooks/mysql/libraries/mysql.rb

   1 require "chef/mixin/command"
   2 require "rexml/document"
   3
   4 class Chef
   5   class MySQL
   6     include Chef::Mixin::Command
   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=#{options[:user]}")
  30         args.push("--password=#{options[:password]}") if options[:password]
  31       else
  32         args.push("--defaults-file=/etc/mysql/debian.cnf")
  33       end
  34
  35       # Build the other arguments
  36       args.push("--execute=\"#{options[:command]}\"") if options[:command]
  37
  38       # Get the database to use
  39       database = options[:database] || "mysql"
  40
  41       # Build the command to run
  42       command = "/usr/bin/mysql #{args.join(' ')} #{database}"
  43
  44       # Escape backticks in the command
  45       command.gsub!(/`/, "\\\\`")
  46
  47       # Run the command
  48       run_command(:command => command, :user => "root", :group => "root")
  49     end
  50
  51     def query(sql, options = {})
  52       # Get the database to use
  53       database = options[:database] || "mysql"
  54
  55       # Construct the command string
  56       command = "/usr/bin/mysql --defaults-file=/etc/mysql/debian.cnf --xml --execute='#{sql}' #{database}"
  57
  58       # Run the query
  59       status, stdout, stderr = output_of_command(command, :user => "root", :group => "root")
  60       handle_command_failures(status, "STDOUT: #{stdout}\nSTDERR: #{stderr}", :output_on_failure => true)
  61
  62       # Parse the output
  63       document = REXML::Document.new(stdout)
  64
  65       # Create
  66       records = []
  67
  68       # Loop over the rows in the result set
  69       document.root.each_element("/resultset/row") do |row|
  70         # Create a record
  71         record = {}
  72
  73         # Loop over the fields, adding them to the record
  74         row.each_element("field") do |field|
  75           name = field.attributes["name"].downcase
  76           value = field.text
  77
  78           record[name.to_sym] = value
  79         end
  80
  81         # Add the record to the record list
  82         records << record
  83       end
  84
  85       # Return the record list
  86       records
  87     end
  88
  89     def users
  90       @users ||= query("SELECT * FROM user").each_with_object({}) do |user, users|
  91         name = "'#{user[:user]}'@'#{user[:host]}'"
  92
  93         users[name] = USER_PRIVILEGES.each_with_object({}) do |privilege, privileges|
  94           privileges[privilege] = user["#{privilege}_priv".to_sym] == "Y"
  95         end
  96       end
  97     end
  98
  99     def databases
 100       @databases ||= query("SHOW databases").each_with_object({}) do |database, databases|
 101         databases[database[:database]] = {
 102           :permissions => {}
 103         }
 104       end
 105
 106       query("SELECT * FROM db").each do |record|
 107         database = @databases[record[:db]]
 108
 109         next unless database
 110
 111         user = "'#{record[:user]}'@'#{record[:host]}'"
 112
 113         database[:permissions][user] = DATABASE_PRIVILEGES.each_with_object([]) do |privilege, privileges|
 114           privileges << privilege if record["#{privilege}_priv".to_sym] == "Y"
 115         end
 116       end
 117
 118       @databases
 119     end
 120
 121     def canonicalise_user(user)
 122       local, host = user.split("@")
 123
 124       host = "%" unless host
 125
 126       local = "'#{local}'" unless local =~ /^'.*'$/
 127       host = "'#{host}'" unless host =~ /^'.*'$/
 128
 129       "#{local}@#{host}"
 130     end
 131
 132     def privilege_name(privilege)
 133       case privilege
 134       when :grant
 135         "GRANT OPTION"
 136       when :create_tmp_table
 137         "CREATE TEMPORARY TABLES"
 138       else
 139         privilege.to_s.upcase.tr("_", " ")
 140       end
 141     end
 142   end
 143 end