Update LanguageMenuUIController.cs

Simple-Station · May 1, 2024 · e66df36 · e66df36
1 parent a938832
commit e66df36
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/Content.Client/UserInterface/Systems/Language/LanguageMenuUIController.cs b/Content.Client/UserInterface/Systems/Language/LanguageMenuUIController.cs
@@ -12,13 +12,15 @@
 using Robust.Shared.Input.Binding;
 using Robust.Shared.Utility;
 using static Robust.Client.UserInterface.Controls.BaseButton;
+using Robust.Client.GameObjects;
 
 namespace Content.Client.UserInterface.Systems.Language;
 
 [UsedImplicitly]
 public sealed class LanguageMenuUIController : UIController, IOnStateEntered<GameplayState>, IOnStateExited<GameplayState>
 {
     public LanguageMenuWindow? _languageWindow;
+    private IEntityManager _entManager;
     private MenuButton? LanguageButton => UIManager.GetActiveUIWidgetOrNull<MenuBar.Widgets.GameTopMenuBar>()?.LanguageButton;
 
     /// <summary>
@@ -27,6 +29,11 @@ public sealed class LanguageMenuUIController : UIController, IOnStateEntered<Gam
     /// </summary>
     public Action<(string current, List<string> spoken, List<string> understood)>? LanguagesUpdatedHook;
 
+    public LanguageMenuUIController()
+    {
+        _entManager = IoCManager.Resolve<IEntityManager>();
+    }
+
     public override void Initialize()
     {
         LanguagesUpdatedHook += (args) =>
@@ -42,8 +49,7 @@ public void OnStateEntered(GameplayState state)
     {
         DebugTools.Assert(_languageWindow == null);
 
-        var entManager = IoCManager.Resolve<IEntityManager>();
-        var clientLanguageSystem = entManager.EntitySysManager.GetEntitySystem<LanguageSystem>();
+        var clientLanguageSystem = _entManager.EntitySysManager.GetEntitySystem<LanguageSystem>();
         clientLanguageSystem.LanguagesUpdatedHook += LanguagesUpdatedHook;
 
         _languageWindow = UIManager.CreateWindow<LanguageMenuWindow>();
@@ -60,8 +66,7 @@ public void OnStateExited(GameplayState state)
             _languageWindow = null;
         }
 
-        var entManager = IoCManager.Resolve<IEntityManager>();
-        var clientLanguageSystem = entManager.EntitySysManager.GetEntitySystem<LanguageSystem>();
+        var clientLanguageSystem = _entManager.EntitySysManager.GetEntitySystem<LanguageSystem>();
         clientLanguageSystem.LanguagesUpdatedHook -= LanguagesUpdatedHook;
 
         CommandBinds.Unregister<LanguageMenuUIController>();