require restart attribute

1 files changed, 9 insertions, 1 deletions

diff --git a/src/main/java/dev/isxander/yacl/gui/YACLScreen.java b/src/main/java/dev/isxander/yacl/gui/YACLScreen.java
index ccd3929..986a4f6 100644
--- a/src/main/java/dev/isxander/yacl/gui/YACLScreen.java
+++ b/src/main/java/dev/isxander/yacl/gui/YACLScreen.java
@@ -69,7 +69,12 @@ public class YACLScreen extends Screen {
             saveButtonMessage = null;
 
             if (pendingChanges()) {
-                OptionUtils.forEachOptions(config, Option::applyValue);
+                AtomicBoolean requiresRestart = new AtomicBoolean(false);
+                OptionUtils.forEachOptions(config, option -> {
+                    if (option.requiresRestart())
+                        requiresRestart.set(true);
+                    option.applyValue();
+                });
                 OptionUtils.forEachOptions(config, option -> {
                     if (option.changed()) {
                         YACLConstants.LOGGER.error("Option '{}' was saved as '{}' but the changes don't seem to have applied.", option.name().getString(), option.pendingValue());
@@ -77,6 +82,9 @@ public class YACLScreen extends Screen {
                     }
                 });
                 config.saveFunction().run();
+                if (requiresRestart.get()) {
+                    client.setScreen(new RequireRestartScreen(this));
+                }
             } else close();
         });
         actionDim.expand(-actionDim.width() / 2 - 2, 0).move(-actionDim.width() / 2 - 2, -22);