55 lines
2.5 KiB
Java
55 lines
2.5 KiB
Java
package net.minecraft.world.item.trading;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.Optional;
|
|
import java.util.function.UnaryOperator;
|
|
import net.minecraft.core.Holder;
|
|
import net.minecraft.core.component.DataComponentExactPredicate;
|
|
import net.minecraft.core.component.DataComponentGetter;
|
|
import net.minecraft.network.RegistryFriendlyByteBuf;
|
|
import net.minecraft.network.codec.ByteBufCodecs;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
import net.minecraft.util.ExtraCodecs;
|
|
import net.minecraft.world.item.Item;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.level.ItemLike;
|
|
|
|
public record ItemCost(Holder<Item> item, int count, DataComponentExactPredicate components, ItemStack itemStack) {
|
|
public static final Codec<ItemCost> CODEC = RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
Item.CODEC.fieldOf("id").forGetter(ItemCost::item),
|
|
ExtraCodecs.POSITIVE_INT.fieldOf("count").orElse(1).forGetter(ItemCost::count),
|
|
DataComponentExactPredicate.CODEC.optionalFieldOf("components", DataComponentExactPredicate.EMPTY).forGetter(ItemCost::components)
|
|
)
|
|
.apply(instance, ItemCost::new)
|
|
);
|
|
public static final StreamCodec<RegistryFriendlyByteBuf, ItemCost> STREAM_CODEC = StreamCodec.composite(
|
|
Item.STREAM_CODEC, ItemCost::item, ByteBufCodecs.VAR_INT, ItemCost::count, DataComponentExactPredicate.STREAM_CODEC, ItemCost::components, ItemCost::new
|
|
);
|
|
public static final StreamCodec<RegistryFriendlyByteBuf, Optional<ItemCost>> OPTIONAL_STREAM_CODEC = STREAM_CODEC.apply(ByteBufCodecs::optional);
|
|
|
|
public ItemCost(ItemLike item) {
|
|
this(item, 1);
|
|
}
|
|
|
|
public ItemCost(ItemLike item, int count) {
|
|
this(item.asItem().builtInRegistryHolder(), count, DataComponentExactPredicate.EMPTY);
|
|
}
|
|
|
|
public ItemCost(Holder<Item> item, int count, DataComponentExactPredicate components) {
|
|
this(item, count, components, createStack(item, count, components));
|
|
}
|
|
|
|
public ItemCost withComponents(UnaryOperator<DataComponentExactPredicate.Builder> components) {
|
|
return new ItemCost(this.item, this.count, ((DataComponentExactPredicate.Builder)components.apply(DataComponentExactPredicate.builder())).build());
|
|
}
|
|
|
|
private static ItemStack createStack(Holder<Item> item, int count, DataComponentExactPredicate components) {
|
|
return new ItemStack(item, count, components.asPatch());
|
|
}
|
|
|
|
public boolean test(ItemStack stack) {
|
|
return stack.is(this.item) && this.components.test((DataComponentGetter)stack);
|
|
}
|
|
}
|