package net.minecraft.world.entity.boss.enderdragon.phases;

import net.minecraft.world.damagesource.DamageSource;
import net.minecraft.world.entity.boss.enderdragon.EnderDragon;
import net.minecraft.world.entity.projectile.arrow.AbstractArrow;
import net.minecraft.world.entity.projectile.hurtingprojectile.windcharge.WindCharge;

public abstract class AbstractDragonSittingPhase extends AbstractDragonPhaseInstance {
	public AbstractDragonSittingPhase(final EnderDragon dragon) {
		super(dragon);
	}

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

	@Override
	public float onHurt(final DamageSource source, final float damage) {
		if (!(source.getDirectEntity() instanceof AbstractArrow) && !(source.getDirectEntity() instanceof WindCharge)) {
			return super.onHurt(source, damage);
		} else {
			source.getDirectEntity().igniteForSeconds(1.0F);
			return 0.0F;
		}
	}
}