mcproc 2016.2.20

data/lib/god/contacts/prowl.rb

@@ -0,0 +1,57 @@
+# Send a notice to Prowl (http://prowl.weks.net/).
+#
+# apikey - The String API key.
+
+CONTACT_DEPS[:prowl] = ['prowly']
+CONTACT_DEPS[:prowl].each do |d|
+  require d
+end
+
+module God
+  module Contacts
+    class Prowl < Contact
+
+      class << self
+        attr_accessor :apikey
+      end
+
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'apikey' must be specified", self) if self.apikey.nil?
+        valid
+      end
+
+      attr_accessor :apikey
+
+      def notify(message, time, priority, category, host)
+        result = Prowly.notify do |n|
+          n.apikey      = arg(:apikey)
+          n.priority    = map_priority(priority.to_i)
+          n.application = category || "God"
+          n.event       = "on " + host.to_s
+          n.description = message.to_s + " at " + time.to_s
+        end
+
+        if result.succeeded?
+          self.info = "sent prowl notification to #{self.name}"
+        else
+          self.info = "failed to send prowl notification to #{self.name}: #{result.message}"
+        end
+      rescue Object => e
+        applog(nil, :info, "failed to send prowl notification to #{self.name}: #{e.message}")
+        applog(nil, :debug, e.backtrace.join("\n"))
+      end
+
+      def map_priority(priority)
+        case priority
+           when 1 then Prowly::Notification::Priority::EMERGENCY
+           when 2 then Prowly::Notification::Priority::HIGH
+           when 3 then Prowly::Notification::Priority::NORMAL
+           when 4 then Prowly::Notification::Priority::MODERATE
+           when 5 then Prowly::Notification::Priority::VERY_LOW
+           else Prowly::Notification::Priority::NORMAL
+        end
+      end
+    end
+  end
+end

data/lib/god/contacts/scout.rb

@@ -0,0 +1,55 @@
+# Send a notice to Scout (http://scoutapp.com/).
+#
+# client_key - The String client key.
+# plugin_id  - The String plugin id.
+
+require 'net/http'
+require 'uri'
+
+module God
+  module Contacts
+
+    class Scout < Contact
+      class << self
+        attr_accessor :client_key, :plugin_id
+        attr_accessor :format
+      end
+
+      self.format = lambda do |message, priority, category, host|
+        text  = "Message: #{message}\n"
+        text += "Host: #{host}\n"         if host
+        text += "Priority: #{priority}\n" if priority
+        text += "Category: #{category}\n" if category
+        return text
+      end
+
+      attr_accessor :client_key, :plugin_id
+
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'client_key' must be specified", self) unless arg(:client_key)
+        valid &= complain("Attribute 'plugin_id' must be specified", self) unless arg(:plugin_id)
+        valid
+      end
+
+      def notify(message, time, priority, category, host)
+        data = {
+          :client_key => arg(:client_key),
+          :plugin_id => arg(:plugin_id),
+          :format => 'xml',
+          'alert[subject]' => message,
+          'alert[body]' => Scout.format.call(message, priority, category, host)
+        }
+
+        uri = URI.parse('http://scoutapp.com/alerts/create')
+        Net::HTTP.post_form(uri, data)
+
+        self.info = "sent scout alert to plugin ##{plugin_id}"
+      rescue => e
+        applog(nil, :info, "failed to send scout alert to plugin ##{plugin_id}: #{e.message}")
+        applog(nil, :debug, e.backtrace.join("\n"))
+      end
+    end
+
+  end
+end

data/lib/god/contacts/sensu.rb

@@ -0,0 +1,59 @@
+# ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+# 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))
+        u.close
+      end
+
+      def notify(message, time, priority, category, host)
+        data = {
+          :category => category,
+          :message => message,
+          :priority => priority,
+          :host => host,
+          :time => time,
+        }
+        parcel = { 'name' => 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), 'executed' => Time.now.to_i }
+        sensu_client_socket parcel.to_json
+        self.info = "notified sensu: #{arg(:check_name)}"
+      end
+    end
+  end
+end
+

