From 776dadb050b4e7d76de9fa16161d2cda4dd66e5b Mon Sep 17 00:00:00 2001
From: Aaron <51387595+AzureAaron@users.noreply.github.com>
Date: Sun, 3 Sep 2023 16:36:08 -0400
Subject: YACL Config

---
 .../skyblock/shortcut/ShortcutsConfigScreen.java         | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

(limited to 'src/main/java/me/xmrvizzy/skyblocker/skyblock/shortcut')

diff --git a/src/main/java/me/xmrvizzy/skyblocker/skyblock/shortcut/ShortcutsConfigScreen.java b/src/main/java/me/xmrvizzy/skyblocker/skyblock/shortcut/ShortcutsConfigScreen.java
index dec23d36..043582c4 100644
--- a/src/main/java/me/xmrvizzy/skyblocker/skyblock/shortcut/ShortcutsConfigScreen.java
+++ b/src/main/java/me/xmrvizzy/skyblocker/skyblock/shortcut/ShortcutsConfigScreen.java
@@ -18,9 +18,15 @@ public class ShortcutsConfigScreen extends Screen {
     private ButtonWidget buttonDone;
     private boolean initialized;
     private double scrollAmount;
-
+    private final Screen parent;
+    
     public ShortcutsConfigScreen() {
+    	this(null);
+    }
+
+    public ShortcutsConfigScreen(Screen parent) {
         super(Text.translatable("skyblocker.shortcuts.config"));
+        this.parent = parent;
     }
 
     @Override
@@ -89,13 +95,15 @@ public class ShortcutsConfigScreen extends Screen {
         if (client != null && shortcutsConfigListWidget.hasChanges()) {
             client.setScreen(new ConfirmScreen(confirmedAction -> {
                 if (confirmedAction) {
-                    super.close();
+                	this.client.setScreen(parent);
+                    if (parent == null) super.close();
                 } else {
                     client.setScreen(this);
                 }
-            }, Text.translatable("text.cloth-config.quit_config"), Text.translatable("text.cloth-config.quit_config_sure"), Text.translatable("text.cloth-config.quit_discard"), ScreenTexts.CANCEL));
+            }, Text.translatable("text.skyblocker.quit_config"), Text.translatable("text.skyblocker.quit_config_sure"), Text.translatable("text.skyblocker.quit_discard"), ScreenTexts.CANCEL));
         } else {
-            super.close();
+            this.client.setScreen(parent);
+            if (parent == null) super.close();
         }
     }
 
-- 
cgit