net.minestom.server.item.component.EnchantmentList Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of minestom-snapshots Show documentation
1.20.4 Lightweight Minecraft server
There is a newer version: 7320437640
package net.minestom.server.item.component;

import net.kyori.adventure.nbt.BinaryTag;
import net.kyori.adventure.nbt.CompoundBinaryTag;
import net.minestom.server.item.enchant.Enchantment;
import net.minestom.server.network.NetworkBuffer;
import net.minestom.server.registry.DynamicRegistry;
import net.minestom.server.utils.nbt.BinaryTagSerializer;
import net.minestom.server.utils.validate.Check;
import org.jetbrains.annotations.NotNull;

import java.util.HashMap;
import java.util.Map;

import static net.kyori.adventure.nbt.StringBinaryTag.stringBinaryTag;

public record EnchantmentList(@NotNull Map, Integer> enchantments,
                              boolean showInTooltip) {
    public static final EnchantmentList EMPTY = new EnchantmentList(Map.of(), true);

    public static NetworkBuffer.Type NETWORK_TYPE = new NetworkBuffer.Type<>() {
        @Override
        public void write(@NotNull NetworkBuffer buffer, @NotNull EnchantmentList value) {
            buffer.write(NetworkBuffer.VAR_INT, value.enchantments.size());
            for (Map.Entry, Integer> entry : value.enchantments.entrySet()) {
                buffer.write(Enchantment.NETWORK_TYPE, entry.getKey());
                buffer.write(NetworkBuffer.VAR_INT, entry.getValue());
            }
            buffer.write(NetworkBuffer.BOOLEAN, value.showInTooltip);
        }

        @Override
        public @NotNull EnchantmentList read(@NotNull NetworkBuffer buffer) {
            int size = buffer.read(NetworkBuffer.VAR_INT);
            Check.argCondition(size < 0 || size > Short.MAX_VALUE, "Invalid enchantment list size: {0}", size);
            Map, Integer> enchantments = new HashMap<>(size);
            for (int i = 0; i < size; i++) {
                DynamicRegistry.Key enchantment = buffer.read(Enchantment.NETWORK_TYPE);
                enchantments.put(enchantment, buffer.read(NetworkBuffer.VAR_INT));
            }
            boolean showInTooltip = buffer.read(NetworkBuffer.BOOLEAN);
            return new EnchantmentList(enchantments, showInTooltip);
        }
    };
    public static BinaryTagSerializer NBT_TYPE = new BinaryTagSerializer<>() {
        @Override
        public @NotNull BinaryTag write(@NotNull Context context, @NotNull EnchantmentList value) {
            CompoundBinaryTag.Builder levels = CompoundBinaryTag.builder();
            for (Map.Entry, Integer> entry : value.enchantments.entrySet()) {
                levels.put(entry.getKey().name(), BinaryTagSerializer.INT.write(context, entry.getValue()));
            }

            return CompoundBinaryTag.builder()
                    .put("levels", levels.build())
                    .putBoolean("show_in_tooltip", value.showInTooltip)
                    .build();
        }

        @Override
        public @NotNull EnchantmentList read(@NotNull Context context, @NotNull BinaryTag raw) {
            if (!(raw instanceof CompoundBinaryTag tag))
                throw new IllegalArgumentException("Enchantment list must be a compound tag");

            // We have two variants of the enchantment list, one with {levels: {...}, show_in_tooltip: boolean} and one with {...}.
            CompoundBinaryTag levels = tag.keySet().contains("levels") ? tag.getCompound("levels") : tag;
            Map, Integer> enchantments = new HashMap<>(levels.size());
            for (Map.Entry entry : levels) {
                DynamicRegistry.Key enchantment = Enchantment.NBT_TYPE.read(context, stringBinaryTag(entry.getKey()));
                int level = BinaryTagSerializer.INT.read(entry.getValue());
                if (level > 0) enchantments.put(enchantment, level);
            }

            // Doesnt matter which variant we chose, the default will work.
            boolean showInTooltip = tag.getBoolean("show_in_tooltip", true);

            return new EnchantmentList(enchantments, showInTooltip);
        }
    };

    public EnchantmentList {
        enchantments = Map.copyOf(enchantments);
    }

    public EnchantmentList(@NotNull Map, Integer> enchantments) {
        this(enchantments, true);
    }

    public EnchantmentList(@NotNull DynamicRegistry.Key enchantment, int level) {
        this(Map.of(enchantment, level), true);
    }

    public boolean has(@NotNull DynamicRegistry.Key enchantment) {
        return enchantments.containsKey(enchantment);
    }

    public int level(@NotNull DynamicRegistry.Key enchantment) {
        return enchantments.getOrDefault(enchantment, 0);
    }

    public @NotNull EnchantmentList with(@NotNull DynamicRegistry.Key enchantment, int level) {
        Map, Integer> newEnchantments = new HashMap<>(enchantments);
        newEnchantments.put(enchantment, level);
        return new EnchantmentList(newEnchantments, showInTooltip);
    }

    public @NotNull EnchantmentList remove(@NotNull DynamicRegistry.Key enchantment) {
        Map, Integer> newEnchantments = new HashMap<>(enchantments);
        newEnchantments.remove(enchantment);
        return new EnchantmentList(newEnchantments, showInTooltip);
    }

    public @NotNull EnchantmentList withTooltip(boolean showInTooltip) {
        return new EnchantmentList(enchantments, showInTooltip);
    }
}