data/lib/god/contacts/slack.rb

@@ -0,0 +1,98 @@
+# Send a message to a Slack channel
+#
+# account        - The name of your Slack account (visible in URL, e.g. foo.slack.com)
+# token          - The token of the webhook created in Slack
+# channel        - The name of the channel to send the message to, prefixed with #
+# notify_channel - Whether to send an "@channel" in the message, to alert everyone in the channel
+# format         - An optional format string to change how the alert is displayed
+
+require 'net/http'
+require 'uri'
+
+CONTACT_DEPS[:slack] = ['json']
+CONTACT_DEPS[:slack].each do |d|
+  require d
+end
+
+module God
+  module Contacts
+
+    class Slack < Contact
+      class << self
+        attr_accessor :url, :channel, :notify_channel, :format, :username, :emoji
+      end
+
+      self.channel        = "#general"
+      self.notify_channel = false
+      self.format         = "%{priority} alert on %{host}: %{message} (%{category}, %{time})"
+
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'url' must be specified", self) unless arg(:url)
+        valid
+      end
+
+      attr_accessor :url, :channel, :notify_channel, :format, :username, :emoji
+
+      def text(data)
+        text = ""
+        text << "<!channel> " if arg(:notify_channel)
+
+        if RUBY_VERSION =~ /^1\.8/
+          text << arg(:format).gsub(/%\{(\w+)\}/) do |match|
+            data[$1.to_sym]
+          end
+        else
+          text << arg(:format) % data
+        end
+
+        text
+      end
+
+      def notify(message, time, priority, category, host)
+        text = text({
+          :message => message,
+          :time => time,
+          :priority => priority,
+          :category => category,
+          :host => host
+        })
+
+        request(text)
+      end
+
+      def api_url
+        URI.parse arg(:url)
+      end
+
+      def request(text)
+        http = Net::HTTP.new(api_url.host, api_url.port)
+        http.use_ssl = true
+
+        req = Net::HTTP::Post.new(api_url.request_uri)
+        req.body = {
+          :link_names => 1,
+          :text => text,
+          :channel => arg(:channel)
+        }.tap { |payload|
+          payload[:username] = arg(:username) if arg(:username)
+          payload[:icon_emoji] = arg(:emoji) if arg(:emoji)
+        }.to_json
+
+        res = http.request(req)
+
+        case res
+          when Net::HTTPSuccess
+            self.info = "successfully notified slack on channel #{arg(:channel)}"
+          else
+            self.info = "failed to send webhook to #{arg(:url)}: #{res.error!}"
+        end
+      rescue Object => e
+        applog(nil, :info, "failed to send webhook to #{arg(:url)}: #{e.message}")
+        applog(nil, :debug, e.backtrace.join("\n"))
+      end
+
+    end
+
+  end
+end

data/lib/god/contacts/statsd.rb

@@ -0,0 +1,46 @@
+# Send a notice to statsd
+#
+# host - statsd host
+# port - statsd port (optional)
+
+require 'statsd-ruby'
+
+module God
+  module Contacts
+
+    class Statsd < Contact
+      class << self
+        attr_accessor :host, :port
+      end
+
+      attr_accessor :host, :port
+
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'statsd_host' must be specified", self) unless arg(:host)
+        valid
+      end
+
+      def notify(message, time, priority, category, hostname)
+        statsd = ::Statsd.new host, (port ? port.to_i : 8125) # 8125 is the default statsd port
+
+        hostname.gsub! /\./, '_'
+        app = message.gsub /([^\s]*).*/, '\1'
+
+        [
+            'cpu out of bounds',
+            'memory out of bounds',
+            'process is flapping'
+        ].each do |event_type|
+          statsd.increment "god.#{event_type.gsub(/\s/, '_')}.#{hostname}.#{app}" if message.include? event_type
+        end
+
+        self.info = 'sent statsd alert'
+      rescue => e
+        applog(nil, :info, "failed to send statsd alert: #{e.message}")
+        applog(nil, :debug, e.backtrace.join("\n"))
+      end
+    end
+
+  end
+end

