From a97b3fee99dc25b5961ee616589365a5582f66ae Mon Sep 17 00:00:00 2001
From: MetalStormGhost <ghosti@smytky.org>
Date: Sat, 16 Sep 2023 23:34:05 +0300
Subject: [PATCH] Will now dynamically generate offmapZones for supply cargo
 aircraft.

---
 game/pretense/pretenseluagenerator.py      | 27 ++++++++++++++++++++++
 resources/plugins/pretense/init_body_2.lua | 15 +-----------
 2 files changed, 28 insertions(+), 14 deletions(-)

diff --git a/game/pretense/pretenseluagenerator.py b/game/pretense/pretenseluagenerator.py
index cb80b397..d4005d08 100644
--- a/game/pretense/pretenseluagenerator.py
+++ b/game/pretense/pretenseluagenerator.py
@@ -350,6 +350,32 @@ class PretenseLuaGenerator(LuaGenerator):
                     f"    cm: addConnection('{cp.name}', '{closest_cps[1].name}')\n"
                 )
 
+        lua_string_supply = "local redSupply = {\n"
+        # Generate ConnectionManager connections
+        for cp_side in range(1, 3):
+            for cp in self.game.theater.controlpoints:
+                if isinstance(cp, OffMapSpawn):
+                    continue
+                cp_side_captured = cp_side == 2
+                if cp_side_captured != cp.captured:
+                    continue
+                cp_name_trimmed = "".join([i for i in cp.name.lower() if i.isalnum()])
+                for mission_type in self.game.pretense_air[cp_side][cp_name_trimmed]:
+                    if mission_type == FlightType.TRANSPORT.name:
+                        for air_group in self.game.pretense_air[cp_side][
+                            cp_name_trimmed
+                        ][mission_type]:
+                            lua_string_supply += f"'{air_group}',"
+            lua_string_supply += "}\n"
+            if cp_side < 2:
+                lua_string_supply += "local blueSupply = {\n"
+        lua_string_supply += "local offmapZones = {\n"
+        for cp in self.game.theater.controlpoints:
+            if isinstance(cp, Airfield):
+                cp_name_trimmed = "".join([i for i in cp.name.lower() if i.isalnum()])
+                lua_string_supply += f"   zones.{cp_name_trimmed},\n"
+        lua_string_supply += "}\n"
+
         init_body_1_file = open("./resources/plugins/pretense/init_body_1.lua", "r")
         init_body_1 = init_body_1_file.read()
 
@@ -373,6 +399,7 @@ class PretenseLuaGenerator(LuaGenerator):
             + init_body_1
             + lua_string_jtac
             + init_body_2
+            + lua_string_supply
             + init_footer
         )
 
diff --git a/resources/plugins/pretense/init_body_2.lua b/resources/plugins/pretense/init_body_2.lua
index 9d2798d8..9994dce9 100644
--- a/resources/plugins/pretense/init_body_2.lua
+++ b/resources/plugins/pretense/init_body_2.lua
@@ -58,18 +58,5 @@ end, {}, timer.getTime()+30)
 
 
 --supply injection
-local blueSupply = {'offmap-supply-blue-1','offmap-supply-blue-2','offmap-supply-blue-3','offmap-supply-blue-4','offmap-supply-blue-5'}
-local redSupply = {'offmap-supply-red-1','offmap-supply-red-2','offmap-supply-red-3','offmap-supply-red-4','offmap-supply-red-5'}
-local offmapZones = {
-	zones.batumi, 
-	zones.sochi,
-	zones.nalchik, 
-	zones.beslan,
-	zones.mozdok,
-	zones.mineralnye,
---  zones.senaki,
---	zones.sukhumi,
---	zones.gudauta,
---	zones.kobuleti,
-}
+