using System; using System.Collections.Generic; using System.Linq; using Content.Server.Chemistry.Components.SolutionManager; using Content.Server.Chemistry.EntitySystems; using Content.Server.Hands.Components; using Content.Server.Power.Components; using Content.Server.UserInterface; using Content.Shared.ActionBlocker; using Content.Shared.Chemistry.Components; using Content.Shared.Chemistry.Dispenser; using Content.Shared.Containers.ItemSlots; using Content.Shared.FixedPoint; using Content.Shared.Interaction; using Content.Shared.Popups; using Content.Shared.Sound; using JetBrains.Annotations; using Robust.Server.GameObjects; using Robust.Shared.Audio; using Robust.Shared.GameObjects; using Robust.Shared.IoC; using Robust.Shared.Localization; using Robust.Shared.Log; using Robust.Shared.Player; using Robust.Shared.Prototypes; using Robust.Shared.Serialization.Manager.Attributes; using Robust.Shared.ViewVariables; namespace Content.Server.Chemistry.Components { /// /// Contains all the server-side logic for reagent dispensers. See also . /// This includes initializing the component based on prototype data, and sending and receiving messages from the client. /// Messages sent to the client are used to update update the user interface for a component instance. /// Messages sent from the client are used to handle ui button presses. /// [RegisterComponent] [ComponentReference(typeof(IActivate))] [ComponentReference(typeof(SharedReagentDispenserComponent))] public class ReagentDispenserComponent : SharedReagentDispenserComponent, IActivate { private static ReagentInventoryComparer _comparer = new(); public static string SolutionName = "reagent"; [Dependency] private readonly IPrototypeManager _prototypeManager = default!; [ViewVariables] [DataField("pack")] private string _packPrototypeId = ""; [DataField("clickSound")] private SoundSpecifier _clickSound = new SoundPathSpecifier("/Audio/Machines/machine_switch.ogg"); [ViewVariables] private FixedPoint2 _dispenseAmount = FixedPoint2.New(10); [UsedImplicitly] [ViewVariables] private Solution? Solution { get { EntitySystem.Get().TryGetSolution(Owner.Uid, SolutionName, out var solution); return solution; } } [ViewVariables] private bool Powered => !IoCManager.Resolve().TryGetComponent(Owner.Uid, out ApcPowerReceiverComponent? receiver) || receiver.Powered; [ViewVariables] private BoundUserInterface? UserInterface => Owner.GetUIOrNull(ReagentDispenserUiKey.Key); /// /// Called once per instance of this component. Gets references to any other components needed /// by this component and initializes it's UI and other data. /// protected override void Initialize() { base.Initialize(); if (UserInterface != null) { UserInterface.OnReceiveMessage += OnUiReceiveMessage; } InitializeFromPrototype(); } [Obsolete("Component Messages are deprecated, use Entity Events instead.")] public override void HandleMessage(ComponentMessage message, IComponent? component) { #pragma warning disable 618 base.HandleMessage(message, component); #pragma warning restore 618 switch (message) { case PowerChangedMessage powerChanged: OnPowerChanged(powerChanged); break; } } /// /// Checks to see if the pack defined in this components yaml prototype /// exists. If so, it fills the reagent inventory list. /// private void InitializeFromPrototype() { if (string.IsNullOrEmpty(_packPrototypeId)) return; if (!_prototypeManager.TryIndex(_packPrototypeId, out ReagentDispenserInventoryPrototype? packPrototype)) { return; } foreach (var entry in packPrototype.Inventory) { Inventory.Add(new ReagentDispenserInventoryEntry(entry)); } Inventory.Sort(_comparer); } private void OnPowerChanged(PowerChangedMessage e) { UpdateUserInterface(); } /// /// Handles ui messages from the client. For things such as button presses /// which interact with the world and require server action. /// /// A user interface message from the client. private void OnUiReceiveMessage(ServerBoundUserInterfaceMessage obj) { if (obj.Session.AttachedEntity == null) { return; } var msg = (UiButtonPressedMessage) obj.Message; var needsPower = msg.Button switch { UiButton.Eject => false, _ => true, }; if (!PlayerCanUseDispenser(obj.Session.AttachedEntity, needsPower)) return; switch (msg.Button) { case UiButton.Eject: EntitySystem.Get().TryEjectToHands(OwnerUid, BeakerSlot, obj.Session.AttachedEntityUid); break; case UiButton.Clear: TryClear(); break; case UiButton.SetDispenseAmount1: _dispenseAmount = FixedPoint2.New(1); break; case UiButton.SetDispenseAmount5: _dispenseAmount = FixedPoint2.New(5); break; case UiButton.SetDispenseAmount10: _dispenseAmount = FixedPoint2.New(10); break; case UiButton.SetDispenseAmount15: _dispenseAmount = FixedPoint2.New(15); break; case UiButton.SetDispenseAmount20: _dispenseAmount = FixedPoint2.New(20); break; case UiButton.SetDispenseAmount25: _dispenseAmount = FixedPoint2.New(25); break; case UiButton.SetDispenseAmount30: _dispenseAmount = FixedPoint2.New(30); break; case UiButton.SetDispenseAmount50: _dispenseAmount = FixedPoint2.New(50); break; case UiButton.SetDispenseAmount100: _dispenseAmount = FixedPoint2.New(100); break; case UiButton.Dispense: if (BeakerSlot.HasItem) { TryDispense(msg.DispenseIndex); } Logger.Info($"User {obj.Session.UserId.UserId} ({obj.Session.Name}) dispensed {_dispenseAmount}u of {Inventory[msg.DispenseIndex].ID}"); break; default: throw new ArgumentOutOfRangeException(); } ClickSound(); } /// /// Checks whether the player entity is able to use the chem dispenser. /// /// The player entity. /// Returns true if the entity can use the dispenser, and false if it cannot. private bool PlayerCanUseDispenser(IEntity? playerEntity, bool needsPower = true) { //Need player entity to check if they are still able to use the dispenser if (playerEntity == null) return false; var actionBlocker = EntitySystem.Get(); //Check if player can interact in their current state if (!actionBlocker.CanInteract(playerEntity.Uid) || !actionBlocker.CanUse(playerEntity.Uid)) return false; //Check if device is powered if (needsPower && !Powered) return false; return true; } /// /// Gets component data to be used to update the user interface client-side. /// /// Returns a private ReagentDispenserBoundUserInterfaceState GetUserInterfaceState() { var beaker = BeakerSlot.Item; if (beaker == null || !IoCManager.Resolve().TryGetComponent(beaker.Uid, out FitsInDispenserComponent? fits) || !EntitySystem.Get().TryGetSolution(beaker.Uid, fits.Solution, out var solution)) { return new ReagentDispenserBoundUserInterfaceState(Powered, false, FixedPoint2.New(0), FixedPoint2.New(0), string.Empty, Inventory, IoCManager.Resolve().GetComponent(Owner.Uid).EntityName, null, _dispenseAmount); } return new ReagentDispenserBoundUserInterfaceState(Powered, true, solution.CurrentVolume, solution.MaxVolume, IoCManager.Resolve().GetComponent(beaker.Uid).EntityName, Inventory, IoCManager.Resolve().GetComponent(Owner.Uid).EntityName, solution.Contents.ToList(), _dispenseAmount); } public void UpdateUserInterface() { var state = GetUserInterfaceState(); UserInterface?.SetState(state); } /// /// If this component contains an entity with a , remove all of it's reagents / solutions. /// private void TryClear() { var beaker = BeakerSlot.Item; if (beaker == null || !IoCManager.Resolve().TryGetComponent(beaker.Uid, out FitsInDispenserComponent? fits) || !EntitySystem.Get() .TryGetSolution(beaker.Uid, fits.Solution, out var solution)) return; EntitySystem.Get().RemoveAllSolution(beaker.Uid, solution); UpdateUserInterface(); } /// /// If this component contains an entity with a , attempt to dispense the specified reagent to it. /// /// The index of the reagent in Inventory. private void TryDispense(int dispenseIndex) { var beaker = BeakerSlot.Item; if (beaker is null || !IoCManager.Resolve().TryGetComponent(beaker.Uid, out FitsInDispenserComponent? fits) || !EntitySystem.Get() .TryGetSolution(beaker.Uid, fits.Solution, out var solution)) return; EntitySystem.Get() .TryAddReagent(beaker.Uid, solution, Inventory[dispenseIndex].ID, _dispenseAmount, out _); UpdateUserInterface(); } /// /// Called when you click the owner entity with an empty hand. Opens the UI client-side if possible. /// /// Data relevant to the event such as the actor which triggered it. void IActivate.Activate(ActivateEventArgs args) { if (!IoCManager.Resolve().TryGetComponent(args.User.Uid, out ActorComponent? actor)) { return; } if (!IoCManager.Resolve().TryGetComponent(args.User.Uid, out HandsComponent? hands)) { Owner.PopupMessage(args.User, Loc.GetString("reagent-dispenser-component-activate-no-hands")); return; } var activeHandEntity = hands.GetActiveHand?.Owner; if (activeHandEntity == null) { UserInterface?.Open(actor.PlayerSession); } } private void ClickSound() { SoundSystem.Play(Filter.Pvs(Owner), _clickSound.GetSound(), Owner, AudioParams.Default.WithVolume(-2f)); } private class ReagentInventoryComparer : Comparer { public override int Compare(ReagentDispenserInventoryEntry x, ReagentDispenserInventoryEntry y) { return string.Compare(x.ID, y.ID, StringComparison.InvariantCultureIgnoreCase); } } } }