From 064866748042d34c826533846ce0c7f6a4f49697 Mon Sep 17 00:00:00 2001
From: DBotThePony <dbotthepony@yandex.ru>
Date: Sun, 25 Jun 2023 10:54:19 +0700
Subject: [PATCH] Revert "Forge seemingly removed its lootpool deserializer
 patches"

This reverts commit 926b9a2a1708c1b572fd7c223df6a955608bc3da.
---
 .../mc/otm/data/loot/LootPoolAppender.kt      | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/data/loot/LootPoolAppender.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/data/loot/LootPoolAppender.kt
index 9f1ff7ec8..1b473af22 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/data/loot/LootPoolAppender.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/data/loot/LootPoolAppender.kt
@@ -43,14 +43,35 @@ class LootPoolAppender(conditions: Array<out LootItemCondition>, pools: Stream<L
 	companion object {
 		private val lootPoolCodec: Codec<List<LootPool>>
 
+		private val forgeHooksLootContext by lazy {
+			val field = ForgeHooks::class.java.getDeclaredField("lootContext")
+			field.isAccessible = true
+			field
+		}
+
+		private val lootTableContextConstructor by lazy {
+			val clazz = ForgeHooks::class.java.declaredClasses.firstOrNull { it.name.contains("LootTableContext") } ?: throw NoSuchElementException("Unable to find ForgeHooks\$LootTableContext!")
+			val constructor = clazz.getDeclaredConstructor(ResourceLocation::class.java, Boolean::class.java)
+			constructor.isAccessible = true
+			constructor
+		}
+
 		init {
 			val serializer = Deserializers.createLootTableSerializer().create()
+			val notExistingLocation = ResourceLocation("null", "null")
 
 			lootPoolCodec = Codec.list(Codec.PASSTHROUGH.flatXmap({
+				val dequeueHolder = forgeHooksLootContext.get(null) as ThreadLocal<Deque<Any>>
+				val deque = dequeueHolder.get() ?: java.util.ArrayDeque()
+				dequeueHolder.set(deque)
+
 				try {
+					deque.push(lootTableContextConstructor.newInstance(notExistingLocation, true))
 					DataResult.success(serializer.fromJson(it.convert(JsonOps.INSTANCE).value, LootPool::class.java))
 				} catch(err: JsonSyntaxException) {
 					DataResult.error { err.message }
+				} finally {
+					deque.pop()
 				}
 			}, {
 				try {