Up version

changelog.txt

@@ -1,4 +1,13 @@
 ---------------------------------------------------------------------------------------------------
+Version: 2.0.7
+Date: 2024-11-10
+  Optimizations:
+    - Add cleanup of empty global tables
+    - Move global table creation from on_surface_created to on_built_entity to save on event
+      listeners
+    - Review force rename/removal handling
+    - Code cleanup and comments
+---------------------------------------------------------------------------------------------------
 Version: 2.0.6
 Date: 2024-11-10
   Features:

control.lua

@@ -67,6 +67,13 @@ local function add_entity_to_list(event)
 	--Whenever an ammo-turret, vehicle or artillery type entity is built, add it to the global table.
 	local entity = event.created_entity or event.entity
 	local index = entity.surface.name.."_"..entity.force.name
+
+	-- Add table if it doesn't exist and a player of this force is connected
+	if entity.force.connected_players[1] and not storage.ammo_entities[index] then
+		storage.ammo_entities[index] = {}
+	end
+
+	-- Add entity to table
 	if storage.ammo_entities[index] then
 		table.insert(storage.ammo_entities[index], entity)
 	end
@@ -134,20 +141,6 @@ local function remove_force_from_list(event)
 	end
 end
 
-local function add_surface_to_list(event)
-	--Whenever a surface is created, add an empty list to the global table for each force with a connected player.
-	local surface
-	if event.surface_index then
-		surface = game.surfaces[event.surface_index]
-	end
-
-	for _,force in pairs(game.forces) do
-		if surface and surface.valid and not storage.ammo_entities[surface.name.."_"..force.name] and force.connected_players[1] then
-			storage.ammo_entities[surface.name.."_"..force.name] = {}
-		end
-	end
-end
-
 local function remove_surface_from_list(event)
 	--Whenever a surface is renamed or deleted, move/remove all entities in/from the global table.
 	for _,force in pairs(game.forces) do
@@ -195,7 +188,8 @@ local function generate_alerts()
 				ammo_entities = storage.ammo_entities[surface.name.."_"..player.force.name]
 			end
 
-			if ammo_entities and not table_is_empty(ammo_entities) then
+			if ammo_entities then
+				if not table_is_empty(ammo_entities) then
 					for index,entity in pairs(ammo_entities) do
 						if entity.valid then
 							if entity.force == player.force then
@@ -244,6 +238,10 @@ local function generate_alerts()
 							table.remove(ammo_entities, index)
 						end
 					end
+				else
+					-- Cleanup if table is empty
+					storage.ammo_entities[surface.name.."_"..player.force.name] = nil
+				end
 			end
 		end
 	end
@@ -262,7 +260,6 @@ script.on_event(defines.events.on_player_changed_force, remove_force_from_list)
 script.on_event(defines.events.on_force_created, add_force_to_list)
 script.on_event(defines.events.on_forces_merged, remove_force_from_list)
 
-script.on_event(defines.events.on_surface_created, add_surface_to_list)
 script.on_event(defines.events.on_surface_renamed, remove_surface_from_list)
 script.on_event(defines.events.on_pre_surface_deleted, remove_surface_from_list)
 

info.json

@@ -1,6 +1,6 @@
 {
     "name": "Gun_Turret_Alerts",
-    "version": "2.0.6",
+    "version": "2.0.7",
     "title": "Ammo Alerts",
     "author": "Wyrrrd",
     "factorio_version": "2.0",