Fix Night Vision

Content.Client/Overlays/EquipmentHudSystem.cs

@@ -93,7 +93,8 @@ private void OnRoundRestart(RoundRestartCleanupEvent args)
 
     protected virtual void OnRefreshEquipmentHud(EntityUid uid, T component, InventoryRelayedEvent<RefreshEquipmentHudEvent<T>> args)
     {
-        OnRefreshComponentHud(uid, component, args.Args);
+        args.Args.Active = true;
+        args.Args.Components.Add(component);
     }
 
     protected virtual void OnRefreshComponentHud(EntityUid uid, T component, RefreshEquipmentHudEvent<T> args)

Content.Client/Overlays/Switchable/NightVisionSystem.cs

@@ -1,3 +1,4 @@
+using Content.Shared.Inventory;
 using Content.Shared.Inventory.Events;
 using Content.Shared.Overlays.Switchable;
 using Robust.Client.Graphics;
@@ -20,6 +21,26 @@ public override void Initialize()
         _overlay = new BaseSwitchableOverlay<NightVisionComponent>();
     }
 
+    protected override void OnRefreshComponentHud(EntityUid uid,
+        NightVisionComponent component,
+        RefreshEquipmentHudEvent<NightVisionComponent> args)
+    {
+        if (component.IsEquipment)
+            return;
+
+        base.OnRefreshComponentHud(uid, component, args);
+    }
+
+    protected override void OnRefreshEquipmentHud(EntityUid uid,
+        NightVisionComponent component,
+        InventoryRelayedEvent<RefreshEquipmentHudEvent<NightVisionComponent>> args)
+    {
+        if (!component.IsEquipment)
+            return;
+
+        base.OnRefreshEquipmentHud(uid, component, args);
+    }
+
     private void OnToggle(Entity<NightVisionComponent> ent, ref SwitchableOverlayToggledEvent args)
     {
         RefreshOverlay(args.User);

Content.Client/Overlays/Switchable/ThermalVisionOverlay.cs

@@ -143,9 +143,9 @@ private bool CanSee(EntityUid uid, SpriteComponent sprite)
                                   _stealth.GetVisibility(uid, stealth) > 0.5f);
     }
 
-    public void ResetLight()
+    public void ResetLight(bool checkFirstTimePredicted = true)
     {
-        if (_lightEntity == null || !_timing.IsFirstTimePredicted)
+        if (_lightEntity == null || checkFirstTimePredicted && !_timing.IsFirstTimePredicted)
             return;
 
         _entity.DeleteEntity(_lightEntity);

Content.Client/Overlays/Switchable/ThermalVisionSystem.cs

@@ -1,3 +1,4 @@
+using Content.Shared.Inventory;
 using Content.Shared.Inventory.Events;
 using Content.Shared.Overlays.Switchable;
 using Robust.Client.Graphics;
@@ -21,6 +22,26 @@ public override void Initialize()
         _overlay = new BaseSwitchableOverlay<ThermalVisionComponent>();
     }
 
