From: metalgearsloth <31366439+metalgearsloth@users.noreply.github.com>
Date: Wed, 13 Sep 2023 22:03:44 +0000 (+1000)
Subject: Dispose storage windows on state change (#20095)
X-Git-Url: https://git.smokeofanarchy.ru/gitweb.cgi?a=commitdiff_plain;h=91444083cf337d2397562061bbfceef6d2690c9c;p=space-station-14.git

Dispose storage windows on state change (#20095)
---

diff --git a/Content.Client/Storage/UI/StorageUIController.cs b/Content.Client/Storage/UI/StorageUIController.cs
index 9fc7909c34..352ee4f9ed 100644
--- a/Content.Client/Storage/UI/StorageUIController.cs
+++ b/Content.Client/Storage/UI/StorageUIController.cs
@@ -1,10 +1,11 @@
+using Content.Client.Gameplay;
 using Content.Client.Storage.Systems;
 using Content.Shared.Storage;
 using Robust.Client.UserInterface.Controllers;
 
 namespace Content.Client.Storage.UI;
 
-public sealed class StorageUIController : UIController, IOnSystemChanged<StorageSystem>
+public sealed class StorageUIController : UIController, IOnSystemChanged<StorageSystem>, IOnStateExited<GameplayState>
 {
     // This is mainly to keep legacy functionality for now.
     private readonly Dictionary<EntityUid, StorageWindow> _storageWindows = new();
@@ -57,4 +58,14 @@ public sealed class StorageUIController : UIController, IOnSystemChanged<Storage
     {
         system.StorageUpdated -= OnStorageUpdate;
     }
+
+    public void OnStateExited(GameplayState state)
+    {
+        foreach (var window in _storageWindows.Values)
+        {
+            window.Dispose();
+        }
+
+        _storageWindows.Clear();
+    }
 }