microwave 0.1004.1

data/lib/chef/provider/ruby_block.rb

@@ -0,0 +1,34 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: AJ Christensen (<aj@opscode.com>)
+# Copyright:: Copyright (c) 2009 Opscode
+# 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.
+#
+
+class Chef
+  class Provider
+    class RubyBlock < Chef::Provider
+      def load_current_resource
+        true
+      end
+
+      def action_create
+        @new_resource.block.call
+				Chef::Log.info("#{@new_resource} called")
+        @new_resource.updated_by_last_action(true)
+      end
+    end
+  end
+end

data/lib/chef/provider/script.rb

@@ -0,0 +1,55 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 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 'tempfile'
+require 'chef/provider/execute'
+
+class Chef
+  class Provider
+    class Script < Chef::Provider::Execute
+
+      def action_run
+        script_file.puts(@new_resource.code)
+        script_file.close
+
+        set_owner_and_group
+
+        @new_resource.command("\"#{@new_resource.interpreter}\" #{@new_resource.flags} \"#{script_file.path}\"")
+        super
+      ensure
+        unlink_script_file
+      end
+
+      def set_owner_and_group
+        # FileUtils itself implements a no-op if +user+ or +group+ are nil
+        # You can prove this by running FileUtils.chown(nil,nil,'/tmp/file')
+        # as an unprivileged user.
+        FileUtils.chown(@new_resource.user, @new_resource.group, script_file.path)
+      end
+
+      def script_file
+        @script_file ||= Tempfile.open("chef-script")
+      end
+
+      def unlink_script_file
+        @script_file && @script_file.close!
+      end
+
+    end
+  end
+end

data/lib/chef/provider/service.rb

@@ -0,0 +1,122 @@
+#
+# Author:: AJ Christensen (<aj@hjksolutions.com>)
+# Copyright:: Copyright (c) 2008 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/mixin/command'
+require 'chef/provider'
+
+class Chef
+  class Provider
+    class Service < Chef::Provider
+
+      include Chef::Mixin::Command
+
+      def initialize(new_resource, run_context)
+        super
+        @enabled = nil
+      end
+
+      def action_enable
+        if @current_resource.enabled
+          Chef::Log.debug("#{@new_resource} already enabled - nothing to do")
+        else
+          if $testrun || enable_service
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} enabled")
+          end
+        end
+      end
+
+      def action_disable
+        if @current_resource.enabled
+          if $testrun || disable_service
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} disabled")
+          end
+        else
+          Chef::Log.debug("#{@new_resource} already disabled - nothing to do")
+        end
+      end
+
+      def action_start
+        unless @current_resource.running
+          if $testrun || start_service
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} started")
+          end
+        else
+          Chef::Log.debug("#{@new_resource} already running - nothing to do")
+        end
+      end
+
+      def action_stop
+        if @current_resource.running
+          if $testrun || stop_service
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} stopped")
+          end
+        else
+          Chef::Log.debug("#{@new_resource} already stopped - nothing to do")
+        end
+      end
+
+      def action_restart
+        if $testrun || restart_service
+          @new_resource.updated_by_last_action(true)
+          Chef::Log.info("#{@new_resource} restarted")
+        end
+      end
+
+      def action_reload
+        unless (@new_resource.supports[:reload] || @new_resource.reload_command)
+          raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :reload"
+        end
+        if @current_resource.running
+          if $testrun || reload_service
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} reloaded")
+          end
+        end
+      end
+
+      def enable_service
+        raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :enable"
+      end
+
+      def disable_service
+        raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :disable"
+      end
+
+      def start_service
+        raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :start"
+      end
+
+      def stop_service
+        raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :stop"
+      end
+
+      def restart_service
+        raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :restart"
+      end
+
+      def reload_service
+        raise Chef::Exceptions::UnsupportedAction, "#{self.to_s} does not support :restart"
+      end
+
+    end
+  end
+end

data/lib/chef/provider/service/arch.rb

