Added validation if any connected building was destroyed.

NebulaModTeam · Apr 17, 2021 · 7e1b90c · 7e1b90c
1 parent e5e4a24
commit 7e1b90c
Showing 1 changed file with 32 additions and 1 deletion.
diff --git a/NebulaHost/PacketProcessors/Factory/Entity/CreatePrebuildsRequestProcessor.cs b/NebulaHost/PacketProcessors/Factory/Entity/CreatePrebuildsRequestProcessor.cs
@@ -79,6 +79,7 @@ public void ProcessPacket(CreatePrebuildsRequest packet, NebulaConnection conn)
                 GameMain.mainPlayer.mecha.buildArea = float.MaxValue;
                 FactoryManager.IgnoreBasicBuildConditionChecks = true;
                 bool canBuild = pab.CheckBuildConditions();
+                canBuild &= CheckBuildingConnections(pab.buildPreviews, planet.factory);
                 FactoryManager.IgnoreBasicBuildConditionChecks = false;
 
                 if (canBuild)
@@ -104,9 +105,39 @@ public void ProcessPacket(CreatePrebuildsRequest packet, NebulaConnection conn)
                 FactoryManager.EventFactory = null;
                 pab.buildPreviews = tmpList;
                 pab.waitConfirm = tmpConfirm;
-                pab.previewPose.position = tmpPos;
+                pab.previewPose.position = tmpPos; 
                 pab.previewPose.rotation = tmpRot;
             }
         }
+
+        public bool CheckBuildingConnections(List<BuildPreview> buildPreviews, PlanetFactory factory)
+        {
+            //Check if some entity that is suppose to be connected to this building is missing
+            for(int i = 0; i < buildPreviews.Count; i++)
+            {
+                bool isInputOk = true;
+                if (buildPreviews[i].inputObjId > 0)
+                {
+                    isInputOk = factory.entityPool.Length >= buildPreviews[i].inputObjId && factory.entityPool[buildPreviews[i].inputObjId].id != 0;
+                } else if (buildPreviews[i].inputObjId < 0)
+                {
+                    isInputOk = factory.prebuildPool.Length >= -buildPreviews[i].inputObjId && factory.prebuildPool[-buildPreviews[i].inputObjId].id != 0;
+                }
+                bool isOutputOk = true;
+                if (buildPreviews[i].outputObjId > 0)
+                {
+                    isInputOk = factory.entityPool.Length >= buildPreviews[i].outputObjId && factory.entityPool[buildPreviews[i].outputObjId].id != 0;
+                }
+                else if (buildPreviews[i].outputObjId < 0)
+                {
+                    isInputOk = factory.prebuildPool.Length >= -buildPreviews[i].outputObjId && factory.prebuildPool[-buildPreviews[i].outputObjId].id != 0;
+                }
+                if (!isInputOk || !isOutputOk)
+                {
+                    return false;
+                }
+            }
+            return true;
+        }
     }
 }