vaimo-messages-handler 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 89b17c68d79f1ef94b034d9f753b5696d2386e14
-  data.tar.gz: cf312e17e699da7f6d0c250c6495e3a701c45ec9
+  metadata.gz: 943072b1a328bf2fb6c784aa8bf3e18fd11dd573
+  data.tar.gz: 80374526e30581bb5d7bdbce22e1709fb7173c75
 SHA512:
-  metadata.gz: 3122e0f06f520700be49bdca317b42581711dbe29292af81cfc9378cea492771a0799b47773e10855f4703401159ffb3240aef744cabab46413cc414d1d6dfd0
-  data.tar.gz: d5475a8688ffed4ee848bd3de2b3345eda88c10eec7acae77d006358d4ece8d873dc7c1d0f9ba6b1416c540a76f655ad0c543325d67e114df3607d08bba02f35
+  metadata.gz: 73e3f616aabbe948c607582ffe4a3bccb0b6c5eab4ee5259c8854bec0d23c7f6f2f96c63e9a1fb2a526d7939fe2e00d6b3b2efa62b52b49df27b53d94dd92549
+  data.tar.gz: 50b7fd10bef5877c4e74d4fef2fbf1c3f24475fa141e998c79713170bb4d9f7ef8ce0330a0a6134deda6e079733fbe396f1a5afe2b6ba673d56fd2305c83e4db

data/bin/send-message.rb

@@ -28,11 +28,100 @@
 
 require 'date'
 require 'sensu-handler'
-require 'timeout'
+require 'net/http'
+require 'uri'
+require 'json'
 
 # VaimoMessages
 class VaimoMessages < Sensu::Handler
+  option :json_config_name,
+         description: 'Name of the JSON Configuration block (default: messages)',
+         short:       '-j JsonConfig',
+         long:        '--json_config JsonConfig',
+         required:    false,
+         default:     'messages'
+
+  def config_settings
+    settings[config[:json_config_name]]
+  end
+
+  def client_name
+    @event['client']['name']
+  end
+
+  def check_name
+    @event['check']['name']
+  end
+
+  def level
+    @event['action'].eql?('resolve') ? 'normal' : 'error'
+  end
+
+  def web_hook
+    config_settings['web_hook'] || nil
+  end
+
+  def auth_token
+    config_settings['auth_token'] || nil
+  end
+
+  def recipient
+    config_settings['recipient'] || nil
+  end
+
+  def from
+    config_settings['from'] || 'SensuAlert'
+  end
+
+  def post_args
+    {
+      auth_token: auth_token,
+      receiver:   recipient,
+      from:       from,
+      server:     client_name,
+      level:      level,
+      message:    nil,
+      component:  check_name
+    }
+  end
+
+  def parse_response(response)
+    data = JSON.parse(response)
+
+    hash = {}
+    data.each do |key, value|
+      hash[key] = value
+    end
+
+    raise ErrorHandler "Messages WebHook: #{hash['message']}" if hash['error']
+
+    hash
+  end
+
+  def send_message
+    header = ['Content-Type' => 'text/json']
+
+    uri = URI.parse(web_hook)
+
+    http         = Net::HTTP.new(uri.host, uri.port)
+    http.use_ssl = true
+
+    request      = Net::HTTP::Post.new(uri.request_uri, header)
+    request.body = post_args.to_json
+
+    response = http.request(request)
+
+    parse_response(response.body)
+  end
+
   def handle
-    File.open('/tmp/ruby-sensu-data', 'w') { |file| file.write(gets) }
+    raise 'No web_hook configured' if web_hook.nil?
+    raise 'No auth_token configured' if auth_token.nil?
+    raise 'No recipient configured' if recipient.nil?
+
+    send_message
+  end
+
+  class ErrorHandler < StandardError
   end
-end
+end

data/lib/vaimo-messages-handler/version.rb

@@ -2,7 +2,7 @@ module VaimoMessagesHandler
   module Version
     MAJOR = 1
     MINOR = 0
-    PATCH = 1
+    PATCH = 2
 
     VER_STRING = [MAJOR, MINOR, PATCH].compact.join('.')
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vaimo-messages-handler
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Margus Heinmaa