add client_subscription type

Author: jlambert121

lib/puppet/provider/sensu_client_subscription/json.rb

@@ -0,0 +1,30 @@
+require 'rubygems' if RUBY_VERSION < '1.9.0' && Puppet.features.rubygems?
+require 'json' if Puppet.features.json?
+
+Puppet::Type.type(:sensu_client_subscription).provide(:json) do
+  confine :feature => :json
+
+  def initialize(*args)
+    super
+
+    begin
+      @conf = JSON.parse(File.read("/etc/sensu/conf.d/subscription_#{resource[:name]}.json"))
+    rescue
+      @conf = {}
+    end
+  end
+
+  def flush
+    File.open("/etc/sensu/conf.d/subscription_#{resource[:name]}.json", 'w') do |f|
+      f.puts JSON.pretty_generate(@conf)
+    end
+  end
+
+  def create
+    @conf['client'] = {'subscriptions' => [ resource[:name] ] }
+  end
+
+  def destroy
+    @conf = nil
+  end
+end

lib/puppet/type/sensu_client_subscription.rb

@@ -0,0 +1,31 @@
+Puppet::Type.newtype(:sensu_client_subscription) do
+  @doc = ""
+
+  def initialize(*args)
+    super
+
+    self[:notify] = [
+      "Service[sensu-client]",
+    ].select { |ref| catalog.resource(ref) }
+  end
+
+  ensurable do
+    newvalue(:present) do
+      provider.create
+    end
+
+    newvalue(:absent) do
+      provider.destroy
+    end
+
+    defaultto :present
+  end
+
+  newparam(:name) do
+    desc "The subscription name"
+  end
+
+  autorequire(:package) do
+    ['sensu']
+  end
+end