elzar 0.0.1

data/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

data/cookbooks/database/libraries/resource_database.rb

@@ -0,0 +1,111 @@
+#
+# 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/resource'
+
+class Chef
+  class Resource
+    class Database < Chef::Resource
+
+      def initialize(name, run_context=nil)
+        super
+        @resource_name = :database
+        @database_name = name
+        @allowed_actions.push(:create, :drop, :query)
+        @action = :create
+      end
+
+      def database_name(arg=nil)
+        set_or_return(
+          :database_name,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def connection(arg=nil)
+        set_or_return(
+          :connection,
+          arg,
+          :required => true
+        )
+      end
+
+      def sql(arg=nil)
+        set_or_return(
+          :sql,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def sql(arg=nil)
+        set_or_return(
+          :sql,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      # 
+      def template(arg=nil)
+        set_or_return(
+          :template,
+          arg,
+          :kind_of => String,
+          :default => 'DEFAULT'
+        )
+      end
+
+      def encoding(arg=nil)
+        set_or_return(
+          :encoding,
+          arg,
+          :kind_of => String,
+          :default => 'DEFAULT'
+        )
+      end
+
+      def tablespace(arg=nil)
+        set_or_return(
+          :tablespace,
+          arg,
+          :kind_of => String,
+          :default => 'DEFAULT'
+        )
+      end
+
+      def connection_limit(arg=nil)
+        set_or_return(
+          :connection_limit,
+          arg,
+          :kind_of => String,
+          :default => '-1'
+        )
+      end
+
+      def owner(arg=nil)
+        set_or_return(
+          :owner,
+          arg,
+          :kind_of => String
+        )
+      end
+    end
+  end
+end

data/cookbooks/database/libraries/resource_database_user.rb

@@ -0,0 +1,90 @@
+#
+# 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__), 'resource_database')
+
+class Chef
+  class Resource
+    class DatabaseUser < Chef::Resource::Database
+
+      def initialize(name, run_context=nil)
+        super
+        @resource_name = :database_user
+        @username = name
+
+        @database_name = nil
+        @table = nil
+        @host = 'localhost'
+        @privileges = [:all]
+
+        @allowed_actions.push(:create, :drop, :grant)
+        @action = :create
+      end
+
+      def database_name(arg=nil)
+        set_or_return(
+          :database_name,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def username(arg=nil)
+        set_or_return(
+          :username,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def password(arg=nil)
+        set_or_return(
+          :password,
+          arg,
+          :kind_of => String,
+          :required => true
+        )
+      end
+
+      def table(arg=nil)
+        set_or_return(
+          :table,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def host(arg=nil)
+        set_or_return(
+          :host,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def privileges(arg=nil)
+        set_or_return(
+          :privileges,
+          arg,
+          :kind_of => Array
+        )
+      end
+
+    end
+  end
+end

data/cookbooks/database/libraries/resource_mysql_database.rb

@@ -0,0 +1,34 @@
+#
+# 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__), 'resource_database')
+require File.join(File.dirname(__FILE__), 'provider_database_mysql')
+
+class Chef
+  class Resource
+    class MysqlDatabase < Chef::Resource::Database
+
+      def initialize(name, run_context=nil)
+        super
+        @resource_name = :mysql_database
+        @provider = Chef::Provider::Database::Mysql
+      end
+
+    end
+  end
+end

data/cookbooks/database/libraries/resource_mysql_database_user.rb

@@ -0,0 +1,34 @@
+#
+# 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__), 'resource_database_user')
+require File.join(File.dirname(__FILE__), 'provider_database_mysql_user')
+
+class Chef
+  class Resource
+    class MysqlDatabaseUser < Chef::Resource::DatabaseUser
+
+      def initialize(name, run_context=nil)
+        super
+        @resource_name = :mysql_database_user
+        @provider = Chef::Provider::Database::MysqlUser
+      end
+
+    end
+  end
+end

data/cookbooks/database/libraries/resource_postgresql_database.rb

@@ -0,0 +1,35 @@
+#
+# 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__), 'resource_database')
+require File.join(File.dirname(__FILE__), 'provider_database_postgresql')
+
+class Chef
+  class Resource
+    class PostgresqlDatabase < Chef::Resource::Database
+
+      def initialize(name, run_context=nil)
+        super
+        @resource_name = :postgresql_database
+        @provider = Chef::Provider::Database::Postgresql
+      end
+
+    end
+  end
+end

data/cookbooks/database/libraries/resource_postgresql_database_user.rb

@@ -0,0 +1,35 @@
+#
+# 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__), 'resource_database_user')
+require File.join(File.dirname(__FILE__), 'provider_database_postgresql_user')
+
+class Chef
+  class Resource
+    class PostgresqlDatabaseUser < Chef::Resource::DatabaseUser
+
+      def initialize(name, run_context=nil)
+        super
+        @resource_name = :postgresql_database_user
+        @provider = Chef::Provider::Database::PostgresqlUser
+      end
+
+    end
+  end
+end