Cloning rework (#3808)

* Fix cloning

* Fix cloning after clone dies and remove unneeded code

* Fix ignored

Co-authored-by: Silver <silvertorch5@gmail.com>

Content.Server/GameObjects/EntitySystems/CloningSystem.cs

@@ -1,23 +1,107 @@
-using System.Collections.Generic;
+using System.Collections.Generic;
 using System.Linq;
 using Content.Server.GameObjects.Components.Medical;
+using Content.Server.GameObjects.Components.Mobs;
+using Content.Server.GameObjects.Components.Power.ApcNetComponents;
 using Content.Server.Mobs;
 using Content.Shared.GameTicking;
+using Content.Shared.Interfaces.GameObjects.Components;
+using Robust.Server.GameObjects;
 using Robust.Shared.GameObjects;
+using Robust.Shared.Maths;
+using static Content.Shared.GameObjects.Components.Medical.SharedCloningPodComponent;
 
 namespace Content.Server.GameObjects.EntitySystems
 {
     internal sealed class CloningSystem : EntitySystem, IResettingEntitySystem
     {
+        public readonly Dictionary<int, Mind> Minds = new();
+        public readonly Dictionary<Mind, EntityUid> ClonesWaitingForMind = new();
+
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            SubscribeLocalEvent<CloningPodComponent, ActivateInWorldMessage>(HandleActivate);
+            SubscribeLocalEvent<BeingClonedComponent, MindAddedMessage>(HandleMindAdded);
+        }
+
+        public override void Shutdown()
+        {
+            base.Shutdown();
+
+            UnsubscribeLocalEvent<CloningPodComponent, ActivateInWorldMessage>(HandleActivate);
+            UnsubscribeLocalEvent<BeingClonedComponent, MindAddedMessage>(HandleMindAdded);
+        }
+
+        internal void TransferMindToClone(Mind mind)
+        {
+            if (!ClonesWaitingForMind.TryGetValue(mind, out var entityUid) ||
+                !EntityManager.TryGetEntity(entityUid, out var entity) ||
+                !entity.TryGetComponent(out MindComponent? mindComp) ||
+                mindComp.Mind != null)
+                return;
+
+            mind?.TransferTo(entity);
+            mind?.UnVisit();
+        }
+
+        private void HandleActivate(EntityUid uid, CloningPodComponent component, ActivateInWorldMessage args)
+        {
+            if (!component.Powered ||
+                !args.User.TryGetComponent(out IActorComponent? actor))
+            {
+                return;
+            }
+
+            component.UserInterface?.Open(actor.playerSession);
+        }
+
+        private void HandleMindAdded(EntityUid uid, BeingClonedComponent component, MindAddedMessage message)
+        {
+            if (component.Parent == EntityUid.Invalid ||
+                !EntityManager.TryGetEntity(component.Parent, out var parent) ||
+                !parent.TryGetComponent<CloningPodComponent>(out var cloningPodComponent) ||
+                component.Owner != cloningPodComponent.BodyContainer?.ContainedEntity)
+            {
+                component.Owner.RemoveComponent<BeingClonedComponent>();
+                return;
+            }
+
+            cloningPodComponent.UpdateStatus(CloningPodStatus.Cloning);
+        }
+
         public override void Update(float frameTime)
         {
-            foreach (var comp in ComponentManager.EntityQuery<CloningPodComponent>(true))
+            foreach (var (cloning, power) in ComponentManager.EntityQuery<CloningPodComponent, PowerReceiverComponent>(true))
             {
-                comp.Update(frameTime);
+                if (!power.Powered)
+                    return;
+
+                if (cloning.BodyContainer.ContainedEntity != null)
+                {
+                    cloning.CloningProgress += frameTime;
+                    cloning.CloningProgress = MathHelper.Clamp(cloning.CloningProgress, 0f, cloning.CloningTime);
+                }
+
+                if (cloning.CapturedMind?.Session?.AttachedEntity == cloning.BodyContainer.ContainedEntity)
+                {
+                    cloning.Eject();
+                }
+
+                UpdateUserInterface(cloning);
             }
         }
 
-        public readonly Dictionary<int, Mind> Minds = new();
+        public void UpdateUserInterface(CloningPodComponent comp)
+        {
+            var idToUser = GetIdToUser();
+            comp.UserInterface?.SetState(
+                new CloningPodBoundUserInterfaceState(
+                    idToUser,
+                    comp.CloningProgress,
+                    comp.Status == CloningPodStatus.Cloning));
+        }
 
         public void AddToDnaScans(Mind mind)
         {
@@ -40,6 +124,7 @@ namespace Content.Server.GameObjects.EntitySystems
         public void Reset()
         {
             Minds.Clear();
+            ClonesWaitingForMind.Clear();
         }
     }
 }