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 = "%" unless 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