factorio-scenario-ExpCluster/exp_legacy/module/modules/commands/repair.lua

--[[-- Commands Module - Repair
    - Adds a command that allows an admin to repair and revive a large area
    @commands Repair
]]

local Commands = require("modules.exp_legacy.expcore.commands") --- @dep expcore.commands
local config = require("modules.exp_legacy.config.repair") --- @dep config.repair
require("modules.exp_legacy.config.expcore.command_general_parse")

local max_time_to_live = 4294967295 -- unit32 max
--- Repairs entities on your force around you
-- @command repair
-- @tparam number range the range to repair stuff in, there is a max limit to this
Commands.new_command("repair", { "expcom-repair.description" }, "Repairs entities on your force around you")
    :add_param("range", false, "integer-range", 1, config.max_range)
    :register(function(player, range)
        local revive_count = 0
        local heal_count = 0
        local range2 = range ^ 2
        -- Intentionally left as player.surface to allow use in remote view
        local surface = player.surface
        -- Intentionally left as player.position to allow use in remote view
        local center = player.position
        local area = { { x = center.x - range, y = center.y - range }, { x = center.x + range, y = center.y + range } }
        if config.allow_ghost_revive then
            local ghosts = surface.find_entities_filtered{ area = area, type = "entity-ghost", force = player.force }
            for _, ghost in pairs(ghosts) do
                if ghost.valid then
                    local x = ghost.position.x - center.x
                    local y = ghost.position.y - center.y
                    if x ^ 2 + y ^ 2 <= range2 then
                        if config.allow_blueprint_repair or ghost.time_to_live ~= max_time_to_live then
                            revive_count = revive_count + 1
                            if not config.disallow[ghost.ghost_name] then ghost.revive() end
                        end
                    end
                end
            end
        end
        if config.allow_heal_entities then
            local entities = surface.find_entities_filtered{ area = area, force = player.force }
            for _, entity in pairs(entities) do
                if entity.valid then
                    local x = entity.position.x - center.x
                    local y = entity.position.y - center.y
                    if entity.health and entity.get_health_ratio() ~= 1 and x ^ 2 + y ^ 2 <= range2 then
                        heal_count = heal_count + 1
                        entity.health = max_time_to_live
                    end
                end
            end
        end
        return Commands.success{ "expcom-repair.result", revive_count, heal_count }
    end)