Re-organize all projects (#4166)

Content.Server/Cloning/Components/CloningPodComponent.cs

@@ -0,0 +1,186 @@
+using System;
+using Content.Server.CharacterAppearance.Components;
+using Content.Server.EUI;
+using Content.Server.Mind.Components;
+using Content.Server.Power.Components;
+using Content.Server.UserInterface;
+using Content.Shared.Cloning;
+using Content.Shared.MobState;
+using Content.Shared.Notification;
+using Robust.Server.GameObjects;
+using Robust.Server.Player;
+using Robust.Shared.Containers;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Localization;
+using Robust.Shared.Serialization.Manager.Attributes;
+using Robust.Shared.ViewVariables;
+
+namespace Content.Server.Cloning.Components
+{
+    [RegisterComponent]
+    public class CloningPodComponent : SharedCloningPodComponent
+    {
+        [Dependency] private readonly IPlayerManager _playerManager = null!;
+        [Dependency] private readonly EuiManager _euiManager = null!;
+
+        [ViewVariables]
+        public bool Powered => !Owner.TryGetComponent(out PowerReceiverComponent? receiver) || receiver.Powered;
+
+        [ViewVariables]
+        public BoundUserInterface? UserInterface =>
+            Owner.GetUIOrNull(CloningPodUIKey.Key);
+
+        [ViewVariables] public ContainerSlot BodyContainer = default!;
+        [ViewVariables] public Mind.Mind? CapturedMind;
+        [ViewVariables] public float CloningProgress = 0;
+        [DataField("cloningTime")]
+        [ViewVariables] public float CloningTime = 30f;
+        // Used to prevent as many duplicate UI messages as possible
+        [ViewVariables] public bool UiKnownPowerState = false;
+
+        [ViewVariables]
+        public CloningPodStatus Status;
+
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            if (UserInterface != null)
+            {
+                UserInterface.OnReceiveMessage += OnUiReceiveMessage;
+            }
+
+            BodyContainer = ContainerHelpers.EnsureContainer<ContainerSlot>(Owner, $"{Name}-bodyContainer");
+
+            //TODO: write this so that it checks for a change in power events for GORE POD cases
+            EntitySystem.Get<CloningSystem>().UpdateUserInterface(this);
+        }
+
+        public override void OnRemove()
+        {
+            if (UserInterface != null)
+            {
+                UserInterface.OnReceiveMessage -= OnUiReceiveMessage;
+            }
+
+            base.OnRemove();
+        }
+
+        private void UpdateAppearance()
+        {
+            if (Owner.TryGetComponent(out AppearanceComponent? appearance))
+            {
+                appearance.SetData(CloningPodVisuals.Status, Status);
+            }
+        }
+
+        private void OnUiReceiveMessage(ServerBoundUserInterfaceMessage obj)
+        {
+            if (obj.Message is not CloningPodUiButtonPressedMessage message) return;
+
+            switch (message.Button)
+            {
+                case UiButton.Clone:
+                    if (BodyContainer.ContainedEntity != null)
+                    {
+                        obj.Session.AttachedEntity?.PopupMessageCursor(Loc.GetString("cloning-pod-component-msg-occupied"));
+                        return;
+                    }
+
+                    if (message.ScanId == null)
+                    {
+                        obj.Session.AttachedEntity?.PopupMessageCursor(Loc.GetString("cloning-pod-component-msg-no-selection"));
+                        return;
+                    }
+
+                    var cloningSystem = EntitySystem.Get<CloningSystem>();
+
+                    if (!cloningSystem.IdToDNA.TryGetValue(message.ScanId.Value, out var dna))
+                    {
+                        obj.Session.AttachedEntity?.PopupMessageCursor(Loc.GetString("cloning-pod-component-msg-bad-selection"));
+                        return; // ScanId is not in database
+                    }
+
+                    var mind = dna.Mind;
+
+                    if (cloningSystem.ClonesWaitingForMind.TryGetValue(mind, out var cloneUid))
+                    {
+                        if (Owner.EntityManager.TryGetEntity(cloneUid, out var clone) &&
+                            clone.TryGetComponent<IMobStateComponent>(out var cloneState) &&
+                            !cloneState.IsDead() &&
+                            clone.TryGetComponent(out MindComponent? cloneMindComp) &&
+                            (cloneMindComp.Mind == null || cloneMindComp.Mind == mind))
+                        {
+                            obj.Session.AttachedEntity?.PopupMessageCursor(Loc.GetString("cloning-pod-component-msg-already-cloning"));
+                            return; // Mind already has clone
+                        }
+
+                        cloningSystem.ClonesWaitingForMind.Remove(mind);
+                    }
+
+                    if (mind.OwnedEntity != null &&
+                        mind.OwnedEntity.TryGetComponent<IMobStateComponent>(out var state) &&
+                        !state.IsDead())
+                    {
+                        obj.Session.AttachedEntity?.PopupMessageCursor(Loc.GetString("cloning-pod-component-msg-already-alive"));
+                        return; // Body controlled by mind is not dead
+                    }
+
+                    // Yes, we still need to track down the client because we need to open the Eui
+                    if (mind.UserId == null || !_playerManager.TryGetSessionById(mind.UserId.Value, out var client))
+                    {
+                        obj.Session.AttachedEntity?.PopupMessageCursor(Loc.GetString("cloning-pod-component-msg-user-offline"));
+                        return; // If we can't track down the client, we can't offer transfer. That'd be quite bad.
+                    }
+
+                    var mob = Owner.EntityManager.SpawnEntity("HumanMob_Content", Owner.Transform.MapPosition);
+
+                    mob.GetComponent<HumanoidAppearanceComponent>().UpdateFromProfile(dna.Profile);
+                    mob.Name = dna.Profile.Name;
+
+                    var cloneMindReturn = mob.AddComponent<BeingClonedComponent>();
+                    cloneMindReturn.Mind = mind;
+                    cloneMindReturn.Parent = Owner.Uid;
+
+                    BodyContainer.Insert(mob);
+                    CapturedMind = mind;
+                    cloningSystem.ClonesWaitingForMind.Add(mind, mob.Uid);
+
+                    UpdateStatus(CloningPodStatus.NoMind);
+
+                    var acceptMessage = new AcceptCloningEui(mind);
+                    _euiManager.OpenEui(acceptMessage, client);
+
+                    break;
+
+                case UiButton.Eject:
+                    Eject();
+                    break;
+
+                default:
+                    throw new ArgumentOutOfRangeException();
+            }
+        }
+
+        public void Eject()
+        {
+            var entity = BodyContainer.ContainedEntity;
+            if (entity == null || CloningProgress < CloningTime)
+                return;
+
+            entity.RemoveComponent<BeingClonedComponent>();
+            BodyContainer.Remove(entity!);
+            CapturedMind = null;
+            CloningProgress = 0f;
+            UpdateStatus(CloningPodStatus.Idle);
+        }
+
+        public void UpdateStatus(CloningPodStatus status)
+        {
+            Status = status;
+            UpdateAppearance();
+            EntitySystem.Get<CloningSystem>().UpdateUserInterface(this);
+        }
+    }
+}