package net.minecraft.advancements.critereon;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Collection;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import net.minecraft.advancements.CriteriaTriggers;
import net.minecraft.advancements.Criterion;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.world.entity.Entity;
import net.minecraft.world.level.storage.loot.LootContext;

public class ChanneledLightningTrigger extends SimpleCriterionTrigger<ChanneledLightningTrigger.TriggerInstance> {
	@Override
	public Codec<ChanneledLightningTrigger.TriggerInstance> codec() {
		return ChanneledLightningTrigger.TriggerInstance.CODEC;
	}

	public void trigger(ServerPlayer player, Collection<? extends Entity> entityTriggered) {
		List<LootContext> list = (List<LootContext>)entityTriggered.stream()
			.map(entity -> EntityPredicate.createContext(player, entity))
			.collect(Collectors.toList());
		this.trigger(player, triggerInstance -> triggerInstance.matches(list));
	}

	public record TriggerInstance(Optional<ContextAwarePredicate> player, List<ContextAwarePredicate> victims) implements SimpleCriterionTrigger.SimpleInstance {
		public static final Codec<ChanneledLightningTrigger.TriggerInstance> CODEC = RecordCodecBuilder.create(
			instance -> instance.group(
					EntityPredicate.ADVANCEMENT_CODEC.optionalFieldOf("player").forGetter(ChanneledLightningTrigger.TriggerInstance::player),
					EntityPredicate.ADVANCEMENT_CODEC.listOf().optionalFieldOf("victims", List.of()).forGetter(ChanneledLightningTrigger.TriggerInstance::victims)
				)
				.apply(instance, ChanneledLightningTrigger.TriggerInstance::new)
		);

		public static Criterion<ChanneledLightningTrigger.TriggerInstance> channeledLightning(EntityPredicate.Builder... victims) {
			return CriteriaTriggers.CHANNELED_LIGHTNING.createCriterion(new ChanneledLightningTrigger.TriggerInstance(Optional.empty(), EntityPredicate.wrap(victims)));
		}

		public boolean matches(Collection<? extends LootContext> victims) {
			for (ContextAwarePredicate contextAwarePredicate : this.victims) {
				boolean bl = false;

				for (LootContext lootContext : victims) {
					if (contextAwarePredicate.matches(lootContext)) {
						bl = true;
						break;
					}
				}

				if (!bl) {
					return false;
				}
			}

			return true;
		}

		@Override
		public void validate(CriterionValidator validator) {
			SimpleCriterionTrigger.SimpleInstance.super.validate(validator);
			validator.validateEntities(this.victims, ".victims");
		}
	}
}