166 lines
7 KiB
Java
166 lines
7 KiB
Java
package net.minecraft.world.item.component;
|
|
|
|
import com.google.common.collect.ImmutableList;
|
|
import com.google.common.collect.Lists;
|
|
import com.google.common.collect.ImmutableList.Builder;
|
|
import com.google.gson.JsonElement;
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.DataResult;
|
|
import com.mojang.serialization.JsonOps;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.List;
|
|
import java.util.Optional;
|
|
import java.util.function.Consumer;
|
|
import net.minecraft.ChatFormatting;
|
|
import net.minecraft.commands.CommandSourceStack;
|
|
import net.minecraft.core.HolderLookup;
|
|
import net.minecraft.core.component.DataComponentGetter;
|
|
import net.minecraft.core.component.DataComponents;
|
|
import net.minecraft.network.RegistryFriendlyByteBuf;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.network.chat.ComponentSerialization;
|
|
import net.minecraft.network.chat.ComponentUtils;
|
|
import net.minecraft.network.codec.ByteBufCodecs;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
import net.minecraft.server.network.Filterable;
|
|
import net.minecraft.util.ExtraCodecs;
|
|
import net.minecraft.util.GsonHelper;
|
|
import net.minecraft.util.StringUtil;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.item.Item;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.item.TooltipFlag;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public record WrittenBookContent(Filterable<String> title, String author, int generation, List<Filterable<Component>> pages, boolean resolved)
|
|
implements BookContent<Component, WrittenBookContent>,
|
|
TooltipProvider {
|
|
public static final WrittenBookContent EMPTY = new WrittenBookContent(Filterable.passThrough(""), "", 0, List.of(), true);
|
|
public static final int PAGE_LENGTH = 32767;
|
|
public static final int TITLE_LENGTH = 16;
|
|
public static final int TITLE_MAX_LENGTH = 32;
|
|
public static final int MAX_GENERATION = 3;
|
|
public static final int MAX_CRAFTABLE_GENERATION = 2;
|
|
public static final Codec<Component> CONTENT_CODEC = ComponentSerialization.flatRestrictedCodec(32767);
|
|
public static final Codec<List<Filterable<Component>>> PAGES_CODEC = pagesCodec(CONTENT_CODEC);
|
|
public static final Codec<WrittenBookContent> CODEC = RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
Filterable.codec(Codec.string(0, 32)).fieldOf("title").forGetter(WrittenBookContent::title),
|
|
Codec.STRING.fieldOf("author").forGetter(WrittenBookContent::author),
|
|
ExtraCodecs.intRange(0, 3).optionalFieldOf("generation", 0).forGetter(WrittenBookContent::generation),
|
|
PAGES_CODEC.optionalFieldOf("pages", List.of()).forGetter(WrittenBookContent::pages),
|
|
Codec.BOOL.optionalFieldOf("resolved", false).forGetter(WrittenBookContent::resolved)
|
|
)
|
|
.apply(instance, WrittenBookContent::new)
|
|
);
|
|
public static final StreamCodec<RegistryFriendlyByteBuf, WrittenBookContent> STREAM_CODEC = StreamCodec.composite(
|
|
Filterable.streamCodec(ByteBufCodecs.stringUtf8(32)),
|
|
WrittenBookContent::title,
|
|
ByteBufCodecs.STRING_UTF8,
|
|
WrittenBookContent::author,
|
|
ByteBufCodecs.VAR_INT,
|
|
WrittenBookContent::generation,
|
|
Filterable.streamCodec(ComponentSerialization.STREAM_CODEC).apply(ByteBufCodecs.list()),
|
|
WrittenBookContent::pages,
|
|
ByteBufCodecs.BOOL,
|
|
WrittenBookContent::resolved,
|
|
WrittenBookContent::new
|
|
);
|
|
|
|
public WrittenBookContent(Filterable<String> title, String author, int generation, List<Filterable<Component>> pages, boolean resolved) {
|
|
if (generation >= 0 && generation <= 3) {
|
|
this.title = title;
|
|
this.author = author;
|
|
this.generation = generation;
|
|
this.pages = pages;
|
|
this.resolved = resolved;
|
|
} else {
|
|
throw new IllegalArgumentException("Generation was " + generation + ", but must be between 0 and 3");
|
|
}
|
|
}
|
|
|
|
private static Codec<Filterable<Component>> pageCodec(Codec<Component> codec) {
|
|
return Filterable.codec(codec);
|
|
}
|
|
|
|
public static Codec<List<Filterable<Component>>> pagesCodec(Codec<Component> codec) {
|
|
return pageCodec(codec).listOf();
|
|
}
|
|
|
|
@Nullable
|
|
public WrittenBookContent tryCraftCopy() {
|
|
return this.generation >= 2 ? null : new WrittenBookContent(this.title, this.author, this.generation + 1, this.pages, this.resolved);
|
|
}
|
|
|
|
public static boolean resolveForItem(ItemStack stack, CommandSourceStack source, @Nullable Player player) {
|
|
WrittenBookContent writtenBookContent = stack.get(DataComponents.WRITTEN_BOOK_CONTENT);
|
|
if (writtenBookContent != null && !writtenBookContent.resolved()) {
|
|
WrittenBookContent writtenBookContent2 = writtenBookContent.resolve(source, player);
|
|
if (writtenBookContent2 != null) {
|
|
stack.set(DataComponents.WRITTEN_BOOK_CONTENT, writtenBookContent2);
|
|
return true;
|
|
}
|
|
|
|
stack.set(DataComponents.WRITTEN_BOOK_CONTENT, writtenBookContent.markResolved());
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
@Nullable
|
|
public WrittenBookContent resolve(CommandSourceStack source, @Nullable Player player) {
|
|
if (this.resolved) {
|
|
return null;
|
|
} else {
|
|
Builder<Filterable<Component>> builder = ImmutableList.builderWithExpectedSize(this.pages.size());
|
|
|
|
for (Filterable<Component> filterable : this.pages) {
|
|
Optional<Filterable<Component>> optional = resolvePage(source, player, filterable);
|
|
if (optional.isEmpty()) {
|
|
return null;
|
|
}
|
|
|
|
builder.add((Filterable<Component>)optional.get());
|
|
}
|
|
|
|
return new WrittenBookContent(this.title, this.author, this.generation, builder.build(), true);
|
|
}
|
|
}
|
|
|
|
public WrittenBookContent markResolved() {
|
|
return new WrittenBookContent(this.title, this.author, this.generation, this.pages, true);
|
|
}
|
|
|
|
private static Optional<Filterable<Component>> resolvePage(CommandSourceStack source, @Nullable Player player, Filterable<Component> pages) {
|
|
return pages.resolve(component -> {
|
|
try {
|
|
Component component2 = ComponentUtils.updateForEntity(source, component, player, 0);
|
|
return isPageTooLarge(component2, source.registryAccess()) ? Optional.empty() : Optional.of(component2);
|
|
} catch (Exception var4) {
|
|
return Optional.of(component);
|
|
}
|
|
});
|
|
}
|
|
|
|
private static boolean isPageTooLarge(Component page, HolderLookup.Provider registryAccess) {
|
|
DataResult<JsonElement> dataResult = ComponentSerialization.CODEC.encodeStart(registryAccess.createSerializationContext(JsonOps.INSTANCE), page);
|
|
return dataResult.isSuccess() && GsonHelper.encodesLongerThan(dataResult.getOrThrow(), 32767);
|
|
}
|
|
|
|
public List<Component> getPages(boolean filtered) {
|
|
return Lists.transform(this.pages, filterable -> (Component)filterable.get(filtered));
|
|
}
|
|
|
|
public WrittenBookContent withReplacedPages(List<Filterable<Component>> list) {
|
|
return new WrittenBookContent(this.title, this.author, this.generation, list, false);
|
|
}
|
|
|
|
@Override
|
|
public void addToTooltip(Item.TooltipContext context, Consumer<Component> tooltipAdder, TooltipFlag flag, DataComponentGetter componentGetter) {
|
|
if (!StringUtil.isBlank(this.author)) {
|
|
tooltipAdder.accept(Component.translatable("book.byAuthor", this.author).withStyle(ChatFormatting.GRAY));
|
|
}
|
|
|
|
tooltipAdder.accept(Component.translatable("book.generation." + this.generation).withStyle(ChatFormatting.GRAY));
|
|
}
|
|
}
|