From 849bea21ce83abceb4eaf634ad0667555aa30ea7 Mon Sep 17 00:00:00 2001
From: Mister_Nebula <41904486+misternebula@users.noreply.github.com>
Date: Wed, 8 Dec 2021 10:59:36 +0000
Subject: [PATCH] hide quit button

---
 QSB/Menus/MenuManager.cs | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/QSB/Menus/MenuManager.cs b/QSB/Menus/MenuManager.cs
index 73308e3d..b34e525a 100644
--- a/QSB/Menus/MenuManager.cs
+++ b/QSB/Menus/MenuManager.cs
@@ -301,6 +301,11 @@ namespace QSB.Menus
 				SetButtonActive(ResumeGameButton, false);
 				SetButtonActive(NewGameButton, false);
 			}
+
+			if (QSBSceneManager.IsInUniverse)
+			{
+				SetButtonActive(QuitButton, false);
+			}
 		}
 
 		private void OnConnected()
@@ -329,9 +334,10 @@ namespace QSB.Menus
 			};
 			OpenInfoPopup(text, "OK");
 
-			DisconnectButton.gameObject.SetActive(false);
-			ClientButton.SetActive(true);
-			HostButton?.gameObject.SetActive(true);
+			SetButtonActive(DisconnectButton, false);
+			SetButtonActive(ClientButton, true);
+			SetButtonActive(HostButton, true);
+			SetButtonActive(QuitButton, true);
 		}
 
 		private void OnDisconnected(NetworkError error)
@@ -350,6 +356,7 @@ namespace QSB.Menus
 
 			SetButtonActive(DisconnectButton, false);
 			SetButtonActive(ClientButton, true);
+			SetButtonActive(QuitButton, true);
 			SetButtonActive(HostButton, true);
 			SetButtonActive(ResumeGameButton, StandaloneProfileManager.SharedInstance.currentProfileGameSave.loopCount > 1);
 			SetButtonActive(NewGameButton, true);
@@ -374,6 +381,7 @@ namespace QSB.Menus
 					SetButtonActive(HostButton, true);
 					SetButtonActive(ResumeGameButton, StandaloneProfileManager.SharedInstance.currentProfileGameSave.loopCount > 1);
 					SetButtonActive(NewGameButton, true);
+					SetButtonActive(QuitButton, true);
 					break;
 				default:
 					text = $"Internal QNet client error!\n\nNetworkError:{error}";