tbd-station-14/Content.Client/GameObjects/Components/Chemistry/InjectorComponent.cs

using Content.Client.UserInterface.Stylesheets;
using Content.Client.Utility;
using Content.Shared.Chemistry;
using Content.Shared.GameObjects.Components.Chemistry;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.Controls;
using Robust.Shared.GameObjects;
using Robust.Shared.Localization;
using Robust.Shared.Timing;
using Robust.Shared.ViewVariables;

namespace Content.Client.GameObjects.Components.Chemistry
{
    /// <summary>
    /// Client behavior for injectors & syringes. Used for item status on injectors
    /// </summary>
    [RegisterComponent]
    public class InjectorComponent : SharedInjectorComponent, IItemStatus
    {
        [ViewVariables] private ReagentUnit CurrentVolume { get; set; }
        [ViewVariables] private ReagentUnit TotalVolume { get; set; }
        [ViewVariables] private InjectorToggleMode CurrentMode { get; set; }
        [ViewVariables(VVAccess.ReadWrite)] private bool _uiUpdateNeeded;

        //Add/remove item status code
        Control IItemStatus.MakeControl() => new StatusControl(this);
        void IItemStatus.DestroyControl(Control control) { }

        //Handle net updates
        public override void HandleComponentState(ComponentState curState, ComponentState nextState)
        {
            if (curState is not InjectorComponentState state)
            {
                return;
            }

            CurrentVolume = state.CurrentVolume;
            TotalVolume = state.TotalVolume;
            CurrentMode = state.CurrentMode;
            _uiUpdateNeeded = true;
        }

        /// <summary>
        /// Item status control for injectors
        /// </summary>
        private sealed class StatusControl : Control
        {
            private readonly InjectorComponent _parent;
            private readonly RichTextLabel _label;

            public StatusControl(InjectorComponent parent)
            {
                _parent = parent;
                _label = new RichTextLabel { StyleClasses = { StyleNano.StyleClassItemStatus } };
                AddChild(_label);

                parent._uiUpdateNeeded = true;
            }

            protected override void Update(FrameEventArgs args)
            {
                base.Update(args);
                if (!_parent._uiUpdateNeeded)
                {
                    return;
                }

                _parent._uiUpdateNeeded = false;

                //Update current volume and injector state
                var modeStringLocalized = _parent.CurrentMode switch
                {
                    InjectorToggleMode.Draw => Loc.GetString("Draw"),
                    InjectorToggleMode.Inject => Loc.GetString("Inject"),
                    _ => Loc.GetString("Invalid")
                };
                _label.SetMarkup(Loc.GetString("Volume: [color=white]{0}/{1}[/color] | [color=white]{2}[/color]",
                    _parent.CurrentVolume, _parent.TotalVolume, modeStringLocalized));
            }
        }
    }
}