Added a fluid tag getter

InnovativeOnlineIndustries · Apr 3, 2021 · 35cdae6 · 35cdae6
1 parent ffd48b2
commit 35cdae6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/com/hrznstudio/titanium/util/TagUtil.java b/src/main/java/com/hrznstudio/titanium/util/TagUtil.java
@@ -84,6 +84,13 @@ public static ITag<EntityType<?>> getEntityTypeTag(ResourceLocation resourceLoca
         return EntityTypeTags.getTagById(resourceLocation.toString());
     }
 
+    public static ITag<Fluid> getFluidTag(ResourceLocation resourceLocation) {
+        if (FluidTags.getCollection().getRegisteredTags().contains(resourceLocation)) {
+            return FluidTags.getCollection().get(resourceLocation);
+        }
+        return FluidTags.makeWrapperTag(resourceLocation.toString());
+    }
+
     public static ItemStack getItemWithPreference(ITag<Item> tag){
         if (tag.getAllElements().isEmpty()) return ItemStack.EMPTY;
         List<Item> elements = tag.getAllElements();