package net.minecraft.world.entity.ai.goal;

import java.util.EnumSet;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.Mob;

public class OcelotAttackGoal extends Goal {
	private final Mob mob;
	private LivingEntity target;
	private int attackTime;

	public OcelotAttackGoal(Mob mob) {
		this.mob = mob;
		this.setFlags(EnumSet.of(Goal.Flag.MOVE, Goal.Flag.LOOK));
	}

	@Override
	public boolean canUse() {
		LivingEntity livingEntity = this.mob.getTarget();
		if (livingEntity == null) {
			return false;
		} else {
			this.target = livingEntity;
			return true;
		}
	}

	@Override
	public boolean canContinueToUse() {
		if (!this.target.isAlive()) {
			return false;
		} else {
			return this.mob.distanceToSqr(this.target) > 225.0 ? false : !this.mob.getNavigation().isDone() || this.canUse();
		}
	}

	@Override
	public void stop() {
		this.target = null;
		this.mob.getNavigation().stop();
	}

	@Override
	public boolean requiresUpdateEveryTick() {
		return true;
	}

	@Override
	public void tick() {
		this.mob.getLookControl().setLookAt(this.target, 30.0F, 30.0F);
		double d = this.mob.getBbWidth() * 2.0F * (this.mob.getBbWidth() * 2.0F);
		double e = this.mob.distanceToSqr(this.target.getX(), this.target.getY(), this.target.getZ());
		double f = 0.8;
		if (e > d && e < 16.0) {
			f = 1.33;
		} else if (e < 225.0) {
			f = 0.6;
		}

		this.mob.getNavigation().moveTo(this.target, f);
		this.attackTime = Math.max(this.attackTime - 1, 0);
		if (!(e > d)) {
			if (this.attackTime <= 0) {
				this.attackTime = 20;
				this.mob.doHurtTarget(this.target);
			}
		}
	}
}