package net.minecraft.core.component.predicates;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Optional;
import net.minecraft.advancements.critereon.SingleComponentItemPredicate;
import net.minecraft.core.HolderSet;
import net.minecraft.core.RegistryCodecs;
import net.minecraft.core.component.DataComponentType;
import net.minecraft.core.component.DataComponents;
import net.minecraft.core.registries.Registries;
import net.minecraft.world.item.equipment.trim.ArmorTrim;
import net.minecraft.world.item.equipment.trim.TrimMaterial;
import net.minecraft.world.item.equipment.trim.TrimPattern;

public record TrimPredicate(Optional<HolderSet<TrimMaterial>> material, Optional<HolderSet<TrimPattern>> pattern)
	implements SingleComponentItemPredicate<ArmorTrim> {
	public static final Codec<TrimPredicate> CODEC = RecordCodecBuilder.create(
		instance -> instance.group(
				RegistryCodecs.homogeneousList(Registries.TRIM_MATERIAL).optionalFieldOf("material").forGetter(TrimPredicate::material),
				RegistryCodecs.homogeneousList(Registries.TRIM_PATTERN).optionalFieldOf("pattern").forGetter(TrimPredicate::pattern)
			)
			.apply(instance, TrimPredicate::new)
	);

	@Override
	public DataComponentType<ArmorTrim> componentType() {
		return DataComponents.TRIM;
	}

	public boolean matches(ArmorTrim armorTrim) {
		return this.material.isPresent() && !((HolderSet)this.material.get()).contains(armorTrim.material())
			? false
			: !this.pattern.isPresent() || ((HolderSet)this.pattern.get()).contains(armorTrim.pattern());
	}
}