Only show the button for standard single player

qe201020335 · Nov 15, 2024 · 066f13d · 066f13d
1 parent 953309c
commit 066f13d
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/FailButton/HarmonyPatches/AddButtonToPauseMenu.cs b/FailButton/HarmonyPatches/AddButtonToPauseMenu.cs
@@ -17,10 +17,11 @@ static class AddButtonToPauseMenu
 
         static void Postfix(PauseMenuManager __instance)
         {
-            __instance.StartCoroutine(DoTheFunny(__instance));
+            var gameplayManager = Resources.FindObjectsOfTypeAll<StandardLevelGameplayManager>().FirstOrDefault();
+            if (gameplayManager != null) __instance.StartCoroutine(DoTheFunny(__instance, gameplayManager));
         }
 
-        static IEnumerator DoTheFunny(PauseMenuManager __instance)
+        static IEnumerator DoTheFunny(PauseMenuManager __instance, StandardLevelGameplayManager gameplayManager)
         {
             if (_b != null)
                 yield break;
@@ -52,13 +53,8 @@ static IEnumerator DoTheFunny(PauseMenuManager __instance)
 
             b.onClick.AddListener(() =>
             {
-                var s = Resources.FindObjectsOfTypeAll<StandardLevelGameplayManager>().FirstOrDefault();
-
-                if (s == null)
-                    return;
-
-                s._initData.SetField(nameof(s._initData.continueGameplayWith0Energy), false);
-                s.HandleGameEnergyDidReach0();
+                gameplayManager._initData.SetField(nameof(gameplayManager._initData.continueGameplayWith0Energy), false);
+                gameplayManager.HandleGameEnergyDidReach0();
             });
 
             yield return null;