vrr/ntopng
2
0
Fork 0
mirror of https://github.com/ntop/ntopng.git synced 2026-04-29 15:39:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Refactors user_scripts into checks (lua)

Browse source
This commit is contained in:
Simone Mainardi 2021-06-16 18:02:22 +02:00
parent 3c3aa5a25f
commit 76fd315d1b
222 changed files with 980 additions and 981 deletions
Download patch file Download diff file Expand all files Collapse all files

126
scripts/lua/get_user_script_config.lua
View file

@ -1,126 +0,0 @@
--
-- (C) 2019-21 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_utils"
local json = require("dkjson")
local user_scripts = require("user_scripts")
local alert_consts = require("alert_consts")
local alert_exclusions = require "alert_exclusions"
sendHTTPContentTypeHeader('application/json')
local subdir = _GET["script_subdir"] or "host"
local factory = _GET["factory"]
local script_key = _GET["script_key"]
local script_type = user_scripts.getScriptType(subdir)
if(script_type == nil) then
traceError(TRACE_ERROR, TRACE_CONSOLE, "Bad subdir: " .. subdir)
return
end
if(script_key == nil) then
traceError(TRACE_ERROR, TRACE_CONSOLE, "Missing script_key parameter")
return
end
local config_set
if factory and factory == "true" then
config_set = user_scripts.getFactoryConfig()
else
config_set = user_scripts.getConfigset()
end
-- ################################################
local script = user_scripts.loadModule(getSystemInterfaceId(), script_type, subdir, script_key)
if(script == nil) then
traceError(TRACE_ERROR, TRACE_CONSOLE, "Unknown user script: " .. script_key)
return
end
local result = {
hooks = {},
gui = {},
metadata = {},
filters = {}
}
if(script.gui) then
local known_fields = {i18n_title=1, i18n_description=1, i18n_field_unit=1, input_builder=1}
for field, val in pairs(script.gui) do
if not known_fields[field] then
result.gui[field] = val
end
end
if(script.gui.i18n_field_unit) then
result.gui.fields_unit = i18n(script.gui.i18n_field_unit)
end
result.gui.input_builder = script.gui.input_builder
end
if (script.default_value) then
result.metadata.default_value = script.default_value
end
if (script.is_alert) then
result.metadata.is_alert = script.is_alert
end
-- Getting filter configurations
if script.alert_id and (subdir == "flow" or subdir == "host") then
result.filters = { current_filters = {}}
local current_filters
if subdir == "flow" then
current_filters = alert_exclusions.flow_alerts_get_excluded_hosts(script.alert_id)
else
current_filters = alert_exclusions.host_alerts_get_excluded_hosts(script.alert_id)
end
for current_ip, _ in pairs(current_filters or {}) do
result.filters.current_filters[#result.filters.current_filters + 1] = {ip = current_ip}
end
else
local filter_conf = config_set["filters"]
if filter_conf and filter_conf[subdir] and filter_conf[subdir][script_key] and filter_conf[subdir][script_key]["filter"] then
result.filters = filter_conf[subdir][script_key]["filter"]
end
if not result.filters or table.len(result.filters) == 0 then
-- No configuration found, trying to check if there is a default filter configured
result.filters = user_scripts.getDefaultFilters(interface.getId(), subdir, script_key)
end
end
-------------------------------
local hooks_config = user_scripts.getScriptConfig(config_set, script, subdir)
-- script.template:render(hooks_config)
for hook, config in pairs(hooks_config) do
local granularity_info = alert_consts.alerts_granularities[hook]
local label = nil
if(granularity_info) then
label = i18n(granularity_info.i18n_title)
end
result.hooks[hook] = table.merge(config, {
label = label or "",
})
end
-- ################################################
print(json.encode(result))