Auto Save/Apply SeedCracker Seed

Author: cev-api

README.md

@@ -93,3 +93,5 @@ 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.

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"