From 05a276e89d5f7efaf9792da4dec357743adb8cda Mon Sep 17 00:00:00 2001
From: Micle <micle@micle.dev>
Date: Sat, 7 Jun 2025 01:51:55 +0100
Subject: [PATCH] Added feature removal for non-swamp near river feature.

---
 .../worldgen/ModBiomeModifiers.java                      | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/main/java/dev/micle/firefly_bush_backport/worldgen/ModBiomeModifiers.java b/src/main/java/dev/micle/firefly_bush_backport/worldgen/ModBiomeModifiers.java
index d9e18ae..49f640a 100644
--- a/src/main/java/dev/micle/firefly_bush_backport/worldgen/ModBiomeModifiers.java
+++ b/src/main/java/dev/micle/firefly_bush_backport/worldgen/ModBiomeModifiers.java
@@ -16,10 +16,13 @@ import net.minecraftforge.common.world.BiomeModifier;
 import net.minecraftforge.common.world.ForgeBiomeModifiers;
 import net.minecraftforge.registries.ForgeRegistries;
 
+import java.util.Set;
+
 public class ModBiomeModifiers {
     public static final ResourceKey<BiomeModifier> ADD_PATCH_FIREFLY_BUSH_SWAMP = registerKey("add_patch_firefly_bush_swamp");
     public static final ResourceKey<BiomeModifier> ADD_PATCH_FIREFLY_BUSH_NEAR_WATER_SWAMP = registerKey("add_patch_firefly_bush_near_water_swamp");
     public static final ResourceKey<BiomeModifier> ADD_PATCH_FIREFLY_BUSH_NEAR_WATER = registerKey("add_patch_firefly_bush_near_water");
+    public static final ResourceKey<BiomeModifier> REMOVE_PATCH_FIREFLY_BUSH_NEAR_WATER = registerKey("remove_patch_firefly_bush_near_water");
 
     public static void bootstrap(BootstapContext<BiomeModifier> context) {
         HolderGetter<PlacedFeature> placedFeatures = context.lookup(Registries.PLACED_FEATURE);
@@ -42,6 +45,12 @@ public class ModBiomeModifiers {
                 HolderSet.direct(placedFeatures.getOrThrow(ModPlacedFeatures.PATCH_FIREFLY_BUSH_NEAR_WATER_PLACED_KEY)),
                 GenerationStep.Decoration.VEGETAL_DECORATION
         ));
+
+        context.register(REMOVE_PATCH_FIREFLY_BUSH_NEAR_WATER, new ForgeBiomeModifiers.RemoveFeaturesBiomeModifier(
+                biomes.getOrThrow(Tags.Biomes.IS_SWAMP),
+                HolderSet.direct(placedFeatures.getOrThrow(ModPlacedFeatures.PATCH_FIREFLY_BUSH_NEAR_WATER_PLACED_KEY)),
+                Set.of(GenerationStep.Decoration.VEGETAL_DECORATION)
+        ));
     }
 
     private static ResourceKey<BiomeModifier> registerKey(String name) {