working itemhoarder

src/main/java/de/jottyfan/minecraft/quickiefabric/blockentity/ItemHoarderBlockEntity.java

@@ -1,130 +1,123 @@
 package de.jottyfan.minecraft.quickiefabric.blockentity;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.Iterator;
 import java.util.List;
-import java.util.ListIterator;
-import java.util.Map;
 
 import net.minecraft.block.BlockState;
-import net.minecraft.block.entity.BlockEntity;
+import net.minecraft.block.entity.LootableContainerBlockEntity;
+import net.minecraft.entity.Entity;
+import net.minecraft.entity.ItemEntity;
+import net.minecraft.entity.player.PlayerInventory;
+import net.minecraft.inventory.Inventories;
 import net.minecraft.item.ItemStack;
 import net.minecraft.nbt.CompoundTag;
-import net.minecraft.nbt.ListTag;
+import net.minecraft.screen.ScreenHandler;
-import net.minecraft.nbt.Tag;
 import net.minecraft.text.Text;
+import net.minecraft.text.TranslatableText;
+import net.minecraft.util.Tickable;
+import net.minecraft.util.collection.DefaultedList;
+import net.minecraft.util.math.BlockPos;
+import net.minecraft.util.math.Box;
+import net.minecraft.world.World;
 
 /**
  * 
  * @author jotty
  *
  */
