Enabling multiple init schemes and platforms for service resource

Squashed commit of the following:

commit 15f213281c01db0ccd5b3c7fe0992d4eeb4ae4ba
Author: Tyler Flint <[email protected]>
Date:   Thu Jun 11 15:05:10 2015 -0600

    runit admin is sv not svc

commit 9b7c78c55ee872f47b60e0c627c71a2bf6ce0e74
Author: Tyler Flint <[email protected]>
Date:   Thu Jun 11 15:03:16 2015 -0600

    adding in control logic for runit init scheme

commit df41cad9cffbdd5ccf22ffb81fd88b16e67ae012
Author: Tyler Flint <[email protected]>
Date:   Thu Jun 11 14:49:26 2015 -0600

    attempt to provide a general pattern for handling dynamic init schemes across multiple platforms

Authored by Tyler Flint <[email protected]>
Reviewed by Greg Linton <[email protected]>

lib/hooky/error.rb

@@ -3,5 +3,6 @@ module Error
     class UnexpectedExit < StandardError; end
     class UnknownAction < StandardError; end
     class UnsupportedPlatform < StandardError; end
+    class UnsupportedOption < StandardError; end
   end
 end

lib/hooky/resource/base.rb

@@ -90,6 +90,16 @@ def only_if(&block)
         end
       end
 
+      protected
+
+      def run_command!(cmd, expect_code=0)
+        `#{cmd}`
+        code = $?.exitstatus
+        if code != expect_code
+          raise Hooky::Error::UnexpectedExit, "#{cmd} failed with exit code '#{code}'"
+        end
+      end
+
     end
   end
 end

lib/hooky/resource/service.rb

@@ -4,12 +4,23 @@ class Service < Base
 
       field :recursive
       field :service_name
+      field :init
 
       actions :enable, :disable, :start, :stop, :restart, :reload
       default_action :enable
 
       def initialize(name)
         service_name(name) unless service_name
+
+        # if init scheme is not provided, try to set reasonable defaults
+        if not init
+          case platform.name
+          when 'smartos'
+            init(:smf)
+          when 'ubuntu'
+            init(:upstart)
+          end
+        end
       end
 
       def run(action)
@@ -32,26 +43,44 @@ def run(action)
       protected
 
       def enable!
-        run_command! "svcadm enable -s #{"-r" if recursive} #{service_name}"
+        case init
+        when :smf
+          run_command! "svcadm enable -s #{"-r" if recursive} #{service_name}"
+        when :runit
+          run_command! "sv start #{service_name}"
+        else
+          Hooky::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
+        end
       end
 
       def disable!
-        run_command! "svcadm disable -s #{service_name}"
+        case init
+        when :smf
+          run_command! "svcadm disable -s #{service_name}"
+        when :runit
+          run_command! "sv stop #{service_name}"
+        else
+          Hooky::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
+        end
       end
 
       def restart!
-        run_command! "svcadm restart #{service_name}"
+        case init
+        when :smf
+          run_command! "svcadm restart #{service_name}"
+        when :runit
+          disable!; enable!
+        else
+          Hooky::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
+        end
       end
 
       def reload!
-        run_command! "svcadm refresh #{service_name}"
-      end
-
-      def run_command!(cmd, expect_code=0)
-        `#{cmd}`
-        code = $?.exitstatus
-        if code != expect_code
-          raise Hooky::Error::UnexpectedExit, "#{cmd} failed with exit code '#{code}'"
+        case init
+        when :smf
+          run_command! "svcadm refresh #{service_name}"
+        else
+          Hooky::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
         end
       end