require 'singleton'
require 'json'
require 'date'

# Defining state constants
module State
  UP=1
  DOWN=2
  WARNING=3
  NA=0
end

module HelperMethods
  # Checks if all hosts are up
  def all_hosts_up?(hosts)
    hosts.each do |host|
      return false unless
          status_data['hosts'].has_key?(host) and
              status_data['hosts'][host]['current_state'] == 0
    end

    true
  end

  def host_up?(host)
    status_data['hosts'].has_key?(host) and
        status_data['hosts'][host]['current_state'] == 0
  end

  def host_flapping?(host)
    status_data['hosts'].has_key?(host) and
        status_data['hosts'][host]['is_flapping'] != 0
  end

  # Checks if the service is up
  def service_up?(host, service)
    status_data['services'].has_key?(host) and
        status_data['services'][host].has_key?(service) and
        status_data['services'][host][service]['current_state'] == 0
  end

  def service_flapping?(host, service)
    status_data['services'].has_key?(host) and
        status_data['services'][host].has_key?(service) and
        status_data['services'][host][service]['is_flapping'] != 0
  end

  def has_service?(host, service)
    status_data['services'][host].has_key?(service)
  end

  def default(host, service = nil)
    if service == nil
      if host_flapping? host
        State::WARNING
      elsif host_up? host
        State::UP
      else
        State::DOWN
      end
    else
      return State::NA unless has_service? host, service

      if service_flapping? host, service
        State::WARNING
      elsif service_up? host, service
        State::UP
      else
        State::DOWN
      end
    end
  end
end

class ServiceRegistry
  CACHE_SECONDS = 600
  StatusSource = File.join(File.dirname(__FILE__), '..', 'data', 'status.json')

  include Singleton
  include HelperMethods
  attr_reader :load_date

  def initialize
    @cache_locked = false
    @next_name = nil
    @current_category = nil
  end

  def name(n)
    @next_name = n
  end

  def service(name, &block)
    @services[name] = {}
    @services[name][:name] = @next_name || name

    begin
      @services[name][:status] = block.call
    rescue Exception => e
      @services[name][:status] = State::NA
      $stderr.puts e
    end

    @next_name = nil
  end

  def services
    update?
    @services
  end

  def status_data
    update?
    @status_data
  end

  def columns
    update?
    @columns
  end

  def categories
    update?
    @categories
  end

  def category(name, &block)
    @categories[name] = {}
    @current_category = name
    @categories[name][:services] = block.call
    @columns[@categories[name][:column]] << name
  end

  def column(id)
    @categories[@current_category][:column] = id
  end

  def update!
    @cache_locked = true
    @services = {}
    @categories = {}
    @columns = {1 => [], 2 => [], 3 => []}
    @status_data = JSON.parse(File.read(StatusSource))
    load(File.join(File.dirname(__FILE__), '..', 'data', 'services.rb'))
    @load_date = DateTime.now
    @cache_locked = false
  rescue Exception => e
    $stderr.puts e
    @services = {}
    @categories = {}
    @columns = {1 => [], 2 => [], 3 => []}
    @load_date = DateTime.new(2000, 1, 1)
    @cache_locked = false
  end

  private
  def update?
    if not @load_date.nil? and not @cache_locked and ((DateTime.now - @load_date) * 60 * 60 * 24).to_i > CACHE_SECONDS
      update!
    end
  end
end

def Services(&block)
  ServiceRegistry.instance.instance_eval(&block)
end