hookit 0.7.11

data/lib/hookit/resource/link.rb

@@ -0,0 +1,61 @@
+module Hookit
+  module Resource
+    class Link < Base
+
+      field :owner
+      field :group
+      field :link_type
+      field :target_file
+      field :to
+
+      actions :create, :delete
+      default_action :create
+
+      def initialize(name)
+        target_file name unless target_file
+        link_type :symbolic
+        super
+      end
+
+      def run(action)
+        case action
+        when :create
+          create!
+          chown!
+        when :delete
+          delete!
+        end
+      end
+
+      protected
+
+      def create!
+        args = ['f']
+        args << 'sn' if link_type == :symbolic
+        cmd = "ln -#{args.join} #{to} #{target_file}"
+        `#{cmd}`
+        code = $?.exitstatus
+        if code != 0
+          raise Hookit::Error::UnexpectedExit, "#{cmd} failed with exit code '#{code}'"
+        end
+      end
+
+      def delete!
+        cmd = "rm -f #{target_file}"
+        `#{cmd}`
+        code = $?.exitstatus
+        if code != 0
+          raise Hookit::Error::UnexpectedExit, "#{cmd} failed with exit code '#{code}'"
+        end
+      end
+
+      def chown!
+        return unless owner or group
+        if ::File.exists? target_file
+          `chown #{(group.nil?) ? owner : "#{owner}:#{group}"} #{target_file}`
+        end
+      end
+
+    end
+  end
+end

data/lib/hookit/resource/logrotate.rb

@@ -0,0 +1,38 @@
+module Hookit
+  module Resource
+    class Logrotate < Base
+
+      field :path
+      field :filesize
+      field :max_size
+      field :count
+
+      actions :create
+      default_action :create
+
+      def initialize(name)
+        path name unless path
+        super
+      end
+
+      def run(action)
+        case action
+        when :create
+          create!
+        end
+      end
+
+      protected
+
+      def create!
+        case platform.os
+        when 'sun'
+          `logadm -c -w #{path} -s #{filesize ||= '10m'} -S #{max_size ||= '500m'} -C #{count ||= '10'} -N`
+        else
+          raise Hookit::Error::UnsupportedPlatform, "unsupported platform '#{platform.name}'"
+        end
+      end
+
+    end
+  end
+end

data/lib/hookit/resource/mount.rb

@@ -0,0 +1,77 @@
+module Hookit
+  module Resource
+    class Mount < Base
+
+      field :device
+      # field :device_type
+      # field :dump
+      field :fstype
+      field :mount_point
+      field :options
+      field :pass
+      field :supports
+
+      actions :mount, :umount, :remount, :enable, :disable
+      default_action :mount
+
+      def initialize(name)
+        mount_point(name) unless mount_point
+        pass('-') unless pass
+        super
+      end
+
+      def run(action)
+        case action
+        when :mount
+          mount!
+        when :umount
+          umount!
+        when :remount
+          umount!
+          mount!
+        when :enable
+          disable!
+          enable!
+        when :disable
+          disable!
+        end
+      end
+
+      protected
+
+      def mount!
+        ::FileUtils.mkdir_p(mount_point)
+        run_command! "mount -O -F #{fstype} -o retry=5,timeo=300 #{options!(as_arg=true)} #{device} #{mount_point}"
+      end
+
+      def umount!
+        run_command! "umount #{mount_point}"
+      end
+
+      def enable!
+        entry = "#{device}\t#{device =~ /^\/dev/ ? device : "-"}\t#{mount_point}\t#{fstype}\t#{pass}\tyes\t#{options!}"
+        `echo "#{entry}" >> /etc/vfstab`
+      end
+
+      def disable!
+        `egrep -v "#{device}.*#{mount_point}" /etc/vfstab > /tmp/vfstab.tmp; mv -f /tmp/vfstab.tmp /etc/vfstab`
+      end
+
+      def options!(as_arg=false)
+        options = self.options.kind_of?(Array) ? self.options.join(',') : self.options
+        if as_arg
+          options ? (return "-o #{options}") : (return "")
+        end
+        options != "" ? (return "#{options}") : (return "-")
+      end
+
+      def run_command!(cmd, expect_code=0)
+        `#{cmd}`
+        code = $?.exitstatus
+        if code != expect_code
+          raise Hookit::Error::UnexpectedExit, "#{cmd} failed with exit code '#{code}'"
+        end
+      end
+    end
+  end
+end

data/lib/hookit/resource/package.rb

