z_mirror/Krastorio2

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

Caleb HeuerChange to four-space indentation07409ca62026-01-05

local flib_gui = require("__flib__.gui")
local flib_math = require("__flib__.math")
local flib_table = require("__flib__.table")

local util = require("scripts.util")

-- the full energy buffer on normal quality
local energy_per_teleport =
    prototypes.entity["kr-planetary-teleporter"].electric_energy_source_prototype.buffer_capacity

--- @class PlanetaryTeleporterGui
--- @field elems table<string, LuaGuiElement>
--- @field entity LuaEntity
--- @field data PlanetaryTeleporterData
--- @field player LuaPlayer
--- @field search_open boolean
--- @field search_query string

--- @param player LuaPlayer
--- @param from LuaEntity
--- @param to LuaEntity
local function teleport_player(player, from, to)
    from.energy = from.energy - energy_per_teleport
    to.energy = to.energy - energy_per_teleport
    local position = to.position
    position.y = position.y + 1.1
    player.teleport(position, to.surface)
    from.surface.play_sound({
        path = "kr-planetary-teleporter-effect-sound",
        volume_modifier = 0.8,
        position = from.position,
    })
    to.surface.play_sound({
        path = "kr-planetary-teleporter-effect-sound",
        position = to.position,
        volume_modifier = 1,
    })
end

--- @param self PlanetaryTeleporterGui
local function filter_destinations(self)
    local search_query = self.search_query
    for _, frame in pairs(self.elems.destinations_table.children) do
        if search_query == "" then
            frame.visible = true
            goto continue
        end
        local unit_number = tonumber(frame.name) --[[@as uint]]
        local data = storage.planetary_teleporter[unit_number]
        if not data then
            frame.destroy()
            goto continue
        end
        if data.name and string.find(string.lower(data.name), search_query, nil, true) then
            frame.visible = true
        else
            frame.visible = false
        end
        ::continue::
    end
end

--- @param self PlanetaryTeleporterGui
--- @param by_click boolean
local function toggle_search(self, by_click)
    local search_button = self.elems.search_button
    if not by_click then
        search_button.toggled = not search_button.toggled
    end
    self.search_open = search_button.toggled

    local search_textfield = self.elems.search_textfield
    search_textfield.visible = self.search_open
    if self.search_open then
        search_textfield.focus()
        search_textfield.select_all()
    else
        self.search_query = ""
    end

    filter_destinations(self)
end

--- @param self PlanetaryTeleporterGui
local function destroy_gui(self)
    local window = self.elems.kr_planetary_teleporter_window
    if window.valid then
        window.destroy()
        if self.player.opened == window then
            self.player.opened = nil
        end
    end
    storage.planetary_teleporter_gui[self.player.index] = nil
end

--- @param self PlanetaryTeleporterGui
local function update_name_label(self)
    self.elems.name_label.caption = self.data.name or { "gui.kr-planetary-teleporter-unnamed" }
end

--- @param e EventData.on_gui_closed|EventData.on_gui_click
local function on_window_closed(e)
    local self = storage.planetary_teleporter_gui[e.player_index]
    if not self then
        return
    end
    if self.search_open then
        toggle_search(self, false)
        self.player.opened = self.elems.kr_planetary_teleporter_window
        return
    end
    destroy_gui(self)
end

--- @param e EventData.on_gui_text_changed
local function on_search_textfield_text_changed(e)
    local self = storage.planetary_teleporter_gui[e.player_index]
    if not self then
        return
    end
    self.search_query = e.element.text
    filter_destinations(self)
end

--- @param e EventData.on_gui_click|EventData.CustomInputEvent
local function on_search_button_clicked(e)
    local self = storage.planetary_teleporter_gui[e.player_index]
    if not self then
        return
    end
    toggle_search(self, true)
end

--- @param e EventData.on_gui_click
local function on_rename_button_clicked(e)
    local self = storage.planetary_teleporter_gui[e.player_index]
    if not self or not self.elems.kr_planetary_teleporter_window.valid then
        return
    end
    local textfield = self.elems.name_textfield
    local label = self.elems.name_label
    if textfield.visible then
        local text = textfield.text
        if #text > 0 then
            self.data.name = text
        else
            self.data.name = nil
        end
        update_name_label(self)
        label.visible = true
        textfield.visible = false
    else
        label.visible = false
        textfield.text = self.data.name or ""
        textfield.visible = true
        textfield.select_all()
        textfield.focus()
    end
