using Content.Shared.GameTicking; using Content.Shared.Mood; using Content.Shared.Overlays; using Robust.Client.Graphics; using Robust.Shared.Player; namespace Content.Client.Overlays; public sealed class SaturationScaleSystem : EntitySystem { [Dependency] private readonly IOverlayManager _overlayMan = default!; [Dependency] private readonly ISharedPlayerManager _playerMan = default!; private SaturationScaleOverlay _overlay = default!; public override void Initialize() { base.Initialize(); _overlay = new(); SubscribeLocalEvent(OnInit); SubscribeLocalEvent(OnShutdown); SubscribeLocalEvent(OnPlayerAttached); SubscribeLocalEvent(OnPlayerDetached); SubscribeNetworkEvent(RoundRestartCleanup); } private void RoundRestartCleanup(RoundRestartCleanupEvent ev) { _overlayMan.RemoveOverlay(_overlay); } private void OnPlayerDetached(EntityUid uid, SaturationScaleOverlayComponent component, PlayerDetachedEvent args) { _overlayMan.RemoveOverlay(_overlay); } private void OnPlayerAttached(EntityUid uid, SaturationScaleOverlayComponent component, PlayerAttachedEvent args) { _overlayMan.AddOverlay(_overlay); } private void OnShutdown(EntityUid uid, SaturationScaleOverlayComponent component, ComponentShutdown args) { if (uid != _playerMan.LocalEntity) return; _overlayMan.RemoveOverlay(_overlay); } private void OnInit(EntityUid uid, SaturationScaleOverlayComponent component, ComponentInit args) { if (uid != _playerMan.LocalEntity) return; _overlayMan.AddOverlay(_overlay); } }