@@ -0,0 +1,74 @@
+module Hookit
+  module Resource
+    class Package < Base
+
+      field :package_name
+      field :source
+      field :version
+      field :scope
+
+      actions :install
+      default_action :install
+
+      def initialize(name)
+        package_name(name) unless source
+        scope :default unless scope
+        super
+      end
+
+      def run(action)
+        case action
+        when :install
+          install!
+        end
+      end
+
+      def install!
+        begin
+          install_package
+        rescue Hookit::Error::UnexpectedExit
+          if not registry("pkgsrc.#{scope}.updated")
+            update_pkg_db
+            registry("pkgsrc.#{scope}.updated", true)
+            retry
+          else
+            raise
+          end
+        end
+      end
+
+      protected
+
+      def install_package
+        `#{pkgin} -y in #{package}`
+
+        code = $?.exitstatus
+        if not code == 0
+          raise Hookit::Error::UnexpectedExit, "pkgin in #{package} failed with exit code '#{code}'"
+        end
+      end
+
+      def update_pkg_db
+        `#{pkgin} -y up`
+      end
+
+      def package
+        if version
+          "#{package_name}-#{version}"
+        else
+          package_name
+        end
+      end
+
+      def pkgin
+        case scope
+        when :default
+          "/opt/local/bin/pkgin"
+        when :gopagoda
+          "/opt/gopagoda/bin/pkgin"
+        end
+      end
+
+    end
+  end
+end

data/lib/hookit/resource/rsync.rb

@@ -0,0 +1,67 @@
+module Hookit
+  module Resource
+    class Rsync < Base
+
+      field :source
+      field :destination
+      field :wrapper
+      field :archive
+      field :recursive
+      field :checksum
+      field :compress
+
+      actions :sync
+      default_action :sync
+
+      def initialize(name)
+        source name unless source
+        super
+      end
+
+      def run(action)
+        case action
+        when :sync
+          sync!
+        end
+      end
+
+      def sync!
+        run_command! "rsync -q#{archive!}#{recursive!}#{checksum!}#{compress!} #{wrapper!} #{source} #{destination}"
+      end
+
+      def archive!
+        (return "a") if archive
+        ""
+      end
+
+      def recursive!
+        (return "r") if archive
+        ""
+      end
+
+      def checksum!
+        (return "c") if archive
+        ""
+      end
+
+      def compress!
+        (return "z") if archive
+        ""
+      end
+
+      def wrapper!
+        (return "-e '#{wrapper}'") if wrapper
+        ""
+      end
+
+      def run_command!(cmd, expect_code=0)
+        `#{cmd}`
+        code = $?.exitstatus
+        if code != expect_code
+          raise Hookit::Error::UnexpectedExit, "#{cmd} failed with exit code '#{code}'"
+        end
+      end
+
+    end
+  end
+end

data/lib/hookit/resource/scp.rb

@@ -0,0 +1,87 @@
+module Hookit
+  module Resource
+    class Scp < Base
+
+      field :source
+      field :destination
+      field :port
+      field :recursive
+      field :config
+      field :cipher
+      field :identity
+      field :ssh_options
+      field :compression
+      field :preserve
+
+      actions :copy
+      default_action :copy
+
+      def initialize(name)
+        source(name) unless source
+        preserve(true) unless preserve
+        recursive(true) unless recursive
+        super
+      end
+
+      def run(action)
+        case action
+        when :copy
+          copy!
+        end
+      end
+
+      def copy!
+        run_command!("scp -q#{preserve!}#{recursive!}B#{compression!} #{config!} #{port!} #{cipher!} #{identity!} #{ssh_options!} #{source} #{destination}")
+      end
+
+      def cipher!
+        (return "-c #{cipher}") if cipher
+        ""
+      end
+
+      def compression!
+        (return "C") if compression
+        ""
+      end
+
+      def config!
+        (return "-F #{config}") if config
+        ""
+      end
+
+      def identity!
+        (return "-i #{identity}") if identity
+        ""
+      end
+
+      def port!
+        (return "-P #{port}") if port
+        ""
+      end
+
+      def preserve!
+        (return "p") if preserve
+        ""
+      end
+
+      def recursive!
+        (return "r") if recursive
+        ""
+      end
+
+      def ssh_options!
+        (return "-o #{ssh_options}") if ssh_options
+        ""
+      end
+
+      def run_command!(cmd, expect_code=0)
+        `#{cmd}`
+        code = $?.exitstatus
+        if code != expect_code
+          raise Hookit::Error::UnexpectedExit, "#{cmd} failed with exit code '#{code}'"
+        end
+      end
+
+    end
+  end
+end

data/lib/hookit/resource/service.rb

