feat: Add BalmRegistries.getAttribute

TwelveIterationMods · Sep 13, 2023 · dbe234c · dbe234c
1 parent 54d1292
commit dbe234c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmRegistries.java b/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);
+    }
 }
diff --git a/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmRegistries.java b/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);
+    }
 }
diff --git a/shared/src/main/java/net/blay09/mods/balm/api/BalmRegistries.java b/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);
 }