From 91444083cf337d2397562061bbfceef6d2690c9c Mon Sep 17 00:00:00 2001
From: metalgearsloth <31366439+metalgearsloth@users.noreply.github.com>
Date: Thu, 14 Sep 2023 08:03:44 +1000
Subject: [PATCH] Dispose storage windows on state change (#20095)

---
 Content.Client/Storage/UI/StorageUIController.cs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

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();
+    }
 }
-- 
2.51.2