Implement new fixbiome command

Author: robotgryphon

build.gradle

@@ -42,7 +42,7 @@ minecraft {
     mappings channel: 'official', version: mappings_version
     // makeObfSourceJar = false // an Srg named sources jar is made by default. uncomment this to disable.
 
-    // accessTransformer = file('src/main/resources/META-INF/accesstransformer.cfg')
+    accessTransformer = file('src/main/resources/META-INF/accesstransformer.cfg')
 
     // Default run configurations.
     // These can be tweaked, removed, or duplicated as needed.

src/main/java/dev/compactmods/machines/api/core/Messages.java

@@ -8,4 +8,5 @@ public abstract class Messages {
     public static final ResourceLocation MACHINE_SPAWNPOINT_SET = new ResourceLocation(Constants.MOD_ID, "spawnpoint_set");
     public static final ResourceLocation TELEPORT_OUT_OF_BOUNDS = new ResourceLocation(Constants.MOD_ID, "teleport_oob");
     public static final ResourceLocation HOW_DID_YOU_GET_HERE = new ResourceLocation(Constants.MOD_ID, "how_did_you_get_here");
+    public static final ResourceLocation FIXBIOME_IN_BAD_DIMENSION = new ResourceLocation(Constants.MOD_ID, "fixbiome_bad_dim");
 }

src/main/java/dev/compactmods/machines/command/CMCommandRoot.java

@@ -10,6 +10,7 @@ public class CMCommandRoot {
     public static void register(CommandDispatcher<CommandSource> dispatcher) {
         final LiteralArgumentBuilder<CommandSource> root = LiteralArgumentBuilder.literal(CompactMachines.MOD_ID);
         root.then(CMEjectSubcommand.register());
+        root.then(CMFixBiomeSubcommand.register());
         dispatcher.register(root);
     }
 }

src/main/java/dev/compactmods/machines/command/CMFixBiomeSubcommand.java

@@ -0,0 +1,92 @@
+package dev.compactmods.machines.command;
+
+import java.util.Arrays;
+import com.mojang.brigadier.builder.ArgumentBuilder;
+import com.mojang.brigadier.context.CommandContext;
+import dev.compactmods.machines.CompactMachines;
+import dev.compactmods.machines.api.core.Messages;
+import dev.compactmods.machines.core.Registration;
+import dev.compactmods.machines.util.TranslationUtil;
+import net.minecraft.command.CommandSource;
+import net.minecraft.command.Commands;
+import net.minecraft.command.arguments.ResourceLocationArgument;
+import net.minecraft.command.arguments.SuggestionProviders;
+import net.minecraft.network.play.server.SChunkDataPacket;
+import net.minecraft.server.MinecraftServer;
+import net.minecraft.util.RegistryKey;
+import net.minecraft.util.ResourceLocation;
+import net.minecraft.util.math.BlockPos;
+import net.minecraft.util.math.vector.Vector3d;
+import net.minecraft.util.registry.Registry;
+import net.minecraft.world.biome.Biome;
+import net.minecraft.world.biome.BiomeContainer;
+import net.minecraft.world.biome.Biomes;
+import net.minecraft.world.chunk.Chunk;
+import net.minecraft.world.server.ServerWorld;
+import net.minecraftforge.fml.network.PacketDistributor;
+
+public class CMFixBiomeSubcommand {
+    public static ArgumentBuilder<CommandSource, ?> register() {
+        return Commands.literal("fixbiome")
+                .executes(CMFixBiomeSubcommand::fixBiomeDefault)
+                .then(
+                        Commands.argument("biome", ResourceLocationArgument.id())
+                                .suggests(SuggestionProviders.AVAILABLE_BIOMES)
+                                .executes(CMFixBiomeSubcommand::specificBiome)
+                );
+    }
+
+
+    private static int specificBiome(CommandContext<CommandSource> ctx) {
+        if(!ctx.getSource().getLevel().dimension().equals(Registration.COMPACT_DIMENSION)) {
+            ctx.getSource().sendFailure(TranslationUtil.message(Messages.FIXBIOME_IN_BAD_DIMENSION));
+            return -1;
+        }
+
+        final ResourceLocation biomeId = ResourceLocationArgument.getId(ctx, "biome");
+        updateBiomeData(ctx, biomeId);
+        return 0;
+    }
+
+
+    private static int fixBiomeDefault(CommandContext<CommandSource> ctx) {
+        if(!ctx.getSource().getLevel().dimension().equals(Registration.COMPACT_DIMENSION)) {
+            ctx.getSource().sendFailure(TranslationUtil.message(Messages.FIXBIOME_IN_BAD_DIMENSION));
+            return -1;
+        }
+
+        final ResourceLocation biomeId = Biomes.PLAINS.location();
+        updateBiomeData(ctx, biomeId);
+        return 0;
+    }
+
+    private static void updateBiomeData(CommandContext<CommandSource> ctx, ResourceLocation biomeId) {
+        final MinecraftServer server = ctx.getSource().getServer();
+        final ServerWorld level = server.getLevel(Registration.COMPACT_DIMENSION);
+        if (level == null) {
+            CompactMachines.LOGGER.error("Error: Compact dimension not registered.");
+            return;
+        }
+
+        final Vector3d position = ctx.getSource().getPosition();
+        final BlockPos excAt = new BlockPos(position.x, position.y, position.z);
+        final Chunk chunkAt = level.getChunkAt(excAt);
+
+        final BiomeContainer biomes = chunkAt.getBiomes();
+        if (biomes != null) {
+            final Biome newBiome = server.registryAccess()
+                    .registryOrThrow(Registry.BIOME_REGISTRY)
+                    .get(RegistryKey.create(Registry.BIOME_REGISTRY, biomeId));
+
+            Arrays.fill(biomes.biomes, newBiome);
+
+            chunkAt.setUnsaved(true);
+
+            SChunkDataPacket pkt = new SChunkDataPacket(chunkAt, 65535);
+
+            PacketDistributor.TRACKING_CHUNK
+                    .with(() -> level.getChunkAt(excAt))
+                    .send(pkt);
+        }
+    }
+}

src/main/java/dev/compactmods/machines/level/BiomeHelper.java

@@ -0,0 +1,18 @@
+package dev.compactmods.machines.level;
+
+import net.minecraft.util.math.BlockPos;
+import net.minecraft.util.math.MathHelper;
+import net.minecraft.world.biome.BiomeContainer;
+
+public class BiomeHelper {
+    private static final int WIDTH_BITS = (int)Math.round(Math.log(16.0D) / Math.log(2.0D)) - 2;
+
+    public static int biomeIndex(BlockPos pos) {
+        int l = (pos.getX() >> 2) & BiomeContainer.HORIZONTAL_MASK;
+        int m = MathHelper.clamp(pos.getY() >> 2, 0, BiomeContainer.VERTICAL_MASK);
+        int n = (pos.getZ() >> 2) & BiomeContainer.HORIZONTAL_MASK;
+        return m << WIDTH_BITS + WIDTH_BITS
+                | n << WIDTH_BITS
+                | l;
+    }
+}

src/main/resources/META-INF/accesstransformer.cfg

@@ -0,0 +1 @@
+public net.minecraft.world.biome.BiomeContainer field_227054_f_ # biomes

src/main/resources/assets/compactmachines/lang/en_us.json

@@ -16,6 +16,7 @@
   "message.compactmachines.spawnpoint_set": "New spawn point set.",
   "message.compactmachines.no_machine_data": "No machine data loaded; report this.",
   "message.compactmachines.teleport_oob": "An otherworldly force prevents your teleportation.",
+  "message.compactmachines.fixbiome_bad_dim": "Cannot use fixbiome command outside of a machine room.",
 
   "block.compactmachines.machine": "Compact Machine",
   "block.compactmachines.machine_tiny": "Compact Machine (Tiny)",