fix: invites after disconnecting from game

olavim · Jul 17, 2021 · 3594b4a · 3594b4a
1 parent 0b572b3
commit 3594b4a
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 19 deletions.
diff --git a/RoundsWithFriends/PrivateRoomHandler.cs b/RoundsWithFriends/PrivateRoomHandler.cs
@@ -48,28 +48,13 @@ public bool IsOpen
             }
         }
 
-        private static void OnSceneLoad(Scene scene, LoadSceneMode mode)
-        {
-            // Map changes also cause scene loads. We don't want to open the lobby during those...
-            if (scene.name == "Main")
-            {
-                SceneManager.sceneLoaded -= PrivateRoomHandler.OnSceneLoad;
-                PrivateRoomHandler.instance.Open();
-
-                if (PhotonNetwork.IsMasterClient)
-                {
-                    PrivateRoomHandler.RestoreSettings();
-                }
-            }
-        }
-
         private static void SaveSettings()
         {
             PrivateRoomHandler.PrevHandlerID = GameModeManager.CurrentHandlerID;
             PrivateRoomHandler.PrevSettings = GameModeManager.CurrentHandler.Settings;
         }
 
-        private static void RestoreSettings()
+        public static void RestoreSettings()
         {
             PrivateRoomHandler.instance.SyncMethod(nameof(PrivateRoomHandler.SetGameSettings), null, PrivateRoomHandler.PrevHandlerID, PrivateRoomHandler.PrevSettings);
             PrivateRoomHandler.PrevHandlerID = null;
@@ -624,9 +609,6 @@ public static void StartGame()
                 PrivateRoomHandler.SaveSettings();
             }
 
-            // The main scene is reloaded after the game. After the reload is done, we want to reopen the lobby.
-            SceneManager.sceneLoaded += PrivateRoomHandler.OnSceneLoad;
-
             PhotonNetwork.LocalPlayer.SetProperty("ready", false);
             PhotonNetwork.LocalPlayer.SetProperty("readyOrder", -1);
 

diff --git a/RoundsWithFriends/RoundEndHandler.cs b/RoundsWithFriends/RoundEndHandler.cs
@@ -145,10 +145,27 @@ public static void Exit()
                 }
             }
 
+            // Reopen lobby after main scene is loaded
+            SceneManager.sceneLoaded += RoundEndHandler.OnSceneLoad;
+
             gm.GameMode.StopAllCoroutines();
             SceneManager.LoadScene(SceneManager.GetActiveScene().name);
 
             RoundEndHandler.instance.waitingForHost = false;
         }
+
+        private static void OnSceneLoad(Scene scene, LoadSceneMode mode)
+        {
+            if (scene.name == "Main")
+            {
+                SceneManager.sceneLoaded -= RoundEndHandler.OnSceneLoad;
+                PrivateRoomHandler.instance.Open();
+
+                if (PhotonNetwork.IsMasterClient)
+                {
+                    PrivateRoomHandler.RestoreSettings();
+                }
+            }
+        }
     }
 }