Using GameState to determine if we quit or return to menu

Author: LPLafontaineB

Assets/BossRoom/Scripts/Client/UI/UIQuitPanel.cs

@@ -1,3 +1,4 @@
+using System;
 using Unity.Multiplayer.Samples.BossRoom.Shared;
 using Unity.Multiplayer.Samples.BossRoom.Shared.Infrastructure;
 using UnityEngine;
@@ -13,18 +14,31 @@ public class UIQuitPanel : MonoBehaviour
 
         ApplicationController m_ApplicationController;
 
-        private bool m_QuitMode = true;
+        bool m_QuitMode = true;
 
         [Inject]
-        private void InjectDependencies(ApplicationController applicationController)
+        void InjectDependencies(ApplicationController applicationController)
         {
             m_ApplicationController = applicationController;
         }
 
-        void OnEnable()
+        void Awake()
         {
-            m_QuitMode = SceneManager.GetActiveScene().name == "MainMenu";
-            m_QuitButtonText.text = m_QuitMode ? "Exit Game?" : "Return to menu?";
+            SceneManager.sceneLoaded += SceneManagerOnsceneLoaded;
+        }
+
+        void SceneManagerOnsceneLoaded(Scene scene, LoadSceneMode loadSceneMode)
+        {
+            var currentGameState = FindObjectOfType<GameStateBehaviour>();
+            if (currentGameState != null)
+            {
+                m_QuitMode = (currentGameState.ActiveState == GameState.MainMenu);
+                m_QuitButtonText.text = m_QuitMode ? "Exit Game?" : "Return to menu?";
+            }
+            else
+            {
+                Debug.LogError($"Scene {scene.name} does not contain a GameStateBehavior");
+            }
         }
 
         public void Quit()