package net.minecraft.client.renderer.fog.environment;

import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Camera;
import net.minecraft.client.DeltaTracker;
import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.client.renderer.fog.FogData;
import net.minecraft.core.Holder;
import net.minecraft.util.Mth;
import net.minecraft.world.effect.MobEffect;
import net.minecraft.world.effect.MobEffectInstance;
import net.minecraft.world.effect.MobEffects;
import net.minecraft.world.entity.LivingEntity;

@Environment(EnvType.CLIENT)
public class BlindnessFogEnvironment extends MobEffectFogEnvironment {
	@Override
	public Holder<MobEffect> getMobEffect() {
		return MobEffects.BLINDNESS;
	}

	@Override
	public void setupFog(final FogData fog, final Camera camera, final ClientLevel level, final float renderDistance, final DeltaTracker deltaTracker) {
		if (camera.entity() instanceof LivingEntity livingEntity) {
			MobEffectInstance effect = livingEntity.getEffect(this.getMobEffect());
			if (effect != null) {
				float distance = effect.isInfiniteDuration() ? 5.0F : Mth.lerp(Math.min(1.0F, effect.getDuration() / 20.0F), renderDistance, 5.0F);
				fog.environmentalStart = distance * 0.25F;
				fog.environmentalEnd = distance;
				fog.skyEnd = distance * 0.8F;
				fog.cloudEnd = distance * 0.8F;
			}
		}
	}

	@Override
	public float getModifiedDarkness(final LivingEntity entity, float darkness, final float partialTickTime) {
		MobEffectInstance instance = entity.getEffect(this.getMobEffect());
		if (instance != null) {
			if (instance.endsWithin(19)) {
				darkness = Math.max(instance.getDuration() / 20.0F, darkness);
			} else {
				darkness = 1.0F;
			}
		}

		return darkness;
	}
}