Auto Save/Apply SeedCracker Seed

Author: cev-api

README.md

@@ -26,7 +26,9 @@ This change was accepted upstream, however in my fork I have adjusted the size o
 
 ### World Presets
 
-If the server you're on uses anything other than the default world preset (Large Biomes, Single Biome, ~~Amplified~~, ~~Superflat~~) this will greatly change the world generation. Change the preset to match the server in order to produce an accurate seedmap. Note that Amplified and Superflat biomes are not implemented yet and are placeholders.
+If the server you're on uses anything other than the default world preset (Large Biomes, Single Biome, ~~Amplified~~, ~~Superflat~~) this will greatly change the world generation. Change the preset to match the server in order to produce an accurate seedmap. 
+
+Note that Amplified and Superflat biomes are not implemented yet and are placeholders.
 
 - ```/sm:preset list``` — show available presets  
 - ```/sm:preset set <id>``` — set SeedMapper’s preset 
@@ -93,3 +95,7 @@ Can now export the entire loot table for the map you're viewing by clicking ```E
 
 Exported data will be located in ```SeedMapper/loot/<Server IP>_<Seed>-<Date/Time>.json```
 
+### Auto Apply SeedCracker Seed
+If the server you're in already has a seed in the database it will be auto applied and saved.
+
+You can enable/disable this with ```/sm:config AutoApplySeedCrackerSeed true/false```

src/main/java/dev/xpple/seedmapper/config/Configs.java

@@ -13,9 +13,11 @@
 import net.minecraft.util.Util;
 import dev.xpple.seedmapper.render.esp.EspStyle;
 
+import java.net.SocketAddress;
 import java.util.EnumSet;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Objects;
 import java.util.function.Supplier;
 
 import static dev.xpple.seedmapper.util.ChatBuilder.*;
@@ -34,10 +36,52 @@ public static void save() {
     @Config(putter = @Config.Putter("none"), adder = @Config.Adder(value = "addSavedSeed", type = long.class))
     public static Map<String, Long> SavedSeeds = new HashMap<>();
     private static void addSavedSeed(long seed) {
-        String key = Minecraft.getInstance().getConnection().getConnection().getRemoteAddress().toString();
+        String key = getCurrentServerKey();
+        if (key == null) {
+            return;
+        }
         SavedSeeds.put(key, seed);
     }
 
+    @Config
+    public static boolean AutoApplySeedCrackerSeed = true;
+
+    private static String getCurrentServerKey() {
+        Minecraft minecraft = Minecraft.getInstance();
+        if (minecraft == null || minecraft.getConnection() == null || minecraft.getConnection().getConnection() == null) {
+            return null;
+        }
+        SocketAddress remoteAddress = minecraft.getConnection().getConnection().getRemoteAddress();
+        return remoteAddress != null ? remoteAddress.toString() : null;
+    }
+
+    public static void applySeedForCurrentServer(long seed, boolean storeAsSavedSeed) {
+        String key = getCurrentServerKey();
+        boolean changed = false;
+        if (storeAsSavedSeed && key != null && !Objects.equals(SavedSeeds.get(key), seed)) {
+            SavedSeeds.put(key, seed);
+            changed = true;
+        }
+        if (!Objects.equals(Seed, seed)) {
+            Seed = seed;
+            changed = true;
+        }
+        if (changed) {
+            save();
+        }
+    }
+
+    public static void loadSavedSeedForCurrentServer() {
+        String key = getCurrentServerKey();
+        if (key == null) {
+            return;
+        }
+        Long savedSeed = SavedSeeds.get(key);
+        if (savedSeed != null) {
+            applySeedForCurrentServer(savedSeed, false);
+        }
+    }
+
     @Config
     public static SeedResolutionArgument.SeedResolution SeedResolutionOrder = new SeedResolutionArgument.SeedResolution();
 

src/main/java/dev/xpple/seedmapper/mixin/ChatComponentMixin.java

@@ -0,0 +1,25 @@
+package dev.xpple.seedmapper.mixin;
+
+import dev.xpple.seedmapper.util.SavedSeedChatCatcher;
+import net.minecraft.client.GuiMessageTag;
+import net.minecraft.client.gui.components.ChatComponent;
+import net.minecraft.network.chat.Component;
+import net.minecraft.network.chat.MessageSignature;
+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.callback.CallbackInfo;
+
+@Mixin(ChatComponent.class)
+public class ChatComponentMixin {
+
+    @Inject(method = "addMessage(Lnet/minecraft/network/chat/Component;)V", at = @At("HEAD"))
+    private void seedmapper$captureSimple(Component component, CallbackInfo ci) {
+        SavedSeedChatCatcher.capture(component);
+    }
+
+    @Inject(method = "addMessage(Lnet/minecraft/network/chat/Component;Lnet/minecraft/network/chat/MessageSignature;Lnet/minecraft/client/GuiMessageTag;)V", at = @At("HEAD"))
+    private void seedmapper$captureSigned(Component component, MessageSignature signature, GuiMessageTag tag, CallbackInfo ci) {
+        SavedSeedChatCatcher.capture(component);
+    }
+}

src/main/java/dev/xpple/seedmapper/mixin/ClientPacketListenerMixin.java

@@ -1,5 +1,6 @@
 package dev.xpple.seedmapper.mixin;
 
+import dev.xpple.seedmapper.config.Configs;
 import dev.xpple.seedmapper.render.RenderManager;
 import net.minecraft.client.multiplayer.ClientPacketListener;
 import net.minecraft.network.protocol.game.ClientboundLoginPacket;
@@ -16,6 +17,11 @@ private void onHandleLogin(ClientboundLoginPacket packet, CallbackInfo ci) {
         RenderManager.clear();
     }
 
+    @Inject(method = "handleLogin", at = @At("TAIL"))
+    private void seedmapper$applySavedSeed(ClientboundLoginPacket packet, CallbackInfo ci) {
+        Configs.loadSavedSeedForCurrentServer();
+    }
+
     @Inject(method = "handleRespawn", at = @At("HEAD"))
     private void onHandleRespawn(ClientboundRespawnPacket packet, CallbackInfo ci) {
         RenderManager.clear();

src/main/java/dev/xpple/seedmapper/util/SavedSeedChatCatcher.java

@@ -0,0 +1,31 @@
+package dev.xpple.seedmapper.util;
+
+import dev.xpple.seedmapper.config.Configs;
+import net.minecraft.network.chat.Component;
+
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public final class SavedSeedChatCatcher {
+
+    private static final Pattern FOUND_SEED_PATTERN = Pattern.compile("(?i)found world seed \\[(-?\\d+)]\\s+from database");
+
+    private SavedSeedChatCatcher() {
+    }
+
+    public static void capture(Component component) {
+        if (component == null || !Configs.AutoApplySeedCrackerSeed) {
+            return;
+        }
+        String plain = component.getString();
+        Matcher matcher = FOUND_SEED_PATTERN.matcher(plain);
+        if (!matcher.find()) {
+            return;
+        }
+        try {
+            long seed = Long.parseLong(matcher.group(1));
+            Configs.applySeedForCurrentServer(seed, true);
+        } catch (NumberFormatException ignored) {
+        }
+    }
+}

src/main/resources/mixins.seedmapper.json

@@ -3,6 +3,7 @@
   "package": "dev.xpple.seedmapper.mixin",
   "compatibilityLevel": "JAVA_21",
   "client": [
+    "ChatComponentMixin",
     "ClientPacketListenerMixin",
     "ChatScreenMixin",
     "betterconfig.ConfigCommandClientMixin"