Fishing net now uses its own tag to check for valid fish.

critters now have "can breathe underwater" tag
hybridlabs · Feb 7, 2024 · 9af864e · 9af864e
1 parent d1db116
commit 9af864e
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/src/main/kotlin/dev/hybridlabs/aquatic/data/server/tag/EntityTypeTagProvider.kt b/src/main/kotlin/dev/hybridlabs/aquatic/data/server/tag/EntityTypeTagProvider.kt
@@ -6,6 +6,7 @@ import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput
 import net.fabricmc.fabric.api.datagen.v1.provider.FabricTagProvider
 import net.minecraft.entity.EntityType
 import net.minecraft.registry.RegistryWrapper
+import net.minecraft.registry.tag.EntityTypeTags
 import java.util.concurrent.CompletableFuture
 
 class EntityTypeTagProvider(output: FabricDataOutput, registriesFuture: CompletableFuture<RegistryWrapper.WrapperLookup>) : FabricTagProvider.EntityTypeTagProvider(output, registriesFuture) {
@@ -207,6 +208,11 @@ class EntityTypeTagProvider(output: FabricDataOutput, registriesFuture: Completa
                 HybridAquaticEntityTypes.STARFISH,
             )
 
+        getOrCreateTagBuilder(EntityTypeTags.CAN_BREATHE_UNDER_WATER)
+            .addTag(HybridAquaticEntityTags.CRAB)
+            .addTag(HybridAquaticEntityTags.CRITTER)
+            .addTag(HybridAquaticEntityTags.SHRIMP)
+
         // jellyfish
         getOrCreateTagBuilder(HybridAquaticEntityTags.JELLYFISH)
             .add(
@@ -220,5 +226,12 @@ class EntityTypeTagProvider(output: FabricDataOutput, registriesFuture: Completa
                 HybridAquaticEntityTypes.COMPASS_JELLYFISH,
                 HybridAquaticEntityTypes.BLUE_JELLYFISH,
             )
+
+        // fishing net
+        getOrCreateTagBuilder(HybridAquaticEntityTags.CAN_USE_FISHING_NET_ON)
+            .addTag(HybridAquaticEntityTags.JELLYFISH)
+            .addTag(HybridAquaticEntityTags.CRITTER)
+            .addTag(HybridAquaticEntityTags.SMALL_PREY)
+            .addTag(HybridAquaticEntityTags.MEDIUM_PREY)
     }
 }
diff --git a/src/main/kotlin/dev/hybridlabs/aquatic/item/FishingNetItem.kt b/src/main/kotlin/dev/hybridlabs/aquatic/item/FishingNetItem.kt
@@ -18,7 +18,7 @@ import java.util.*
 class FishingNetItem(settings: Settings?): Item(settings) {
 
     override fun useOnEntity(stack: ItemStack, user: PlayerEntity, entity: LivingEntity, hand: Hand): ActionResult {
-        val validFishForNet = entity.type.isIn(HybridAquaticEntityTags.JELLYFISH) || entity.type.isIn(HybridAquaticEntityTags.CRITTER) || entity.type.isIn(HybridAquaticEntityTags.SMALL_PREY) || entity.type.isIn(HybridAquaticEntityTags.MEDIUM_PREY)
+        val validFishForNet = entity.type.isIn(HybridAquaticEntityTags.CAN_USE_FISHING_NET_ON)
 
         if (!alreadyHasFish(stack) && validFishForNet) {
             writeEntityToNet(entity, user, hand)

diff --git a/src/main/kotlin/dev/hybridlabs/aquatic/loot/entry/MessageInABottleItemEntry.kt b/src/main/kotlin/dev/hybridlabs/aquatic/loot/entry/MessageInABottleItemEntry.kt
@@ -52,6 +52,7 @@ class MessageInABottleItemEntry(
          */
         val CODEC: Codec<MessageInABottleItemEntry> = RecordCodecBuilder.create { instance ->
             val default = addLeafFields(instance)
+
             instance.group(
                     default.t1(),
                     default.t2(),

diff --git a/src/main/kotlin/dev/hybridlabs/aquatic/tag/HybridAquaticEntityTags.kt b/src/main/kotlin/dev/hybridlabs/aquatic/tag/HybridAquaticEntityTags.kt
@@ -28,6 +28,8 @@ object HybridAquaticEntityTags {
     val FISHES = create("fishes")
     val SHARKS = create("sharks")
 
+    val CAN_USE_FISHING_NET_ON = create("can_use_fishing_net_on")
+
     private fun create(id: String): TagKey<EntityType<*>> {
         return TagKey.of(RegistryKeys.ENTITY_TYPE, Identifier(HybridAquatic.MOD_ID, id))
     }