package net.minecraft.world.effect;

import java.util.List;
import net.minecraft.core.Holder;
import net.minecraft.network.chat.Component;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.util.Mth;
import net.minecraft.util.StringUtil;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.phys.Vec3;
import org.jspecify.annotations.Nullable;

public final class MobEffectUtil {
	public static Component formatDuration(final MobEffectInstance instance, final float scale, final float tickrate) {
		if (instance.isInfiniteDuration()) {
			return Component.translatable("effect.duration.infinite");
		} else {
			int duration = Mth.floor(instance.getDuration() * scale);
			return Component.literal(StringUtil.formatTickDuration(duration, tickrate));
		}
	}

	public static boolean hasDigSpeed(final LivingEntity mob) {
		return mob.hasEffect(MobEffects.HASTE) || mob.hasEffect(MobEffects.CONDUIT_POWER);
	}

	public static int getDigSpeedAmplification(final LivingEntity mob) {
		int a = 0;
		int b = 0;
		if (mob.hasEffect(MobEffects.HASTE)) {
			a = mob.getEffect(MobEffects.HASTE).getAmplifier();
		}

		if (mob.hasEffect(MobEffects.CONDUIT_POWER)) {
			b = mob.getEffect(MobEffects.CONDUIT_POWER).getAmplifier();
		}

		return Math.max(a, b);
	}

	public static boolean hasWaterBreathing(final LivingEntity mob) {
		return mob.hasEffect(MobEffects.WATER_BREATHING) || mob.hasEffect(MobEffects.CONDUIT_POWER) || mob.hasEffect(MobEffects.BREATH_OF_THE_NAUTILUS);
	}

	public static boolean shouldEffectsRefillAirsupply(final LivingEntity mob) {
		return !mob.hasEffect(MobEffects.BREATH_OF_THE_NAUTILUS) || mob.hasEffect(MobEffects.WATER_BREATHING) || mob.hasEffect(MobEffects.CONDUIT_POWER);
	}

	public static List<ServerPlayer> addEffectToPlayersAround(
		final ServerLevel level,
		@Nullable final Entity source,
		final Vec3 position,
		final double radius,
		final MobEffectInstance effectInstance,
		final int displayEffectLimit
	) {
		Holder<MobEffect> effect = effectInstance.getEffect();
		List<ServerPlayer> players = level.getPlayers(
			input -> input.gameMode.isSurvival()
				&& (source == null || !source.isAlliedTo(input))
				&& position.closerThan(input.position(), radius)
				&& (
					!input.hasEffect(effect)
						|| input.getEffect(effect).getAmplifier() < effectInstance.getAmplifier()
						|| input.getEffect(effect).endsWithin(displayEffectLimit - 1)
				)
		);
		players.forEach(player -> player.addEffect(new MobEffectInstance(effectInstance), source));
		return players;
	}
}