crystall-punk-14/Content.Client/Crayon/CrayonComponent.cs

using Content.Client.Items.Components;
using Content.Client.Message;
using Content.Client.Stylesheets;
using Content.Shared.Crayon;
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.Crayon
{
    [RegisterComponent]
    public class CrayonComponent : SharedCrayonComponent, IItemStatus
    {
        [ViewVariables(VVAccess.ReadWrite)] private bool _uiUpdateNeeded;
        [ViewVariables(VVAccess.ReadWrite)] private string Color => _color;
        [ViewVariables] private int Charges { get; set; }
        [ViewVariables] private int Capacity { get; set; }

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

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

            _color = state.Color;
            SelectedState = state.State;
            Charges = state.Charges;
            Capacity = state.Capacity;

            _uiUpdateNeeded = true;
        }

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

            public StatusControl(CrayonComponent 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("crayon-drawing-label",
                    ("color",_parent.Color),
                    ("state",_parent.SelectedState),
                    ("charges", _parent.Charges),
                    ("capacity",_parent.Capacity)));
            }
        }
    }
}