first commit

NitroxClient/GameLogic/Spawning/WorldEntities/WorldEntitySpawnerResolver.cs

@@ -0,0 +1,74 @@
+using System.Collections.Generic;
+using NitroxClient.GameLogic.Spawning.Metadata;
+using NitroxModel.DataStructures.GameLogic.Entities;
+using NitroxModel_Subnautica.DataStructures;
+
+namespace NitroxClient.GameLogic.Spawning.WorldEntities;
+
+public class WorldEntitySpawnerResolver
+{
+    private readonly DefaultWorldEntitySpawner defaultEntitySpawner = new();
+    private readonly VehicleWorldEntitySpawner vehicleWorldEntitySpawner;
+
+    private readonly PrefabPlaceholderEntitySpawner prefabPlaceholderEntitySpawner;
+    private readonly PlaceholderGroupWorldEntitySpawner placeholderGroupWorldEntitySpawner;
+    private readonly PlayerWorldEntitySpawner playerWorldEntitySpawner;
+    private readonly SerializedWorldEntitySpawner serializedWorldEntitySpawner;
+    private readonly GeyserWorldEntitySpawner geyserWorldEntitySpawner;
+    private readonly ReefbackEntitySpawner reefbackEntitySpawner;
+    private readonly ReefbackChildEntitySpawner reefbackChildEntitySpawner;
+    private readonly CreatureRespawnEntitySpawner creatureRespawnEntitySpawner;
+
+    private readonly Dictionary<TechType, IWorldEntitySpawner> customSpawnersByTechType = new();
+
+    public WorldEntitySpawnerResolver(EntityMetadataManager entityMetadataManager, PlayerManager playerManager, LocalPlayer localPlayer, Entities entities, SimulationOwnership simulationOwnership)
+    {
+        customSpawnersByTechType[TechType.Crash] = new CrashEntitySpawner();
+        customSpawnersByTechType[TechType.EscapePod] = new EscapePodWorldEntitySpawner(localPlayer);
+        customSpawnersByTechType[TechType.Creepvine] = new CreepvineEntitySpawner(defaultEntitySpawner);
+
+        vehicleWorldEntitySpawner = new VehicleWorldEntitySpawner(entities);
+        prefabPlaceholderEntitySpawner = new PrefabPlaceholderEntitySpawner(defaultEntitySpawner);
+        placeholderGroupWorldEntitySpawner = new PlaceholderGroupWorldEntitySpawner(entities, this, defaultEntitySpawner, entityMetadataManager, prefabPlaceholderEntitySpawner);
+        playerWorldEntitySpawner = new PlayerWorldEntitySpawner(playerManager, localPlayer);
+        serializedWorldEntitySpawner = new SerializedWorldEntitySpawner();
+        geyserWorldEntitySpawner = new GeyserWorldEntitySpawner(entities);
+        reefbackChildEntitySpawner = new ReefbackChildEntitySpawner();
+        reefbackEntitySpawner = new ReefbackEntitySpawner(reefbackChildEntitySpawner);
+        creatureRespawnEntitySpawner = new CreatureRespawnEntitySpawner(simulationOwnership);
+    }
+
+    public IWorldEntitySpawner ResolveEntitySpawner(WorldEntity entity)
+    {
+        switch (entity)
+        {
+            case PrefabPlaceholderEntity:
+                return prefabPlaceholderEntitySpawner;
+            case PlaceholderGroupWorldEntity:
+                return placeholderGroupWorldEntitySpawner;
+            case PlayerWorldEntity:
+                return playerWorldEntitySpawner;
+            case VehicleWorldEntity:
+                return vehicleWorldEntitySpawner;
+            case SerializedWorldEntity:
+                return serializedWorldEntitySpawner;
+            case GeyserWorldEntity:
+                return geyserWorldEntitySpawner;
+            case ReefbackEntity:
+                return reefbackEntitySpawner;
+            case ReefbackChildEntity:
+                return reefbackChildEntitySpawner;
+            case CreatureRespawnEntity:
+                return creatureRespawnEntitySpawner;
+        }
+
+        TechType techType = entity.TechType.ToUnity();
+
+        if (customSpawnersByTechType.TryGetValue(techType, out IWorldEntitySpawner value))
+        {
+            return value;
+        }
+
+        return defaultEntitySpawner;
+    }
+}