using System; using System.Collections.Generic; using Content.Server.Body; using Content.Server.Body.Mechanisms; using Content.Shared.Body.Mechanism; using Content.Shared.Body.Surgery; using Content.Shared.Interfaces; using Content.Shared.Interfaces.GameObjects.Components; using Robust.Server.GameObjects; using Robust.Server.GameObjects.Components.UserInterface; using Robust.Server.Interfaces.Player; using Robust.Shared.GameObjects; using Robust.Shared.Interfaces.GameObjects; using Robust.Shared.IoC; using Robust.Shared.Localization; using Robust.Shared.Log; using Robust.Shared.Prototypes; using Robust.Shared.Serialization; using Robust.Shared.ViewVariables; namespace Content.Server.GameObjects.Components.Body { /// /// Component representing a dropped, tangible entity. /// [RegisterComponent] public class DroppedMechanismComponent : Component, IAfterInteract { #pragma warning disable 649 [Dependency] private readonly ISharedNotifyManager _sharedNotifyManager; [Dependency] private IPrototypeManager _prototypeManager; #pragma warning restore 649 public sealed override string Name => "DroppedMechanism"; private readonly Dictionary _optionsCache = new Dictionary(); private BodyManagerComponent _bodyManagerComponentCache; private int _idHash; private IEntity _performerCache; private BoundUserInterface _userInterface; [ViewVariables] public Mechanism ContainedMechanism { get; private set; } void IAfterInteract.AfterInteract(AfterInteractEventArgs eventArgs) { if (eventArgs.Target == null) { return; } CloseAllSurgeryUIs(); _optionsCache.Clear(); _performerCache = null; _bodyManagerComponentCache = null; if (eventArgs.Target.TryGetComponent(out var bodyManager)) { SendBodyPartListToUser(eventArgs, bodyManager); } else if (eventArgs.Target.TryGetComponent(out var droppedBodyPart)) { if (droppedBodyPart.ContainedBodyPart == null) { Logger.Debug( "Installing a mechanism was attempted on an IEntity with a DroppedBodyPartComponent that doesn't have a BodyPart in it!"); throw new InvalidOperationException("A DroppedBodyPartComponent exists without a BodyPart in it!"); } if (!droppedBodyPart.ContainedBodyPart.TryInstallDroppedMechanism(this)) { _sharedNotifyManager.PopupMessage(eventArgs.Target, eventArgs.User, Loc.GetString("You can't fit it in!")); } } } public override void Initialize() { base.Initialize(); _userInterface = Owner.GetComponent() .GetBoundUserInterface(GenericSurgeryUiKey.Key); _userInterface.OnReceiveMessage += UserInterfaceOnOnReceiveMessage; } public void InitializeDroppedMechanism(Mechanism data) { ContainedMechanism = data; Owner.Name = Loc.GetString(ContainedMechanism.Name); if (Owner.TryGetComponent(out SpriteComponent component)) { component.LayerSetRSI(0, data.RSIPath); component.LayerSetState(0, data.RSIState); } } public override void ExposeData(ObjectSerializer serializer) { // This is a temporary way to have spawnable hard-coded DroppedMechanismComponent prototypes // In the future (when it becomes possible) DroppedMechanismComponent should be auto-generated from // the Mechanism prototypes var debugLoadMechanismData = ""; base.ExposeData(serializer); serializer.DataField(ref debugLoadMechanismData, "debugLoadMechanismData", ""); if (serializer.Reading && debugLoadMechanismData != "") { _prototypeManager.TryIndex(debugLoadMechanismData, out MechanismPrototype data); var mechanism = new Mechanism(data); mechanism.EnsureInitialize(); InitializeDroppedMechanism(mechanism); } } private void SendBodyPartListToUser(AfterInteractEventArgs eventArgs, BodyManagerComponent bodyManager) { // Create dictionary to send to client (text to be shown : data sent back if selected) var toSend = new Dictionary(); foreach (var (key, value) in bodyManager.Parts) { // For each limb in the target, add it to our cache if it is a valid option. if (value.CanInstallMechanism(ContainedMechanism)) { _optionsCache.Add(_idHash, value); toSend.Add(key + ": " + value.Name, _idHash++); } } if (_optionsCache.Count > 0) { OpenSurgeryUI(eventArgs.User.GetComponent().playerSession); UpdateSurgeryUIBodyPartRequest(eventArgs.User.GetComponent().playerSession, toSend); _performerCache = eventArgs.User; _bodyManagerComponentCache = bodyManager; } else // If surgery cannot be performed, show message saying so. { _sharedNotifyManager.PopupMessage(eventArgs.Target, eventArgs.User, Loc.GetString("You see no way to install the {0}.", Owner.Name)); } } /// /// Called after the client chooses from a list of possible BodyParts that can be operated on. /// private void HandleReceiveBodyPart(int key) { CloseSurgeryUI(_performerCache.GetComponent().playerSession); // TODO: sanity checks to see whether user is in range, user is still able-bodied, target is still the same, etc etc if (!_optionsCache.TryGetValue(key, out var targetObject)) { _sharedNotifyManager.PopupMessage(_bodyManagerComponentCache.Owner, _performerCache, Loc.GetString("You see no useful way to use the {0} anymore.", Owner.Name)); return; } var target = targetObject as BodyPart; _sharedNotifyManager.PopupMessage( _bodyManagerComponentCache.Owner, _performerCache, !target.TryInstallDroppedMechanism(this) ? Loc.GetString("You can't fit it in!") : Loc.GetString("You jam the {1} inside {0:them}.", _performerCache, ContainedMechanism.Name)); // TODO: {1:theName} } private void OpenSurgeryUI(IPlayerSession session) { _userInterface.Open(session); } private void UpdateSurgeryUIBodyPartRequest(IPlayerSession session, Dictionary options) { _userInterface.SendMessage(new RequestBodyPartSurgeryUIMessage(options), session); } private void CloseSurgeryUI(IPlayerSession session) { _userInterface.Close(session); } private void CloseAllSurgeryUIs() { _userInterface.CloseAll(); } private void UserInterfaceOnOnReceiveMessage(ServerBoundUserInterfaceMessage message) { switch (message.Message) { case ReceiveBodyPartSurgeryUIMessage msg: HandleReceiveBodyPart(msg.SelectedOptionId); break; } } } }