server_maint 0.0.3 → 0.0.4

data/lib/cookbooks/database/libraries/provider_database_mysql.rb

@@ -0,0 +1,103 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+require 'chef/provider'
+
+class Chef
+  class Provider
+    class Database
+      class Mysql < Chef::Provider
+        include Chef::Mixin::ShellOut
+
+        def load_current_resource
+          Gem.clear_paths
+          require 'mysql'
+          @current_resource = Chef::Resource::Database.new(@new_resource.name)
+          @current_resource.database_name(@new_resource.database_name)
+          @current_resource
+        end
+
+        def action_create
+          unless exists?
+            begin
+              Chef::Log.debug("#{@new_resource}: Creating database #{new_resource.database_name}")
+              create_sql = "CREATE DATABASE #{new_resource.database_name}"
+              create_sql += " CHARACTER SET = #{new_resource.encoding}" if new_resource.encoding
+              create_sql += " COLLATE = #{new_resource.collation}" if new_resource.collation
+              Chef::Log.debug("#{@new_resource}: Performing query [#{create_sql}]")
+              db.query(create_sql)
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_drop
+          if exists?
+            begin
+              Chef::Log.debug("#{@new_resource}: Dropping database #{new_resource.database_name}")
+              db.query("drop database #{new_resource.database_name}")
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_query
+          if exists?
+            begin
+              db.select_db(@new_resource.database_name) if @new_resource.database_name
+              Chef::Log.debug("#{@new_resource}: Performing query [#{new_resource.sql_query}]")
+              db.query(@new_resource.sql_query)
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        private
+        def exists?
+          db.list_dbs.include?(@new_resource.database_name)
+        end
+
+        def db
+          @db ||= begin
+            connection = ::Mysql.new(
+              @new_resource.connection[:host],
+              @new_resource.connection[:username],
+              @new_resource.connection[:password],
+              nil,
+              @new_resource.connection[:port] || 3306
+            )
+            connection.set_server_option ::Mysql::OPTION_MULTI_STATEMENTS_ON
+            connection
+          end
+        end
+
+        def close
+          @db.close rescue nil
+          @db = nil
+        end
+
+      end
+    end
+  end
+end

data/lib/cookbooks/database/libraries/provider_database_mysql_user.rb

@@ -0,0 +1,76 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+require File.join(File.dirname(__FILE__), 'provider_database_mysql')
+
+class Chef
+  class Provider
+    class Database
+      class MysqlUser < Chef::Provider::Database::Mysql
+        include Chef::Mixin::ShellOut
+
+        def load_current_resource
+          Gem.clear_paths
+          require 'mysql'
+          @current_resource = Chef::Resource::DatabaseUser.new(@new_resource.name)
+          @current_resource.username(@new_resource.name)
+          @current_resource
+        end
+
+        def action_create
+          unless exists?
+            begin
+              db.query("CREATE USER '#{@new_resource.username}'@'#{@new_resource.host}' IDENTIFIED BY '#{@new_resource.password}'")
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_drop
+          if exists?
+            begin
+              db.query("DROP USER '#{@new_resource.username}'@'#{@new_resource.host}'")
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_grant
+          begin
+            grant_statement = "GRANT #{@new_resource.privileges.join(', ')} ON #{@new_resource.database_name || "*"}.#{@new_resource.table || "*"} TO '#{@new_resource.username}'@'#{@new_resource.host}' IDENTIFIED BY '#{@new_resource.password}'"
+            Chef::Log.info("#{@new_resource}: granting access with statement [#{grant_statement}]")
+            db.query(grant_statement)
+            @new_resource.updated_by_last_action(true)
+          ensure
+            close
+          end
+        end
+
+        private
+        def exists?
+          db.query("select User,host from mysql.user where User='#{@new_resource.username}' AND host = '#{@new_resource.host}'").num_rows != 0
+        end
+
+      end
+    end
+  end
+end

data/lib/cookbooks/database/libraries/provider_database_postgresql.rb

