package net.minecraft.advancements.critereon;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import net.minecraft.core.component.DataComponentType;
import net.minecraft.core.component.DataComponents;
import net.minecraft.world.item.ItemStack;

public record ItemDamagePredicate(MinMaxBounds.Ints durability, MinMaxBounds.Ints damage) implements SingleComponentItemPredicate<Integer> {
	public static final Codec<ItemDamagePredicate> CODEC = RecordCodecBuilder.create(
		instance -> instance.group(
				MinMaxBounds.Ints.CODEC.optionalFieldOf("durability", MinMaxBounds.Ints.ANY).forGetter(ItemDamagePredicate::durability),
				MinMaxBounds.Ints.CODEC.optionalFieldOf("damage", MinMaxBounds.Ints.ANY).forGetter(ItemDamagePredicate::damage)
			)
			.apply(instance, ItemDamagePredicate::new)
	);

	@Override
	public DataComponentType<Integer> componentType() {
		return DataComponents.DAMAGE;
	}

	public boolean matches(ItemStack stack, Integer value) {
		return !this.durability.matches(stack.getMaxDamage() - value) ? false : this.damage.matches(value);
	}

	public static ItemDamagePredicate durability(MinMaxBounds.Ints damage) {
		return new ItemDamagePredicate(damage, MinMaxBounds.Ints.ANY);
	}
}