tbd-station-14/Content.Client/GameObjects/Components/StackComponent.cs

#nullable enable

using Content.Client.UserInterface.Stylesheets;
using Content.Client.Utility;
using Content.Shared.GameObjects.Components;
using Robust.Client.GameObjects;
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
{
    [RegisterComponent]
    [ComponentReference(typeof(SharedStackComponent))]
    public class StackComponent : SharedStackComponent, IItemStatus
    {
        [ViewVariables(VVAccess.ReadWrite)] private bool _uiUpdateNeeded;
        [ComponentDependency] private readonly AppearanceComponent? _appearanceComponent = default!;

        public Control MakeControl() => new StatusControl(this);

        public override int Count
        {
            get => base.Count;
            set
            {
                var valueChanged = value != Count;
                base.Count = value;

                if (valueChanged)
                {
                    _appearanceComponent?.SetData(StackVisuals.Actual, Count);

                }

                _uiUpdateNeeded = true;
            }
        }

        public override void Initialize()
        {
            base.Initialize();

            if (!Owner.Deleted)
            {
                _appearanceComponent?.SetData(StackVisuals.MaxCount, MaxCount);
                _appearanceComponent?.SetData(StackVisuals.Hide, false);
            }
        }

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

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

                parent._uiUpdateNeeded = true;
            }

            protected override void FrameUpdate(FrameEventArgs args)
            {
                base.FrameUpdate(args);

                if (!_parent._uiUpdateNeeded)
                {
                    return;
                }

                _parent._uiUpdateNeeded = false;

                _label.SetMarkup(Loc.GetString("comp-stack-status", ("count", _parent.Count)));
            }
        }
    }
}