From 28909812bbfe2560746171a3c7a280ef19329039 Mon Sep 17 00:00:00 2001
From: "J.T. McQuigg" <J.T.McQuigg12@gmail.com>
Date: Wed, 29 Nov 2023 22:27:54 -0500
Subject: [PATCH] Datagen addOptionalTag/s methods that allow passing the
 TagKey itself instead of passing the location (#9807)

---
 .../common/extensions/IForgeTagAppender.java        | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/main/java/net/minecraftforge/common/extensions/IForgeTagAppender.java b/src/main/java/net/minecraftforge/common/extensions/IForgeTagAppender.java
index d3f64324b6..6568d3ae5b 100644
--- a/src/main/java/net/minecraftforge/common/extensions/IForgeTagAppender.java
+++ b/src/main/java/net/minecraftforge/common/extensions/IForgeTagAppender.java
@@ -25,6 +25,19 @@ public interface IForgeTagAppender<T>
         return builder;
     }
 
+    default TagsProvider.TagAppender<T> addOptionalTag(TagKey<T> value) {
+        return self().addOptionalTag(value.location());
+    }
+
+    @SuppressWarnings("unchecked")
+    default TagsProvider.TagAppender<T> addOptionalTags(TagKey<T>... values) {
+        TagsProvider.TagAppender<T> builder = self();
+        for (TagKey<T> value : values) {
+            builder.addOptionalTag(value.location());
+        }
+        return builder;
+    }
+
     default TagsProvider.TagAppender<T> replace() {
         return replace(true);
     }