From 91c70885b89bacfed0e8f1806f5f45bddf1113c3 Mon Sep 17 00:00:00 2001
From: Hunter19823 <hspragg@asu.edu>
Date: Mon, 4 Mar 2024 15:16:20 -0700
Subject: [PATCH] Added a mixin to ShapedRecipeMixin that appends NBT data to
 ItemStack when parsing result.

---
 .../core/mixin/fabric/ShapedRecipeMixin.java  | 54 +++++++++++++++++++
 .../main/resources/kubejs-fabric.mixins.json  |  1 +
 2 files changed, 55 insertions(+)
 create mode 100644 fabric/src/main/java/dev/latvian/mods/kubejs/core/mixin/fabric/ShapedRecipeMixin.java

diff --git a/fabric/src/main/java/dev/latvian/mods/kubejs/core/mixin/fabric/ShapedRecipeMixin.java b/fabric/src/main/java/dev/latvian/mods/kubejs/core/mixin/fabric/ShapedRecipeMixin.java
new file mode 100644
index 000000000..0f040c599
--- /dev/null
+++ b/fabric/src/main/java/dev/latvian/mods/kubejs/core/mixin/fabric/ShapedRecipeMixin.java
@@ -0,0 +1,54 @@
+package dev.latvian.mods.kubejs.core.mixin.fabric;
+
+import com.google.gson.JsonObject;
+import com.google.gson.JsonSyntaxException;
+import com.mojang.brigadier.exceptions.CommandSyntaxException;
+import dev.latvian.mods.rhino.mod.util.NBTUtils;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.nbt.TagParser;
+import net.minecraft.util.GsonHelper;
+import net.minecraft.world.item.ItemStack;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
+
+@Mixin(net.minecraft.world.item.crafting.ShapedRecipe.class)
+public class ShapedRecipeMixin {
+	@Inject(
+		method = "itemStackFromJson",
+		at = @At(
+			value = "RETURN",
+			remap = false
+		),
+		cancellable = false
+	)
+	private static void itemStackFromJson(JsonObject jsonObject, CallbackInfoReturnable<ItemStack> cir) {
+		if(cir.getReturnValue() == null || cir.getReturnValue() == ItemStack.EMPTY) {
+			return;
+		}
+
+		if(jsonObject.has("nbt")) {
+			var json = jsonObject.get("nbt");
+			// Process null
+			if (json == null || json.isJsonNull()) {
+				return;
+			}
+
+			try {
+				CompoundTag tag = null;
+				if (json.isJsonObject()) {
+					// We use a normal .toString() to convert the json to string, and read it as SNBT.
+					// Using DynamicOps would mess with the type of integers and cause things like damage comparisons to fail...
+					tag = TagParser.parseTag(json.toString());
+				} else {
+					// Assume it's a string representation of the NBT
+					tag = TagParser.parseTag(GsonHelper.convertToString(json, "nbt"));
+				}
+				cir.getReturnValue().getOrCreateTag().merge(tag);
+			} catch (CommandSyntaxException commandSyntaxException) {
+				throw new JsonSyntaxException("Invalid nbt tag: " + commandSyntaxException.getMessage());
+			}
+		}
+	}
+}
diff --git a/fabric/src/main/resources/kubejs-fabric.mixins.json b/fabric/src/main/resources/kubejs-fabric.mixins.json
index a961b44e0..974621e10 100644
--- a/fabric/src/main/resources/kubejs-fabric.mixins.json
+++ b/fabric/src/main/resources/kubejs-fabric.mixins.json
@@ -8,6 +8,7 @@
 		"DifferenceIngredientMixin",
 		"IngredientMatchingMixin",
 		"NbtIngredientMixin",
+		"ShapedRecipeMixin",
 		"tools.shears.BlockInteractShearsMixin",
 		"tools.shears.EntityInteractShearsMixin",
 		"tools.shears.MatchToolMixin",