94 lines
3.4 KiB
Java
94 lines
3.4 KiB
Java
package net.minecraft.advancements.critereon;
|
|
|
|
import com.google.common.collect.ImmutableMap;
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
import java.util.function.Predicate;
|
|
import net.minecraft.advancements.critereon.ItemSubPredicate.Type;
|
|
import net.minecraft.advancements.critereon.MinMaxBounds.Ints;
|
|
import net.minecraft.core.HolderGetter;
|
|
import net.minecraft.core.HolderSet;
|
|
import net.minecraft.core.RegistryCodecs;
|
|
import net.minecraft.core.component.DataComponentHolder;
|
|
import net.minecraft.core.component.DataComponentPredicate;
|
|
import net.minecraft.core.registries.Registries;
|
|
import net.minecraft.tags.TagKey;
|
|
import net.minecraft.world.item.Item;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.ItemLike;
|
|
|
|
public record ItemPredicate(Optional<HolderSet<Item>> items, Ints count, DataComponentPredicate components, Map<Type<?>, ItemSubPredicate> subPredicates)
|
|
implements Predicate<ItemStack> {
|
|
public static final Codec<ItemPredicate> CODEC = RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
RegistryCodecs.homogeneousList(Registries.ITEM).optionalFieldOf("items").forGetter(ItemPredicate::items),
|
|
Ints.CODEC.optionalFieldOf("count", Ints.ANY).forGetter(ItemPredicate::count),
|
|
DataComponentPredicate.CODEC.optionalFieldOf("components", DataComponentPredicate.EMPTY).forGetter(ItemPredicate::components),
|
|
ItemSubPredicate.CODEC.optionalFieldOf("predicates", Map.of()).forGetter(ItemPredicate::subPredicates)
|
|
)
|
|
.apply(instance, ItemPredicate::new)
|
|
);
|
|
|
|
public boolean test(ItemStack itemStack) {
|
|
if (this.items.isPresent() && !itemStack.is((HolderSet<Item>)this.items.get())) {
|
|
return false;
|
|
} else if (!this.count.matches(itemStack.getCount())) {
|
|
return false;
|
|
} else if (!this.components.test((DataComponentHolder)itemStack)) {
|
|
return false;
|
|
} else {
|
|
for (ItemSubPredicate itemSubPredicate : this.subPredicates.values()) {
|
|
if (!itemSubPredicate.matches(itemStack)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
|
|
public static class Builder {
|
|
private Optional<HolderSet<Item>> items = Optional.empty();
|
|
private Ints count = Ints.ANY;
|
|
private DataComponentPredicate components = DataComponentPredicate.EMPTY;
|
|
private final ImmutableMap.Builder<Type<?>, ItemSubPredicate> subPredicates = ImmutableMap.builder();
|
|
|
|
private Builder() {
|
|
}
|
|
|
|
public static ItemPredicate.Builder item() {
|
|
return new ItemPredicate.Builder();
|
|
}
|
|
|
|
public ItemPredicate.Builder of(HolderGetter<Item> holderGetter, ItemLike... itemLikes) {
|
|
this.items = Optional.of(HolderSet.direct(itemLike -> itemLike.asItem().builtInRegistryHolder(), itemLikes));
|
|
return this;
|
|
}
|
|
|
|
public ItemPredicate.Builder of(HolderGetter<Item> holderGetter, TagKey<Item> tagKey) {
|
|
this.items = Optional.of(holderGetter.getOrThrow(tagKey));
|
|
return this;
|
|
}
|
|
|
|
public ItemPredicate.Builder withCount(Ints count) {
|
|
this.count = count;
|
|
return this;
|
|
}
|
|
|
|
public <T extends ItemSubPredicate> ItemPredicate.Builder withSubPredicate(Type<T> type, T itemSubPredicate) {
|
|
this.subPredicates.put(type, itemSubPredicate);
|
|
return this;
|
|
}
|
|
|
|
public ItemPredicate.Builder hasComponents(DataComponentPredicate components) {
|
|
this.components = components;
|
|
return this;
|
|
}
|
|
|
|
public ItemPredicate build() {
|
|
return new ItemPredicate(this.items, this.count, this.components, this.subPredicates.build());
|
|
}
|
|
}
|
|
}
|