data/lib/god/contacts/twitter.rb

@@ -0,0 +1,51 @@
+# Send a notice to a Twitter account (http://twitter.com/).
+#
+# consumer_token  - The String OAuth consumer token (defaults to God's
+#                   existing consumer token).
+# consumer_secret - The String OAuth consumer secret (defaults to God's
+#                   existing consumer secret).
+# access_token    - The String OAuth access token.
+# access_secret   - The String OAuth access secret.
+
+CONTACT_DEPS[:twitter] = ['twitter']
+CONTACT_DEPS[:twitter].each do |d|
+  require d
+end
+
+module God
+  module Contacts
+    class Twitter < Contact
+      class << self
+        attr_accessor :consumer_token, :consumer_secret,
+                      :access_token, :access_secret
+      end
+
+      self.consumer_token = 'gOhjax6s0L3mLeaTtBWPw'
+      self.consumer_secret = 'yz4gpAVXJHKxvsGK85tEyzQJ7o2FEy27H1KEWL75jfA'
+
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'consumer_token' must be specified", self) unless arg(:consumer_token)
+        valid &= complain("Attribute 'consumer_secret' must be specified", self) unless arg(:consumer_secret)
+        valid &= complain("Attribute 'access_token' must be specified", self) unless arg(:access_token)
+        valid &= complain("Attribute 'access_secret' must be specified", self) unless arg(:access_secret)
+        valid
+      end
+
+      attr_accessor :consumer_token, :consumer_secret,
+                    :access_token, :access_secret
+
+      def notify(message, time, priority, category, host)
+        oauth = ::Twitter::OAuth.new(arg(:consumer_token), arg(:consumer_secret))
+        oauth.authorize_from_access(arg(:access_token), arg(:access_secret))
+
+        ::Twitter::Base.new(oauth).update(message)
+
+        self.info = "sent twitter update"
+      rescue => e
+        applog(nil, :info, "failed to send twitter update: #{e.message}")
+        applog(nil, :debug, e.backtrace.join("\n"))
+      end
+    end
+  end
+end

data/lib/god/contacts/webhook.rb

@@ -0,0 +1,74 @@
+# Send a notice to a webhook.
+#
+# url    - The String webhook URL.
+# format - The Symbol format [ :form | :json ] (default: :form).
+
+require 'net/http'
+require 'uri'
+
+CONTACT_DEPS[:webhook] = ['json']
+CONTACT_DEPS[:webhook].each do |d|
+  require d
+end
+
+module God
+  module Contacts
+
+    class Webhook < Contact
+      class << self
+        attr_accessor :url, :format
+      end
+
+      self.format = :form
+
+      def valid?
+        valid = true
+        valid &= complain("Attribute 'url' must be specified", self) unless arg(:url)
+        valid &= complain("Attribute 'format' must be one of [ :form | :json ]", self) unless [:form, :json].include?(arg(:format))
+        valid
+      end
+
+      attr_accessor :url, :format
+
+      def notify(message, time, priority, category, host)
+        data = {
+          :message => message,
+          :time => time,
+          :priority => priority,
+          :category => category,
+          :host => host
+        }
+
+        uri = URI.parse(arg(:url))
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true if uri.scheme == "https"
+
+        req = nil
+        res = nil
+
+        case arg(:format)
+          when :form
+            req = Net::HTTP::Post.new(uri.request_uri)
+            req.set_form_data(data)
+          when :json
+            req = Net::HTTP::Post.new(uri.request_uri)
+            req.body = data.to_json
+        end
+
+        res = http.request(req)
+
+        case res
+          when Net::HTTPSuccess
+            self.info = "sent webhook to #{arg(:url)}"
+          else
+            self.info = "failed to send webhook to #{arg(:url)}: #{res.error!}"
+        end
+      rescue Object => e
+        applog(nil, :info, "failed to send webhook to #{arg(:url)}: #{e.message}")
+        applog(nil, :debug, e.backtrace.join("\n"))
+      end
+
+    end
+
+  end
+end