z_mirror/Krastorio2

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

Caleb HeuerRun stylua02265e002025-11-03

local flib_position = require("__flib__.position")

--- @class RadioactivityPlayerData
--- @field entity boolean
--- @field damaged boolean
--- @field inventory boolean
--- @field last_position MapPosition

local default_entities = { "uranium-ore" }
local default_items = {
  "nuclear-fuel",
  "uranium-235",
  "uranium-238",
  "uranium-fuel-cell",
  "uranium-ore",
  "depleted-uranium-fuel-cell",
}
local range = 7

--- @param player LuaPlayer
local function check_inventory(player)
  if not storage.radioactivity.enabled or not next(storage.radioactivity.entities) then
    return
  end

  local player_data = storage.radioactivity.players[player.index]
  if not player_data then
    return
  end

  local character = player.character
  if not character or not character.valid then
    player_data.inventory = false
    return
  end

  local inventory = character.get_main_inventory()
  if not inventory then
    return
  end

  for _, item_name in pairs(storage.radioactivity.items) do
    -- FIXME: This does not check trash inventory
    if inventory.get_item_count_filtered({ name = item_name }) > 0 then
      player_data.inventory = true
      return
    end
  end

  player_data.inventory = false
end

--- @param player LuaPlayer
local function check_around_player(player)
  if not storage.radioactivity.enabled or not next(storage.radioactivity.items) then
    return
  end

  local player_data = storage.radioactivity.players[player.index]
  if not player_data then
    return
  end

  if not player.character or not player.character.valid then
    player_data.entity = false
    return
  end

  local position = flib_position.floor(player.physical_position)
  local last_position = player_data.last_position
  if flib_position.eq(position, last_position) then
    return
  end
  player_data.last_position = position

  player_data.entity = player.surface.count_entities_filtered({
    name = storage.radioactivity.entities,
    radius = range,
    position = position,
  }) > 0
end

--- @param player LuaPlayer
local function add_player(player)
  storage.radioactivity.players[player.index] = {
    entity = false,
    damaged = false,
    inventory = false,
    last_position = { x = 0, y = 0 },
  }
end

--- @param player_index uint
local function remove_player(player_index)
  storage.radioactivity.players[player_index] = nil
end

local function update_and_damage()
  if not storage.radioactivity.enabled then
    return
  end

  for player_index, player_data in pairs(storage.radioactivity.players) do
    if not player_data.entity and not player_data.inventory then
      goto continue
    end

    local player = game.get_player(player_index)
    if not player or not player.connected or not player.character then
      goto continue
    end

    player.add_custom_alert(
      player.character,
      { type = "virtual", name = "kr-nuclear-2" },
      { "gui-alert-tooltip.kr-taking-radioactive-damage" },
      false
    )

    -- Damage the player
    -- TODO: Account for armor and energy shields
    local base_damage = 7.25
    player_data.damaged = player.character.damage(base_damage, "enemy", "kr-radioactive") > 0

    ::continue::
  end
end

local function update_sounds()
  if not storage.radioactivity.enabled then
    return
  end

  for player_index, player_data in pairs(storage.radioactivity.players) do
    if not player_data.entity and not player_data.inventory then
      goto continue
    end

    if not player_data.damaged then
      goto continue
    end

    local player = game.get_player(player_index)
    if not player or not player.connected then
      goto continue
    end

    player.play_sound({
      path = "kr-radioactive",
      position = player.physical_position,
      volume_modifier = 0.5,
    })

    ::continue::
  end
end

--- @param e EventData.on_player_changed_position|EventData.on_player_changed_surface
local function on_player_moved(e)
  local player = game.get_player(e.player_index)
  if player then
    check_around_player(player)
  end
end

--- @param e EventData.on_player_main_inventory_changed|EventData.on_player_trash_inventory_changed
local function on_player_inventory_changed(e)
  local player = game.get_player(e.player_index)
  if player then
    check_inventory(player)
  end
end

--- @param e EventData.on_player_died|EventData.on_player_respawned
local function on_player_alive_state_changed(e)
  local player = game.get_player(e.player_index)
  if player then
    check_around_player(player)
    check_inventory(player)
  end
end

--- @param e EventData.on_player_created
local function on_player_created(e)
  local player = game.get_player(e.player_index)
  if not player then
    return
  end

  add_player(player)
end

--- @param e EventData.on_player_removed
local function on_player_removed(e)
  remove_player(e.player_index)
end

--- @param e EventData.on_runtime_mod_setting_changed
local function on_runtime_mod_setting_changed(e)
  if e.setting ~= "kr-enable-radioactivity" or not storage.radioactivity then
    return
  end

  storage.radioactivity.enabled = settings.global["kr-enable-radioactivity"].value --[[@as boolean]]
end

--- @class RadioactivityHandler : event_handler
local radioactivity = {}

function radioactivity.on_init()
  storage.radioactivity = {
    enabled = settings.global["kr-enable-radioactivity"].value --[[@as boolean]],
    --- @type table<uint, RadioactivityPlayerData>
    players = {},
  }
  radioactivity.on_configuration_changed()
end

function radioactivity.on_configuration_changed()
  storage.radioactivity.entities = {}
  for _, entity_name in pairs(default_entities) do
    if prototypes.entity[entity_name] then
      table.insert(storage.radioactivity.entities, entity_name)
    end
  end
  storage.radioactivity.items = {}
  for _, item_name in pairs(default_items) do
    if prototypes.item[item_name] then
      table.insert(storage.radioactivity.items, item_name)
    end
  end
end

remote.add_interface("kr-radioactivity", {
  --- @param name string
  add_entity = function(name)
    if not storage.radioactivity then
      return
    end
    if not name or type(name) ~= "string" then
      error("`name` must be a string.")
    end

    if not prototypes.entity[name] then
      error("Entity `" .. name .. "` does not exist.")
    end

    table.insert(storage.radioactivity.entities, name)
  end,
  --- @param name string
  add_item = function(name)
    if not storage.radioactivity then
      return
    end
    if not name or type(name) ~= "string" then
      error("`name` must be a string.")
    end

    if not prototypes.item[name] then
      error("Item `" .. name .. "` does not exist.")
    end

    table.insert(storage.radioactivity.items, name)
  end,
})

radioactivity.events = {
  [defines.events.on_player_changed_position] = on_player_moved,
  [defines.events.on_player_changed_surface] = on_player_moved,
  [defines.events.on_player_created] = on_player_created,
  [defines.events.on_player_cursor_stack_changed] = on_player_inventory_changed,
  [defines.events.on_player_died] = on_player_alive_state_changed,
  [defines.events.on_player_main_inventory_changed] = on_player_inventory_changed,
  [defines.events.on_player_removed] = on_player_removed,
  [defines.events.on_player_respawned] = on_player_alive_state_changed,
  [defines.events.on_player_toggled_map_editor] = on_player_moved,
  [defines.events.on_player_trash_inventory_changed] = on_player_inventory_changed,
  [defines.events.on_runtime_mod_setting_changed] = on_runtime_mod_setting_changed,
  [defines.events.on_tick] = update_sounds,
}

radioactivity.on_nth_tick = {
  [20] = update_and_damage,
}

return radioactivity