--
-- (C) 2021-24 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/alert_store/?.lua;" .. package.path

require "ntop_utils"
local os_utils = require "os_utils"

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

local alert_store_utils = {}

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

-- @brief Returns an array of all available alert_store Lua class instances
function alert_store_utils.all_instances_factory()
    local alert_store_dir = os_utils.fixPath(dirs.installdir .. "/scripts/lua/modules/alert_store/")
    local res = {}

    for alert_store_file in pairs(ntop.readdir(alert_store_dir)) do
        -- Load all sub-classes of alert_store
        if alert_store_file:match("_alert_store%.lua$") then
            if not ntop.isPro() then
                if string.find(alert_store_file, "snmp") then
                    goto continue
                end
            end

            -- Note: skip special instances without a family (e.g. 'all')
            if alert_store_file:match("all_alert_store%.lua$") then
                goto continue
            end

            local file_info = string.split(alert_store_file, "%.")
            local instance_name = file_info[1]

            local instance = require(instance_name).new()

            if instance then
                local family_name = instance:get_family()

                if family_name then
                    res[family_name] = instance
                end
            end

            ::continue::
        end
    end

    return res
end

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

-- @brief Call instance:db_cleanup for every available alert_store instance
function alert_store_utils.housekeeping(ifid)
    -- NOTE: Clickhouse housekeeping is done daily in clickhouseDeleteOldPartitions
    if not ntop.isClickHouseEnabled() then
        local all_instances = alert_store_utils.all_instances_factory()
        for _, instance in pairs(all_instances) do
            instance:housekeeping(ifid)
        end

        -- Reclaims unused disk space and defragments tables and indices.
        -- Should be called as disk space and defragmentation are not run
        -- automatically by sqlite.
        local q = string.format("VACUUM")
        local vacuum = interface.alert_store_query(q)
    end
end

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

return alert_store_utils
