control.lua

---@type event_handler_lib
event_handler = require "event_handler"
util = require "util"

gui = require "scripts.flib-gui"
CustomInput = require "scripts.custom-input"
Search = require "scripts.search"
SearchResults = require "scripts.search-results"
ResultLocation = require "scripts.result-location"
SearchGui = require "scripts.search-gui"
Interop = require "scripts.remote"

---@alias ItemName string
---@alias EntityName string
---@alias SurfaceName string

---@class (exact) SearchGuiRefs
---@field frame LuaGuiElement frame
---@field pin_button LuaGuiElement sprite-button
---@field close_button LuaGuiElement sprite-button
---@field subheader_title LuaGuiElement label
---@field sort_results_dropdown LuaGuiElement drop-down
---@field all_surfaces LuaGuiElement checkbox
---@field all_qualities LuaGuiElement checkbox
---@field item_select LuaGuiElement choose-elem-button
---@field include_consumers LuaGuiElement checkbox
---@field include_machines LuaGuiElement checkbox
---@field include_inventories LuaGuiElement checkbox
---@field include_logistics LuaGuiElement checkbox
---@field include_modules LuaGuiElement checkbox
---@field include_entities LuaGuiElement checkbox
---@field include_ground_items LuaGuiElement checkbox
---@field include_requesters LuaGuiElement checkbox
---@field include_signals LuaGuiElement checkbox
---@field include_map_tags LuaGuiElement checkbox
---@field searching_label LuaGuiElement label
---@field search_progressbar LuaGuiElement progressbar
---@field result_flow LuaGuiElement flow
---@field recent_items_flow LuaGuiElement flow
---@field highlighted_button? LuaGuiElement sprite-button

---@class (exact) PlayerData
---@field refs SearchGuiRefs
---@field pinned? boolean
---@field ignore_close? boolean
---@field sort_results_by "name"|"distance"|"count"

---@class (exact) SearchGuiState
---@field all_qualities boolean
---@field all_surfaces boolean
---@field consumers boolean
---@field producers boolean
---@field storage boolean
---@field logistics boolean
---@field modules boolean
---@field requesters boolean
---@field ground_items boolean
---@field entities boolean
---@field signals boolean
---@field map_tags boolean

---@class (exact) CurrentSurfaceSearchData
---@field surface LuaSurface
---@field surface_data SurfaceData
---@field surface_statistics SurfaceStatistics
---@field chunk_iterator LuaChunkIterator

---@class (exact) SearchData
---@field blocking boolean
---@field tick_triggered GameTick
---@field force LuaForce
---@field state SearchGuiState
---@field target_item SignalID
---@field type_list string[]
---@field neutral_type_list string[]
---@field player LuaPlayer
---@field data table<SurfaceName, SurfaceData>
---@field statistics table<SurfaceName, SurfaceStatistics>
---@field not_started_surfaces LuaSurface[]
---@field search_complete boolean
---@field current_surface_search_data? CurrentSurfaceSearchData Used for non-blocking searches
---@field total_chunk_count number
---@field completed_chunk_count number

---@class (exact) EntityGroup
---@field count number
---@field avg_position MapPosition
---@field selection_box BoundingBox
---@field entity_name EntityName
---@field selection_boxes BoundingBox[]
---@field localised_name LocalisedString
---@field recipe_list? table<string, {localised_name: LocalisedString, count: number}>
---@field item_count? number
---@field fluid_count? number
---@field module_count? number
---@field request_count? number
---@field signal_count? number
---@field resource_count? number
---@field distance? number

---@alias SurfaceDataCategoryName "consumers"|"producers"|"storage"|"logistics"|"modules"|"requesters"|"ground_items"|"entities"|"signals"|"map_tags"
---@alias SurfaceData table<SurfaceDataCategoryName, EntityGroup[]>

---@alias SurfaceStatisticsCategoryName "consumers_count"|"producers_count"|"item_count"|"fluid_count"|"module_count"|"entity_count"|"resource_count"|"ground_count"|"request_count"|"signal_count"|"tag_count"
---@alias SurfaceStatistics table <SurfaceStatisticsCategoryName, number>

---@class ResultLocationData
---@field position MapPosition
---@field surface SurfaceName
---@field selection_boxes BoundingBox[]
---@field group_selection_box BoundingBox

DEBOUNCE_TICKS = 60

Control = {}

function map_to_list(map)
  local i = 1
  local list = {}
  for name, _ in pairs(map) do
    list[i] = name
    i = i + 1
  end
  return list
end

---@return LuaSurface[]
function filtered_surfaces()
  -- Skip certain modded surfaces that won't have assemblers/chests placed on them
  local surfaces = {}
  for _, surface in pairs(game.surfaces) do
    local surface_name = surface.name
    if string.sub(surface_name, -12) ~= "-transformer"  -- Power Overload
        and string.sub(surface_name, 1, 17) ~= "nullius-landfill-"  -- Nullius
        and string.sub(surface_name, 0, 8) ~= "starmap-"  -- Space Exploration
        and string.sub(surface_name, 0, 15) ~= "EE_TESTSURFACE_"  -- Editor Extensions
        and string.sub(surface_name, 0, 10) ~= "BPL_TheLab"  -- Blueprint Designer Lab
        and string.sub(surface_name, 0, 9) ~= "bpsb-lab-"  -- Blueprint Sandboxes
        and surface.name ~= "IR-limbo"  -- Industrial Revolution 2/3
        and surface_name ~= "aai-signals"  -- AAI Signals
        and surface_name ~= "secret_companion_surface_please_dont_touch"  -- Companion Drones
      then
      table.insert(surfaces, surface)
    end
  end
  return surfaces
