--
-- (C) 2019 - ntop.org
--

local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_utils"
require "alert_utils"

local alerts_api = require("alerts_api")

local do_trace          = false
local config_alerts     = nil
local ifname            = nil
local available_modules = nil

-- #################################################################

-- The function below ia called once (#pragma once)
function setup(str_granularity)
   if(do_trace) then print("alert.lua:setup("..str_granularity..") called\n") end
   ifname = interface.setActiveInterfaceId(tonumber(interface.getId()))
   config_alerts = getHostsConfiguredAlertThresholds(ifname, str_granularity)

   -- Load the threshold checking functions
   available_modules = alerts_api.load_check_modules("host")
end

-- #################################################################

-- The function below is called once per host
function checkHostAlerts(granularity)
  local info = host.getFullInfo()
  local host_key   = info.ip.."@"..info.vlan
  local host_config = config_alerts[host_key] or {}
  local global_config = config_alerts["local_hosts"] or {}
  local has_configured_alerts = (table.len(host_config or global_confi) > 0)

  if has_configured_alerts then
    for _, check in pairs(available_modules) do
      local config = host_config[check.key] or global_config[check.key]

      if config then
        check.check_function({
          granularity = granularity,
          alert_entity = alerts_api.hostAlertEntity(host_key),
          entity_info = info,
          alert_config = config,
          check_module = check,
        })
      end
    end
  end
end