Add a load more cookbooks

[chef.git] / cookbooks / mysql / providers / database.rb

diff --git a/cookbooks/mysql/providers/database.rb b/cookbooks/mysql/providers/database.rb
new file mode 100644 (file)
index 0000000..29b38f7
--- /dev/null
+++ b/cookbooks/mysql/providers/database.rb
@@ -0,0 +1,87 @@
+#
+# Cookbook Name:: mysql
+# Provider:: mysql_database
+#
+# Copyright 2013, OpenStreetMap Foundation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+def load_current_resource
+  @mysql = Chef::MySQL.new
+
+  @current_resource = Chef::Resource::MysqlDatabase.new(new_resource.name)
+  @current_resource.database(new_resource.database)
+  if mysql_database = @mysql.databases[@current_resource.database]
+    @current_resource.permissions(mysql_database[:permissions])
+  end
+  @current_resource
+end
+
+action :create do
+  unless @mysql.databases.include?(new_resource.database)
+    converge_by("create #{new_resource}") do
+      Chef::Log.info("Creating #{new_resource}")
+      @mysql.execute(:command => "CREATE DATABASE `#{new_resource.database}`")
+    end
+  end
+
+  new_permissions = Hash[new_resource.permissions.collect do |user,privileges|
+    [@mysql.canonicalise_user(user), privileges]
+  end]
+
+  @current_resource.permissions.each do |user,privileges|
+    unless new_permissions[user]
+      converge_by("revoke all for #{user} on #{new_resource}") do
+        Chef::Log.info("Revoking all for #{user} on #{new_resource}")
+        @mysql.execute(:command => "REVOKE ALL ON `#{new_resource.database}`.* FROM #{user}")
+      end
+    end
+  end
+
+  new_permissions.each do |user,new_privileges|
+    current_privileges = @current_resource.permissions[user] || {}
+    new_privileges = Array(new_privileges)
+
+    if new_privileges.include?(:all)
+      new_privileges |= (Chef::MySQL::DATABASE_PRIVILEGES - [:grant])
+    end
+
+    Chef::MySQL::DATABASE_PRIVILEGES.each do |privilege|
+      if new_privileges.include?(privilege)
+        unless current_privileges.include?(privilege)
+          converge_by("grant #{privilege} for #{user} on mysql database #{new_resource}") do
+            Chef::Log.info("Granting #{privilege} for #{user} on mysql database #{new_resource}")
+            @mysql.execute(:command => "GRANT #{@mysql.privilege_name(privilege)} ON `#{new_resource.database}`.* TO #{user}")
+          end
+        end
+      else
+        if current_privileges.include?(privilege)
+          converge_by("revoke #{privilege} for #{user} on #{new_resource}") do
+            Chef::Log.info("Revoking #{privilege} for #{user} on #{new_resource}")
+            @mysql.execute(:command => "REVOKE #{@mysql.privilege_name(privilege)} ON `#{new_resource.database}`.* FROM #{user}")
+          end
+        end
+      end
+    end
+  end
+end
+
+action :drop do
+  if @mysql.databases.include?(new_resource.database)
+    converge_by("drop #{new_resource}") do
+      Chef::Log.info("Dropping #{new_resource}")
+      @mysql.execute(:command => "DROP DATABASE `#{new_resource.database}`")
+    end
+  end
+end