flapjack 1.2.1 → 1.2.2

data/lib/flapjack/gateways/email/alert.html.erb

@@ -32,10 +32,12 @@
       <td><%= @alert.state_title_case %></td>
     </tr>
 
-    <tr>
-      <td><strong>Summary</strong></td>
-      <td><%= @alert.summary %></td>
-    </tr>
+    <% if @alert.summary %>
+      <tr>
+        <td><strong>Summary</strong></td>
+        <td><%= @alert.summary %></td>
+      </tr>
+    <% end %>
 
     <% if @alert.details %>
       <tr>

data/lib/flapjack/gateways/email/alert.text.erb

@@ -5,7 +5,9 @@ Monitoring has detected the following:
 Entity:           <%= @alert.entity %>
 Check:            <%= @alert.check %>
 State:            <%= @alert.state_title_case %>
+<% if @alert.summary -%>
 Summary:          <%= @alert.summary %>
+<% end -%>
 <% if @alert.details -%>
 Details:          <%= @alert.details %>
 <% end -%>

data/lib/flapjack/gateways/jabber.rb

@@ -241,6 +241,7 @@ module Flapjack
           msg = "commands: \n" +
                 "  ACKID <id> <comment> [duration: <time spec>]\n" +
                 "  ack entities /pattern/ <comment> [duration: <time spec>]\n" +
+                "  maint entities /pattern/ <comment> [(start-in|start-at): <time spec>] [duration: <time spec>]\n" +
                 "  status entities /pattern/\n" +
                 "  ack checks /check_pattern/ on /entity_pattern/ <comment> [duration: <time spec>]\n" +
                 "  status checks /check_pattern/ on /entity_pattern/\n" +
@@ -423,6 +424,66 @@ module Flapjack
           end
         end
 
+        when /^(?:maint )?entities\s+\/(.+)\/(?:\s*(.*?)(?:\s*start-in:.*?(\w+.*?))?(?:\s*start-at:.*?(\w+.*?))?(?:\s*duration:.*?(\w+.*?))?)$/i
+          entity_pattern   = $1.strip
+          comment          = $2 ? $2.strip : nil
+          start_in         = $3 ? $3.strip : nil
+          start_at         = $4 ? $4.strip : nil
+          duration_str     = $5 ? $5.strip : '1 hour'
+          maint_duration   = ChronicDuration.parse(duration_str)
+          entity_names     = Flapjack::Data::Entity.find_all_name_matching(entity_pattern, :redis => @redis)
+
+          if comment.nil? || (comment.length == 0)
+            comment = "#{from}: Set via chatbot"
+          else
+            comment = "#{from}: #{comment}"
+          end
+
+          maint_start = case
+          when start_in
+            Time.now.to_i + ChronicDuration.parse(start_in)
+          when start_at
+            Chronic.parse(start_at).to_i
+          else
+            Time.now.to_i
+          end
+
+          if entity_names
+            number_found = entity_names.length
+            case
+            when number_found == 0
+              msg = "found no entities matching /#{entity_pattern}/"
+            when number_found >= 1
+              entities = entity_names.map {|name|
+                Flapjack::Data::Entity.find_by_name(name, :redis => @redis)
+              }.compact.inject({}) {|memo, entity|
+                memo[entity] = entity.check_list.sort
+                memo
+              }
+              if entities.length >= 1
+                entities.each_pair do |entity,check_list|
+                  check_list.each do |check|
+                    entity_check = Flapjack::Data::EntityCheck.for_entity(entity, check, :redis => @redis)
+                    entity_check.create_scheduled_maintenance(
+                      maint_start,
+                      maint_duration,
+                      :summary => comment,
+                      )
+                    entity_check.update_current_scheduled_maintenance
+                  end
+                end
+                msg = entities.inject("Maint list:\n") {|memo,kv|
+                  kv[1].each {|e| memo << "#{kv[0].name}:#{e}\n" }
+                  memo
+                }
+            else
+              msg = "found no matching entities with active checks"
+            end
+          else
+            msg = "that doesn't seem to be a valid pattern - /#{pattern}/"
+          end
+        end
+
         when /^(?:status )?entities\s+\/(.+)\/.*$/im
           entity_pattern  = $1 ? $1.strip : nil
           entity_names    = Flapjack::Data::Entity.find_all_name_matching(entity_pattern, :redis => @redis)
@@ -766,4 +827,3 @@ module Flapjack
 
   end
 end
