z_mirror/Krastorio2

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

Polka37Space age fixes + K2SO compatibility (#724)c13e36142026-06-25

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.valid ~= true then
      goto continue
    end
    --Limit to same surface if space age is not enabled
    if not script.active_mods["space-age"] and 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,
          surface_index = data.surface.index,
          {
            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