only worked on the first world in each mc session

src/main/java/misterx/diamondgen/StartGen.java

@@ -4,20 +4,21 @@
 import kaptainwutax.seedutils.mc.seed.ChunkSeeds;
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.world.ClientWorld;
+import net.minecraft.entity.player.PlayerEntity;
 import net.minecraft.util.math.BlockPos;
 
 import java.util.Random;
 
 public class StartGen {
     public SimOreGen simOreGen = new SimOreGen();
     public ClientWorld world = MinecraftClient.getInstance().world;
+    public PlayerEntity player = MinecraftClient.getInstance().player;
     public ChunkGenerated chunkList = new ChunkGenerated();
-    private long currentSeed = 0L;
+    public long currentSeed = 0L;
 
     StartGen(long seed) {
         this.currentSeed = seed;
     }
-    public void setCurrentSeed(long seed) {this.currentSeed = seed;}
 
     public void getStartingPos(int BlockX, int BlockZ) {
         if(!chunkList.Check(BlockX,BlockZ)) {

src/main/java/misterx/diamondgen/Util.java

@@ -7,29 +7,30 @@
 
 public class Util {
     private static final MinecraftClient client = MinecraftClient.getInstance();
-    private static final PlayerEntity player = MinecraftClient.getInstance().player;
 
     public static int distanceToPlayer(BlockPos posOre) {
-        assert player != null;
-        BlockPos posPlayer = player.getBlockPos();
-        int x = Math.abs(Math.abs(posOre.getX())-Math.abs(posPlayer.getX()));
-        int z = Math.abs(Math.abs(posOre.getZ())-Math.abs(posPlayer.getZ()));
-        return x + z;
+        if(client.player != null) {
+            BlockPos posPlayer = client.player.getBlockPos();
+            int x = Math.abs(Math.abs(posOre.getX()) - Math.abs(posPlayer.getX()));
+            int z = Math.abs(Math.abs(posOre.getZ()) - Math.abs(posPlayer.getZ()));
+            return x + z;
+        }
+        return 1001;
     }
     public static void reload() {
         int renderdistance = client.options.viewDistance;
 
-        int playerChunkX = (int) (Math.round(player.getX()) >> 4);
-        int playerChunkZ = (int) (Math.round(player.getZ()) >> 4);
+        int playerChunkX = (int) (Math.round(client.player.getX()) >> 4);
+        int playerChunkZ = (int) (Math.round(client.player.getZ()) >> 4);
         for(int i = playerChunkX - renderdistance;i < playerChunkX + renderdistance; i++) {
             for(int j = playerChunkZ - renderdistance;j < playerChunkZ + renderdistance; j++) {
                 DiamondGen.gen.getStartingPos(i << 4,j<<4);
             }
         }
     }
      public static boolean isOpaque(BlockPos pos) {
-        if(client.world != null) {
-            return client.world.getBlockState(pos).isOpaque();
+        if(DiamondGen.gen.world != null) {
+            return DiamondGen.gen.world.getBlockState(pos).isOpaque();
         }
         return false;
     }

src/main/java/misterx/diamondgen/mixin/ClientWorldMixin.java

@@ -0,0 +1,17 @@
+package misterx.diamondgen.mixin;
+
+import misterx.diamondgen.DiamondGen;
+import net.minecraft.client.world.ClientWorld;
+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(ClientWorld.class)
+public class ClientWorldMixin {
+
+    @Inject(method = "disconnect", at = @At("HEAD"))
+    private void disconnect(CallbackInfo ci) {
+        DiamondGen.clear(DiamondGen.gen.currentSeed);
+    }
+}

src/main/resources/diamondgen.mixins.json

@@ -4,13 +4,13 @@
   "package": "misterx.diamondgen.mixin",
   "compatibilityLevel": "JAVA_8",
   "mixins": [
-
   ],
   "client": [
+    "ClientPlayerEntityMixin",
     "ClientPlayNetworkHandlerMixin",
+    "ClientWorldMixin",
     "DummyProfilerMixin",
-    "GameRendererMixin",
-    "ClientPlayerEntityMixin"
+    "GameRendererMixin"
   ],
   "server": [
   ],