end

--- @param e EventData.on_gui_confirmed
local function on_name_textfield_confirmed(e)
    -- Just pass through the logic because it's all the same anyway.
    on_rename_button_clicked(e --[[@as EventData.on_gui_click]])
end

--- @param e EventData.on_gui_click
local function on_teleport_button_clicked(e)
    local self = storage.planetary_teleporter_gui[e.player_index]
    if not self or not self.elems.kr_planetary_teleporter_window.valid then
        return
    end
    local target_unit_number = e.element.tags.unit_number --[[@as UnitNumber]]
    assert(target_unit_number, "target_unit_number is nil")
    local target_data = storage.planetary_teleporter[target_unit_number]
    assert(target_data, "target_data is nil")
    local players = storage.planetary_teleporter_players[
        target_data.entities.turret.unit_number --[[@as uint]]
    ]
    if players then
        util.flying_text_with_sound(self.player, { "gui.kr-planetary-teleporter-destination-blocked" })
        return
    end
    local source_unit_number = self.entity.unit_number
    assert(source_unit_number, "source_unit_number is nil")
    local source_data = storage.planetary_teleporter[source_unit_number]
    teleport_player(self.player, source_data.entities.base, target_data.entities.base)
    destroy_gui(self)
end

--- @type table<defines.entity_status, { sprite: SpritePath, label: LocalisedString }>
local status_sprites = {
    [defines.entity_status.charging] = { label = { "entity-status.charging" }, sprite = "utility/status_yellow" },
    [defines.entity_status.fully_charged] = {
        label = { "entity-status.fully-charged" },
        sprite = "utility/status_working",
    },
    [defines.entity_status.no_power] = { label = { "entity-status.no-power" }, sprite = "utility/status_not_working" },
    [defines.entity_status.normal] = { label = { "entity-status.normal" }, sprite = "utility/status_working" },
    [defines.entity_status.marked_for_deconstruction] = {
        label = { "entity-status.marked-for-deconstruction" },
        sprite = "utility/status_not_working",
    },
}

--- @param self PlanetaryTeleporterGui
local function update_gui(self)
    if not self.entity.valid then
        destroy_gui(self)
        return
    end
    local buffer_capacity = self.entity.prototype.electric_energy_source_prototype.buffer_capacity
    local status = self.entity.status
    local status_properties = status_sprites[status] or {}
    self.elems.status_icon.sprite = status_properties.sprite
    self.elems.status_label.caption = status_properties.label

    local charge = self.entity.energy / buffer_capacity
    self.elems.charge_progressbar.value = charge
    self.elems.charge_progressbar.caption = { "format-percent", flib_math.round(charge * 100) }

    local players = storage.planetary_teleporter_players[self.data.entities.turret.unit_number]
    local standing_on_teleporter = players and flib_table.find(players, self.player.index) or false
    local too_many_players = players and #players > 1 or false
    local source_fully_charged = flib_math.round(charge, 0.1) >= 1
    local fully_operational = standing_on_teleporter and source_fully_charged and not too_many_players

    if fully_operational then
        self.elems.destinations_status_icon.sprite = "flib_indicator_green"
        self.elems.destinations_status_label.caption = { "gui.kr-planetary-teleporter-ready" }
    else
        self.elems.destinations_status_icon.sprite = "flib_indicator_red"
        if not standing_on_teleporter then
            self.elems.destinations_status_label.caption = { "gui.kr-planetary-teleporter-not-on-teleporter" }
        elseif too_many_players then
            self.elems.destinations_status_label.caption = { "gui.kr-planetary-teleporter-too-many-players" }
        else
            self.elems.destinations_status_label.caption = { "gui.kr-planetary-teleporter-not-fully-charged" }
        end
    end

    update_name_label(self)

    for _, frame in pairs(self.elems.destinations_table.children) do
        local unit_number = tonumber(frame.name) --[[@as uint]]
        local dest_data = storage.planetary_teleporter[unit_number]
        if not dest_data then
            goto continue
        end
        local dest_entity = dest_data.entities.base
        local dest_charge = dest_entity.energy / buffer_capacity
        local dest_fully_charged = flib_math.round(dest_charge, 0.1) >= 1
        local minimap = frame.minimap_frame.minimap --[[@as LuaGuiElement]]
        minimap.progressbar.value = dest_charge
        minimap.progressbar.visible = not dest_fully_charged
        minimap.minimap_button.enabled = fully_operational and dest_fully_charged
        if not dest_fully_charged then
            minimap.minimap_button.tooltip = { "format-percent", flib_math.round(dest_charge * 100) }
        elseif fully_operational then
            minimap.minimap_button.tooltip = { "gui.kr-planetary-teleporter-teleport" }
        else
            minimap.minimap_button.tooltip = nil
        end
        ::continue::
    end