end

local function update_surface_count()
  -- Hides 'All surfaces' button
  local multiple_surfaces = #filtered_surfaces() > 1

  if multiple_surfaces ~= storage.multiple_surfaces then
    for _, player_data in pairs(storage.players) do
      local all_surfaces = player_data.refs.all_surfaces
      all_surfaces.visible = multiple_surfaces
    end
  end

  storage.multiple_surfaces = multiple_surfaces
end

local function generate_item_to_entity_table()
  -- Make map of {item_name -> list[entity_name]}
  -- First try items_to_place_this, then mineable_properties
  -- Exception in mineable_properties is we don't want to include entity_name when type == "simple-entity" if item_name is a resource
  -- This prevents things like rocks showing when searching for stone

  local resource_prototypes = prototypes.get_entity_filtered({{filter = "type", type = "resource"}})
  local is_resource = {}
  for _, resource in pairs(resource_prototypes) do
    is_resource[resource.name] = true
  end


  local item_to_entities_maps = {}
  for _, prototype in pairs(prototypes.entity) do
    local items_to_place_this = prototype.items_to_place_this
    if items_to_place_this then
      for _, item in pairs(items_to_place_this) do
        local item_name = item.name
        local associated_entities_map = item_to_entities_maps[item_name] or {}
        associated_entities_map[prototype.name] = true
        item_to_entities_maps[item_name] = associated_entities_map
      end
    end
    local properties = prototype.mineable_properties
    if properties.minable and properties.products then
      for _, item in pairs(prototype.mineable_properties.products) do
        local item_name = item.name
        -- Filter out rocks
        if prototype.type ~= "simple-entity" or not is_resource[item_name] then
          local associated_entities_map = item_to_entities_maps[item_name] or {}
          associated_entities_map[prototype.name] = true
          item_to_entities_maps[item_name] = associated_entities_map
        end
      end
    end
  end

  for _, prototype in pairs(prototypes.item) do
    local place_result = prototype.place_result
    if place_result then
      local item_name = prototype.name
      local associated_entities_map = item_to_entities_maps[item_name] or {}
      associated_entities_map[place_result.name] = true
      item_to_entities_maps[item_name] = associated_entities_map
    end
    local plant_result = prototype.plant_result
    if plant_result then
      local item_name = prototype.name
      local associated_entities_map = item_to_entities_maps[item_name] or {}
      associated_entities_map[plant_result.name] = true
      item_to_entities_maps[item_name] = associated_entities_map
    end
  end

  -- Hardcode some Pyanodons associations
  if script.active_mods["pypetroleumhandling"] then
    --  `if` checks in case something removed those items or playing an older version of Py
    if item_to_entities_maps["raw-gas"] then
      item_to_entities_maps["raw-gas"]["bitumen-seep"] = true
    end
    if item_to_entities_maps["tar"] then
      item_to_entities_maps["tar"]["bitumen-seep"] = true
    end
    if item_to_entities_maps["crude-oil"] then
      item_to_entities_maps["crude-oil"]["bitumen-seep"] = true
    end
  end

  local item_to_entities = {}
  for item_name, entity_map in pairs(item_to_entities_maps) do
    item_to_entities[item_name] = map_to_list(entity_map)
  end

  ---@type table<ItemName, EntityName[]>
  storage.item_to_entities = item_to_entities
end

local function setup_interop_callback()
  Interop.set_on_external_item_callback(function(player_index, item_name, source)
    Interop.add_recent_item(player_index, item_name, source)
    local player = game.get_player(player_index)
    if player then
      local player_data = storage.players[player_index]
      if player_data and player_data.refs.frame.valid and player_data.refs.frame.visible then
        SearchGui.update_recent_panel(player_data)
      end
    end
  end)
end

local function on_init()
  ---@type table<PlayerIndex, PlayerData>
  storage.players = {}
  ---@type table<PlayerIndex, SearchData>
  storage.current_searches = {}
  ---@type boolean
  storage.multiple_surfaces = false
  ---@type table<PlayerIndex, {item_name: string, source: string}[]>
  storage.recent_external_items = {}
  update_surface_count()
  generate_item_to_entity_table()
  Interop.subscribe_to_events()
  setup_interop_callback()
end

local function on_load()
  Interop.subscribe_to_events()
  setup_interop_callback()
end

local function on_configuration_changed()
  -- Destroy all GUIs
  for player_index, player_data in pairs(storage.players) do
    local player = game.get_player(player_index)
    if player then
      SearchGui.destroy(player, player_data)
    else
      storage.players[player_index] = nil
    end
  end

  -- Stop in-progress non-blocking searches
  storage.current_searches = {}

  storage.multiple_surfaces = false
  if not storage.recent_external_items then
    storage.recent_external_items = {}
  end
  update_surface_count()
  generate_item_to_entity_table()
  Interop.subscribe_to_events()
  setup_interop_callback()
end

Control.on_init = on_init
Control.on_load = on_load
Control.on_configuration_changed = on_configuration_changed
Control.events = {
  [defines.events.on_surface_created] = update_surface_count,
  [defines.events.on_surface_deleted] = update_surface_count,
}

event_handler.add_libraries{
  gui --[[@as event_handler]],
  Control,
  CustomInput,
  Search,
  SearchResults,
  ResultLocation,
  SearchGui,
}