package com.latexcatdrone.ai; import com.latexcatdrone.task.Task; import net.minecraft.block.BlockState; import net.minecraft.block.Blocks; import net.minecraft.entity.ai.pathing.Path; import net.minecraft.entity.player.PlayerEntity; import net.minecraft.item.ItemStack; import net.minecraft.item.Items; import net.minecraft.server.network.ServerPlayerEntity; import net.minecraft.util.Hand; import net.minecraft.util.hit.BlockHitResult; import net.minecraft.util.math.BlockPos; import net.minecraft.util.math.Direction; import net.minecraft.util.math.Vec3d; import net.minecraft.world.World; public class AIController { private final Task task; private boolean active = false; private int actionTimer = 0; private AIState currentState = AIState.GATHERING; private BlockPos targetPos = null; public AIController(Task task) { this.task = task; } public void start(ServerPlayerEntity player) { active = true; currentState = AIState.GATHERING; } public boolean isActive() { return active; } public boolean isTaskComplete() { return !active; } public void tick(ServerPlayerEntity player) { if (!active) return; actionTimer++; if (task.isCompleted(player)) { active = false; return; } if (actionTimer % 20 == 0) { performAction(player); } } private void performAction(ServerPlayerEntity player) { World world = player.getWorld(); switch (task.getType()) { case BUILD -> performBuildAction(player, world); case COLLECT -> performCollectAction(player, world); case CRAFT -> performCraftAction(player); case EXPLORE -> performExploreAction(player, world); } } private void performBuildAction(ServerPlayerEntity player, World world) { if (targetPos == null) { targetPos = player.getBlockPos().add(5, 0, 5); } ItemStack buildMaterial = findBuildMaterial(player); if (buildMaterial.isEmpty()) { gatherWood(player, world); return; } BlockPos placePos = findPlacePosition(player, world); if (placePos != null && world.getBlockState(placePos).isAir()) { player.getInventory().selectedSlot = findBuildMaterialSlot(player); BlockHitResult hitResult = new BlockHitResult( Vec3d.ofCenter(placePos), Direction.UP, placePos.down(), false ); player.interactionManager.interactBlock( player, world, player.getStackInHand(Hand.MAIN_HAND), Hand.MAIN_HAND, hitResult ); } } private void performCollectAction(ServerPlayerEntity player, World world) { gatherWood(player, world); } private void performCraftAction(ServerPlayerEntity player) { player.openHandledScreen(player.currentScreenHandler); } private void performExploreAction(ServerPlayerEntity player, World world) { BlockPos orePos = findNearbyOre(player, world); if (orePos != null) { mineBlock(player, world, orePos); } else { moveRandomly(player); } } private void gatherWood(ServerPlayerEntity player, World world) { BlockPos treePos = findNearbyTree(player, world); if (treePos != null) { mineBlock(player, world, treePos); } } private BlockPos findNearbyTree(PlayerEntity player, World world) { BlockPos playerPos = player.getBlockPos(); for (int x = -10; x <= 10; x++) { for (int y = -5; y <= 5; y++) { for (int z = -10; z <= 10; z++) { BlockPos pos = playerPos.add(x, y, z); BlockState state = world.getBlockState(pos); if (state.isOf(Blocks.OAK_LOG) || state.isOf(Blocks.BIRCH_LOG) || state.isOf(Blocks.SPRUCE_LOG) || state.isOf(Blocks.DARK_OAK_LOG)) { return pos; } } } } return null; } private BlockPos findNearbyOre(PlayerEntity player, World world) { BlockPos playerPos = player.getBlockPos(); for (int x = -15; x <= 15; x++) { for (int y = -10; y <= 10; y++) { for (int z = -15; z <= 15; z++) { BlockPos pos = playerPos.add(x, y, z); if (world.getBlockState(pos).isOf(Blocks.DIAMOND_ORE) || world.getBlockState(pos).isOf(Blocks.DEEPSLATE_DIAMOND_ORE)) { return pos; } } } } return null; } private void mineBlock(ServerPlayerEntity player, World world, BlockPos pos) { player.interactionManager.tryBreakBlock(pos); } private ItemStack findBuildMaterial(PlayerEntity player) { for (int i = 0; i < player.getInventory().size(); i++) { ItemStack stack = player.getInventory().getStack(i); if (stack.isOf(Items.COBBLESTONE) || stack.isOf(Items.OAK_PLANKS) || stack.isOf(Items.SPRUCE_PLANKS) || stack.isOf(Items.BIRCH_PLANKS)) { return stack; } } return ItemStack.EMPTY; } private int findBuildMaterialSlot(PlayerEntity player) { for (int i = 0; i < player.getInventory().size(); i++) { ItemStack stack = player.getInventory().getStack(i); if (stack.isOf(Items.COBBLESTONE) || stack.isOf(Items.OAK_PLANKS)) { return i; } } return 0; } private BlockPos findPlacePosition(PlayerEntity player, World world) { BlockPos center = player.getBlockPos(); for (int x = -5; x <= 5; x++) { for (int y = 0; y <= 3; y++) { for (int z = -5; z <= 5; z++) { BlockPos pos = center.add(x, y, z); if (world.getBlockState(pos).isAir() && !world.getBlockState(pos.down()).isAir()) { return pos; } } } } return null; } private void moveRandomly(ServerPlayerEntity player) { double randomX = (Math.random() - 0.5) * 0.2; double randomZ = (Math.random() - 0.5) * 0.2; player.setVelocity(randomX, player.getVelocity().y, randomZ); } private enum AIState { GATHERING, BUILDING, CRAFTING, MOVING } }