62 lines
2.8 KiB
Java
62 lines
2.8 KiB
Java
|
package net.minecraft.world.item.trading;
|
||
|
|
|
||
|
|
import com.mojang.serialization.Codec;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
||
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder.Instance;
|
||
|
|
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(
|
||
|
|
p_390864_ -> p_390864_.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(p_390864_, 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 p_333321_) {
|
||
|
|
this(p_333321_, 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
public ItemCost(ItemLike p_332783_, int p_331715_) {
|
||
|
|
this(p_332783_.asItem().builtInRegistryHolder(), p_331715_, DataComponentExactPredicate.EMPTY);
|
||
|
|
}
|
||
|
|
|
||
|
|
public ItemCost(Holder<Item> p_334519_, int p_331202_, DataComponentExactPredicate p_397399_) {
|
||
|
|
this(p_334519_, p_331202_, p_397399_, createStack(p_334519_, p_331202_, p_397399_));
|
||
|
|
}
|
||
|
|
|
||
|
|
public ItemCost withComponents(UnaryOperator<DataComponentExactPredicate.Builder> p_328625_) {
|
||
|
|
return new ItemCost(this.item, this.count, p_328625_.apply(DataComponentExactPredicate.builder()).build());
|
||
|
|
}
|
||
|
|
|
||
|
|
private static ItemStack createStack(Holder<Item> p_329043_, int p_329370_, DataComponentExactPredicate p_393981_) {
|
||
|
|
return new ItemStack(p_329043_, p_329370_, p_393981_.asPatch());
|
||
|
|
}
|
||
|
|
|
||
|
|
public boolean test(ItemStack p_331178_) {
|
||
|
|
return p_331178_.is(this.item) && this.components.test((DataComponentGetter)p_331178_);
|
||
|
|
}
|
||
|
|
}
|