fix

Rxup · Feb 19, 2024 · 48906fb · 48906fb
1 parent 10e173b
commit 48906fb
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/Content.Client/Backmen/Sponsors/SponsorsManager.cs b/Content.Client/Backmen/Sponsors/SponsorsManager.cs
@@ -8,7 +8,6 @@ namespace Content.Client.Backmen.Sponsors;
 public sealed class SponsorsManager : IClientSponsorsManager
 {
     [Dependency] private readonly IClientNetManager _netMgr = default!;
-    [Dependency] private readonly IEntityManager _entityManager = default!;
 
     public void Initialize()
     {
@@ -18,7 +17,7 @@ public void Initialize()
 
     private void RxWhitelist(Shared.Backmen.MsgWhitelist message)
     {
-        _entityManager.System<WL.WhitelistSystem>().Whitelisted = message.Whitelisted;
+        Whitelisted = message.Whitelisted;
     }
 
     private void OnUpdate(MsgSponsorInfo message)
@@ -56,4 +55,5 @@ private void Reset()
 
     public HashSet<string> Prototypes { get; } = new();
     public int Tier { get; private set; } = 0;
+    public bool Whitelisted { get; private set; } = false;
 }
diff --git a/Content.Client/Backmen/WL/WhitelistSystem.cs b/Content.Client/Backmen/WL/WhitelistSystem.cs
@@ -1,11 +1,14 @@
-using Content.Shared.Backmen.WL;
+using Content.Corvax.Interfaces.Client;
+using Content.Shared.Backmen.WL;
 using Content.Shared.Species.Components;
 using Robust.Shared.Network;
 
 namespace Content.Client.Backmen.WL;
 
 public sealed class WhitelistSystem  : SharedWhitelistSystem
 {
+    [Dependency] private readonly IClientSponsorsManager _sponsorsManager = default!;
+
     public override void Initialize()
     {
         base.Initialize();
@@ -16,5 +19,5 @@ public override void ProcessReform(EntityUid child, Entity<ReformComponent> sour
         // noop
     }
 
-    public bool Whitelisted { get; internal set; }
+    public bool Whitelisted => _sponsorsManager.Whitelisted;
 }
diff --git a/Content.Client/Corvax/IClientSponsorsManager.cs b/Content.Client/Corvax/IClientSponsorsManager.cs
@@ -7,4 +7,5 @@ public interface IClientSponsorsManager : ISharedSponsorsManager
 {
     public HashSet<string> Prototypes { get; }
     public int Tier { get; }
+    public bool Whitelisted { get; }
 }