--
-- (C) 2013-20 - ntop.org
--

dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_utils"
local ts_utils = require("ts_utils")
local info = ntop.getInfo() 
local page_utils = require("page_utils")
local alerts_api = require("alerts_api")
local format_utils = require("format_utils")

local plugins_utils = require "plugins_utils"
local notification_endpoint_configs = plugins_utils.loadModule("notification_endpoints", "notification_endpoint_configs")
local notification_endpoint_recipients = plugins_utils.loadModule("notification_endpoints", "notification_endpoint_recipients")

sendHTTPContentTypeHeader('text/html')

page_utils.print_header(i18n("about.about_x", { product=info.product }))

if not isAdministrator() then
  return
end

dofile(dirs.installdir .. "/scripts/lua/inc/menu.lua")

local res

---------------------------
-- TEST ENDPOINT CONFIGS --
---------------------------

res = notification_endpoint_configs.reset_endpoint_configs()
assert(res["status"] == "OK")

res = notification_endpoint_configs.delete_endpoint_config("nonexisting_config_name")
assert(res["status"] == "failed" and res["error"]["type"] == "endpoint_config_not_existing")

res = notification_endpoint_configs.add_endpoint_config("nonexisting", nil, nil)
assert(res["status"] == "failed" and res["error"]["type"] == "endpoint_not_existing")

res = notification_endpoint_configs.add_endpoint_config("email", nil, nil)
assert(res["status"] == "failed" and res["error"]["type"] == "invalid_endpoint_conf_name")

res = notification_endpoint_configs.add_endpoint_config("email", "ntop_email", nil)

assert(res["status"] == "failed" and res["error"]["type"] == "invalid_conf_params")

-- see email.lua for mandatory conf_params
local conf_params = {
   smtp_server_name = "mail.ntop.org"
}

res = notification_endpoint_configs.add_endpoint_config("email", "ntop_email", conf_params)
assert(res["status"] == "failed" and res["error"]["type"] == "missing_mandatory_param")

conf_params = {
   smtp_server_name = "mail.ntop.org",
   sender = "tester@ntop.org"
}
res = notification_endpoint_configs.add_endpoint_config("email", "ntop_email", conf_params)
assert(res["status"] == "OK")

-- Duplicate addition
res = notification_endpoint_configs.add_endpoint_config("email", "ntop_email", conf_params)
assert(res["status"] == "failed" and res["error"]["type"] == "endpoint_config_already_existing")

-- Delete the endpoint
res = notification_endpoint_configs.delete_endpoint_config("ntop_email")
assert(res["status"] == "OK")

-- Add also some optional params
conf_params = {
   smtp_server_name = "mail.ntop.org",
   sender = "tester@ntop.org",
   username = "ntopuser",
   password = "ntoppassword"
}

res = notification_endpoint_configs.add_endpoint_config("email", "ntop_email", conf_params)
assert(res["status"] == "OK")

res = notification_endpoint_configs.delete_endpoint_config("ntop_email")
assert(res["status"] == "OK")

-- Add some garbage and make sure it is not written
conf_params["garbage"] = "trash"

res = notification_endpoint_configs.add_endpoint_config("email", "ntop_email", conf_params)
assert(res["status"] == "OK")

res = notification_endpoint_configs.get_endpoint_config("ntop_email")
assert(res["status"] == "OK")
assert(res["endpoint_key"] == "email")
assert(res["endpoint_conf_name"] == "ntop_email")
assert(res["endpoint_conf"])
assert(not res["endpoint_conf"]["garbage"])

for k, v in pairs(res["endpoint_conf"]) do
   assert(conf_params[k])
   assert(conf_params[k] == v)
end

-- Edit the config
conf_params["smtp_server_name"] = "mail2.ntop.org"
res = notification_endpoint_configs.edit_endpoint_config_params("ntop_email", conf_params)
assert(res["status"] == "OK")

res = notification_endpoint_configs.get_endpoint_config("ntop_email")
assert(res["status"] == "OK")
assert(res["endpoint_key"] == "email")
assert(res["endpoint_conf_name"] == "ntop_email")
assert(res["endpoint_conf"])
assert(res["endpoint_conf"]["smtp_server_name"] == "mail2.ntop.org")

-- Add another endpoint
conf_params = {
   smtp_server_name = "mail.google.com",
   sender = "tester@google.com",
   username = "googleuser",
   password = "googlepassword"
}

res = notification_endpoint_configs.add_endpoint_config("email", "google_email", conf_params)
assert(res["status"] == "OK")

-- Get all configs
res = notification_endpoint_configs.get_endpoint_configs()
assert(#res == 2)

------------------------------
-- TEST ENDPOINT RECIPIENTS --
------------------------------

res = notification_endpoint_recipients.add_endpoint_recipient("nonexisting_config_name", nil, nil)
assert(res["status"] == "failed" and res["error"]["type"] == "endpoint_config_not_existing")

res = notification_endpoint_recipients.add_endpoint_recipient("ntop_email", nil, nil)
assert(res["status"] == "failed" and res["error"]["type"] == "invalid_endpoint_recipient_name")

res = notification_endpoint_recipients.add_endpoint_recipient("ntop_email", "sysadmins", nil)
assert(res["status"] == "failed" and res["error"]["type"] == "invalid_recipient_params")

res = notification_endpoint_recipients.add_endpoint_recipient("ntop_email", "sysadmins", {})
assert(res["status"] == "failed" and res["error"]["type"] == "missing_mandatory_param")

local recipient_params = {
   to = "ci@ntop.org"
}

res = notification_endpoint_recipients.add_endpoint_recipient("ntop_email", "sysadmins", recipient_params)
assert(res["status"] == "OK")

res = notification_endpoint_recipients.add_endpoint_recipient("ntop_email", "sysadmins", recipient_params)
assert(res["status"] == "failed" and res["error"]["type"] == "endpoint_recipient_already_existing")

recipient_params["garbage"] = "trash"

dofile(dirs.installdir .. "/scripts/lua/inc/footer.lua")