@@ -0,0 +1,131 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Author:: Lamont Granquist (<lamont@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+require 'chef/provider'
+
+class Chef
+  class Provider
+    class Database
+      class Postgresql < Chef::Provider
+        include Chef::Mixin::ShellOut
+
+        def load_current_resource
+          Gem.clear_paths
+          require 'pg'
+          @current_resource = Chef::Resource::Database.new(@new_resource.name)
+          @current_resource.database_name(@new_resource.database_name)
+          @current_resource
+        end
+
+        def action_create
+          unless exists?
+            begin
+              encoding = @new_resource.encoding
+              if encoding != "DEFAULT"
+                encoding = "'#{@new_resource.encoding}'"
+              end
+              Chef::Log.debug("#{@new_resource}: Creating database #{new_resource.database_name}")
+              create_sql = "CREATE DATABASE #{new_resource.database_name}"
+              create_sql += " TEMPLATE = #{new_resource.template}" if new_resource.template
+              create_sql += " ENCODING = #{encoding}" if new_resource.encoding
+              create_sql += " TABLESPACE = #{new_resource.tablespace}" if new_resource.tablespace
+              create_sql += " LC_CTYPE = '#{new_resource.collation}' LC_COLLATE = '#{new_resource.collation}'" if new_resource.collation
+              create_sql += " CONNECTION LIMIT = #{new_resource.connection_limit}" if new_resource.connection_limit
+              create_sql += " OWNER = #{new_resource.owner}" if new_resource.owner
+              Chef::Log.debug("#{@new_resource}: Performing query [#{create_sql}]")
+              db("template1").query(create_sql)
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_drop
+          if exists?
+            begin
+              Chef::Log.debug("#{@new_resource}: Dropping database #{new_resource.database_name}")
+              db("template1").query("drop database #{new_resource.database_name}")
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_query
+          if exists?
+            begin
+              Chef::Log.debug("#{@new_resource}: Performing query [#{new_resource.sql_query}]")
+              db(@new_resource.database_name).query(@new_resource.sql_query)
+              Chef::Log.debug("#{@new_resource}: query [#{new_resource.sql_query}] succeeded")
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        private
+
+        def exists?
+          begin
+            Chef::Log.debug("#{@new_resource}: checking if database #{@new_resource.database_name} exists")
+            ret = db("template1").query("select * from pg_database where datname = '#{@new_resource.database_name}'").num_tuples != 0
+            ret ? Chef::Log.debug("#{@new_resource}: database #{@new_resource.database_name} exists") :
+                  Chef::Log.debug("#{@new_resource}: database #{@new_resource.database_name} does not exist")
+          ensure
+            close
+          end
+          ret
+        end
+
+        #
+        # Specifying the database in the connection parameter for the postgres resource is not recommended.
+        #
+        # - action_create/drop/exists will use the "template1" database to do work by default.
+        # - action_query will use the resource database_name.
+        # - specifying a database in the connection will override this behavior
+        #
+        def db(dbname = nil)
+          close if @db
+          dbname = @new_resource.connection[:database] if @new_resource.connection[:database]
+          host = @new_resource.connection[:host]
+          port = @new_resource.connection[:port] || 5432
+          user = @new_resource.connection[:username] || "postgres"
+          Chef::Log.debug("#{@new_resource}: connecting to database #{dbname} on #{host}:#{port} as #{user}")
+          password = @new_resource.connection[:password] || node[:postgresql][:password][:postgres]
+          @db = ::PGconn.new(
+            :host => host,
+            :port => port,
+            :dbname => dbname,
+            :user => user,
+            :password => password
+          )
+        end
+
+        def close
+          @db.close rescue nil
+          @db = nil
+        end
+
+      end
+    end
+  end
+end

data/lib/cookbooks/database/libraries/provider_database_postgresql_user.rb

@@ -0,0 +1,83 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Author:: Lamont Granquist (<lamont@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+require File.join(File.dirname(__FILE__), 'provider_database_postgresql')
+
+class Chef
+  class Provider
+    class Database
+      class PostgresqlUser < Chef::Provider::Database::Postgresql
+        include Chef::Mixin::ShellOut
+
+        def load_current_resource
+          Gem.clear_paths
+          require 'pg'
+          @current_resource = Chef::Resource::DatabaseUser.new(@new_resource.name)
+          @current_resource.username(@new_resource.name)
+          @current_resource
+        end
+
+        def action_create
+          unless exists?
+            begin
+              db("template1").query("CREATE USER #{@new_resource.username} WITH PASSWORD '#{@new_resource.password}'")
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_drop
+          if exists?
+            begin
+              db("template1").query("DROP USER #{@new_resource.username}")
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_grant
+          begin
+            # FIXME: grants on individual tables
+            grant_statement = "GRANT #{@new_resource.privileges.join(', ')} ON DATABASE #{@new_resource.database_name} TO #{@new_resource.username}"
+            Chef::Log.info("#{@new_resource}: granting access with statement [#{grant_statement}]")
+            db(@new_resource.database_name).query(grant_statement)
+            @new_resource.updated_by_last_action(true)
+          ensure
+            close
+          end
+        end
+
+        private
+        def exists?
+          begin
+            exists = db("template1").query("select * from pg_user where usename='#{@new_resource.username}'").num_tuples != 0
+          ensure
+            close
+          end
+          exists
+        end
+
+      end
+    end
+  end
+end

data/lib/cookbooks/database/libraries/provider_database_sql_server.rb

@@ -0,0 +1,109 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+require 'chef/provider'
+
+class Chef
+  class Provider
+    class Database
+      class SqlServer < Chef::Provider
+        include Chef::Mixin::ShellOut
+
+        def load_current_resource
+          Gem.clear_paths
+          require 'tiny_tds'
+          @current_resource = Chef::Resource::Database.new(@new_resource.name)
+          @current_resource.database_name(@new_resource.database_name)
+          @current_resource
+        end
+
+        def action_create
+          unless exists?
+            begin
+              Chef::Log.debug("#{@new_resource}: Creating database #{new_resource.database_name}")
+              db.execute("CREATE DATABASE [#{new_resource.database_name}]").do
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_drop
+          if exists?
+            begin
+              Chef::Log.debug("#{@new_resource}: Dropping database #{new_resource.database_name}")
+              db.execute("DROP DATABASE #{new_resource.database_name}").do
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        def action_query
+          if exists?
+            begin
+              #db.select_db(@new_resource.database_name) if @new_resource.database_name
+              Chef::Log.debug("#{@new_resource}: Performing query [#{new_resource.sql_query}]")
+              db.execute(@new_resource.sql_query).do
+              @new_resource.updated_by_last_action(true)
+            ensure
+              close
+            end
+          end
+        end
+
+        private
+        def exists?
+          exists = false
+          begin
+            result = db.execute("SELECT name FROM sys.databases")
+            result.each do |row|
+              if row['name'] =~ /#{@new_resource.database_name}/i
+                exists = true
+                break
+              end
+            end
+            result.cancel
+          ensure
+            close
+          end
+          exists
+        end
+
+        def db
+          @db ||= begin
+            ::TinyTds::Client.new(
+              :host => @new_resource.connection[:host],
+              :username => @new_resource.connection[:username],
+              :password => @new_resource.connection[:password],
+              :port => @new_resource.connection[:port] || 1433
+            )
+          end
+        end
+
+        def close
+          @db.close rescue nil
+          @db = nil
+        end
+
+      end
+    end
+  end
+end

data/lib/cookbooks/database/libraries/provider_database_sql_server_user.rb

@@ -0,0 +1,106 @@
+#
+# Author:: Seth Chisamore (<schisamo@opscode.com>)
+# Copyright:: Copyright (c) 2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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.
+#
+
+require File.join(File.dirname(__FILE__), 'provider_database_sql_server')
+
+class Chef
+  class Provider
+    class Database
+      class SqlServerUser < Chef::Provider::Database::SqlServer
+        include Chef::Mixin::ShellOut
+
+        def load_current_resource
+          Gem.clear_paths
+          require 'tiny_tds'
+          @current_resource = Chef::Resource::DatabaseUser.new(@new_resource.name)
+          @current_resource.username(@new_resource.name)
+          @current_resource
+        end
+
+        def action_create
+          begin
+            unless exists?(:logins)
+              db.execute("CREATE LOGIN #{@new_resource.username} WITH PASSWORD = '#{@new_resource.password}', CHECK_POLICY = OFF").do
+              @new_resource.updated_by_last_action(true)
+            end
+            unless exists?(:users)
+              if @new_resource.database_name
+                Chef::Log.info("#{@new_resource} creating user in '#{@new_resource.database_name}' database context.")
+                db.execute("USE #{@new_resource.database_name}").do
+              else
+                Chef::Log.info("#{@new_resource} database_name not provided, creating user in global context.")
+              end
+              db.execute("CREATE USER #{@new_resource.username} FOR LOGIN #{@new_resource.username}").do
+              @new_resource.updated_by_last_action(true)
+            end
+          ensure
+            close
+          end
+        end
+
+        def action_drop
+          begin
+            if exists?(:users)
+              db.execute("DROP USER #{@new_resource.username}").do
+              @new_resource.updated_by_last_action(true)
+            end
+            if exists?(:logins)
+              db.execute("DROP LOGIN #{@new_resource.username}").do
+              @new_resource.updated_by_last_action(true)
+            end
+          ensure
+            close
+          end
+        end
+
+        def action_grant
+          begin
+            if @new_resource.password
+              action_create
+            end
+            Chef::Application.fatal!('Please provide a database_name, SQL Server does not support global GRANT statements.') unless @new_resource.database_name
+            grant_statement = "GRANT #{@new_resource.privileges.join(', ')} ON DATABASE::#{@new_resource.database_name} TO #{@new_resource.username}"
+            Chef::Log.info("#{@new_resource} granting access with statement [#{grant_statement}]")
+            db.execute("USE #{@new_resource.database_name}").do
+            db.execute(grant_statement).do
+            @new_resource.updated_by_last_action(true)
+          ensure
+            close
+          end
+        end
+
+        private
+        def exists?(type=:users)
+          case type
+          when :users
+            table = "database_principals"
+            if @new_resource.database_name
+              Chef::Log.debug("#{@new_resource} searching for existing user in '#{@new_resource.database_name}' database context.")
+              db.execute("USE #{@new_resource.database_name}").do
+            end
+          when :logins
+            table = "server_principals"
+          end
+
+          result = db.execute("SELECT name FROM sys.#{table} WHERE name='#{@new_resource.username}'")
+          result.each.any?
+        end
+      end
+    end
+  end
+end