@@ -0,0 +1,89 @@
+module Hookit
+  module Resource
+    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)
+        case action
+        when :enable
+          enable!
+        when :disable
+          disable!
+        when :start
+          enable!
+        when :stop
+          disable!
+        when :restart
+          restart!
+        when :reload
+          reload!
+        end
+      end
+
+      protected
+
+      def enable!
+        case init
+        when :smf
+          run_command! "svcadm enable -s #{"-r" if recursive} #{service_name}"
+        when :runit
+          run_command! "sv start #{service_name}"
+        else
+          Hookit::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
+        end
+      end
+
+      def disable!
+        case init
+        when :smf
+          run_command! "svcadm disable -s #{service_name}"
+        when :runit
+          run_command! "sv stop #{service_name}"
+        else
+          Hookit::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
+        end
+      end
+
+      def restart!
+        case init
+        when :smf
+          run_command! "svcadm restart #{service_name}"
+        when :runit
+          disable!; enable!
+        else
+          Hookit::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
+        end
+      end
+
+      def reload!
+        case init
+        when :smf
+          run_command! "svcadm refresh #{service_name}"
+        else
+          Hookit::Error::UnsupportedOption, "Unsupported init schema '#{init}'"
+        end
+      end
+
+    end
+  end
+end

data/lib/hookit/resource/socket.rb

@@ -0,0 +1,67 @@
+module Hookit
+  module Resource
+    class Socket < Base
+
+      field :port
+      field :interface
+      field :service
+      field :max_checks
+
+      actions :listening, :no_connections, :reset
+      default_action :listening
+
+      def initialize(name)
+        service name unless service
+        max_checks 3 unless max_checks
+        super
+      end
+
+      def run(action)
+        case action
+        when :listening
+          check_listening!
+        when :no_connections
+          check_no_connections!
+        when :reset
+          reset!
+        end
+      end
+
+      protected
+
+      def check_listening!
+        # increment check
+        registry("#{service}.listening", registry("#{service}.listening").to_i + 1)
+
+        if `netstat -an | grep '\*\.#{port}' | grep LISTEN`.empty?
+          count = registry("#{service}.listening").to_i
+          if count <= max_checks
+            exit(count + 10)
+          else
+            $stderr.puts "ERROR: timed out waiting for #{service} to listen"
+            exit(Hookit::Exit::ERROR)
+          end
+        end
+
+      end
+
+      def check_no_connections!
+        # increment check
+        registry("#{service}.no_connections", registry("#{service}.no_connections").to_i + 1)
+
+        unless `netstat -an | grep 'ESTABLISHED' | awk '{ print $1 }' | grep "$(ifconfig #{interface} | grep inet | awk '{ print $2 }')\.#{port}"`.empty?
+          count = registry("#{service}.no_connections").to_i
+          if count <= max_checks
+            exit(count + 10)
+          end
+        end
+      end
+
+      def reset!
+        registry("#{service}.listening", 0)
+        registry("#{service}.no_connections", 0)
+      end
+
+    end
+  end
+end

data/lib/hookit/resource/template.rb

@@ -0,0 +1,87 @@
+require 'tilt'
+require 'erubis'
+require 'erb'
+require 'oj'
+require 'multi_json'
+
+module Hookit
+  module Resource
+    class Template < Base
+      
+      field :path
+      field :source
+      field :variables
+      field :mode
+      field :owner
+      field :group
+
+      actions :create, :create_if_missing, :delete, :touch
+      default_action :create
+
+      def initialize(name)
+        path name unless path
+        source "#{::File.basename(name)}.erb"
+        super
+      end
+
+      def run(action)
+        case action
+        when :create
+          create!
+          chown!
+          chmod!
+        when :create_if_missing
+          create_if_missing!
+          chown!
+          chmod!
+        when :delete
+          delete!
+        when :touch
+          touch!
+        end
+      end
+
+      protected
+
+      def create!
+        ::File.write path, render
+      end
+
+      def create_if_missing!
+        if not ::File.exists? path
+          create!
+        end
+      end
+
+      def delete!
+        ::File.delete path
+      end
+
+      def chown!
+        return unless owner or group
+        `chown #{(group.nil?) ? owner : "#{owner}:#{group}"} #{path}`
+      end
+
+      def chmod!
+        ::File.chmod(mode, path) if mode
+      end
+
+      def touch!
+        `touch -c #{path}`
+      end
+
+      def render
+        Tilt.new("#{template_dir}/#{source}").render(self, variables)
+      end
+
+      def template_dir
+        "#{module_root}/templates"
+      end
+
+      def module_root
+        dict[:module_root]
+      end
+
+    end
+  end
+end