z_mirror/EditorExtensions

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

Caleb HeuerRemove usage of flib migration modulecabb4ba2026-06-08

local infinity_accumulator = require("scripts.infinity-accumulator")
local infinity_loader = require("scripts.infinity-loader")
local inventory_filters = require("scripts.inventory-filters")
local inventory_sync = require("scripts.inventory-sync")
local linked_belt = require("scripts.linked-belt")
local util = require("scripts.util")

local version_migrations = {
  ["2.0.0"] = function()
    -- Preserve testing lab state
    local testing_lab_state = {}
    for player_index, player_table in pairs(storage.players) do
      local player = game.get_player(player_index)
      if player and player_table.lab_state and player_table.normal_state then
        testing_lab_state[player_index] = {
          normal = player_table.normal_state,
          lab = player_table.lab_state,
          player = player,
          refresh = false,
        }
      end
    end
    -- NUKE EVERYTHING
    storage = { testing_lab_state = testing_lab_state, wagons = storage.wagons }
    rendering.clear("EditorExtensions")
    for _, player in pairs(game.players) do
      for _, gui in pairs({ player.gui.top, player.gui.left, player.gui.center, player.gui.screen, player.gui.relative }) do
        for _, child in pairs(gui.children) do
          if child.get_mod() == "EditorExtensions" then
            child.destroy()
          end
        end
      end
    end
    -- Start over
    infinity_accumulator.on_init()
    infinity_loader.on_init()
    inventory_filters.on_init()
    inventory_sync.on_init()
    linked_belt.on_init()

    -- Migrate infinity loaders
    for _, surface in pairs(game.surfaces) do
      for _, combinator in pairs(surface.find_entities_filtered({ name = "ee-infinity-loader-dummy-combinator" })) do
        local cb = combinator.get_control_behavior() --[[@as LuaConstantCombinatorControlBehavior?]]
        if not cb then
          goto continue
        end
        local section = cb.get_section(1)
        if not section then
          goto continue
        end
        local filter_1, filter_2 = section.filters[1], section.filters[2]
        local loader = combinator.surface.create_entity({
          name = "ee-infinity-loader",
          direction = combinator.direction,
          position = combinator.position,
          force = combinator.force,
          last_user = combinator.last_user,
          fast_replace = true,
          create_build_effect_smoke = false,
        })
        if not loader then
          error("Failed to create infinity loader replacement.")
        end
        if filter_1 and filter_1.value then
          loader.set_filter(1, { name = filter_1.value.name, quality = filter_1.value.quality, amount = filter_1.min })
        end
        if filter_2 and filter_2.value then
          loader.set_filter(2, { name = filter_2.value.name, quality = filter_2.value.quality, amount = filter_2.min })
        end
        infinity_loader.snap(loader)
        combinator.destroy()
        ::continue::
      end
    end
  end,
  ["2.3.0"] = function()
    storage.aggregate_filters = nil
    storage.infinity_pipe_amount_type = nil
    for _, gui in pairs(storage.infinity_pipe_gui or {}) do
      local window = gui.elems.ee_infinity_pipe_window
      if window and window.valid then
        window.destroy()
      end
    end
    storage.infinity_pipe_gui = nil
  end,
  ["2.3.1"] = function()
    for surface_name, surface in pairs(game.surfaces) do
      if string.find(surface_name, "EE_TESTSURFACE_") then
        for force_name, force in pairs(game.forces) do
          if not string.find(force_name, "EE_TESTFORCE_") then
            force.set_surface_hidden(surface, true)
          end
        end
      end
    end
  end,
  --- @param old_version string
  ["2.4.0"] = function(old_version)
    if helpers.compare_versions(old_version, "2.0.0") > 0 then
      return
    end
    for _, surface in pairs(game.surfaces) do
      for _, loader in pairs(surface.find_entities_filtered({ name = "ee-infinity-loader" })) do
        loader.set_filter(2, loader.get_filter(1))
      end
    end
  end,
  ["2.5.0"] = function()
    local to_remove = {}
    for unit_number, wagon_data in pairs(storage.wagons) do
      local proxy = wagon_data.proxy
      if proxy and proxy.valid and proxy.name == "ee-infinity-wagon-chest" then
        local wagon = proxy.surface.find_entity("ee-infinity-cargo-wagon", proxy.position)
        if wagon then
          wagon.infinity_container_filters = proxy.infinity_container_filters
          wagon.remove_unfiltered_items = proxy.remove_unfiltered_items
          proxy.destroy()
        end
        table.insert(to_remove, unit_number)
      end
    end
    for _, unit_number in pairs(to_remove) do
      storage.wagons[unit_number] = nil
    end
  end,
  ["2.5.1"] = function()
    local in_debug_world = util.in_debug_world() or util.in_testing_scenario()
    for _, surface in pairs(game.surfaces) do
      if (in_debug_world and surface.name == "nauvis") or string.find(surface.name, "EE_TESTSURFACE") then
        surface.ignore_surface_conditions = true
      end
    end
  end,
}

--- @class migrations_handler: event_handler
local migrations = {}

--- @param e ConfigurationChangedData
function migrations.on_configuration_changed(e)
  local old_version = e.mod_changes and e.mod_changes[script.mod_name] and e.mod_changes[script.mod_name].old_version
  if not old_version then
    return
  end

  for version, callback in pairs(version_migrations) do
    if helpers.compare_versions(version, old_version) > 0 then
      callback(old_version)
    end
  end
end

return migrations