--
-- (C) 2014-21 - ntop.org
--

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

local json = require "dkjson"
local rest_utils = require "rest_utils"
local user_scripts = require "user_scripts"
local alert_utils = require "alert_utils"
local alert_exclusions = require "alert_exclusions"

local alert_rest_utils = {}

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

-- @brief exclude an alert using the parameters that the POST has
function _exclude_flow_alert(additional_filters, delete_alerts)
   local success = false

   local alert_key = tonumber(_POST["alert_key"])
   local alert_addr = _POST["alert_addr"]

   if alert_key and alert_addr then
      success = true
   end

   if success then
      if alert_addr then
	 alert_exclusions.disable_alert(alert_addr, alert_key)
	 if delete_alerts == "true" then
	    alert_utils.deleteFlowAlertsMatching(alert_addr, alert_key)
	 end
      end
   end

   if success then
      rc = rest_utils.consts.success.ok
      rest_utils.answer(rc)
   else
      rc = rest_utils.consts.err.invalid_args
      rest_utils.answer(rc)
   end
end

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

-- @brief exclude an alert using the parameters that the POST has
function alert_rest_utils.exclude_alert()
   -- POST parameters
   local additional_filters = _POST["filters"]
   local subdir = _POST["subdir"]
   local script_key = _POST["script_key"]
   local delete_alerts = _POST["delete_alerts"] or "false"
   
   -- Parameters used by the various functions
   local success = ""
   local new_filter  = {}
   local update_err = ""

   -- Parameters used for the rest answer
   local rc = ""
   local res = ""

   if subdir == "flow" then
      return _exclude_flow_alert(additional_filters, delete_alerts)
   end
   
   -- Checking that all parameters where given to the POST
   if not additional_filters or not subdir or not script_key then
      rest_utils.answer(rest_utils.consts.err.invalid_args)
      return
   end

   -- Getting the parameters
   success, new_filter = user_scripts.parseFilterParams(additional_filters, subdir, false)
      
   if success then
      success, update_err = user_scripts.updateScriptConfig(script_key, subdir, nil, nil, new_filter)
   else
      -- Error while parsing the params, error is printed
      update_err = new_filter
   end

   if success then
      if delete_alerts == "true" then
	 alert_utils.deleteAlertsMatchingUserScriptFilter(subdir, script_key, new_filter.new_filters[1])                                                                            
      end
      
      rc = rest_utils.consts.success.ok
      rest_utils.answer(rc)
   else
      rc = rest_utils.consts.err.invalid_args
      res = update_err
      rest_utils.answer(rc, res)
   end
end

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

return alert_rest_utils