tbd-station-14/Content.Client/Light/Components/HandheldLightComponent.cs

using Content.Client.Items.Components;
using Content.Shared.Hands.Components;
using Content.Shared.Light.Component;
using Robust.Client.Graphics;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.Controls;
using Robust.Shared.Analyzers;
using Robust.Shared.GameObjects;
using Robust.Shared.Maths;
using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.Timing;
using System.Collections.Generic;
using static Robust.Client.UserInterface.Controls.BoxContainer;
using static Robust.Shared.GameObjects.SharedSpriteComponent;

namespace Content.Client.Light.Components
{
    [RegisterComponent]
    [Friend(typeof(HandheldLightSystem))]
    public sealed class HandheldLightComponent : SharedHandheldLightComponent, IItemStatus
    {
        public byte? Level;
        public bool Activated;

        /// <summary>
        ///     Whether to automatically set item-prefixes when toggling the flashlight.
        /// </summary>
        /// <remarks>
        ///     Flashlights should probably be using explicit unshaded sprite, in-hand and clothing layers, this is
        ///     mostly here for backwards compatibility.
        /// </remarks>
        [DataField("addPrefix")]
        public bool AddPrefix = false;

        /// <summary>
        ///     Sprite layer that will have it's visibility toggled when this item is toggled.
        /// </summary>
        [DataField("layer")]
        public string Layer = "light";

        /// <summary>
        ///     Layers to add to the sprite of the player that is holding this entity.
        /// </summary>
        [DataField("inhandVisuals")]
        public Dictionary<HandLocation, List<PrototypeLayerData>> InhandVisuals = new();

        /// <summary>
        ///     Layers to add to the sprite of the player that is wearing this entity.
        /// </summary>
        [DataField("clothingVisuals")]
        public readonly Dictionary<string, List<PrototypeLayerData>> ClothingVisuals = new();

        public Color Color { get; internal set; }

        public Control MakeControl()
        {
            return new StatusControl(this);
        }

        private sealed class StatusControl : Control
        {
            private const float TimerCycle = 1;

            private readonly HandheldLightComponent _parent;
            private readonly PanelContainer[] _sections = new PanelContainer[StatusLevels - 1];

            private float _timer;

            private static readonly StyleBoxFlat StyleBoxLit = new()
            {
                BackgroundColor = Color.LimeGreen
            };

            private static readonly StyleBoxFlat StyleBoxUnlit = new()
            {
                BackgroundColor = Color.Black
            };

            public StatusControl(HandheldLightComponent parent)
            {
                _parent = parent;

                var wrapper = new BoxContainer
                {
                    Orientation = LayoutOrientation.Horizontal,
                    SeparationOverride = 4,
                    HorizontalAlignment = HAlignment.Center
                };

                AddChild(wrapper);

                for (var i = 0; i < _sections.Length; i++)
                {
                    var panel = new PanelContainer {MinSize = (20, 20)};
                    wrapper.AddChild(panel);
                    _sections[i] = panel;
                }
            }

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

                _timer += args.DeltaSeconds;
                _timer %= TimerCycle;

                var level = _parent.Level;

                for (var i = 0; i < _sections.Length; i++)
                {
                    if (i == 0)
                    {
                        if (level == 0 || level == null)
                        {
                            _sections[0].PanelOverride = StyleBoxUnlit;
                        }
                        else if (level == 1)
                        {
                            // Flash the last light.
                            _sections[0].PanelOverride = _timer > TimerCycle / 2 ? StyleBoxLit : StyleBoxUnlit;
                        }
                        else
                        {
                            _sections[0].PanelOverride = StyleBoxLit;
                        }

                        continue;
                    }

                    _sections[i].PanelOverride = level >= i + 2 ? StyleBoxLit : StyleBoxUnlit;
                }
            }
        }
    }
}