crystall-punk-14/Content.Client/_CP14/NightVision/CP14ClientNightVisionSystem.cs

using Content.Shared._CP14.NightVision;
using Robust.Client.Player;
using Robust.Shared.Player;

namespace Content.Client._CP14.NightVision;

public sealed class CP14ClientNightVisionSystem : CP14SharedNightVisionSystem
{
    [Dependency] private readonly SharedTransformSystem _transform = default!;
    [Dependency] private readonly IPlayerManager _playerManager = default!;

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

        SubscribeLocalEvent<CP14NightVisionComponent, CP14ToggleNightVisionEvent>(OnToggleNightVision);
        SubscribeLocalEvent<CP14NightVisionComponent, PlayerDetachedEvent>(OnPlayerDetached);
    }

    protected override void OnRemove(Entity<CP14NightVisionComponent> ent, ref ComponentRemove args)
    {
        base.OnRemove(ent, ref args);

        NightVisionOff(ent);
    }

    private void OnPlayerDetached(Entity<CP14NightVisionComponent> ent, ref PlayerDetachedEvent args)
    {
        NightVisionOff(ent);
    }

    private void OnToggleNightVision(Entity<CP14NightVisionComponent> ent, ref CP14ToggleNightVisionEvent args)
    {
        NightVisionToggle(ent);
    }

    private void NightVisionOn(Entity<CP14NightVisionComponent> ent)
    {
        if (_playerManager.LocalSession?.AttachedEntity != ent)
            return;

        var nightVisionLight = Spawn(ent.Comp.LightPrototype, Transform(ent).Coordinates);
        _transform.SetParent(nightVisionLight, ent);
        _transform.SetWorldRotation(nightVisionLight, _transform.GetWorldRotation(ent));
        ent.Comp.LocalLightEntity = nightVisionLight;
    }

    private void NightVisionOff(Entity<CP14NightVisionComponent> ent)
    {
        QueueDel(ent.Comp.LocalLightEntity);
        ent.Comp.LocalLightEntity = null;
    }

    private void NightVisionToggle(Entity<CP14NightVisionComponent> ent)
    {
        if (ent.Comp.LocalLightEntity == null)
        {
            NightVisionOn(ent);
        }
        else
        {
            NightVisionOff(ent);
        }
    }
}