package com.example.villagersettlement.blockentity; import net.minecraft.block.*; import net.minecraft.block.entity.BlockEntity; import net.minecraft.block.entity.ChestBlockEntity; import net.minecraft.item.ItemStack; import net.minecraft.item.Items; import net.minecraft.server.world.ServerWorld; import net.minecraft.storage.ReadView; import net.minecraft.storage.WriteView; import net.minecraft.util.math.BlockPos; import net.minecraft.util.math.ChunkPos; import java.util.UUID; public class FarmerWorkstationBlockEntity extends BlockEntity { private UUID assignedVillagerUuid; private BlockPos chestPos; private int workCooldown = 0; private static final int WORK_INTERVAL = 80; private static final int SEARCH_RANGE_CHUNKS = 2; public FarmerWorkstationBlockEntity(BlockPos pos, BlockState state) { super(ModBlockEntities.FARMER_WORKSTATION, pos, state); } @Override protected void writeData(WriteView view) { super.writeData(view); if (assignedVillagerUuid != null) { view.putLong("VillagerMost", assignedVillagerUuid.getMostSignificantBits()); view.putLong("VillagerLeast", assignedVillagerUuid.getLeastSignificantBits()); } if (chestPos != null) { view.putInt("ChestX", chestPos.getX()); view.putInt("ChestY", chestPos.getY()); view.putInt("ChestZ", chestPos.getZ()); } view.putInt("WorkCooldown", workCooldown); } @Override protected void readData(ReadView view) { super.readData(view); long most = view.getLong("VillagerMost", 0L); long least = view.getLong("VillagerLeast", 0L); if (most != 0L || least != 0L) { assignedVillagerUuid = new UUID(most, least); } if (view.getInt("ChestX", Integer.MIN_VALUE) != Integer.MIN_VALUE) { int x = view.getInt("ChestX", 0); int y = view.getInt("ChestY", 0); int z = view.getInt("ChestZ", 0); chestPos = new BlockPos(x, y, z); } workCooldown = view.getInt("WorkCooldown", 0); } public void tick(ServerWorld world) { if (assignedVillagerUuid == null || chestPos == null) { return; } workCooldown--; if (workCooldown <= 0) { performWork(world); workCooldown = WORK_INTERVAL; } } private void performWork(ServerWorld world) { BlockPos targetCrop = findRipeCrop(world); if (targetCrop != null) { harvestCrop(world, targetCrop); return; } BlockPos targetFarmland = findEmptyFarmland(world); if (targetFarmland != null) { plantCrop(world, targetFarmland); } } private void harvestCrop(ServerWorld world, BlockPos cropPos) { BlockState state = world.getBlockState(cropPos); Block.dropStacks(state, world, cropPos); world.setBlockState(cropPos, Blocks.AIR.getDefaultState()); BlockEntity chestEntity = world.getBlockEntity(chestPos); if (chestEntity instanceof ChestBlockEntity chest) { ItemStack droppedStack = new ItemStack(state.getBlock().asItem()); for (int i = 0; i < chest.size(); i++) { ItemStack slotStack = chest.getStack(i); if (slotStack.isEmpty()) { chest.setStack(i, droppedStack); break; } else if (ItemStack.areItemsAndComponentsEqual(slotStack, droppedStack)) { int space = slotStack.getMaxCount() - slotStack.getCount(); if (space > 0) { slotStack.increment(Math.min(space, droppedStack.getCount())); break; } } } } } private void plantCrop(ServerWorld world, BlockPos farmlandPos) { BlockPos plantPos = farmlandPos.up(); if (!world.getBlockState(plantPos).isAir()) { return; } world.setBlockState(plantPos, Blocks.WHEAT.getDefaultState()); } private BlockPos findRipeCrop(ServerWorld world) { ChunkPos centerChunk = new ChunkPos(pos); for (int cx = -SEARCH_RANGE_CHUNKS; cx <= SEARCH_RANGE_CHUNKS; cx++) { for (int cz = -SEARCH_RANGE_CHUNKS; cz <= SEARCH_RANGE_CHUNKS; cz++) { ChunkPos searchChunk = new ChunkPos(centerChunk.x + cx, centerChunk.z + cz); int startX = searchChunk.getStartX(); int startZ = searchChunk.getStartZ(); for (int x = startX; x < startX + 16; x++) { for (int z = startZ; z < startZ + 16; z++) { for (int y = world.getBottomY(); y < world.getTopY(); y++) { BlockPos checkPos = new BlockPos(x, y, z); BlockState state = world.getBlockState(checkPos); if (state.getBlock() instanceof CropBlock crop) { if (crop.isMature(state)) { return checkPos; } } } } } } } return null; } private BlockPos findEmptyFarmland(ServerWorld world) { ChunkPos centerChunk = new ChunkPos(pos); for (int cx = -SEARCH_RANGE_CHUNKS; cx <= SEARCH_RANGE_CHUNKS; cx++) { for (int cz = -SEARCH_RANGE_CHUNKS; cz <= SEARCH_RANGE_CHUNKS; cz++) { ChunkPos searchChunk = new ChunkPos(centerChunk.x + cx, centerChunk.z + cz); int startX = searchChunk.getStartX(); int startZ = searchChunk.getStartZ(); for (int x = startX; x < startX + 16; x++) { for (int z = startZ; z < startZ + 16; z++) { for (int y = world.getBottomY(); y < world.getTopY(); y++) { BlockPos checkPos = new BlockPos(x, y, z); BlockState state = world.getBlockState(checkPos); if (state.getBlock() == Blocks.FARMLAND) { BlockPos above = checkPos.up(); if (world.getBlockState(above).isAir()) { return checkPos; } } } } } } } return null; } public void assignVillager(UUID uuid) { this.assignedVillagerUuid = uuid; markDirty(); } public void setChestPos(BlockPos pos) { this.chestPos = pos; markDirty(); } public boolean hasAssignedVillager() { return assignedVillagerUuid != null; } public UUID getAssignedVillagerUuid() { return assignedVillagerUuid; } public void onRemoved(ServerWorld world) { assignedVillagerUuid = null; chestPos = null; } }