Add tag support in biome process condition

Author: Technici4n

docs/ADDING_RECIPES.md

@@ -38,6 +38,7 @@ The easiest way to add process conditions is via KubeJS, similarly to how inputs
 Here is the list of currently supported conditions:
 - `dimension(dimension key)`: Requires the machine to be in the specified dimension.
 - `biome(biome key)`: Requires the machine to be in the specified biome.
+- `biomeTag(biome tag key)`: Requires the machine to be in a biome of the specified tag.
 - `adjacentBlock(block, position)`: Requires a specific block to be next to the machine.
   - Position indicates where the block should be.
   - For multiblocks, the position is always relative to the controller.

src/main/java/aztech/modern_industrialization/compat/kubejs/recipe/ProcessConditionHelper.java

@@ -28,9 +28,11 @@
 import aztech.modern_industrialization.machines.recipe.condition.CustomProcessCondition;
 import aztech.modern_industrialization.machines.recipe.condition.DimensionProcessCondition;
 import aztech.modern_industrialization.machines.recipe.condition.MachineProcessCondition;
+import com.mojang.datafixers.util.Either;
 import net.minecraft.core.registries.Registries;
 import net.minecraft.resources.ResourceKey;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.tags.TagKey;
 import net.minecraft.world.level.block.Block;
 
 public interface ProcessConditionHelper {
@@ -45,7 +47,11 @@ default ProcessConditionHelper adjacentBlock(Block block, String relativePositio
     }
 
     default ProcessConditionHelper biome(ResourceLocation biome) {
-        return processCondition(new BiomeProcessCondition(ResourceKey.create(Registries.BIOME, biome)));
+        return processCondition(new BiomeProcessCondition(Either.left(ResourceKey.create(Registries.BIOME, biome))));
+    }
+
+    default ProcessConditionHelper biomeTag(ResourceLocation tag) {
+        return processCondition(new BiomeProcessCondition(Either.right(TagKey.create(Registries.BIOME, tag))));
     }
 
     default ProcessConditionHelper customCondition(String id) {

src/main/java/aztech/modern_industrialization/machines/recipe/condition/BiomeProcessCondition.java

@@ -25,35 +25,61 @@
 
 import aztech.modern_industrialization.MIText;
 import aztech.modern_industrialization.machines.recipe.MachineRecipe;
+import aztech.modern_industrialization.proxy.CommonProxy;
+import com.mojang.datafixers.util.Either;
 import com.mojang.serialization.MapCodec;
 import java.util.List;
+import net.minecraft.Util;
 import net.minecraft.core.registries.Registries;
 import net.minecraft.network.RegistryFriendlyByteBuf;
 import net.minecraft.network.chat.Component;
+import net.minecraft.network.chat.MutableComponent;
+import net.minecraft.network.codec.ByteBufCodecs;
 import net.minecraft.network.codec.StreamCodec;
 import net.minecraft.resources.ResourceKey;
+import net.minecraft.resources.ResourceLocation;
+import net.minecraft.tags.TagKey;
 import net.minecraft.world.level.biome.Biome;
+import net.neoforged.neoforge.common.util.NeoForgeExtraCodecs;
 
-public record BiomeProcessCondition(ResourceKey<Biome> biome) implements MachineProcessCondition {
-    static final MapCodec<BiomeProcessCondition> CODEC = ResourceKey.codec(Registries.BIOME)
-            .fieldOf("biome")
+public record BiomeProcessCondition(Either<ResourceKey<Biome>, TagKey<Biome>> biome) implements MachineProcessCondition {
+    static final MapCodec<BiomeProcessCondition> CODEC = NeoForgeExtraCodecs
+            .xor(
+                    ResourceKey.codec(Registries.BIOME).fieldOf("biome"),
+                    TagKey.codec(Registries.BIOME).fieldOf("tag"))
             .xmap(BiomeProcessCondition::new, BiomeProcessCondition::biome);
     static final StreamCodec<RegistryFriendlyByteBuf, BiomeProcessCondition> STREAM_CODEC = StreamCodec.composite(
-            ResourceKey.streamCodec(Registries.BIOME),
+            ByteBufCodecs.either(
+                    ResourceKey.streamCodec(Registries.BIOME),
+                    ResourceLocation.STREAM_CODEC.map(rl -> TagKey.create(Registries.BIOME, rl), TagKey::location)),
             BiomeProcessCondition::biome,
             BiomeProcessCondition::new);
 
     @Override
     public boolean canProcessRecipe(Context context, MachineRecipe recipe) {
         var entityBiome = context.getLevel().getBiome(context.getBlockEntity().getBlockPos());
-        return entityBiome.is(biome);
+        return biome.map(entityBiome::is, entityBiome::is);
+    }
+
+    private static MutableComponent biomeId(ResourceLocation biomeLocation) {
+        return Component.translatable(Util.makeDescriptionId("biome", biomeLocation));
     }
 
     @Override
     public void appendDescription(List<Component> list) {
-        var loc = biome.location();
-        var biomeComponent = Component.translatable("biome.%s.%s".formatted(loc.getNamespace(), loc.getPath()));
-        list.add(MIText.RequiresBiome.text(biomeComponent));
+        biome.ifLeft(rk -> {
+            list.add(MIText.RequiresBiome.text(biomeId(rk.location())));
+        }).ifRight(tag -> {
+            var holderLookup = CommonProxy.INSTANCE.getClientPlayer().registryAccess();
+            var biomeNames = holderLookup.lookupOrThrow(tag.registry()).get(tag).map(named -> {
+                return named.stream()
+                        .map(holder -> biomeId(holder.unwrapKey().orElseThrow().location()))
+                        .reduce((a, b) -> a.append(", ").append(b))
+                        .orElseThrow();
+            }).orElse(Component.literal("???"));
+
+            list.add(MIText.RequiresBiome.text(biomeNames));
+        });
     }
 
     @Override