Crayons ECS (#6364)

Content.Client/Crayon/CrayonSystem.cs

@@ -0,0 +1,72 @@
+using Content.Client.Items;
+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.GameStates;
+using Robust.Shared.Localization;
+using Robust.Shared.Timing;
+
+namespace Content.Client.Crayon;
+
+public sealed class CrayonSystem : EntitySystem
+{
+    // Didn't do in shared because I don't think most of the server stuff can be predicted.
+    public override void Initialize()
+    {
+        base.Initialize();
+        SubscribeLocalEvent<CrayonComponent, ComponentHandleState>(OnCrayonHandleState);
+        SubscribeLocalEvent<CrayonComponent, ItemStatusCollectMessage>(OnCrayonItemStatus);
+    }
+
+    private static void OnCrayonHandleState(EntityUid uid, CrayonComponent component, ref ComponentHandleState args)
+    {
+        if (args.Current is not CrayonComponentState state) return;
+
+        component._color = state.Color;
+        component.SelectedState = state.State;
+        component.Charges = state.Charges;
+        component.Capacity = state.Capacity;
+
+        component.UIUpdateNeeded = true;
+    }
+
+    private static void OnCrayonItemStatus(EntityUid uid, CrayonComponent component, ItemStatusCollectMessage args)
+    {
+        args.Controls.Add(new StatusControl(component));
+    }
+
+    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)));
+        }
+    }
+}