package net.minecraft.advancements.critereon;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Optional;
import net.minecraft.core.component.DataComponentType;
import net.minecraft.core.component.DataComponents;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.component.FireworkExplosion;
import net.minecraft.world.item.component.Fireworks;

public record ItemFireworksPredicate(
	Optional<CollectionPredicate<FireworkExplosion, ItemFireworkExplosionPredicate.FireworkPredicate>> explosions, MinMaxBounds.Ints flightDuration
) implements SingleComponentItemPredicate<Fireworks> {
	public static final Codec<ItemFireworksPredicate> CODEC = RecordCodecBuilder.create(
		instance -> instance.group(
				CollectionPredicate.codec(ItemFireworkExplosionPredicate.FireworkPredicate.CODEC)
					.optionalFieldOf("explosions")
					.forGetter(ItemFireworksPredicate::explosions),
				MinMaxBounds.Ints.CODEC.optionalFieldOf("flight_duration", MinMaxBounds.Ints.ANY).forGetter(ItemFireworksPredicate::flightDuration)
			)
			.apply(instance, ItemFireworksPredicate::new)
	);

	@Override
	public DataComponentType<Fireworks> componentType() {
		return DataComponents.FIREWORKS;
	}

	public boolean matches(ItemStack stack, Fireworks value) {
		return this.explosions.isPresent() && !((CollectionPredicate)this.explosions.get()).test((Iterable)value.explosions())
			? false
			: this.flightDuration.matches(value.flightDuration());
	}
}