Minestom/src/main/java/net/minestom/server/item/component/DyedItemColor.java

package net.minestom.server.item.component;

import net.kyori.adventure.nbt.BinaryTag;
import net.kyori.adventure.nbt.CompoundBinaryTag;
import net.kyori.adventure.nbt.IntBinaryTag;
import net.minestom.server.color.Color;
import net.minestom.server.network.NetworkBuffer;
import net.minestom.server.utils.nbt.BinaryTagSerializer;
import org.jetbrains.annotations.NotNull;

public record DyedItemColor(@NotNull Color color, boolean showInTooltip) {
    public static DyedItemColor LEATHER = new DyedItemColor(new Color(-6265536), true);

    public static final NetworkBuffer.Type<DyedItemColor> NETWORK_TYPE = new NetworkBuffer.Type<>() {
        @Override
        public void write(@NotNull NetworkBuffer buffer, DyedItemColor value) {
            buffer.write(NetworkBuffer.COLOR, value.color);
            buffer.write(NetworkBuffer.BOOLEAN, value.showInTooltip);
        }

        @Override
        public DyedItemColor read(@NotNull NetworkBuffer buffer) {
            return new DyedItemColor(buffer.read(NetworkBuffer.COLOR), buffer.read(NetworkBuffer.BOOLEAN));
        }
    };

    public static final BinaryTagSerializer<DyedItemColor> NBT_TYPE = new BinaryTagSerializer<>() {
        @Override
        public @NotNull BinaryTag write(@NotNull DyedItemColor value) {
            return CompoundBinaryTag.builder()
                    .putInt("color", value.color.asRGB())
                    .putBoolean("show_in_tooltip", value.showInTooltip)
                    .build();
        }

        @Override
        public @NotNull DyedItemColor read(@NotNull BinaryTag tag) {
            if (tag instanceof CompoundBinaryTag compoundTag) {
                int color = compoundTag.getInt("color");
                boolean showInTooltip = compoundTag.getBoolean("show_in_tooltip", true);
                return new DyedItemColor(new Color(color), showInTooltip);
            } else if (tag instanceof IntBinaryTag intTag) {
                return new DyedItemColor(new Color(intTag.intValue()), true);
            }
            return new DyedItemColor(new Color(0), false);
        }
    };

    public @NotNull DyedItemColor withColor(@NotNull Color color) {
        return new DyedItemColor(color, showInTooltip);
    }

    public @NotNull DyedItemColor withTooltip(boolean showInTooltip) {
        return new DyedItemColor(color, showInTooltip);
    }

}