Large Refactor

use less intrusive mixins (replaced some redirects with modifyconstant/-variable)
fix the "still saving the last world" screen never actually rendering
fix savelock not applying on old worlds
make savelock apply later for faster resets
remove/replace unused and unnecessary code
dont render menu.quitworld in multiplayer
more open dependencies
more vanilla / less copypaste code
and some more stuff

Author: jan-leila

build.gradle

@@ -25,15 +25,11 @@ dependencies {
 
 processResources {
     inputs.property "version", project.version
+    filteringCharset "UTF-8"
 
-//    from(sourceSets.main.resources.srcDirs) {
-//        include "fabric.mod.json"
-//        expand "version": project.version
-//    }
-//
-//    from(sourceSets.main.resources.srcDirs) {
-//        exclude "fabric.mod.json"
-//    }
+    filesMatching("fabric.mod.json") {
+        expand "version": project.version
+    }
 }
 
 // ensure that the encoding is set to UTF-8, no matter what the system default is

gradle.properties

@@ -6,6 +6,6 @@ minecraft_version=1.16.1
 yarn_mappings=1.16.1+build.21
 loader_version=0.11.3
 # Mod Properties
-mod_version=1.4.1
+mod_version=1.5
 maven_group=fast-reset-1.16.1-v
 archives_base_name=fast-reset-1.16.1-v

src/main/java/fast_reset/client/Client.java

@@ -0,0 +1,62 @@
+package fast_reset.client;
+
+import net.fabricmc.api.ClientModInitializer;
+import net.fabricmc.loader.api.FabricLoader;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+import java.io.*;
+import java.nio.file.Files;
+
+public class FastReset implements ClientModInitializer {
+
+    public static final Logger LOGGER = LogManager.getLogger();
+    private static final File configurationFile = FabricLoader.getInstance().getConfigDir().resolve("fastReset").resolve("settings.txt").toFile();
+
+    public static boolean saveOnQuit = true;
+    public static Boolean saving = false;
+    public static final Object saveLock = new Object();
+
+    public static int buttonLocation = 0;
+
+    public static void updateButtonLocation() {
+        buttonLocation = (buttonLocation + 1) % 3;
+        save();
+    }
+
+    private static void save() {
+        try {
+            Files.write(configurationFile.toPath(), String.valueOf(buttonLocation).getBytes());
+        } catch (IOException e) {
+            LOGGER.error("Failed to save FastReset config", e);
+        }
+    }
+
+    @Override
+    public void onInitializeClient() {
+        LOGGER.info("Using Fast Reset");
+
+        if (!configurationFile.exists()) {
+            try {
+                if (!configurationFile.getParentFile().mkdirs()) {
+                    throw new IOException("couldn't make config folder");
+                }
+                if (!configurationFile.createNewFile()) {
+                    throw new IOException("couldn't make config file");
+                }
+                save();
+            } catch (IOException e) {
+                LOGGER.error("Failed to create FastReset config", e);
+            }
+            return;
+        }
+
+        try {
+            BufferedReader reader = new BufferedReader(new FileReader(configurationFile));
+            buttonLocation = Integer.parseInt(reader.readLine()) % 3;
+            reader.close();
+        } catch (IOException | NumberFormatException e) {
+            LOGGER.error("Failed to load FastReset config", e);
+        }
+    }
+}

src/main/java/fast_reset/client/FastReset.java

@@ -1,20 +1,14 @@
 package fast_reset.client.mixin;
 
-import fast_reset.client.Client;
+import fast_reset.client.FastReset;
 import net.minecraft.client.gui.screen.GameMenuScreen;
-import net.minecraft.client.gui.screen.SaveLevelScreen;
 import net.minecraft.client.gui.screen.Screen;
-import net.minecraft.client.gui.screen.TitleScreen;
-import net.minecraft.client.gui.screen.multiplayer.MultiplayerScreen;
 import net.minecraft.client.gui.widget.ButtonWidget;
-import net.minecraft.realms.RealmsBridge;
 import net.minecraft.text.Text;
 import net.minecraft.text.TranslatableText;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
-import org.spongepowered.asm.mixin.injection.Inject;
-import org.spongepowered.asm.mixin.injection.Redirect;
-import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.ModifyVariable;
 
 @Mixin(GameMenuScreen.class)
 public class GameMenuMixin extends Screen {
@@ -23,26 +17,17 @@ protected GameMenuMixin(Text title) {
         super(title);
     }
 
-    private static final int bottomRightWidth = 102;
-
-    // kill save on the shutdown
-    @Redirect(method = "initWidgets", at = @At(value = "NEW", target = "net/minecraft/client/gui/widget/ButtonWidget", ordinal=7))
-    private ButtonWidget createExitButton(int defaultX, int defaultY, int defaultWidth, int height, Text message, ButtonWidget.PressAction onPress){
-        int x = Client.buttonLocation == 2 ? (int) (this.width - (bottomRightWidth * 1.5) - 4) : defaultX;
-        int y = Client.buttonLocation == 2 ? this.height - 24 : defaultY;
-        int width = Client.buttonLocation == 2 ? (int) (bottomRightWidth * 1.5) : defaultWidth;
-
-        return new ButtonWidget(x, y, width, height, message, onPress);
-    }
+    @ModifyVariable(method = "initWidgets", at = @At(value = "STORE", ordinal = 1))
+    private ButtonWidget createExitButton(ButtonWidget saveButton) {
+        if (!this.client.isInSingleplayer()) {
+            return saveButton;
+        }
 
-    @Inject(method = "initWidgets", at=@At(value ="TAIL"))
-    private void createSaveButton(CallbackInfo ci){
         int height = 20;
-
         int width;
         int x;
         int y;
-        switch(Client.buttonLocation){
+        switch (FastReset.buttonLocation){
             // bottom right build
             case 0:
                 width = 102;
@@ -52,40 +37,27 @@ private void createSaveButton(CallbackInfo ci){
             // center build
             case 1:
                 width = 204;
-                x = this.width / 2 - width/2;
+                x = this.width / 2 - width / 2;
                 y = this.height / 4 + 148 - height;
                 break;
+            // replace s&q build
             case 2:
             default:
                 width = 204;
-                x = this.width / 2 - width/2;
+                x = this.width / 2 - width / 2;
                 y = this.height / 4 + 124 - height;
+
+                saveButton.x = (int) (this.width - (102 * 1.5) - 4);
+                saveButton.y = this.height - 24;
+                saveButton.setWidth((int) (102 * 1.5));
                 break;
         }
 
-
         this.addButton(new ButtonWidget(x, y, width, height, new TranslatableText("menu.quitWorld"), (buttonWidgetX) -> {
-            Client.saveOnQuit = false;
-
-            boolean bl = this.client.isInSingleplayer();
-            boolean bl2 = this.client.isConnectedToRealms();
-            buttonWidgetX.active = false;
-            this.client.world.disconnect();
-            if (bl) {
-                this.client.disconnect(new SaveLevelScreen(new TranslatableText("menu.savingLevel")));
-            } else {
-                this.client.disconnect();
-            }
-
-            if (bl) {
-                this.client.openScreen(new TitleScreen());
-            } else if (bl2) {
-                RealmsBridge realmsBridge = new RealmsBridge();
-                realmsBridge.switchToRealms(new TitleScreen());
-            } else {
-                this.client.openScreen(new MultiplayerScreen(new TitleScreen()));
-            }
-            Client.saveOnQuit = true;
+            FastReset.saveOnQuit = false;
+            saveButton.onPress();
+            FastReset.saveOnQuit = true;
         }));
+        return saveButton;
     }
 }

src/main/java/fast_reset/client/client/Client.java

@@ -0,0 +1,34 @@
+package fast_reset.client.mixin;
+
+import fast_reset.client.FastReset;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.gui.screen.SaveLevelScreen;
+import net.minecraft.client.gui.screen.Screen;
+import net.minecraft.text.LiteralText;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(MinecraftClient.class)
+public abstract class MinecraftClientMixin {
+
+    @Shadow public abstract void method_29970(Screen screen);
+
+    @Inject(method = "startIntegratedServer(Ljava/lang/String;Lnet/minecraft/util/registry/RegistryTracker$Modifiable;Ljava/util/function/Function;Lcom/mojang/datafixers/util/Function4;ZLnet/minecraft/client/MinecraftClient$WorldLoadAction;)V", at = @At(value = "INVOKE", target = "Lnet/minecraft/server/MinecraftServer;startServer(Ljava/util/function/Function;)Lnet/minecraft/server/MinecraftServer;"))
+    private void worldWait(CallbackInfo ci) {
+        synchronized (FastReset.saving) {
+            if (!FastReset.saving) {
+                FastReset.LOGGER.info("no save lock active");
+                return;
+            }
+        }
+
+        this.method_29970(new SaveLevelScreen(new LiteralText("still saving the last world")));
+
+        synchronized(FastReset.saveLock) {
+            FastReset.LOGGER.info("done waiting for save lock");
+        }
+    }
+}