end

local function on_tick()
    for _, gui in pairs(storage.planetary_teleporter_gui) do
        update_gui(gui)
    end
end

--- @param e EventData.on_gui_opened
local function on_gui_opened(e)
    local entity = e.entity
    if not entity or not entity.valid or entity.name ~= "kr-planetary-teleporter" then
        return
    end

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

    local destination_buttons = {}
    for unit_number, data in pairs(storage.planetary_teleporter) do
        if unit_number == entity.unit_number or data.force ~= entity.force or data.surface ~= entity.surface then
            goto continue
        end
        destination_buttons[#destination_buttons + 1] = {
            type = "frame",
            name = unit_number,
            style = "train_with_minimap_frame",
            direction = "vertical",
            {
                type = "frame",
                name = "minimap_frame",
                style = "deep_frame_in_shallow_frame",
                {
                    type = "minimap",
                    name = "minimap",
                    style = "kr_planetary_teleporter_destination_minimap",
                    position = data.entities.base.position,
                    {
                        type = "button",
                        name = "minimap_button",
                        style = "kr_planetary_teleporter_destination_minimap_button",
                        tags = { unit_number = unit_number },
                        handler = { [defines.events.on_gui_click] = on_teleport_button_clicked },
                    },
                    {
                        type = "progressbar",
                        name = "progressbar",
                        style = "kr_planetary_teleporter_destination_charge_bar",
                        ignored_by_interaction = true,
                    },
                    {
                        type = "label",
                        style = "kr_planetary_teleporter_destination_label",
                        style_mods = { top_margin = 8, right_margin = 8 },
                        caption = data.name or { "gui.kr-planetary-teleporter-unnamed" },
                        ignored_by_interaction = true,
                    },
                },
            },
        }
        ::continue::
    end

    local elems = flib_gui.add(player.gui.screen, {
        type = "frame",
        name = "kr_planetary_teleporter_window",
        direction = "vertical",
        elem_mods = { auto_center = true },
        handler = { [defines.events.on_gui_closed] = on_window_closed },
        {
            type = "flow",
            style = "flib_titlebar_flow",
            drag_target = "kr_planetary_teleporter_window",
            {
                type = "label",
                style = "flib_frame_title",
                caption = { "entity-name.kr-planetary-teleporter" },
                ignored_by_interaction = true,
            },
            { type = "empty-widget", style = "flib_titlebar_drag_handle", ignored_by_interaction = true },
            {
                type = "textfield",
                name = "search_textfield",
                style = "flib_titlebar_search_textfield",
                visible = false,
                handler = { [defines.events.on_gui_text_changed] = on_search_textfield_text_changed },
            },
            {
                type = "sprite-button",
                name = "search_button",
                style = "frame_action_button",
                sprite = "utility/search",
                tooltip = { "gui.flib-search-instruction" },
                auto_toggle = true,
                handler = { [defines.events.on_gui_click] = on_search_button_clicked },
            },
            {
                type = "sprite-button",
                style = "frame_action_button",
                sprite = "utility/close",
                tooltip = { "gui.close-instruction" },
                handler = { [defines.events.on_gui_click] = on_window_closed },
            },
        },
        {
            type = "flow",
            style_mods = { horizontal_spacing = 12 },
            {
                type = "frame",
                style = "inside_shallow_frame",
                direction = "vertical",
                {
                    type = "frame",
                    style = "subheader_frame",
                    style_mods = { width = 424 },
                    {
                        type = "label",
                        name = "name_label",
                        style = "subheader_caption_label",
                        style_mods = { horizontally_squashable = true },
                    },
                    {
                        type = "textfield",
                        name = "name_textfield",
                        visible = false,
                        handler = { [defines.events.on_gui_confirmed] = on_name_textfield_confirmed },
                    },
                    {
                        type = "sprite-button",
                        style = "mini_button_aligned_to_text_vertically_when_centered",
                        sprite = "utility/rename_icon",
                        handler = { [defines.events.on_gui_click] = on_rename_button_clicked },
                    },
                },
                {
                    type = "flow",
                    style_mods = { padding = 12, top_padding = 8, vertical_spacing = 8 },
                    direction = "vertical",
                    {
                        type = "flow",
                        style_mods = { vertical_align = "center" },
                        { type = "sprite", name = "status_icon", style = "flib_indicator" },
                        { type = "label", name = "status_label" },
                    },
                    {
                        type = "frame",
                        style = "deep_frame_in_shallow_frame",
                        { type = "entity-preview", style = "wide_entity_button", elem_mods = { entity = entity } },
                    },
                    {
                        type = "progressbar",
                        name = "charge_progressbar",
                        style = "production_progressbar",
                        style_mods = { horizontally_stretchable = true },
                        caption = "foo",
                        value = 0.2,
                    },
                    { type = "empty-widget", style = "entity_frame_filler" },
                },
            },
            {
                type = "frame",
                style = "inside_deep_frame",
                direction = "vertical",
                {
                    type = "frame",
                    style = "subheader_frame",
                    {
                        type = "label",
                        style = "subheader_caption_label",
                        caption = { "gui.kr-planetary-teleporter-destinations" },
                    },
                    { type = "empty-widget", style = "flib_horizontal_pusher" },
                    {
                        type = "flow",
                        name = "destinations_status_flow",
                        style_mods = { vertical_align = "center", right_margin = 8 },
                        {
                            type = "sprite",
                            name = "destinations_status_icon",
                            style = "flib_indicator",
                            sprite = "flib_indicator_red",
                        },
                        {
                            type = "label",
                            name = "destinations_status_label",
                            caption = { "gui.kr-planetary-teleporter-not-on-teleporter" },
                        },
                    },
                },
                {
                    type = "scroll-pane",
                    style = "kr_planetary_teleporter_destinations_scroll_pane",
                    vertical_scroll_policy = "auto-and-reserve-space",
                    {
                        type = "table",
                        name = "destinations_table",
                        style = "slot_table",
                        column_count = 3,
                        children = destination_buttons,
                    },
                },
            },
        },
    })

    player.opened = elems.kr_planetary_teleporter_window

    --- @type PlanetaryTeleporterGui
    storage.planetary_teleporter_gui[e.player_index] = {
        elems = elems,
        data = storage.planetary_teleporter[entity.unit_number],
        entity = entity,
        player = player,
        search_open = false,
        search_query = "",
    }
