From d223a1623d2bc6eb77182c7ea8edd5217fe8c5ce Mon Sep 17 00:00:00 2001
From: NoMathExpectation <85624722+NoMathExpectation@users.noreply.github.com>
Date: Wed, 19 Jun 2024 01:19:11 +0800
Subject: [PATCH] Make confirm menu more vanilla on preferences.

---
 Source/LevelExtension.cs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/Source/LevelExtension.cs b/Source/LevelExtension.cs
index 9d198a8..9e75686 100644
--- a/Source/LevelExtension.cs
+++ b/Source/LevelExtension.cs
@@ -168,6 +168,7 @@ public static bool PlayerHasImportantCollectible(this Level level)
 
         private static void QuickRestartReplacement(this Level level, bool minimal = false)
         {
+            // hooked just before the original give up method
             if (level.PlayerHasImportantCollectible())
             {
                 level.GiveUpGolden(0, minimal, "menu_pause_restartarea".DialogClean(), menu =>
@@ -251,6 +252,7 @@ private static void OnCreatePauseMenuButtons(Level level, TextMenu menu, bool mi
                     button.Pressed(() =>
                     {
                         menu.RemoveSelf();
+                        level.PauseMainMenuOpen = false;
                         level.GiveUpGolden(menu.IndexOf(button), minimal, label, m2 =>
                         {
                             m2.RemoveSelf();
@@ -266,9 +268,10 @@ private static void OnCreatePauseMenuButtons(Level level, TextMenu menu, bool mi
                     button.Pressed(() =>
                     {
                         menu.RemoveSelf();
+                        level.PauseMainMenuOpen = false;
                         level.GiveUpGolden(menu.IndexOf(button), minimal, label, m2 =>
                         {
-                            m2.RemoveSelf();
+                            m2.Focused = false;
                             origPressed();
                         });
                     });
@@ -280,6 +283,7 @@ private static void OnCreatePauseMenuButtons(Level level, TextMenu menu, bool mi
                     button.Pressed(() =>
                     {
                         menu.RemoveSelf();
+                        level.PauseMainMenuOpen = false;
                         level.GiveUpGolden(menu.IndexOf(button), minimal, label, m2 =>
                         {
                             m2.Focused = false;
@@ -294,6 +298,7 @@ private static void OnCreatePauseMenuButtons(Level level, TextMenu menu, bool mi
                     button.Pressed(() =>
                     {
                         menu.RemoveSelf();
+                        level.PauseMainMenuOpen = false;
                         level.GiveUpGolden(menu.IndexOf(button), minimal, label, m2 =>
                         {
                             m2.Focused = false;