first commit

NitroxClient/GameLogic/Spawning/InstalledBatteryEntitySpawner.cs

@@ -0,0 +1,89 @@
+using System.Collections;
+using NitroxClient.Communication;
+using NitroxClient.GameLogic.Spawning.Abstract;
+using NitroxClient.GameLogic.Spawning.WorldEntities;
+using NitroxClient.MonoBehaviours;
+using NitroxClient.Unity.Helper;
+using NitroxModel.DataStructures.GameLogic;
+using NitroxModel.DataStructures.GameLogic.Entities;
+using NitroxModel.DataStructures.Util;
+using NitroxModel.Packets;
+using NitroxModel_Subnautica.DataStructures;
+using UnityEngine;
+
+namespace NitroxClient.GameLogic.Spawning;
+
+public class InstalledBatteryEntitySpawner : SyncEntitySpawner<InstalledBatteryEntity>
+{
+    protected override IEnumerator SpawnAsync(InstalledBatteryEntity entity, TaskResult<Optional<GameObject>> result)
+    {
+        if (!CanSpawn(entity, out EnergyMixin energyMixin, out string errorLog))
+        {
+            Log.Error(errorLog);
+            result.Set(Optional.Empty);
+            yield break;
+        }
+
+        TaskResult<GameObject> prefabResult = new();
+        yield return DefaultWorldEntitySpawner.RequestPrefab(entity.TechType.ToUnity(), prefabResult);
+        GameObject gameObject = GameObjectHelper.InstantiateWithId(prefabResult.Get(), entity.Id);
+
+        SetupObject(gameObject, energyMixin);
+
+        result.Set(gameObject);
+    }
+
+    protected override bool SpawnSync(InstalledBatteryEntity entity, TaskResult<Optional<GameObject>> result)
+    {
+        if (!DefaultWorldEntitySpawner.TryGetCachedPrefab(out GameObject prefab, entity.TechType.ToUnity()))
+        {
+            return false;
+        }
+        if (!CanSpawn(entity, out EnergyMixin energyMixin, out string errorLog))
+        {
+            Log.Error(errorLog);
+            return true;
+        }
+
+        GameObject gameObject = GameObjectHelper.SpawnFromPrefab(prefab, entity.Id);
+
+        SetupObject(gameObject, energyMixin);
+
+        result.Set(gameObject);
+        return true;
+    }
+
+    protected override bool SpawnsOwnChildren(InstalledBatteryEntity entity) => false;
+
+    private bool CanSpawn(Entity entity, out EnergyMixin energyMixin, out string errorLog)
+    {
+        if (!NitroxEntity.TryGetObjectFrom(entity.ParentId, out GameObject parentObject))
+        {
+            energyMixin = null;
+            errorLog = $"Unable to find parent to install battery {entity}";
+            return false;
+        }
+
+        energyMixin = parentObject.GetComponent<EnergyMixin>();
+
+        if (!energyMixin)
+        {
+            errorLog = $"Unable to find EnergyMixin on parent to install battery {entity}";
+            return false;
+        }
+        errorLog = null;
+        return true;
+    }
+
+    private void SetupObject(GameObject gameObject, EnergyMixin energyMixin)
+    {
+        energyMixin.Initialize();
+        energyMixin.RestoreBattery();
+
+        using (PacketSuppressor<EntityReparented>.Suppress())
+        using (PacketSuppressor<EntitySpawnedByClient>.Suppress())
+        {
+            energyMixin.batterySlot.AddItem(new InventoryItem(gameObject.GetComponent<Pickupable>()));
+        }
+    }
+}