From e28dcf1ee4032fcc58b1ec03ba5c14e66a269825 Mon Sep 17 00:00:00 2001
From: Aleksey <bulldog.seregin@gmail.com>
Date: Wed, 30 Jun 2021 16:23:32 +0300
Subject: [PATCH] Possibility to add NBT data for Anvil recipes output

---
 .../ru/betterend/recipe/builders/AnvilRecipe.java    | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/main/java/ru/betterend/recipe/builders/AnvilRecipe.java b/src/main/java/ru/betterend/recipe/builders/AnvilRecipe.java
index 2a9ed475..0bee9b39 100644
--- a/src/main/java/ru/betterend/recipe/builders/AnvilRecipe.java
+++ b/src/main/java/ru/betterend/recipe/builders/AnvilRecipe.java
@@ -4,9 +4,12 @@ import java.util.Objects;
 
 import com.google.gson.JsonObject;
 
+import com.mojang.brigadier.exceptions.CommandSyntaxException;
 import net.fabricmc.api.EnvType;
 import net.fabricmc.api.Environment;
 import net.minecraft.core.NonNullList;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.nbt.TagParser;
 import net.minecraft.network.FriendlyByteBuf;
 import net.minecraft.resources.ResourceLocation;
 import net.minecraft.tags.Tag;
@@ -276,6 +279,15 @@ public class AnvilRecipe implements Recipe<Container>, BetterEndRecipe {
 			if (output == null) {
 				throw new IllegalStateException("Output item does not exists!");
 			}
+			if (result.has("nbt")) {
+				try {
+					String nbtData = GsonHelper.getAsString(result, "nbt");
+					CompoundTag nbt = TagParser.parseTag(nbtData);
+					output.setTag(nbt);
+				} catch (CommandSyntaxException ex) {
+					BetterEnd.LOGGER.warning("Error parse nbt data for output.", ex);
+				}
+			}
 			int inputCount = GsonHelper.getAsInt(json, "inputCount", 1);
 			int toolLevel = GsonHelper.getAsInt(json, "toolLevel", 1);
 			int anvilLevel = GsonHelper.getAsInt(json, "anvilLevel", 1);