Add jungle pyramid loot

Coffee beans now have a chance of spawning in jungle pyramid chests

Author: Chikorita-Lover

src/main/java/com/chikoritalover/caffeinated/Caffeinated.java

@@ -2,6 +2,15 @@
 
 import com.chikoritalover.caffeinated.registry.*;
 import net.fabricmc.api.ModInitializer;
+import net.fabricmc.fabric.api.loot.v2.LootTableEvents;
+import net.minecraft.item.ItemConvertible;
+import net.minecraft.loot.LootPool;
+import net.minecraft.loot.LootTables;
+import net.minecraft.loot.condition.LootCondition;
+import net.minecraft.loot.condition.RandomChanceLootCondition;
+import net.minecraft.loot.entry.ItemEntry;
+import net.minecraft.loot.provider.number.UniformLootNumberProvider;
+import net.minecraft.util.Identifier;
 import net.minecraft.util.registry.Registry;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -26,5 +35,30 @@ public void onInitialize() {
 		ModSoundEvents.register();
 		ModStatusEffects.register();
 		ModTradeOffers.register();
+
+		registerLootTableEvents();
+	}
+
+	public void registerLootTableEvents() {
+		addLootTablePool(1, 1, 0.5F, LootTables.JUNGLE_TEMPLE_CHEST, ModItems.COFFEE_BEANS, 1, 3);
+	}
+
+	private void addLootTablePool(int minRolls, int maxRolls, float chance, Identifier lootTable, ItemConvertible item) {
+		addLootTablePool(minRolls, maxRolls, chance, lootTable, item, 1, 1);
+	}
+
+	private void addLootTablePool(int minRolls, int maxRolls, float chance, Identifier lootTable, ItemConvertible item, int minCount, int maxCount) {
+		UniformLootNumberProvider lootTableRange = UniformLootNumberProvider.create(minRolls, maxRolls);
+		LootCondition condition = RandomChanceLootCondition.builder(chance).build();
+		LootTableEvents.MODIFY.register((resourceManager, lootManager, id, supplier, setter) -> {
+			if (lootTable.equals(id)) {
+				LootPool lootPool = LootPool.builder()
+						.rolls(lootTableRange)
+						.conditionally(condition)
+						.with(ItemEntry.builder(item).build()).build();
+
+				supplier.pool(lootPool);
+			}
+		});
 	}
 }