feat: Add BalmRegistries.getAttribute

Author: BlayTheNinth

fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmRegistries.java

@@ -9,6 +9,7 @@
 import net.minecraft.tags.TagKey;
 import net.minecraft.world.effect.MobEffect;
 import net.minecraft.world.entity.EntityType;
+import net.minecraft.world.entity.ai.attributes.Attribute;
 import net.minecraft.world.item.Item;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.material.Fluid;
@@ -77,4 +78,9 @@ public Fluid getMilkFluid() {
     public Collection<ResourceLocation> getItemKeys() {
         return BuiltInRegistries.ITEM.keySet();
     }
+
+    @Override
+    public Attribute getAttribute(ResourceLocation key) {
+        return BuiltInRegistries.ATTRIBUTE.get(key);
+    }
 }

forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmRegistries.java

@@ -6,6 +6,7 @@
 import net.minecraft.tags.TagKey;
 import net.minecraft.world.effect.MobEffect;
 import net.minecraft.world.entity.EntityType;
+import net.minecraft.world.entity.ai.attributes.Attribute;
 import net.minecraft.world.item.Item;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.material.Fluid;
@@ -74,4 +75,9 @@ public void enableMilkFluid() {
     public Fluid getMilkFluid() {
         return ForgeMod.MILK.get();
     }
+
+    @Override
+    public Attribute getAttribute(ResourceLocation key) {
+        return ForgeRegistries.ATTRIBUTES.getValue(key);
+    }
 }

shared/src/main/java/net/blay09/mods/balm/api/BalmRegistries.java

@@ -4,6 +4,7 @@
 import net.minecraft.tags.TagKey;
 import net.minecraft.world.effect.MobEffect;
 import net.minecraft.world.entity.EntityType;
+import net.minecraft.world.entity.ai.attributes.Attribute;
 import net.minecraft.world.item.Item;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.material.Fluid;
@@ -34,4 +35,6 @@ public interface BalmRegistries {
     void enableMilkFluid();
 
     Fluid getMilkFluid();
+
+    Attribute getAttribute(ResourceLocation key);
 }