ComfyFactorio/utils/token.lua

local Token = {}

local tokens = {}

local counter = 200
local alternative_counter = 20000

--- Assigns a unquie id for the given var.
-- This function cannot be called after on_init() or on_load() has run as that is a desync risk.
-- Typically this is used to register functions, so the id can be stored in the global table
-- instead of the function. This is becasue closures cannot be safely stored in the global table.
-- @param  var<any>
-- @param  sarg<boolean>
-- @return number the unique token for the variable.
function Token.register(var, sarg)
    if _LIFECYCLE == 8 then
        error('Calling Token.register after on_init() or on_load() has run is a desync risk.', 2)
    end

    if sarg then
        alternative_counter = alternative_counter + 1
        tokens[alternative_counter] = var
        return alternative_counter
    end

    counter = counter + 1

    tokens[counter] = var

    return counter
end

function Token.get(token_id)
    return tokens[token_id]
end

local uid_counter = 100

---@return integer
function Token.uid()
    uid_counter = uid_counter + 1

    return uid_counter
end

return Token