z_mirror/Tapeline

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

Caleb HeuerUse slash requires37bb3aa2022-12-01

local area = require("__flib__/area")
local event = require("__flib__/event")
local migration = require("__flib__/migration")

local constants = require("__Tapeline__/constants")
local migrations = require("__Tapeline__/migrations")
local player_data = require("__Tapeline__/player-data")
local tape = require("__Tapeline__/tape")

--- @param player_table PlayerTable
local function destroy_last_entity(player_table)
  local last_entity = player_table.last_entity
  if last_entity then
    last_entity.destroy()
    player_table.last_entity = nil
  end
end

--- @param tapes table<number, TapeData>
--- @param cursor_position MapPosition
--- @param surface LuaSurface
local function select_tape(tapes, cursor_position, surface)
  local topmost_tape
  local most_recent_update_time = 0
  for i, tape_data in ipairs(tapes) do
    local TapeArea = area.load(tape_data.Area)
    if tape_data.surface == surface and TapeArea:contains_position(cursor_position) then
      if tape_data.last_update_tick > most_recent_update_time then
        most_recent_update_time = tape_data.last_update_tick
        topmost_tape = i
      end
    end
  end

  return topmost_tape
end

--- @param player LuaPlayer
local function holding_tl_tool(player)
  local cursor_stack = player.cursor_stack
  return cursor_stack and cursor_stack.valid_for_read and cursor_stack.name == "tl-tool"
end

--- @param player LuaPlayer
--- @param player_table PlayerTable
local function set_cursor_label(player, player_table)
  local settings, TapeArea
  if player_table.flags.drawing then
    settings = player_table.tape_settings
    TapeArea = area.load(player_table.tapes.drawing.Area)
  elseif player_table.flags.editing then
    local tape_data = player_table.tapes.editing
    settings = tape_data.settings
    TapeArea = area.load(tape_data.Area)
  else
    settings = player_table.tape_settings
  end

  local cursor_stack = player.cursor_stack
  if cursor_stack and cursor_stack.valid_for_read then
    cursor_stack.label = (
      (TapeArea and (TapeArea:width() .. "x" .. TapeArea:height() .. " | ") or "")
      .. constants.mode_labels[settings.mode]
      .. " mode | "
      .. constants.divisor_labels[settings.mode]
      .. " "
      .. settings[settings.mode .. "_divisor"]
    )
  end
end

event.on_init(function()
  --- @type table<uint, PlayerTable>
  global.players = {}

  for _, player in pairs(game.players) do
    player_data.init(player.index)
    player_data.refresh(player, global.players[player.index])
  end
end)

event.on_configuration_changed(function(e)
  if migration.on_config_changed(e, migrations) then
    for i, player_table in pairs(global.players) do
      local player = game.get_player(i) --[[@as LuaPlayer]]
      player_data.refresh(player, player_table)
    end
  end
end)

event.register("tl-get-tool", function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  if player.clear_cursor() then
    player.cursor_stack.set_stack({ name = "tl-tool", count = 100 })
  end
end)

event.register("tl-edit-tape", function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  local player_table = global.players[e.player_index]

  if holding_tl_tool(player) then
    local tapes = player_table.tapes
    local tape_to_edit = select_tape(tapes, e.cursor_position, player.surface)
    if tape_to_edit then
      if player_table.flags.editing then
        tape.exit_edit_mode(player_table)
        set_cursor_label(player, player_table)
      end
      tape.enter_edit_mode(player, player_table, tape_to_edit)
      set_cursor_label(player, player_table)
    end
  end
end)

event.register("tl-delete-tape", function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  local player_table = global.players[e.player_index]

  if holding_tl_tool(player) then
    local tapes = player_table.tapes
    local tape_to_delete = select_tape(tapes, e.cursor_position, player.surface)
    if tape_to_delete then
      tape.delete(player_table, tape_to_delete)
      set_cursor_label(player, player_table)
    end
  end
end)

event.register({
  "tl-increase-divisor",
  "tl-decrease-divisor",
}, function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  if holding_tl_tool(player) then
    local player_table = global.players[e.player_index]
    local settings
    if player_table.flags.editing then
      settings = player_table.tapes.editing.settings
    else
      settings = player_table.tape_settings
    end
    local mode = settings.mode
    local key = mode .. "_divisor"
    local delta = string.find(e.input_name, "increase") and 1 or -1
    local new_divisor = settings[key] + delta
    if new_divisor >= constants.divisor_minimums[mode] then
      settings[key] = new_divisor
      set_cursor_label(player, player_table)
      if player_table.flags.drawing then
        tape.update_draw(player, player_table)
      elseif player_table.flags.editing then
        tape.edit_settings(e.player_index, player_table, mode, new_divisor)
      end
    else
      player.create_local_flying_text({
        text = { "message.tl-minimal-value-is", constants.divisor_minimums[mode] },
        create_at_cursor = true,
      })
      player.play_sound({ path = "utility/cannot_build", volume_modifier = 0.75 })
    end
  end
end)