-public class ItemHoarderBlockEntity extends BlockEntity {
+public class ItemHoarderBlockEntity extends LootableContainerBlockEntity implements Tickable {
-
+	private DefaultedList<ItemStack> stacks;
-	private final String NBT_STACKS = "stacks";
-	private final List<ItemStack> stacks;
 
 	public ItemHoarderBlockEntity() {
 		super(QuickieFabricBlockEntity.ITEMHOARDER);
-		stacks = new ArrayList<>();
+		stacks = DefaultedList.ofSize(54, ItemStack.EMPTY);
-	}
-
-	private final Integer getNextFreeEmptyStackPosition() {
-		Integer emptyStackPosition = null;
-		Iterator<ItemStack> it = stacks.iterator();
-		Integer index = 0;
-		while (emptyStackPosition == null && it.hasNext()) {
-			ItemStack s = it.next();
-			if (s.isEmpty()) {
-				emptyStackPosition = index;
-			}
-			index++;
-		}
-		return emptyStackPosition;
 	}
 
 	// TODO: see https://fabricmc.net/wiki/tutorial:containers for a real chest
-	// TODO: find a way to suck in the surrounding items
 
-//	@Override
+	private final boolean setStackToSlots(ItemStack stack) {
-//	public void tick() {
+		for (ItemStack slot : stacks) {
-//		BlockPos pos = getPos();
+			if (slot.getItem().equals(stack.getItem())) {
-//		World world = getWorld();
+				slot.increment(stack.getCount());
-//		// list all items around here
+				return true;
-//		AxisAlignedBB aabb = new AxisAlignedBB(pos).grow(4);
+			}
-//		List<ItemEntity> itemEntities = world.getEntitiesWithinAABB(ItemEntity.class, aabb);
+		} // if not found, seek for an empty stack instead
-//		boolean dirty = false;
+		for (ItemStack slot : stacks) {
-//		for (ItemEntity itemEntity : itemEntities) {
+			if (slot.isEmpty()) {
-//			if (itemEntity.isAlive()) {
+				Integer index = stacks.indexOf(slot);
-//				ItemStack stack = itemEntity.getItem();
+				stacks.set(index, stack.copy());
-//				if (stack != null) {
+				return true;
-//					Integer emptyStackPosition = getNextFreeEmptyStackPosition();
-//					if (emptyStackPosition == null) {
-//					} else {
-//						stacks.set(emptyStackPosition, stack);
-//						itemEntity.remove();
-//						dirty = true;
-//					}
-//				}
-//			} else {
-//				itemEntity.getItem().getItem().getRegistryName().getPath());
-//			}
-//		}
-//		if (dirty) {
-//			resortStacks();
-//		}
-//	}
-
-	/**
-	 * resort item stacks to be more efficient
-	 */
-	private void resortStacks() {
-		Map<Text, ItemStack> map = new HashMap<>();
-		Integer clearAmount = stacks.size();
-		for (ItemStack stack : stacks) {
-			ItemStack found = map.get(stack.getName());
-			if (found == null) {
-				found = stack;
-				map.put(stack.getName(), found);
-			} else {
-				found.setCount(found.getCount() + stack.getCount());
 			}
 		}
-		for (Integer i = 0; i < clearAmount; i++) {
+		return false;
-			stacks.set(i, ItemStack.EMPTY);
+	}
-		}
+
-		Integer index = 0;
+	@Override
-		for (ItemStack stack : map.values()) {
+	public void tick() {
-			stacks.set(index++, stack);
+		BlockPos pos = getPos();
+		World world = getWorld();
+		float suckradius = 4f; // TODO: make it level up - able and start with 2
+		Box box = new Box(pos).expand(suckradius);
+		List<Entity> entities = world.getEntities(null, box);
+		for (Entity entity : entities) {
+			if (entity instanceof ItemEntity) {
+				ItemEntity itemEntity = (ItemEntity) entity;
+				if (itemEntity.isAlive()) {
+					ItemStack stack = itemEntity.getStack();
+					if (stack != null) {
+						if (setStackToSlots(stack)) {
+							itemEntity.remove();
+						} // else inventory is full
+					}
+				}
+			}
 		}
 	}
 
 	@Override
 	public CompoundTag toTag(CompoundTag tag) {
 		super.toTag(tag);
-		ListTag listTag = new ListTag();
+		if (!this.serializeLootTable(tag)) {
-		for (ItemStack stack : stacks) {
+			Inventories.toTag(tag, this.stacks);
-			listTag.add(stack.toTag(tag));
 		}
-		tag.put(NBT_STACKS, listTag);
 		return tag;
 	}
 
 	@Override
 	public void fromTag(BlockState state, CompoundTag tag) {
 		super.fromTag(state, tag);
-		ListTag listTag = (ListTag) tag.get(NBT_STACKS);
+		this.stacks = DefaultedList.ofSize(this.size(), ItemStack.EMPTY);
-		ListIterator<Tag> i = listTag.listIterator();
+		if (!this.deserializeLootTable(tag)) {
-		stacks.clear();
+			Inventories.fromTag(tag, this.stacks);
-		while (i.hasNext()) {
-			CompoundTag foundTag = (CompoundTag) i.next();
-			ItemStack stack = ItemStack.fromTag(foundTag);
-			stacks.add(stack);
 		}
 	}
 
 	public List<ItemStack> getStacks() {
 		return stacks;
 	}
+
+	@Override
+	public int size() {
+		return 54; // container chest size (9 * 6)
+	}
+
+	@Override
+	protected DefaultedList<ItemStack> getInvStackList() {
+		return stacks;
+	}
+
+	@Override
+	protected void setInvStackList(DefaultedList<ItemStack> list) {
+		this.stacks = list;
+	}
+
+	@Override
+	protected ScreenHandler createScreenHandler(int syncId, PlayerInventory playerInventory) {
+		// TODO: implement, see https://fabricmc.net/wiki/tutorial:containers
+		return null;
+	}
+
+	@Override
+	protected Text getContainerName() {
+		return new TranslatableText("container.itemhoarder");
+	}
 }

src/main/java/de/jottyfan/minecraft/quickiefabric/blocks/BlockItemhoarder.java

@@ -1,12 +1,22 @@
 package de.jottyfan.minecraft.quickiefabric.blocks;
 
+import java.util.ArrayList;
+import java.util.List;
+
 import de.jottyfan.minecraft.quickiefabric.blockentity.ItemHoarderBlockEntity;
 import net.fabricmc.fabric.api.object.builder.v1.block.FabricBlockSettings;
 import net.minecraft.block.Block;
 import net.minecraft.block.BlockEntityProvider;
+import net.minecraft.block.BlockState;
 import net.minecraft.block.Material;
 import net.minecraft.block.entity.BlockEntity;
+import net.minecraft.entity.player.PlayerEntity;
+import net.minecraft.item.ItemStack;
+import net.minecraft.loot.context.LootContext.Builder;
+import net.minecraft.util.ItemScatterer;
+import net.minecraft.util.math.BlockPos;
 import net.minecraft.world.BlockView;
+import net.minecraft.world.World;
 
 /**
  * 
@@ -20,7 +30,26 @@ public class BlockItemhoarder extends Block implements BlockEntityProvider {
 	}
 
 	@Override
-	public BlockEntity createBlockEntity(BlockView arg0) {
+	public BlockEntity createBlockEntity(BlockView blockView) {
 		return new ItemHoarderBlockEntity();
 	}
+
+	@Override
+	public List<ItemStack> getDroppedStacks(BlockState state, Builder builder) {
+		List<ItemStack> list = new ArrayList<>();
+		list.add(new ItemStack(QuickieBlocks.ITEMHOARDER));
+		return list;
+	}
+
+	@Override
+	public void onBreak(World world, BlockPos pos, BlockState state, PlayerEntity player) {
+		BlockEntity blockEntity = world.getBlockEntity(pos);
+		if (blockEntity instanceof ItemHoarderBlockEntity) {
+			ItemHoarderBlockEntity ihbe = (ItemHoarderBlockEntity) blockEntity;
+			for (ItemStack stack : ihbe.getStacks()) {
+				ItemScatterer.spawn(world, pos.getX(), pos.getY(), pos.getZ(), stack);
+			}
+		}
+		super.onBreak(world, pos, state, player);
+	}
 }

src/main/java/de/jottyfan/minecraft/quickiefabric/blocks/BlockLavahoarder.java

@@ -73,7 +73,7 @@ public class BlockLavahoarder extends Block {
 	@Override
 	public void onPlaced(World world, BlockPos pos, BlockState state, LivingEntity placer, ItemStack stack) {
 		Set<String> positions = new HashSet<>();
-		Integer counter = 8;
+		Integer counter = 8; // TODO: make it level up - able
 		findAllAttachedLavaBlocks(positions, pos.up(), world, counter);
 		findAllAttachedLavaBlocks(positions, pos.down(), world, counter);
 		findAllAttachedLavaBlocks(positions, pos.north(), world, counter);