A very basic sensu::check type

Author: rodjek

lib/puppet/provider/sensu_check_config/json.rb

@@ -0,0 +1,68 @@
+require 'json'
+
+Puppet::Type.type(:sensu_check_config).provide(:json) do
+  def initialize(*args)
+    super
+
+    begin
+      @conf = JSON.parse(File.read("/etc/sensu/conf.d/checks_#{resource[:name]}.json"))
+    rescue
+      @conf = {}
+    end
+  end
+
+  def flush
+    File.open("/etc/sensu/conf.d/checks_#{resource[:name]}.json", 'w') do |f|
+      f.puts JSON.pretty_generate(@conf)
+    end
+  end
+
+  def create
+    @conf['checks'] = {}
+    @conf['checks'][resource[:name]] = {}
+    self.handlers = resource[:handlers]
+    self.command = resource[:command]
+    self.interval = resource[:interval]
+    self.subscribers = resource[:subscribers]
+  end
+
+  def destroy
+    @conf = nil
+  end
+
+  def exists?
+    @conf.has_key?('checks') and @conf['checks'].has_key?(resource[:name])
+  end
+
+  def interval
+    @conf['checks'][resource[:name]]['interval'].to_s
+  end
+
+  def interval=(value)
+    @conf['checks'][resource[:name]]['interval'] = value.to_i
+  end
+
+  def handlers
+    @conf['checks'][resource[:name]]['handlers'] || []
+  end
+
+  def handlers=(value)
+    @conf['checks'][resource[:name]]['handlers'] = value
+  end
+
+  def command
+    @conf['checks'][resource[:name]]['command']
+  end
+
+  def command=(value)
+    @conf['checks'][resource[:name]]['command'] = value
+  end
+
+  def subscribers
+    @conf['checks'][resource[:name]]['subscribers'] || []
+  end
+
+  def subscribers=(value)
+    @conf['checks'][resource[:name]]['subscribers'] = value
+  end
+end

lib/puppet/type/sensu_check_config.rb

@@ -0,0 +1,35 @@
+Puppet::Type.newtype(:sensu_check_config) do
+  @doc = ""
+
+  ensurable do
+    newvalue(:present) do
+      provider.create
+    end
+
+    newvalue(:absent) do
+      provider.destroy
+    end
+
+    defaultto :present
+  end
+
+  newparam(:name) do
+    desc "The name of the check"
+  end
+
+  newproperty(:handlers, :array_matching => :all) do
+    desc ""
+  end
+
+  newproperty(:command) do
+    desc ""
+  end
+
+  newproperty(:interval) do
+    desc ""
+  end
+
+  newproperty(:subscribers, :array_matching => :all) do
+    desc ""
+  end
+end

manifests/check.pp

@@ -0,0 +1,14 @@
+define sensu::check(
+                    $command,
+                    $handlers    = [],
+                    $interval    = '60',
+                    $subscribers = []
+                    ) {
+
+  sensu_check_config { $name:
+    command     => $command,
+    handlers    => $handlers,
+    interval    => $interval,
+    subscribers => $subscribers,
+  }
+}