#!/usr/bin/env ruby

require 'flapjack/data/entity_check'
require 'flapjack/filters/base'

module Flapjack
  module Filters

    # * If the service event’s state is a failure, and the time since the ok => failure state change
    #   is below a threshold (e.g. 30 seconds), then don't alert
    # * If the service event’s state is a failure, and the time since the last alert is below a
    #   threshold (5 minutes), and the last notification was not a recovery, then don’t alert
    class Delays
      include Base

      def block?(event)
        failure_delay = 30
        resend_delay  = 300

        result = false

        if (event.type == 'service') and (event.critical? or event.warning?)

          entity_check = Flapjack::Data::EntityCheck.for_event_id(event.id, :redis => @persistence)
          current_time = Time.now.to_i

          if entity_check.failed?
            last_problem_alert   = entity_check.last_problem_notification
            last_change          = entity_check.last_change
            last_notification    = entity_check.last_notification
            last_alert_type      = last_notification[:type]
            last_alert_timestamp = last_notification[:timestamp]

            current_failure_duration = current_time - last_change
            time_since_last_alert    = current_time - last_problem_alert unless last_problem_alert.nil?
            @log.debug("Filter: Delays: last_problem_alert: #{last_problem_alert.to_s}, " +
                       "last_change: #{last_change.to_s}, " +
                       "current_failure_duration: #{current_failure_duration}, " +
                       "time_since_last_alert: #{time_since_last_alert.to_s}")
            if (current_failure_duration < failure_delay)
              result = true
              @log.debug("Filter: Delays: blocking because duration of current failure " +
                         "(#{current_failure_duration}) is less than failure_delay (#{failure_delay})")
            elsif !last_problem_alert.nil? && (time_since_last_alert < resend_delay) &&
              (last_alert_type !~ /recovery/i)

              result = true
              @log.debug("Filter: Delays: blocking because time since last alert for " +
                         "current problem (#{time_since_last_alert}) is less than " +
                         "resend_delay (#{resend_delay}) and last alert type (#{last_alert_type}) was not a recovery")
            else
              @log.debug("Filter: Delays: not blocking because neither of the time comparison " +
                         "conditions were met")
            end
          else
            @log.debug("Filter: Delays: entity_check.failed? returned false ...")
          end
        end

        @log.debug("Filter: Delays: #{result ? "block" : "pass"}")
        result
      end
    end
  end
end