package net.minecraft.advancements.triggers;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Optional;
import net.minecraft.advancements.predicates.ContextAwarePredicate;
import net.minecraft.advancements.predicates.entity.EntityPredicate;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.util.ExtraCodecs;

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

	public void trigger(final ServerPlayer player, final int number) {
		this.trigger(player, t -> t.matches(number));
	}

	public record TriggerInstance(Optional<ContextAwarePredicate> player, Optional<Integer> count) implements SimpleCriterionTrigger.SimpleInstance {
		public static final Codec<SpearMobsTrigger.TriggerInstance> CODEC = RecordCodecBuilder.create(
			i -> i.group(
					EntityPredicate.ADVANCEMENT_CODEC.optionalFieldOf("player").forGetter(SpearMobsTrigger.TriggerInstance::player),
					ExtraCodecs.POSITIVE_INT.optionalFieldOf("count").forGetter(SpearMobsTrigger.TriggerInstance::count)
				)
				.apply(i, SpearMobsTrigger.TriggerInstance::new)
		);

		public static Criterion<SpearMobsTrigger.TriggerInstance> spearMobs(final int requiredCount) {
			return CriteriaTriggers.SPEAR_MOBS_TRIGGER.createCriterion(new SpearMobsTrigger.TriggerInstance(Optional.empty(), Optional.of(requiredCount)));
		}

		public boolean matches(final int requiredCount) {
			return this.count.isEmpty() || requiredCount >= (Integer)this.count.get();
		}
	}
}