+    protected override void OnRefreshComponentHud(EntityUid uid,
+        ThermalVisionComponent component,
+        RefreshEquipmentHudEvent<ThermalVisionComponent> args)
+    {
+        if (component.IsEquipment)
+            return;
+
+        base.OnRefreshComponentHud(uid, component, args);
+    }
+
+    protected override void OnRefreshEquipmentHud(EntityUid uid,
+        ThermalVisionComponent component,
+        InventoryRelayedEvent<RefreshEquipmentHudEvent<ThermalVisionComponent>> args)
+    {
+        if (!component.IsEquipment)
+            return;
+
+        base.OnRefreshEquipmentHud(uid, component, args);
+    }
+
     private void OnToggle(Entity<ThermalVisionComponent> ent, ref SwitchableOverlayToggledEvent args)
     {
         RefreshOverlay(args.User);
@@ -54,6 +75,7 @@ protected override void DeactivateInternal()
     {
         base.DeactivateInternal();
 
+        _thermalOverlay.ResetLight(false);
         UpdateOverlay(null);
         UpdateThermalOverlay(null, 0f);
     }

Content.Shared/Overlays/Switchable/SwitchableOverlayComponent.cs

@@ -12,6 +12,12 @@ public abstract partial class SwitchableOverlayComponent : BaseOverlayComponent
     [DataField]
     public bool DrawOverlay = true;
 
+    /// <summary>
+    /// Whether it should grant equipment enhanced vision or is it mob vision
+    /// </summary>
+    [DataField]
+    public bool IsEquipment;
+
     /// <summary>
     /// If it is greater than 0, overlay isn't toggled but pulsed instead
     /// </summary>

Content.Shared/Overlays/Switchable/SwitchableOverlaySystem.cs

@@ -102,18 +102,22 @@ private void OnHandleState(EntityUid uid, TComp component, ref ComponentHandleSt
 
         component.IsActive = state.IsActive;
 
-        RaiseSwitchableOverlayToggledEvent(uid, uid, component.IsActive);
-        RaiseSwitchableOverlayToggledEvent(uid, Transform(uid).ParentUid, component.IsActive);
+        RaiseSwitchableOverlayToggledEvent(uid,
+            component.IsEquipment ? Transform(uid).ParentUid : uid,
+            component.IsActive);
     }
 
     private void OnGetItemActions(Entity<TComp> ent, ref GetItemActionsEvent args)
     {
-        if (ent.Comp.ToggleAction != null && args.SlotFlags is not SlotFlags.POCKET and not null)
+        if (ent.Comp.IsEquipment && ent.Comp.ToggleAction != null && args.SlotFlags is not SlotFlags.POCKET and not null)
             args.AddAction(ref ent.Comp.ToggleActionEntity, ent.Comp.ToggleAction);
     }
 
     private void OnShutdown(EntityUid uid, TComp component, ComponentShutdown args)
     {
+        if (component.IsEquipment)
+            return;
+
         _actions.RemoveAction(uid, component.ToggleActionEntity);
     }
 
@@ -124,7 +128,7 @@ private void OnInit(EntityUid uid, TComp component, ComponentInit args)
 
     private void OnMapInit(EntityUid uid, TComp component, MapInitEvent args)
     {
-        if (component.ToggleActionEntity == null && component.ToggleAction != null)
+        if (component is { IsEquipment: false, ToggleActionEntity: null, ToggleAction: not null })
             _actions.AddAction(uid, ref component.ToggleActionEntity, component.ToggleAction);
     }
 

Resources/Prototypes/Entities/Clothing/Eyes/glasses.yml

@@ -412,6 +412,7 @@
     sprite: Clothing/Eyes/Glasses/ninjavisor.rsi
   - type: FlashImmunity
   - type: NightVision
+    isEquipment: true
 
 - type: entity #Fake goggles, the latest in anti-valid hunting technology
   parent: ClothingEyesBase

Resources/Prototypes/Entities/Clothing/Eyes/goggles.yml

@@ -11,6 +11,7 @@
   - type: Clothing
     sprite: Clothing/Eyes/Goggles/nightvision.rsi
   - type: NightVision
+    isEquipment: true
   - type: IdentityBlocker
     coverage: EYES
 
@@ -77,6 +78,7 @@
   - type: Clothing
     sprite: Clothing/Eyes/Goggles/thermal.rsi
   - type: ThermalVision
+    isEquipment: true
     pulseTime: 2
     toggleAction: PulseThermalVision
   - type: IdentityBlocker

Resources/Prototypes/Entities/Clothing/Head/hardsuit-helmets.yml

@@ -745,6 +745,7 @@
         Radiation: 0.80
         Caustic: 0.95
   - type: ThermalVision
+    isEquipment: true
     color: "#98EEFB"
     lightRadius: 15
 

Resources/Prototypes/WhiteDream/Entities/Clothing/Cult/armor.yml

@@ -115,6 +115,7 @@
   - type: EyeProtection
   - type: NightVision
     isActive: true
+    isEquipment: true
     toggleAction: null
     activateSound: null
     deactivateSound: null