z_mirror/EditorExtensions

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

Caleb HeuerFix reassigning blueprints not working596431c2025-02-21

local aggregate_chest_names = {
  ["ee-aggregate-chest"] = "ee-aggregate-chest",
  ["ee-aggregate-chest-passive-provider"] = "ee-aggregate-chest-passive-provider",
}

--- @type table<string, InfinityInventoryFilter[]>
local filters = {}
do
  local include_hidden = settings.global["ee-aggregate-include-hidden"].value --[[@as boolean]]
  local i = 0
  for quality_name in pairs(prototypes.quality) do
    local this_filters = {}
    for item_name, prototype in pairs(prototypes.item) do
      if string.find(item_name, "^parameter%-") then
        goto continue
      end
      if prototype.hidden and not include_hidden then
        goto continue
      end
      i = i + 1
      this_filters[i] = {
        name = item_name,
        quality = quality_name,
        count = prototype.stack_size,
        mode = "exactly",
        index = i,
      }
      ::continue::
    end
    filters[quality_name] = this_filters
  end
end
--- Set the filters for the given aggregate chest and removes the bar if there is one
--- @param entity LuaEntity
local function set_filters(entity)
  entity.remove_unfiltered_items = true
  entity.infinity_container_filters = filters[entity.quality.name]
  entity.get_inventory(defines.inventory.chest).set_bar()
end

-- Update the filters of all existing aggregate chests
local function update_all_filters()
  for _, surface in pairs(game.surfaces) do
    for _, entity in pairs(surface.find_entities_filtered({ name = aggregate_chest_names })) do
      set_filters(entity)
    end
  end
end

--- @param e BuiltEvent
local function on_entity_built(e)
  local entity = e.entity or e.destination
  if not entity or not entity.valid or not aggregate_chest_names[entity.name] then
    return
  end
  set_filters(entity)
end

--- @param e EventData.on_player_setup_blueprint
local function on_player_setup_blueprint(e)
  local blueprint = e.stack or e.record
  if not blueprint then
    return
  end

  local entities = blueprint.get_blueprint_entities()
  if not entities then
    return
  end
  local set = false
  for _, entity in pairs(entities) do
    if aggregate_chest_names[entity.name] then
      set = true
      entity.infinity_settings.filters = nil --- @diagnostic disable-line
    end
  end
  if set then
    blueprint.set_blueprint_entities(entities)
  end
end

local aggregate_chest = {}

aggregate_chest.on_configuration_changed = function(_)
  update_all_filters()
end

aggregate_chest.events = {
  [defines.events.on_built_entity] = on_entity_built,
  [defines.events.on_entity_cloned] = on_entity_built,
  [defines.events.on_player_setup_blueprint] = on_player_setup_blueprint,
  [defines.events.on_robot_built_entity] = on_entity_built,
  [defines.events.on_space_platform_built_entity] = on_entity_built,
  [defines.events.script_raised_built] = on_entity_built,
  [defines.events.script_raised_revive] = on_entity_built,
}

return aggregate_chest