event.register({
  "tl-next-mode",
  "tl-previous-mode",
}, function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  if holding_tl_tool(player) then
    local player_table = global.players[e.player_index]
    local settings
    if player_table.flags.editing then
      settings = player_table.tapes.editing.settings
    else
      settings = player_table.tape_settings
    end
    local new_mode = next(constants.modes, settings.mode)
    if not new_mode then
      new_mode = next(constants.modes)
    end
    settings.mode = new_mode
    local divisor = settings[new_mode .. "_divisor"]
    set_cursor_label(player, player_table)
    if player_table.flags.drawing then
      tape.update_draw(player, player_table)
    elseif player_table.flags.editing then
      tape.edit_settings(e.player_index, player_table, new_mode, divisor)
    end
  end
end)

event.register("tl-clear-cursor", function(e)
  local player_table = global.players[e.player_index]
  if player_table.flags.drawing or player_table.flags.editing then
    local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
    player.clear_cursor()
  end
end)

event.on_built_entity(function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  local player_table = global.players[e.player_index]
  local entity = e.created_entity
  local is_ghost = entity.name == "entity-ghost"

  destroy_last_entity(player_table)

  if is_ghost then
    -- instantly revive the entity if it is a ghost
    local _, new_entity = entity.silent_revive()
    if not new_entity then
      return
    end
    entity = new_entity
  end
  -- make the entity invincible to prevent attacks
  entity.destructible = false
  player_table.last_entity = entity

  -- update tape
  if player_table.flags.drawing then
    tape.update_draw(player, player_table, entity.position, is_ghost)
  elseif player_table.flags.editing then
    tape.move(player, player_table, entity.position, entity.surface)
  else
    tape.start_draw(player, player_table, entity.position, entity.surface)
  end

  -- update the cursor
  player.cursor_stack.set_stack({ name = "tl-tool", count = 100 })
  set_cursor_label(player, player_table)
end, {
  { filter = "name", name = "tl-dummy-entity" },
  { filter = "ghost_name", name = "tl-dummy-entity" },
})

event.register({
  defines.events.on_player_selected_area,
  defines.events.on_player_alt_selected_area,
}, function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  local player_table = global.players[e.player_index]
  if player_table.flags.drawing then
    player_table.flags.drawing = false
    destroy_last_entity(player_table)
    tape.complete_draw(player, player_table, e.name == defines.events.on_player_selected_area)
    set_cursor_label(player, player_table)
  elseif player_table.flags.editing then
    destroy_last_entity(player_table)
    tape.complete_move(player_table)
  end
end)

event.on_player_created(function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  player_data.init(e.player_index)
  player_data.refresh(player, global.players[e.player_index])
end)

event.on_player_removed(function(e)
  global.players[e.player_index] = nil
end)

event.on_player_cursor_stack_changed(function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  local player_table = global.players[e.player_index]
  if not player_table then
    return
  end
  local cursor_stack = player.cursor_stack
  local is_empty = not cursor_stack or not cursor_stack.valid_for_read
  if player_table.flags.holding_tool then
    if is_empty or cursor_stack.name ~= "tl-tool" then --- @diagnostic disable-line
      player_table.flags.holding_tool = false
      destroy_last_entity(player_table)
      if player_table.flags.editing then
        tape.exit_edit_mode(player_table)
        player.cursor_stack.set_stack({ name = "tl-tool", count = 100 })
        set_cursor_label(player, player_table)
      else
        if player_table.flags.drawing then
          tape.cancel_draw(player_table)
        end
        if player.controller_type == defines.controllers.character and player_table.flags.increased_build_distance then
          player_table.flags.increased_build_distance = false
          local build_distance = player.character_build_distance_bonus
          if build_distance >= 1000000 then
            -- decrease build distance
            player.character_build_distance_bonus = build_distance - 1000000 --[[@as uint]]
          end
        end
      end
    end
  elseif holding_tl_tool(player) then
    player_table.flags.holding_tool = true
    player.cursor_stack.set_stack({ name = "tl-tool", count = 100 })
    set_cursor_label(player, player_table)
    if player.controller_type == defines.controllers.character and not player_table.flags.increased_build_distance then
      -- increase build distance
      player_table.flags.increased_build_distance = true
      player.character_build_distance_bonus = player.character_build_distance_bonus + 1000000 --[[@as uint]]
    end
  end
end)

-- FIXME: Cancel draw when any controller change occurs (requires API addition)
event.on_pre_player_toggled_map_editor(function(e)
  local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
  local player_table = global.players[e.player_index]
  if not player_table then
    return
  end
  if player_table.flags.increased_build_distance then
    player_table.flags.increased_build_distance = false
    local build_distance = player.character_build_distance_bonus
    if build_distance >= 1000000 then
      -- decrease build distance
      player.character_build_distance_bonus = build_distance - 1000000 --[[@as uint]]
    end
  end
end)

event.on_runtime_mod_setting_changed(function(e)
  local internal = constants.setting_names[e.setting]
  if internal then
    local player = game.get_player(e.player_index) --[[@as LuaPlayer]]
    local player_table = global.players[e.player_index]
    local visual_settings = player_table.visual_settings
    player_data.update_visual_setting(player, visual_settings, e.setting, internal)
    for _, tape_data in ipairs(player_table.tapes) do
      tape.update_visual_settings(tape_data, visual_settings)
    end
  end
end)