-

data/lib/flapjack/gateways/jabber/alert.text.erb

@@ -7,6 +7,6 @@
 <% if ['acknowledgement'].include?(@alert.type) -%>
  has been acknowledged, unscheduled maintenance created for <%= time_period_in_words(@alert.acknowledgement_duration) -%>
 <% end -%>
-<% if @alert.summary && @alert.summary.length > 0 -%>
+<% if @alert.summary && !@alert.summary.empty? -%>
 , <%= @alert.summary -%>
 <% end -%>

data/lib/flapjack/gateways/pagerduty/alert.text.erb

@@ -5,6 +5,6 @@
 <% if ['acknowledgement'].include?(@alert.type) -%>
  has been acknowledged, unscheduled maintenance created for <%= time_period_in_words(@alert.acknowledgement_duration) -%>
 <% end -%>
-<% if @alert.summary && @alert.summary.length > 0 -%>
+<% if @alert.summary && !@alert.summary.empty? -%>
 , <%= @alert.summary -%>
 <% end -%>

data/lib/flapjack/gateways/sms_gammu.rb

@@ -0,0 +1,119 @@
+#!/usr/bin/env ruby
+
+require 'em-synchrony'
+require "em-synchrony/mysql2"
+require 'flapjack/data/alert'
+require 'flapjack/utility'
+
+module Flapjack
+  module Gateways
+    class SmsGammu
+      INSERT_QUERY = <<-SQL
+        INSERT INTO outbox (InsertIntoDB, TextDecoded, DestinationNumber, CreatorID, Class)
+        VALUES ('%s', '%s', '%s', '%s', %s)
+      SQL
+
+      include Flapjack::Utility
+
+      def initialize(opts = {})
+        @config = opts[:config]
+        @logger = opts[:logger]
+        @redis_config = opts[:redis_config] || {}
+
+        if @config.nil? || (@config.respond_to?(:empty?) && @config.empty?)
+          @logger.error "sms_gammu config is missing"
+          return
+        end
+
+        @redis = Flapjack::RedisPool.new(:config => @redis_config, :size => 1, :logger => @logger)
+
+        @logger.info("starting")
+        @logger.debug("new sms_gammu gateway pikelet with the following options: #{@config.inspect}")
+
+        @sent = 0
+      end
+
+      def stop
+        @logger.info("stopping")
+        @should_quit = true
+
+        redis_uri = @redis_config[:path] ||
+          "redis://#{@redis_config[:host] || '127.0.0.1'}:#{@redis_config[:port] || '6379'}/#{@redis_config[:db] || '0'}"
+        shutdown_redis = EM::Hiredis.connect(redis_uri)
+        shutdown_redis.rpush(@config['queue'], Flapjack.dump_json('notification_type' => 'shutdown'))
+      end
+
+      def start
+        @db = Mysql2::EM::Client.new(:host     => @config["mysql_host"],
+                                     :database => @config["mysql_database"],
+                                     :username => @config["mysql_username"],
+                                     :password => @config["mysql_password"])
+
+        queue = @config['queue']
+
+        until @should_quit
+          begin
+            @logger.debug("sms_gammu gateway is going into blpop mode on #{queue}")
+            deliver( Flapjack::Data::Alert.next(queue, :redis => @redis, :logger => @logger) )
+          rescue => e
+            @logger.error "Error generating or dispatching SMS Gammu message: #{e.class}: #{e.message}\n" +
+              e.backtrace.join("\n")
+          end
+        end
+      end
+
+      def deliver(alert)
+        @alert = alert
+        address = @alert.address
+        from    = @config["from"]
+        message = prepare_message
+        errors  = []
+
+        [[from,    "SMS from address is missing"],
+         [address, "SMS address is missing"]].each do |val_err|
+
+          next unless val_err.first.nil? || (val_err.first.respond_to?(:empty?) && val_err.first.empty?)
+          errors << val_err.last
+        end
+
+        unless errors.empty?
+          errors.each {|err| @logger.error err }
+          return
+        end
+
+        send_message(message, from, address)
+      rescue => e
+        @logger.error "Error generating or delivering sms to #{contents['address']}: #{e.class}: #{e.message}"
+        @logger.error e.backtrace.join("\n")
+        raise
+      end
+
+      def prepare_message
+        message_type  = @alert.rollup ? 'rollup' : 'alert'
+        template_path = @config['templates']["#{message_type}.text"]
+        template      = ERB.new(File.read(template_path), nil, '-')
+
+        begin
+          message = template.result(binding).chomp
+          truncate(message, 159)
+        rescue => e
+          @logger.error "Error while excuting the ERB for an sms: " +
+            "ERB being executed: #{template_path}"
+          raise
+        end
+      end
+
+      def send_message(message, from, to)
+        begin
+          @db.query(INSERT_QUERY % [Time.now, message, to, from, 1])
+          @sent += 1
+          @alert.record_send_success!
+          @logger.debug "Sent SMS via Gammu"
+        rescue => e
+          @logger.error "Failed to send SMS via Gammu: #{e.class}, #{e.message}"
+        end
+      end
+
+    end
+  end
+end