@@ -0,0 +1,116 @@
+#
+# Author:: Jan Zimmek (<jan.zimmek@web.de>)
+# Copyright:: Copyright (c) 2010 Jan Zimmek
+# 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/service/init'
+require 'chef/mixin/command'
+
+class Chef::Provider::Service::Arch < Chef::Provider::Service::Init
+
+  def initialize(new_resource, run_context)
+    super
+    @init_command = "/etc/rc.d/#{@new_resource.service_name}"
+  end
+
+  def load_current_resource
+
+    raise Chef::Exceptions::Service unless ::File.exists?("/etc/rc.conf")
+    raise Chef::Exceptions::Service unless ::File.read("/etc/rc.conf").match(/DAEMONS=\((.*)\)/m)
+
+    super
+
+    @current_resource.enabled(daemons.include?(@current_resource.service_name))
+
+    @current_resource
+  end
+
+  # Get list of all daemons from the file '/etc/rc.conf'.
+  # Mutiple lines and background form are supported. Example:
+  #   DAEMONS=(\
+  #     foobar \
+  #     @example \
+  #     !net \
+  #   )
+  def daemons
+    entries = []
+    if ::File.read("/etc/rc.conf").match(/DAEMONS=\((.*)\)/m)
+      entries += $1.gsub(/\\?[\r\n]/, ' ').gsub(/# *[^ ]+/,' ').split(' ') if $1.length > 0
+    end
+
+    yield(entries) if block_given?
+
+    entries
+  end
+
+  # FIXME: Multiple entries of DAEMONS will cause very bad results :)
+  def update_daemons(entries)
+    content = ::File.read("/etc/rc.conf").gsub(/DAEMONS=\((.*)\)/m, "DAEMONS=(#{entries.join(' ')})")
+    ::File.open("/etc/rc.conf", "w") do |f|
+      f.write(content)
+    end
+  end
+
+  def enable_service()
+    new_daemons = []
+    entries = daemons
+
+    if entries.include?(new_resource.service_name) or entries.include?("@#{new_resource.service_name}")
+      # exists and already enabled (or already enabled as a background service)
+      # new_daemons += entries
+    else
+      if entries.include?("!#{new_resource.service_name}")
+        # exists but disabled
+        entries.each do |daemon|
+          if daemon == "!#{new_resource.service_name}"
+            new_daemons << new_resource.service_name
+          else
+            new_daemons << daemon
+          end
+        end
+      else
+        # does not exist
+        new_daemons += entries
+        new_daemons << new_resource.service_name
+      end
+      update_daemons(new_daemons)
+    end
+  end
+
+  def disable_service()
+    new_daemons = []
+    entries = daemons
+
+    if entries.include?("!#{new_resource.service_name}")
+      # exists and disabled
+      # new_daemons += entries
+    else
+      if entries.include?(new_resource.service_name) or entries.include?("@#{new_resource.service_name}")
+        # exists but enabled (or enabled as a back-ground service)
+        # FIXME: Does arch support !@foobar ?
+        entries.each do |daemon|
+          if [new_resource.service_name, "@#{new_resource.service_name}"].include?(daemon)
+            new_daemons << "!#{new_resource.service_name}"
+          else
+            new_daemons << daemon
+          end
+        end
+      end
+      update_daemons(new_daemons)
+    end
+  end
+
+end

data/lib/chef/provider/service/debian.rb

@@ -0,0 +1,130 @@
+#
+# Author:: AJ Christensen (<aj@hjksolutions.com>)
+# Copyright:: Copyright (c) 2008 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/service'
+require 'chef/provider/service/init'
+require 'chef/mixin/command'
+
+class Chef
+  class Provider
+    class Service
+      class Debian < Chef::Provider::Service::Init
+        UPDATE_RC_D_ENABLED_MATCHES = /\/rc[\dS].d\/S|not installed/i
+        UPDATE_RC_D_PRIORITIES = /\/rc([\dS]).d\/([SK])(\d\d)/i
+
+        def load_current_resource
+          super
+
+          @current_resource.priority(get_priority)
+          @current_resource.enabled(service_currently_enabled?(@current_resource.priority))
+          @current_resource
+        end
+
+        def assert_update_rcd_available
+          unless ::File.exists? "/usr/sbin/update-rc.d"
+            raise Chef::Exceptions::Service, "/usr/sbin/update-rc.d does not exist!"
+          end
+        end
+
+        def get_priority
+          assert_update_rcd_available
+          priority = {}
+
+          status = popen4("/usr/sbin/update-rc.d -n -f #{@current_resource.service_name} remove") do |pid, stdin, stdout, stderr|
+
+            [stdout, stderr].each do |iop|
+              iop.each_line do |line|
+                if UPDATE_RC_D_PRIORITIES =~ line
+                  # priority[runlevel] = [ S|K, priority ] 
+                  # S = Start, K = Kill
+                  # debian runlevels: 0 Halt, 1 Singleuser, 2 Multiuser, 3-5 == 2, 6 Reboot
+                  priority[$1] = [($2 == "S" ? :start : :stop), $3]
+                end
+                if line =~ UPDATE_RC_D_ENABLED_MATCHES
+                  enabled = true
+                end
+              end
+            end
+          end
+
+          unless status.exitstatus == 0
+            raise Chef::Exceptions::Service, "/usr/sbin/update-rc.d -n -f #{@current_resource.service_name} failed - #{status.inspect}"
+          end
+          priority
+        end
+
+        def service_currently_enabled?(priority)
+          enabled = false
+
+          priority.each { |runlevel, arguments|
+            Chef::Log.debug("#{@new_resource} runlevel #{runlevel}, action #{arguments[0]}, priority #{arguments[1]}")
+            
+            # if we are in a update-rc.d default startup runlevel && we start in this runlevel
+            if (2..5).include?(runlevel.to_i) && arguments[0] == :start
+              enabled = true
+            end
+          }
+
+          enabled
+        end
+
+        def enable_service()
+          if @new_resource.priority.is_a? Integer
+            run_command(:command => "/usr/sbin/update-rc.d -f #{@new_resource.service_name} remove")
+            run_command(:command => "/usr/sbin/update-rc.d #{@new_resource.service_name} defaults #{@new_resource.priority} #{100 - @new_resource.priority}")
+          elsif @new_resource.priority.is_a? Hash
+            # we call the same command regardless of we're enabling or disabling  
+            # users passing a Hash are responsible for setting their own start priorities
+            set_priority()
+          else # No priority, go with update-rc.d defaults
+            run_command(:command => "/usr/sbin/update-rc.d -f #{@new_resource.service_name} remove")
+            run_command(:command => "/usr/sbin/update-rc.d #{@new_resource.service_name} defaults")
+          end
+
+        end
+
+        def disable_service()
+          if @new_resource.priority.is_a? Integer
+            # Stop processes in reverse order of start using '100 - start_priority'
+            run_command(:command => "/usr/sbin/update-rc.d -f #{@new_resource.service_name} remove")
+            run_command(:command => "/usr/sbin/update-rc.d -f #{@new_resource.service_name} stop #{100 - @new_resource.priority} 2 3 4 5 .")
+          elsif @new_resource.priority.is_a? Hash
+            # we call the same command regardless of we're enabling or disabling  
+            # users passing a Hash are responsible for setting their own stop priorities
+            set_priority()
+          else 
+            # no priority, using '100 - 20 (update-rc.d default)' to stop in reverse order of start
+            run_command(:command => "/usr/sbin/update-rc.d -f #{@new_resource.service_name} remove")
+            run_command(:command => "/usr/sbin/update-rc.d -f #{@new_resource.service_name} stop 80 2 3 4 5 .")
+          end
+        end
+
+        def set_priority()
+          args = ""
+          @new_resource.priority.each do |level, o|
+            action = o[0]
+            priority = o[1]
+            args += "#{action} #{priority} #{level} . "
+          end
+          run_command(:command => "/usr/sbin/update-rc.d -f #{@new_resource.service_name} remove")
+          run_command(:command => "/usr/sbin/update-rc.d #{@new_resource.service_name} #{args}")
+        end
+      end
+    end
+  end
+end