Gurpartap/cognizant

require "cognizant/process/conditions"
require "cognizant/util/rotational_array"

module Cognizant
  class Process
    class ConditionDelegate
      class HistoryValue < Struct.new(:value, :critical); end

      # No need to recreate one every tick.
      EMPTY_ARRAY = [].freeze

      attr_accessor :name

      def initialize(name, options = {}, &block)
        @name = name
        @every = options.delete(:every).to_i

        @times = options.delete(:times) || 1
        @times = [@times, @times] unless @times.is_a?(Array) # handles :times => 5
        @times.map(&:to_i)

        @do = options.has_key?(:do) ? [options.delete(:do)] : [:restart]
        @do = [block] if block

        clear_history!

        @condition = Cognizant::Process::Conditions[@name].new(options)
      end

      def run(pid, tick_number = Time.now.to_i)
        if @last_ran_at.nil? || (@last_ran_at + @every) <= tick_number
          @last_ran_at = tick_number

          value = @condition.run(pid)
          @history << HistoryValue.new(@condition.format_value(value), @condition.check(value))
          # puts self.to_s

          return @do if failed_check?
        end
        EMPTY_ARRAY
      end

      def clear_history!
        @history = Util::RotationalArray.new(@times.last)
      end

      def failed_check?
        @history.count { |v| v and v.critical } >= @times.first
      end

      def to_s
        data = @history.collect { |v| v and "#{v.value}#{'*' unless v.critical}" }.join(", ")
        "#{@name}: [#{data}]\n"
      end
    end
  end
end