New SENSU handling added to "contacts"

lib/god.rb

@@ -95,6 +95,7 @@ def load_contact(name)
 load_contact(:webhook)
 load_contact(:airbrake)
 load_contact(:slack)
+load_contact(:sensu)
 
 $:.unshift File.join(File.dirname(__FILE__), *%w[.. ext god])
 

lib/god/contacts/sensu.rb

@@ -0,0 +1,57 @@
+# ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+# Send a notice to a SENSU client socket, port 3030 on 'localhost' only.
+# ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+# [mandatory]
+# check_name    - a unique check name
+#
+# [optional]
+# status_code   - status codes used are 0 for OK, 1 for WARNING, 2 for CRITICAL, and 3 or greater to indicate UNKNOWN or CUSTOM.
+# handler       - default handler
+#
+
+CONTACT_DEPS[:sensu] = ['json']
+CONTACT_DEPS[:sensu].each do |d|
+  require d
+end
+
+module God
+  module Contacts
+
+    class Sensu < Contact
+      class << self
+        attr_accessor :check_name, :status_code, :handler, :host, :port
+      end
+
+      self.status_code = 2
+      self.handler = 'default'
+      self.host = 'localhost'
+      self.port = 3030
+
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'check_name' must be specified", self) unless arg(:check_name)
+        valid
+      end
+
+      attr_accessor :check_name, :status_code, :handler, :host, :port
+
+      def sensu_client_socket(msg)
+        u = UDPSocket.new
+        u.send(msg + "\n", 0, arg(:host).nil? ? self.host : arg(:host), arg(:port).nil? ? self.port : arg(:port))
+      end
+
+      def notify(message, time, priority, category, host)
+        data = {
+          :category => category,
+          :message => message,
+          :priority => priority,
+          :host => host,
+          :time => time,
+        }
+        parcel = { 'name' => eval("\"" + arg(:check_name) + "\""), 'status' => arg(:status_code).nil? ? self.status_code : arg(:status_code), 'output' => data.to_json, 'handler' => arg(:handler).empty? ? self.handler : arg(:handler) }
+        sensu_client_socket parcel.to_json
+      end
+    end
+  end
+end
+

test/test_sensu.rb

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+require File.dirname(__FILE__) + '/helper'
+
+class TestSensu < Test::Unit::TestCase
+  def test_sensu_notify
+    sensu = God::Contacts::Sensu.new
+    sensu.check_name = "TestSensuContact"
+
+    sensu.notify("Test", Time.now, "Test", "Test", "")
+    assert_equal "sent sensu #{sensu.check_name} notification with status code #{sensu.status_code}", "sent sensu #{sensu.check_name} notification with status code #{sensu.status_code}"
+  end
+end