Fix UI not closing

Simple-Station · Jan 19, 2024 · d34582d · d34582d
1 parent 523e575
commit d34582d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Content.Server/CosmaticDrift/CryoSleep/CryoSleepComponent.cs b/Content.Server/CosmaticDrift/CryoSleep/CryoSleepComponent.cs
@@ -25,4 +25,6 @@ public sealed partial class CryoSleepComponent : Component
     /// </summary>
     [DataField("initialSleepDurationRange")]
     public Vector2 InitialSleepDurationRange = new (5, 10);
+
+    public CryoSleepEui eui = default!;
 }
diff --git a/Content.Server/CosmaticDrift/CryoSleep/CryoSleepSystem.cs b/Content.Server/CosmaticDrift/CryoSleep/CryoSleepSystem.cs
@@ -101,7 +101,8 @@ public bool RespawnUser(EntityUid? toInsert, CryoSleepComponent component, bool
 
         if (success && mindComp?.Session != null)
         {
-            _euiManager.OpenEui(new CryoSleepEui(mind, this), mindComp.Session);
+            component.eui = new CryoSleepEui(mind, this);
+            _euiManager.OpenEui(component.eui, mindComp.Session);
         }
 
         return success;
@@ -192,6 +193,8 @@ private bool EjectBody(EntityUid pod, CryoSleepComponent component)
         component.BodyContainer.Remove(toEject.Value);
         _climb.ForciblySetClimbing(toEject.Value, pod);
 
+        _euiManager.CloseEui(component.eui);
+
         return true;
     }