From 45a9405babe500f876d1be9981dfe187202dc1d9 Mon Sep 17 00:00:00 2001
From: Aleksey <bulldog.seregin@gmail.com>
Date: Thu, 11 Mar 2021 16:41:29 +0300
Subject: [PATCH] End Stone Smelter drop fix

---
 .../ru/betterend/blocks/EndStoneSmelter.java     | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/src/main/java/ru/betterend/blocks/EndStoneSmelter.java b/src/main/java/ru/betterend/blocks/EndStoneSmelter.java
index 533f653b..cfab2cc8 100644
--- a/src/main/java/ru/betterend/blocks/EndStoneSmelter.java
+++ b/src/main/java/ru/betterend/blocks/EndStoneSmelter.java
@@ -1,9 +1,11 @@
 package ru.betterend.blocks;
 
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
 import java.util.Random;
 
+import com.google.common.collect.Lists;
 import net.fabricmc.api.EnvType;
 import net.fabricmc.api.Environment;
 import net.fabricmc.fabric.api.object.builder.v1.block.FabricBlockSettings;
@@ -18,6 +20,7 @@ import net.minecraft.entity.player.PlayerEntity;
 import net.minecraft.item.ItemPlacementContext;
 import net.minecraft.item.ItemStack;
 import net.minecraft.loot.context.LootContext;
+import net.minecraft.loot.context.LootContextParameters;
 import net.minecraft.particle.ParticleTypes;
 import net.minecraft.screen.ScreenHandler;
 import net.minecraft.sound.BlockSoundGroup;
@@ -83,7 +86,18 @@ public class EndStoneSmelter extends BaseBlockWithEntity {
 	
 	@Override
 	public List<ItemStack> getDroppedStacks(BlockState state, LootContext.Builder builder) {
-		return Collections.singletonList(new ItemStack(this));
+		List<ItemStack> drop = Lists.newArrayList(new ItemStack(this));
+		BlockEntity blockEntity = builder.getNullable(LootContextParameters.BLOCK_ENTITY);
+		if (blockEntity instanceof EndStoneSmelterBlockEntity) {
+			EndStoneSmelterBlockEntity smelterBlockEntity = (EndStoneSmelterBlockEntity) blockEntity;
+			for (int i = 0; i < smelterBlockEntity.size(); i++) {
+				ItemStack item = smelterBlockEntity.getStack(i);
+				if (!item.isEmpty()) {
+					drop.add(item);
+				}
+			}
+		}
+		return drop;
 	}
 
 	@Override