ComfyFactorio/utils/spam_protection.lua

local Event = require 'utils.event'
local Global = require 'utils.global'
local Public = {}

local this = {
    prevent_spam = {}, -- the default table where all player indexes will be stored
    default_tick = 10, -- this defines the default tick to check whether or not a user is spamming a button.
    debug_text = false,
    debug_spam = false
}

local main_text = '[Spam Info] '

Global.register(
    this,
    function (t)
        this = t
    end
)

local function debug_text(str)
    if not this.debug_text then
        return
    end
    log(main_text .. str)
end

local function debug_spam(str)
    if not this.debug_spam then
        return
    end
    log(main_text .. str)
end

function Public.reset_spam_table()
    local players = game.connected_players
    this.prevent_spam = {}
    for i = 1, #players do
        local player = players[i]
        this.prevent_spam[player.index] = game.tick
    end
end

function Public.set_new_value(player)
    if this.prevent_spam[player.index] then
        this.prevent_spam[player.index] = game.tick
    end
end

function Public.is_spamming(player, value_to_compare, text)
    if not player or not player.valid then
        player = game.get_player(player)
    end

    if not player or not player.valid then return end

    if not this.prevent_spam[player.index] then
        return false
    end

    if text then
        debug_text('Frame: ' .. text)
    end

    if game.tick_paused then
        return false -- game is paused - shoo
    end

    if this.debug_spam then
        log(serpent.block(debug.traceback()))
    end

    local tick = game.tick
    local value = value_to_compare or this.default_tick
    if this.prevent_spam[player.index] then
        if (tick - this.prevent_spam[player.index]) > value then
            Public.set_new_value(player)
            return false -- is not spamming
        else
            if text then
                debug_spam(player.name .. ' is spamming: ' .. text)
            else
                debug_spam(player.name .. ' is spamming.')
            end
            return true -- is spamming
        end
    end
    return false
end

function Public.get(key)
    if key then
        return this[key]
    else
        return this
    end
end

function Public.set(key, value)
    if key and (value or value == false) then
        this[key] = value
        return this[key]
    elseif key then
        return this[key]
    else
        return this
    end
end

Event.add(
    defines.events.on_player_joined_game,
    function (event)
        local player = game.get_player(event.player_index)
        if not player then
            return
        end

        if not this.prevent_spam[player.index] then
            this.prevent_spam[player.index] = game.tick
        end
    end
)
Event.on_init(
    function ()
        Public.reset_spam_table()
    end
)

return Public