ntopng/scripts/lua/modules/pools/pools_lua_utils.lua

62 lines
2.1 KiB
Lua

--
-- (C) 2017-22 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/pools/?.lua;" .. package.path
require "lua_utils"
local json = require "dkjson"
local os_utils = require "os_utils"
-- ##############################################
local pools_lua_utils = {}
-- ##############################################
-- @brief Returns an array of pool Lua class instances, for all available pools
-- e.g., {interface_pools:create(), local_network_pools:create(), snmp_device_pools:create(), ...}
function pools_lua_utils.all_pool_instances_factory()
local pools_dir = os_utils.fixPath(dirs.installdir .. "/scripts/lua/modules/pools/")
local res = {}
for pool_file in pairs(ntop.readdir(pools_dir)) do
-- Load all sub-classes of pools.lua (and exclude pools.lua itself)
if pool_file:match("_pools%.lua$") and not pool_file:match("^pools.lua$") then
local pool_file_path = os_utils.fixPath(string.format("%s/%s", pools_dir, pool_file))
local pool = dofile(pool_file_path)
-- Make sure pool is actually a pool Lua class by checking if it has method create
-- this is to avoid instantiating modules such as pool_lua_utils.lua and pool_rest_utils.lua
-- which are not classes and thus cannot be instantiated
if pool.create then
-- If it has a method create, then we can instantiate it and add it to the result
local instance = pool:create()
res[#res + 1] = instance
end
end
end
return res
end
-- ##############################################
-- @brief Returns the pool url for a pool with a given `pool_key`
-- @param pool_key A pool key string as found inside `create` method of any pools instance.
-- If no pool key is found, the home of the pool url is returned.
-- @return The pool url
function pools_lua_utils.get_pool_url(pool_key)
local pool_url = ntop.getHttpPrefix().."/lua/admin/manage_pools.lua"
if pool_key then
pool_url = pool_url.."?page="..pool_key
end
return pool_url
end
-- ##############################################
return pools_lua_utils