data/lib/flapjack/gateways/sms_messagenet.rb

@@ -4,6 +4,8 @@ require 'em-synchrony'
 require 'em-synchrony/em-http'
 require 'active_support/inflector'
 
+require 'flapjack/redis_pool'
+
 require 'flapjack/data/alert'
 require 'flapjack/utility'
 
@@ -13,95 +15,121 @@ module Flapjack
 
       MESSAGENET_DEFAULT_URL = 'https://www.messagenet.com.au/dotnet/Lodge.asmx/LodgeSMSMessage'
 
-      class << self
-
-        include Flapjack::Utility
+      include Flapjack::Utility
 
-        def start
-          @sent = 0
-        end
+      def initialize(opts = {})
+        @config = opts[:config]
+        @logger = opts[:logger]
+        @redis_config = opts[:redis_config] || {}
+        @redis = Flapjack::RedisPool.new(:config => @redis_config, :size => 1, :logger => @logger)
 
-        def perform(contents)
-          @logger.debug "Woo, got a notification to send out: #{contents.inspect}"
-          alert = Flapjack::Data::Alert.new(contents, :logger => @logger)
+        @logger.info("starting")
+        @logger.debug("new sms_messagenet gateway pikelet with the following options: #{@config.inspect}")
 
-          endpoint = @config["endpoint"] || MESSAGENET_DEFAULT_URL
-          username = @config["username"]
-          password = @config["password"]
+        @sent = 0
+      end
 
-          address         = alert.address
-          notification_id = alert.notification_id
-          message_type    = alert.rollup ? 'rollup' : 'alert'
+      def stop
+        @logger.info("stopping")
+        @should_quit = true
 
-          my_dir = File.dirname(__FILE__)
-          sms_template_path = case
-          when @config.has_key?('templates') && @config['templates']["#{message_type}.text"]
-            @config['templates']["#{message_type}.text"]
-          else
-            my_dir + "/sms_messagenet/#{message_type}.text.erb"
-          end
-          sms_template = ERB.new(File.read(sms_template_path), nil, '-')
+        redis_uri = @redis_config[:path] ||
+          "redis://#{@redis_config[:host] || '127.0.0.1'}:#{@redis_config[:port] || '6379'}/#{@redis_config[:db] || '0'}"
+        shutdown_redis = EM::Hiredis.connect(redis_uri)
+        shutdown_redis.rpush(@config['queue'], Flapjack.dump_json('notification_type' => 'shutdown'))
+      end
 
-          @alert  = alert
-          bnd     = binding
+      def start
+        queue = @config['queue']
 
+        until @should_quit
           begin
-            message = sms_template.result(bnd).chomp
+            @logger.debug("sms_messagenet gateway is going into blpop mode on #{queue}")
+            deliver( Flapjack::Data::Alert.next(queue, :redis => @redis, :logger => @logger) )
           rescue => e
-            @logger.error "Error while excuting the ERB for an sms: " +
-              "ERB being executed: #{sms_template_path}"
-            raise
-          end
-
-          if @config.nil? || (@config.respond_to?(:empty?) && @config.empty?)
-            @logger.error "Messagenet config is missing"
-            return
+            @logger.error "Error generating or dispatching SMS Messagenet message: #{e.class}: #{e.message}\n" +
+              e.backtrace.join("\n")
           end
+        end
+      end
 
-          errors = []
+      def deliver(alert)
+        endpoint = @config["endpoint"] || MESSAGENET_DEFAULT_URL
+        username = @config["username"]
+        password = @config["password"]
+
+        address         = alert.address
+        notification_id = alert.notification_id
+        message_type    = alert.rollup ? 'rollup' : 'alert'
+
+        my_dir = File.dirname(__FILE__)
+        sms_template_path = case
+        when @config.has_key?('templates') && @config['templates']["#{message_type}.text"]
+          @config['templates']["#{message_type}.text"]
+        else
+          my_dir + "/sms_messagenet/#{message_type}.text.erb"
+        end
+        sms_template = ERB.new(File.read(sms_template_path), nil, '-')
 
