tylerrick-chef 0.10.8

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

@@ -0,0 +1,154 @@
+#
+# Author:: Bryan McLellan (btm@loftninjas.org)
+# Copyright:: Copyright (c) 2009 Bryan McLellan
+# 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/mixin/command'
+
+class Chef
+  class Provider
+    class Service
+      class Freebsd < Chef::Provider::Service::Init
+
+        def load_current_resource
+          @current_resource = Chef::Resource::Service.new(@new_resource.name)
+          @current_resource.service_name(@new_resource.service_name)
+
+          # Determine if we're talking about /etc/rc.d or /usr/local/etc/rc.d
+          if ::File.exists?("/etc/rc.d/#{current_resource.service_name}")
+            @init_command = "/etc/rc.d/#{current_resource.service_name}"
+          elsif ::File.exists?("/usr/local/etc/rc.d/#{current_resource.service_name}")
+            @init_command = "/usr/local/etc/rc.d/#{current_resource.service_name}"
+          else
+            raise Chef::Exceptions::Service, "#{@new_resource}: unable to locate the rc.d script"
+          end
+          Chef::Log.debug("#{@current_resource} found at #{@init_command}")
+
+          if @new_resource.supports[:status]
+            begin
+              if run_command(:command => "#{@init_command} status") == 0
+                @current_resource.running true
+                Chef::Log.debug("#{@new_resource} is running")
+              end
+            rescue Chef::Exceptions::Exec
+              @current_resource.running false
+              nil
+            end
+
+          elsif @new_resource.status_command
+            begin
+              if run_command(:command => @new_resource.status_command) == 0
+                @current_resource.running true
+                Chef::Log.debug("#{@new_resource} is running")
+              end
+            rescue Chef::Exceptions::Exec
+              @current_resource.running false
+              nil
+            end
+
+          else
+            Chef::Log.debug("#{@new_resource} does not support status and you have not specified a status command, falling back to process table inspection")
+
+            if node[:command][:ps].nil? or node[:command][:ps].empty?
+              raise Chef::Exceptions::Service, "#{@new_resource} could not determine how to inspect the process table, please set this nodes 'ps' attribute"
+            end
+
+            status = popen4(node[:command][:ps]) do |pid, stdin, stdout, stderr|
+              r = Regexp.new(@new_resource.pattern)
+              Chef::Log.debug("#{@new_resource} attempting to match #{@new_resource.pattern} (#{r}) against process table")
+              stdout.each_line do |line|
+                if r.match(line)
+                  @current_resource.running true
+                  break
+                end
+              end
+              @current_resource.running false unless @current_resource.running
+            end
+            unless status.exitstatus == 0
+              raise Chef::Exceptions::Service, "Command #{node[:command][:ps]} failed"
+            else
+              Chef::Log.debug("#{@new_resource} #{node[:command][:ps]} exited and parsed successfully, process running: #{@current_resource.running}")
+            end
+          end
+
+          if ::File.exists?("/etc/rc.conf")
+            read_rc_conf.each do |line|
+              case line
+              when /#{Regexp.escape(service_enable_variable_name)}="(\w+)"/
+                if $1 =~ /[Yy][Ee][Ss]/
+                  @current_resource.enabled true
+                elsif $1 =~ /[Nn][Oo][Nn]?[Oo]?[Nn]?[Ee]?/
+                  @current_resource.enabled false
+                end
+              end
+            end
+          end
+          unless @current_resource.enabled
+            Chef::Log.debug("#{@new_resource.name} enable/disable state unknown")
+          end
+
+          @current_resource
+        end
+
+        def read_rc_conf
+          ::File.open("/etc/rc.conf", 'r') { |file| file.readlines }
+        end
+
+        def write_rc_conf(lines)
+          ::File.open("/etc/rc.conf", 'w') do |file|
+            lines.each { |line| file.puts(line) }
+          end
+        end
+
+
+        # The variable name used in /etc/rc.conf for enabling this service
+        def service_enable_variable_name
+          # Look for name="foo" in the shell script @init_command. Use this for determining the variable name in /etc/rc.conf
+          # corresponding to this service
+          # For example: to enable the service mysql-server with the init command /usr/local/etc/rc.d/mysql-server, you need
+          # to set mysql_enable="YES" in /etc/rc.conf
+          makefile = ::File.open(@init_command)
+          makefile.each do |line|
+            case line
+            when /^name="?(\w+)"?/
+              return $1 + "_enable"
+            end
+          end
+          raise Chef::Exceptions::Service, "Could not find name=\"service\" line in #{@init_command}"
+        end
+
+        def set_service_enable(value)
+          lines = read_rc_conf
+          # Remove line that set the old value
+          lines.delete_if { |line| line =~ /#{service_enable_variable_name}/ }
+          # And append the line that sets the new value at the end
+          lines << "#{service_enable_variable_name}=\"#{value}\""
+          write_rc_conf(lines)
+        end
+
+        def enable_service()
+          set_service_enable("YES") unless @current_resource.enabled
+        end
+
+        def disable_service()
+          set_service_enable("NO") if @current_resource.enabled
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,53 @@
+#
+# Author:: Lee Jensen (<ljensen@engineyard.com>)
+# Author:: AJ Christensen (<aj@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 'chef/provider/service'
+require 'chef/mixin/command'
+
+class Chef::Provider::Service::Gentoo < Chef::Provider::Service::Init
+  def load_current_resource
+
+    @new_resource.supports[:status] = true
+    @new_resource.supports[:restart] = true
+
+    super
+    
+    raise Chef::Exceptions::Service unless ::File.exists?("/sbin/rc-update")
+    
+    @current_resource.enabled(
+      Dir.glob("/etc/runlevels/**/#{@current_resource.service_name}").any? do |file|
+        exists = ::File.exists? file
+        readable = ::File.readable? file
+        Chef::Log.debug "#{@new_resource} exists: #{exists}, readable: #{readable}"
+        exists and readable
+      end
+    )
+    Chef::Log.debug "#{@new_resource} enabled: #{@current_resource.enabled}"
+
+    @current_resource
+  end
+  
+  def enable_service()
+    run_command(:command => "/sbin/rc-update add #{@new_resource.service_name} default")
+  end
+  
+  def disable_service()
+    run_command(:command => "/sbin/rc-update del #{@new_resource.service_name} default")
+  end
+end

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

@@ -0,0 +1,71 @@
+#
+# 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/simple'
+require 'chef/mixin/command'
+
+class Chef
+  class Provider
+    class Service
+      class Init < Chef::Provider::Service::Simple
+        
+        def initialize(new_resource, run_context)
+          super
+          @init_command = "/etc/init.d/#{@new_resource.service_name}"
+        end
+
+        def start_service
+          if @new_resource.start_command
+            super
+          else
+            run_command(:command => "#{@init_command} start")
+          end
+        end
+
+        def stop_service
+          if @new_resource.stop_command
+            super
+          else
+            run_command(:command => "#{@init_command} stop")
+          end
+        end
+
+        def restart_service
+          if @new_resource.restart_command
+            super
+          elsif @new_resource.supports[:restart]
+            run_command(:command => "#{@init_command} restart")
+          else
+            stop_service
+            sleep 1
+            start_service
+          end
+        end
+
+        def reload_service
+          if @new_resource.reload_command
+            super
+          elsif @new_resource.supports[:reload]
+            run_command(:command => "#{@init_command} reload")
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,52 @@
+#
+# Author:: Bryan McLellan <btm@loftninjas.org>
+# 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/service'
+require 'chef/provider/service/init'
+require 'chef/mixin/command'
+
+class Chef
+  class Provider
+    class Service
+      class Insserv < Chef::Provider::Service::Init
+
+        def load_current_resource
+          super
+
+          # Look for a /etc/rc.*/SnnSERVICE link to signifiy that the service would be started in a runlevel
+          if Dir.glob("/etc/rc**/S*#{@current_resource.service_name}").empty?
+            @current_resource.enabled false
+          else
+            @current_resource.enabled true 
+          end
+  
+          @current_resource
+        end
+
+        def enable_service()
+          run_command(:command => "/sbin/insserv -r -f #{@new_resource.service_name}")
+          run_command(:command => "/sbin/insserv -d -f #{@new_resource.service_name}")
+        end
+
+        def disable_service()
+          run_command(:command => "/sbin/insserv -r -f #{@new_resource.service_name}")
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+#
+# 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 Invokercd < Chef::Provider::Service::Init
+        
+        def initialize(new_resource, run_context)
+          super
+          @init_command = "/usr/sbin/invoke-rc.d #{@new_resource.service_name}"
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,60 @@
+#
+# 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/shell_out'
+
+class Chef
+  class Provider
+    class Service
+      class Redhat < Chef::Provider::Service::Init
+        include Chef::Mixin::ShellOut
+        
+        CHKCONFIG_ON = /\d:on/
+        
+        def initialize(new_resource, run_context)
+          super
+           @init_command = "/sbin/service #{@new_resource.service_name}"
+           @new_resource.supports[:status] = true
+         end
+        
+        def load_current_resource
+          unless ::File.exists? "/sbin/chkconfig"
+            raise Chef::Exceptions::Service, "/sbin/chkconfig does not exist!"
+          end
+          
+          super
+          
+          chkconfig = shell_out!("/sbin/chkconfig --list #{@current_resource.service_name}", :returns => [0,1])
+          @current_resource.enabled(!!(chkconfig.stdout =~ CHKCONFIG_ON))
+          @current_resource        
+        end
+
+        def enable_service()
+          shell_out! "/sbin/chkconfig #{@new_resource.service_name} on"
+        end
+
+        def disable_service()
+          shell_out! "/sbin/chkconfig #{@new_resource.service_name} off"
+        end
+        
+      end
+    end
+  end
+end

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

@@ -0,0 +1,120 @@
+#
+# Author:: Mathieu Sauve-Frankel <msf@kisoku.net>
+# Copyright:: Copyright (c) 2009 Mathieu Sauve-Frankel
+# 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/mixin/command'
+
+class Chef
+  class Provider
+    class Service
+      class Simple < Chef::Provider::Service
+        def load_current_resource
+          @current_resource = Chef::Resource::Service.new(@new_resource.name)
+          @current_resource.service_name(@new_resource.service_name)
+          if @new_resource.status_command
+            Chef::Log.debug("#{@new_resource} you have specified a status command, running..")
+
+            begin
+              if run_command(:command => @new_resource.status_command) == 0
+                @current_resource.running true
+                Chef::Log.debug("#{@new_resource} is running")
+              end
+            rescue Chef::Exceptions::Exec
+              @current_resource.running false
+              nil
+            end
+
+          elsif @new_resource.supports[:status]
+            Chef::Log.debug("#{@new_resource} supports status, running")
+
+            begin
+              if run_command(:command => "#{@init_command} status") == 0
+                @current_resource.running true
+                Chef::Log.debug("#{@new_resource} is running")
+              end
+            rescue Chef::Exceptions::Exec
+              @current_resource.running false
+              nil
+            end
+          elsif
+            Chef::Log.debug "#{@new_resource} falling back to process table inspection"
+            if ps_cmd.nil? or ps_cmd.empty?
+              raise Chef::Exceptions::Service, "#{@new_resource} could not determine how to inspect the process table, please set this nodes 'command.ps' attribute"
+            end
+            status = popen4(ps_cmd) do |pid, stdin, stdout, stderr|
+              r = Regexp.new(@new_resource.pattern)
+              Chef::Log.debug "#{@new_resource} attempting to match '#{@new_resource.pattern}' (#{r.inspect}) against process list"
+              stdout.each_line do |line|
+                if r.match(line)
+                  @current_resource.running true
+                  break
+                end
+              end
+              @current_resource.running false unless @current_resource.running
+            end
+            unless status.exitstatus == 0
+              raise Chef::Exceptions::Service, "Command #{ps_cmd} failed"
+            else
+              Chef::Log.debug "#{@new_resource} running: #{@current_resource.running}"
+            end
+          end
+
+          @current_resource
+        end
+
+        def start_service
+          if @new_resource.start_command
+            run_command(:command => @new_resource.start_command)
+          else
+            raise Chef::Exceptions::Service, "#{self.to_s} requires that start_command to be set"
+          end
+        end
+
+        def stop_service
+          if @new_resource.stop_command
+            run_command(:command => @new_resource.stop_command)
+          else
+            raise Chef::Exceptions::Service, "#{self.to_s} requires that stop_command to be set"
+          end
+        end
+
+        def restart_service
+          if @new_resource.restart_command
+            run_command(:command => @new_resource.restart_command)
+          else
+            stop_service
+            sleep 1
+            start_service
+          end
+        end
+
+        def reload_service
+          if @new_resource.reload_command
+            run_command(:command => @new_resource.reload_command)
+          else
+            raise Chef::Exceptions::Service, "#{self.to_s} requires that reload_command to be set"
+          end
+        end
+
+        def ps_cmd
+          @run_context.node[:command] && @run_context.node[:command][:ps]
+        end
+      end
+    end
+  end
+end