package com.thebrokenrail.twine.entity;

import com.thebrokenrail.twine.component.StageDataComponent;
import net.minecraft.entity.LivingEntity;
import net.minecraft.entity.ai.goal.FollowTargetGoal;
import net.minecraft.entity.mob.Angerable;
import net.minecraft.entity.mob.HostileEntity;
import net.minecraft.entity.mob.MobEntity;
import net.minecraft.entity.passive.PassiveEntity;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.server.world.ServerWorld;
import net.minecraft.util.math.ChunkPos;

public class FollowPassiveEntityGoal extends FollowTargetGoal<LivingEntity> {
    public FollowPassiveEntityGoal(MobEntity mob) {
        super(mob, LivingEntity.class, 10, false, false, entity -> {
            StageDataComponent component = StageDataComponent.getFromWorld((ServerWorld) mob.getEntityWorld());
            ChunkPos chunkPos = new ChunkPos(mob.getBlockPos());
            int stage = component.findStageOfChunk(chunkPos);

            return entity instanceof PassiveEntity || (entity instanceof PlayerEntity && stage >= 4);
        });
    }

    public boolean canStart() {
        StageDataComponent component = StageDataComponent.getFromWorld((ServerWorld) mob.getEntityWorld());
        ChunkPos chunkPos = new ChunkPos(mob.getBlockPos());
        int stage = component.findStageOfChunk(chunkPos);

        return mob instanceof HostileEntity && (!(mob instanceof Angerable) || stage >= 4) && stage >= 2 && super.canStart();
    }
}