end

--- @param e EventData.CustomInputEvent
local function on_focus_search(e)
    local self = storage.planetary_teleporter_gui[e.player_index]
    if self then
        toggle_search(self, false)
    end
end

--- @class PlanetaryTeleporterGuiHandler : event_handler
local planetary_teleporter_gui = {}

function planetary_teleporter_gui.on_init()
    --- @type table<uint, PlanetaryTeleporterGui>
    storage.planetary_teleporter_gui = {}
end

function planetary_teleporter_gui.on_configuration_changed()
    for _, gui in pairs(storage.planetary_teleporter_gui or {}) do
        if gui.elems.window.valid then
            gui.elems.window.destroy()
        end
    end
    storage.planetary_teleporter_gui = {}
end

planetary_teleporter_gui.events = {
    [defines.events.on_gui_opened] = on_gui_opened,
    [defines.events.on_tick] = on_tick,
    ["kr-linked-focus-search"] = on_focus_search,
}

flib_gui.add_handlers({
    planetary_teleporter_on_name_textfield_confirmed = on_name_textfield_confirmed,
    planetary_teleporter_on_rename_button_clicked = on_rename_button_clicked,
    planetary_teleporter_on_search_button_clicked = on_search_button_clicked,
    planetary_teleporter_on_search_textfield_text_changed = on_search_textfield_text_changed,
    planetary_teleporter_on_teleport_button_clicked = on_teleport_button_clicked,
    planetary_teleporter_on_window_closed = on_window_closed,
})

return planetary_teleporter_gui