fix: fix Primordial Cradle not persisting internal sacrifice values w…

…hen mined
Elenterius · Nov 8, 2023 · ac17061 · ac17061
1 parent 0596b36
commit ac17061
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/datagen/java/com/github/elenterius/biomancy/datagen/loot/ModBlockLoot.java b/src/datagen/java/com/github/elenterius/biomancy/datagen/loot/ModBlockLoot.java
@@ -57,6 +57,14 @@ protected static LootTable.Builder createNameableBioMachineTable(Block block) {
 				)));
 	}
 
+	protected static LootTable.Builder createPrimordialCradleTable(Block block) {
+		return LootTable.lootTable().withPool(applyExplosionCondition(block, LootPool.lootPool().setRolls(ConstantValue.exactly(1))
+				.add(LootItem.lootTableItem(block)
+						.apply(CopyNbtFunction.copyData(ContextNbtProvider.BLOCK_ENTITY).copy("PrimalEnergy", "BlockEntityTag.PrimalEnergy"))
+						.apply(CopyNbtFunction.copyData(ContextNbtProvider.BLOCK_ENTITY).copy("SacrificeHandler", "BlockEntityTag.SacrificeHandler"))
+				)));
+	}
+
 	protected static LootTable.Builder dropWithInventory(Block block) {
 		return LootTable.lootTable().withPool(applyExplosionCondition(block, LootPool.lootPool().setRolls(ConstantValue.exactly(1))
 				.add(LootItem.lootTableItem(block)
@@ -114,7 +122,7 @@ protected static LootTable.Builder createFleshSpikeTable(FleshSpikeBlock block)
 	protected void addTables() {
 		LOGGER.info(logMarker, "registering block loot...");
 
-		dropSelf(ModBlocks.PRIMORDIAL_CRADLE.get());
+		add(ModBlocks.PRIMORDIAL_CRADLE.get(), ModBlockLoot::createPrimordialCradleTable);
 		dropSelf(ModBlocks.TONGUE.get());
 		dropSelf(ModBlocks.MAW_HOPPER.get());
 		add(ModBlocks.STORAGE_SAC.get(), ModBlockLoot::dropWithInventory);