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 Content.Shared.Notification.Managers; 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; protected override void Initialize() { base.Initialize(); if (UserInterface != null) { UserInterface.OnReceiveMessage += OnUiReceiveMessage; } BodyContainer = ContainerHelpers.EnsureContainer(Owner, $"{Name}-bodyContainer"); //TODO: write this so that it checks for a change in power events for GORE POD cases EntitySystem.Get().UpdateUserInterface(this); } protected 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(); 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(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(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().UpdateFromProfile(dna.Profile); mob.Name = dna.Profile.Name; var cloneMindReturn = mob.AddComponent(); 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(); BodyContainer.Remove(entity!); CapturedMind = null; CloningProgress = 0f; UpdateStatus(CloningPodStatus.Idle); } public void UpdateStatus(CloningPodStatus status) { Status = status; UpdateAppearance(); EntitySystem.Get().UpdateUserInterface(this); } } }