tbd-station-14/Content.Server/Light/EntitySystems/HandHeldLightSystem.cs

using System.Collections.Generic;
using System.Linq;
using Content.Server.Light.Components;
using Content.Shared.Verbs;
using JetBrains.Annotations;
using Robust.Shared.GameObjects;
using Robust.Shared.Localization;

namespace Content.Server.Light.EntitySystems
{
    [UsedImplicitly]
    internal sealed class HandHeldLightSystem : EntitySystem
    {
        // TODO: Ideally you'd be able to subscribe to power stuff to get events at certain percentages.. or something?
        // But for now this will be better anyway.
        private HashSet<HandheldLightComponent> _activeLights = new();

        public override void Initialize()
        {
            base.Initialize();
            SubscribeLocalEvent<ActivateHandheldLightMessage>(HandleActivate);
            SubscribeLocalEvent<DeactivateHandheldLightMessage>(HandleDeactivate);
            SubscribeLocalEvent<HandheldLightComponent, GetActivationVerbsEvent>(AddToggleLightVerb);
        }

        public override void Shutdown()
        {
            base.Shutdown();
            _activeLights.Clear();
        }

        private void HandleActivate(ActivateHandheldLightMessage message)
        {
            _activeLights.Add(message.Component);
        }

        private void HandleDeactivate(DeactivateHandheldLightMessage message)
        {
            _activeLights.Remove(message.Component);
        }

        public override void Update(float frameTime)
        {
            foreach (var handheld in _activeLights.ToArray())
            {
                if (handheld.Deleted || handheld.Paused) continue;
                handheld.OnUpdate(frameTime);
            }
        }

        private void AddToggleLightVerb(EntityUid uid, HandheldLightComponent component, GetActivationVerbsEvent args)
        {
            if (!args.CanAccess || !args.CanInteract)
                return;

            Verb verb = new();
            verb.Text = Loc.GetString("verb-toggle-light");
            verb.IconTexture = "/Textures/Interface/VerbIcons/light.svg.192dpi.png";
            verb.Act = component.Activated
                ? () => component.TurnOff()
                : () => component.TurnOn(args.User);

            args.Verbs.Add(verb);
        }
    }
}