-          safe_message = truncate(message, 159)
+        @alert  = alert
+        bnd     = binding
 
-          [[username, "Messagenet username is missing"],
-           [password, "Messagenet password is missing"],
-           [address,  "SMS address is missing"],
-           [notification_id, "Notification id is missing"]].each do |val_err|
+        begin
+          message = sms_template.result(bnd).chomp
+        rescue => e
+          @logger.error "Error while excuting the ERB for an sms: " +
+            "ERB being executed: #{sms_template_path}"
+          raise
+        end
 
-            next unless val_err.first.nil? || (val_err.first.respond_to?(:empty?) && val_err.first.empty?)
-            errors << val_err.last
-          end
+        if @config.nil? || (@config.respond_to?(:empty?) && @config.empty?)
+          @logger.error "Messagenet config is missing"
+          return
+        end
 
-          unless errors.empty?
-            errors.each {|err| @logger.error err }
-            return
-          end
+        errors = []
 
-          query = {'Username'     => username,
-                   'Pwd'          => password,
-                   'PhoneNumber'  => address,
-                   'PhoneMessage' => safe_message}
+        safe_message = truncate(message, 159)
 
-          http = EM::HttpRequest.new(endpoint).get(:query => query)
+        [[username, "Messagenet username is missing"],
+         [password, "Messagenet password is missing"],
+         [address,  "SMS address is missing"],
+         [notification_id, "Notification id is missing"]].each do |val_err|
 
-          @logger.debug "server response: #{http.response}"
+          next unless val_err.first.nil? || (val_err.first.respond_to?(:empty?) && val_err.first.empty?)
+          errors << val_err.last
+        end
 
-          status = (http.nil? || http.response_header.nil?) ? nil : http.response_header.status
-          if (status >= 200) && (status <= 206)
-            @sent += 1
-            alert.record_send_success!
-            @logger.debug "Sent SMS via Messagenet, response status is #{status}, " +
-              "notification_id: #{notification_id}"
-          else
-            @logger.error "Failed to send SMS via Messagenet, response status is #{status}, " +
-              "notification_id: #{notification_id}"
-          end
-        rescue => e
-          @logger.error "Error generating or delivering sms to #{contents['address']}: #{e.class}: #{e.message}"
-          @logger.error e.backtrace.join("\n")
-          raise
+        unless errors.empty?
+          errors.each {|err| @logger.error err }
+          return
         end
 
+        query = {'Username'     => username,
+                 'Pwd'          => password,
+                 'PhoneNumber'  => address,
+                 'PhoneMessage' => safe_message}
+
+        http = EM::HttpRequest.new(endpoint).get(:query => query)
+
+        @logger.debug "server response: #{http.response}"
+
+        status = (http.nil? || http.response_header.nil?) ? nil : http.response_header.status
+        if (status >= 200) && (status <= 206)
+          @sent += 1
+          alert.record_send_success!
+          @logger.debug "Sent SMS via Messagenet, response status is #{status}, " +
+            "notification_id: #{notification_id}"
+        else
+          @logger.error "Failed to send SMS via Messagenet, response status is #{status}, " +
+            "notification_id: #{notification_id}"
+        end
+      rescue => e
+        @logger.error "Error generating or delivering sms to #{alert.address}: #{e.class}: #{e.message}"
+        @logger.error e.backtrace.join("\n")
+        raise
       end
+
     end
   end
 end

data/lib/flapjack/gateways/sms_messagenet/alert.text.erb

@@ -1,5 +1,6 @@
+<% summary = @alert.summary -%>
 <%= @alert.type_sentence_case %>: '<%= @alert.check %>' on <%= @alert.entity -%>
 <% unless ['acknowledgement', 'test'].include?(@alert.notification_type) -%>
  is <%= @alert.state_title_case -%>
 <% end -%>
- at <%= Time.at(@alert.time).strftime('%-d %b %H:%M') %>, <%= @alert.summary -%>
+ at <%= Time.at(@alert.time).strftime('%-d %b %H:%M') %><%= (summary.nil? || summary.empty?) ? '' : ", #{summary}" -%>