SeagullAIStealFromPlayers.java

package com.github.alexthe666.alexsmobs.entity.ai;

import com.github.alexthe666.alexsmobs.config.AMConfig;
import com.github.alexthe666.alexsmobs.entity.EntitySeagull;
import com.github.alexthe666.alexsmobs.misc.AMAdvancementTriggerRegistry;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.world.InteractionHand;
import net.minecraft.world.entity.EntitySelector;
import net.minecraft.world.entity.ai.goal.Goal;
import net.minecraft.world.entity.item.ItemEntity;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.phys.Vec3;
import net.minecraftforge.registries.ForgeRegistries;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;


import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;

public class SeagullAIStealFromPlayers extends Goal {

    private final EntitySeagull seagull;
    private Vec3 fleeVec = null;
    private Player target;
    private int fleeTime = 0;

    public SeagullAIStealFromPlayers(EntitySeagull entitySeagull) {
        this.setFlags(EnumSet.of(Goal.Flag.MOVE, Flag.TARGET));
        this.seagull = entitySeagull;
    }

    @Override
    public boolean canUse() {
        if (seagull.hasEatenSeed) {
            System.out.println("[DEBUG] Seagull " + seagull.getId() + " will not steal: has eaten a seed.");
            return false;
        }
        if (this.seagull.getNoActionTime() >= 100 || seagull.isSitting() || !AMConfig.seagullStealing) {
            // Only print if you are actively debugging
            // System.out.println("[DEBUG] Seagull cannot steal due to action time, sitting, or config.");
            return false;
        }
        if (this.seagull.getRandom().nextInt(12) != 0 || seagull.stealCooldown > 0) {
            // System.out.println("[DEBUG] Seagull cannot steal due to random chance or cooldown.");
            return false;
        }
        if (this.seagull.getMainHandItem().isEmpty()) {
            Player valid = getClosestValidPlayer();
            if (valid != null) {
                System.out.println("[DEBUG] Seagull " + seagull.getId() + " will try to steal from player: " + valid.getName().getString());
                target = valid;
                return true;
            }
        }
        // System.out.println("[DEBUG] Seagull did not find a valid player to steal from.");
        return false;
    }

    private boolean isSeed(ItemStack stack) {
        return stack.is(Items.WHEAT_SEEDS) ||
               stack.is(Items.BEETROOT_SEEDS) ||
               stack.is(Items.MELON_SEEDS) ||
               stack.is(Items.PUMPKIN_SEEDS) ||
               stack.is(Items.TORCHFLOWER_SEEDS) ||
               stack.is(Items.PITCHER_POD);
    }
    

    public void start(){
        this.seagull.aiItemFlag = true;
    }

    public void stop(){
        this.seagull.aiItemFlag = false;
        target = null;
        fleeVec = null;
        fleeTime = 0;
    }

    @Override
    public boolean canContinueToUse() {
        return target != null && !target.isCreative() && (seagull.getMainHandItem().isEmpty() || fleeTime > 0);
    }

    public void tick(){
        seagull.setFlying(true);
        seagull.getMoveControl().setWantedPosition(target.getX(), target.getEyeY(), target.getZ(), 1.2F);
        if(seagull.distanceTo(target) < 2F && seagull.getMainHandItem().isEmpty()){
            if(hasFoods(target)){
                ItemStack foodStack = getFoodItemFrom(target);
                if(!foodStack.isEmpty()){
                    ItemStack copy = foodStack.copy();
                    foodStack.shrink(1);
                    copy.setCount(1);
                    seagull.peck();
                    seagull.setItemInHand(InteractionHand.MAIN_HAND, copy);
                    fleeTime = 60;
                    seagull.stealCooldown = 1500 + seagull.getRandom().nextInt(1500);
                    if(target instanceof ServerPlayer){
                        AMAdvancementTriggerRegistry.SEAGULL_STEAL.trigger((ServerPlayer)target);
                    }
                }else{
                    stop();
                }
            }else{
                stop();
            }
        }
        if(fleeTime > 0){
            if(fleeVec == null){
                fleeVec = seagull.getBlockInViewAway(target.position(), 4);
            }
            if(fleeVec != null){
                seagull.setFlying(true);
                seagull.getMoveControl().setWantedPosition(fleeVec.x, fleeVec.y, fleeVec.z, 1.2F);
                if(seagull.distanceToSqr(fleeVec) < 5){
                    fleeVec = seagull.getBlockInViewAway(fleeVec, 4);
                }
            }
            fleeTime--;
        }
    }

    private Player getClosestValidPlayer(){
        List<Player> list = seagull.level().getEntitiesOfClass(Player.class, seagull.getBoundingBox().inflate(10, 25, 10), EntitySelector.NO_CREATIVE_OR_SPECTATOR);
        Player closest = null;
        if(!list.isEmpty()){
            for(Player player : list){
                if((closest == null || closest.distanceTo(seagull) > player.distanceTo(seagull)) && hasFoods(player)){
                    closest = player;
                }
            }
        }
        return closest;
    }

    private boolean hasFoods(Player player){
        for(int i = 0; i < 9; i++){
            ItemStack stackIn = player.getInventory().items.get(i);
            if(stackIn.isEdible() && !isBlacklisted(stackIn)){
                return true;
            }
        }
        return false;
    }

    private boolean isBlacklisted(ItemStack stack){
        ResourceLocation loc = ForgeRegistries.ITEMS.getKey(stack.getItem());
        if(loc != null){
            for(String str : AMConfig.seagullStealingBlacklist){
                if(loc.toString().equals(str)){
                    return true;
                }
            }
        }
        return false;
    }

    private ItemStack getFoodItemFrom(Player player){
        List<ItemStack> foods = new ArrayList<>();
        for(int i = 0; i < 9; i++){
            ItemStack stackIn = player.getInventory().items.get(i);
            if(stackIn.isEdible() && !isBlacklisted(stackIn)){
                foods.add(stackIn);
            }
        }
        if(!foods.isEmpty()){
            return foods.get(foods.size() <= 1 ? 0 : seagull.getRandom().nextInt(foods.size() - 1));
        }
        return ItemStack.EMPTY;
    }
}