diff --git a/Content.Server/Materials/MaterialSiloSystem.cs b/Content.Server/Materials/MaterialSiloSystem.cs index 649a82f7af..1573da2313 100644 --- a/Content.Server/Materials/MaterialSiloSystem.cs +++ b/Content.Server/Materials/MaterialSiloSystem.cs @@ -4,6 +4,7 @@ using Content.Server.Station.Components; using Content.Shared.DeviceLinking; using Content.Shared.Lathe; using Content.Shared.Materials; +using Robust.Server.GameStates; using Robust.Shared.Timing; namespace Content.Server.Materials; @@ -11,6 +12,7 @@ namespace Content.Server.Materials; public sealed class MaterialSiloSystem : SharedMaterialSiloSystem { [Dependency] private readonly LatheSystem _lathe = default!; + [Dependency] private readonly PvsOverrideSystem _pvs = default!; public override void Initialize() { @@ -18,6 +20,8 @@ public sealed class MaterialSiloSystem : SharedMaterialSiloSystem SubscribeLocalEvent(OnMapInit); SubscribeLocalEvent(OnMaterialAmountChanged); + SubscribeLocalEvent(OnStartup); + SubscribeLocalEvent(OnShutdown); } private void OnMaterialAmountChanged(Entity ent, ref MaterialAmountChangedEvent args) @@ -67,4 +71,14 @@ public sealed class MaterialSiloSystem : SharedMaterialSiloSystem DeviceLink.LinkDefaults(null, silo.Value, utilizer, silo.Value.Comp, sink); } } + + private void OnStartup(Entity ent, ref ComponentStartup args) + { + _pvs.AddGlobalOverride(ent); + } + + private void OnShutdown(Entity ent, ref ComponentShutdown args) + { + _pvs.RemoveGlobalOverride(ent); + } }