z_mirror/Krastorio2

git clone https://git.raiguard.me/z_mirror/Krastorio2

Caleb HeuerChange to four-space indentation07409ca62026-01-05

local util = require("scripts.util")

--- @param e EventData.on_player_placed_equipment
local function on_player_placed_equipment(e)
    local equipment = e.equipment
    if not equipment.valid or equipment.name ~= "kr-energy-absorber-equipment" then
        return
    end
    local grid = e.grid
    if not grid.valid then
        return
    end

    local player = game.get_player(e.player_index)
    if not player then
        return
    end

    local found = 0
    for _, equipment_spec in pairs(grid.get_contents()) do
        if equipment_spec.name == "kr-energy-absorber-equipment" then
            found = equipment_spec.count
            break
        end
    end
    if found <= 1 then
        return
    end

    -- Retrieve the equipment
    local equipment_quality = equipment.quality
    grid.take({ equipment = equipment, by_player = player })

    -- Return the item
    local cursor_stack = player.cursor_stack
    -- If we're holding another absorber
    if cursor_stack and cursor_stack.valid_for_read and cursor_stack.name == "kr-energy-absorber-equipment" then
        cursor_stack.count = cursor_stack.count + 1 --[[@as uint]]
    -- If we're holding something else or the stack is empty
    elseif cursor_stack and player.clear_cursor() then
        cursor_stack.set_stack({ name = "kr-energy-absorber-equipment", count = 1, quality = equipment_quality })
    end

    -- Show the error
    if (storage.error_message_tick[e.player_index] or 0) + 30 < game.ticks_played then
        util.flying_text_with_sound(player, { "message.kr-already-one-energy-absorber-equipment" })
    end
    storage.error_message_tick[e.player_index] = game.ticks_played
end

--- @class EnergyAbsorberHandler : event_handler
local energy_absorber = {}

function energy_absorber.on_init()
    --- Holds the last tick that an energy absorber error message was displayed for each player
    --- @type table<number, number>
    storage.error_message_tick = {}
end

energy_absorber.events = {
    [defines.events.on_player_placed_equipment] = on_player_placed_equipment,
}

return energy_absorber