crystall-punk-14/Content.Client/Chemistry/Components/HyposprayComponent.cs

using Content.Client.Items.Components;
using Content.Client.Message;
using Content.Client.Stylesheets;
using Content.Shared.Chemistry.Components;
using Content.Shared.Chemistry.Reagent;
using Content.Shared.FixedPoint;
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.Chemistry.Components
{
    [RegisterComponent]
    public sealed class HyposprayComponent : SharedHyposprayComponent, IItemStatus
    {
        [ViewVariables] private FixedPoint2 CurrentVolume { get; set; }
        [ViewVariables] private FixedPoint2 TotalVolume { get; set; }
        [ViewVariables(VVAccess.ReadWrite)] private bool _uiUpdateNeeded;

        public override void HandleComponentState(ComponentState? curState, ComponentState? nextState)
        {
            if (curState is not HyposprayComponentState cState)
                return;

            CurrentVolume = cState.CurVolume;
            TotalVolume = cState.MaxVolume;
            _uiUpdateNeeded = true;
        }

        Control IItemStatus.MakeControl()
        {
            return new StatusControl(this);
        }

        private sealed class StatusControl : Control
        {
            private readonly HyposprayComponent _parent;
            private readonly RichTextLabel _label;

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

                Update();
            }

            /// <inheritdoc />
            protected override void FrameUpdate(FrameEventArgs args)
            {
                base.FrameUpdate(args);
                if (!_parent._uiUpdateNeeded)
                {
                    return;
                }
                Update();
            }

            public void Update()
            {

                _parent._uiUpdateNeeded = false;

                _label.SetMarkup(Loc.GetString(
                    "hypospray-volume-text",
                    ("currentVolume", _parent.CurrentVolume),
                    ("totalVolume", _parent.TotalVolume)));
            }
        }
    }
}