fish-station/Content.Client/_Sunrise/Eye/DarkenedVisionSystem.cs

63 lines
No EOL
1.9 KiB
C#

using Robust.Client.Graphics;
using Robust.Shared.Player;
using Robust.Client.Player;
using Content.Client.Overlays;
using Content.Shared.Eye;
using Content.Client.Eye;
using Content.Client.Sunrise.Overlays;
using Content.Client.Sunrise.Eye;
using Content.Shared.Sunrise.Eye;
namespace Content.Client.Sunrise.Eye;
public sealed class DarkenedVisionSystem : SharedDarkenedVisionSystem
{
[Dependency] private readonly IPlayerManager _player = default!;
[Dependency] private readonly IOverlayManager _overlayMan = default!;
private DarkenedVisionOverlay _overlay = default!;
public override void Initialize()
{
base.Initialize();
SubscribeLocalEvent<DarkenedVisionComponent, ComponentStartup>(OnStartup);
SubscribeLocalEvent<DarkenedVisionComponent, ComponentShutdown>(OnShutdown);
SubscribeLocalEvent<DarkenedVisionComponent, LocalPlayerAttachedEvent>(OnAttached);
SubscribeLocalEvent<DarkenedVisionComponent, LocalPlayerDetachedEvent>(OnDetached);
_overlay = new();
}
private void OnAttached(Entity<DarkenedVisionComponent> ent, ref LocalPlayerAttachedEvent args)
{
_overlayMan.AddOverlay(_overlay);
_overlay.DarkenedVision = ent.Comp;
}
private void OnDetached(Entity<DarkenedVisionComponent> ent, ref LocalPlayerDetachedEvent args)
{
_overlayMan.RemoveOverlay(_overlay);
_overlay.DarkenedVision = null;
}
private void OnStartup(Entity<DarkenedVisionComponent> ent, ref ComponentStartup args)
{
if (_player.LocalEntity == ent.Owner)
{
_overlayMan.AddOverlay(_overlay);
_overlay.DarkenedVision = ent.Comp;
}
}
private void OnShutdown(Entity<DarkenedVisionComponent> ent, ref ComponentShutdown args)
{
if (_player.LocalEntity == ent.Owner)
{
_overlayMan.RemoveOverlay(_overlay);
_overlay.DarkenedVision = null;
}
}
}