package net.minecraft.world.effect;

import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.LivingEntity;
import org.jetbrains.annotations.Nullable;

class HealOrHarmMobEffect extends InstantenousMobEffect {
	private final boolean isHarm;

	public HealOrHarmMobEffect(MobEffectCategory category, int color, boolean isHarm) {
		super(category, color);
		this.isHarm = isHarm;
	}

	@Override
	public boolean applyEffectTick(LivingEntity livingEntity, int amplifier) {
		if (this.isHarm == livingEntity.isInvertedHealAndHarm()) {
			livingEntity.heal(Math.max(4 << amplifier, 0));
		} else {
			livingEntity.hurt(livingEntity.damageSources().magic(), 6 << amplifier);
		}

		return true;
	}

	@Override
	public void applyInstantenousEffect(@Nullable Entity source, @Nullable Entity indirectSource, LivingEntity livingEntity, int amplifier, double health) {
		if (this.isHarm == livingEntity.isInvertedHealAndHarm()) {
			int i = (int)(health * (4 << amplifier) + 0.5);
			livingEntity.heal(i);
		} else {
			int i = (int)(health * (6 << amplifier) + 0.5);
			if (source == null) {
				livingEntity.hurt(livingEntity.damageSources().magic(), i);
			} else {
				livingEntity.hurt(livingEntity.damageSources().indirectMagic(source, indirectSource), i);
			}
		}
	}
}