package net.minecraft.world.level.saveddata.maps;

import com.mojang.serialization.Codec;
import io.netty.buffer.ByteBuf;
import java.util.function.Consumer;
import net.minecraft.ChatFormatting;
import net.minecraft.core.component.DataComponentGetter;
import net.minecraft.core.component.DataComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.TooltipFlag;
import net.minecraft.world.item.component.MapPostProcessing;
import net.minecraft.world.item.component.TooltipProvider;

public record MapId(int id) implements TooltipProvider {
    public static final Codec<MapId> CODEC = Codec.INT.xmap(MapId::new, MapId::id);
    public static final StreamCodec<ByteBuf, MapId> STREAM_CODEC = ByteBufCodecs.VAR_INT.map(MapId::new, MapId::id);
    private static final Component LOCKED_TEXT = Component.translatable("filled_map.locked").withStyle(ChatFormatting.GRAY);

    public String key() {
        return "map_" + this.id;
    }

    @Override
    public void addToTooltip(Item.TooltipContext p_395024_, Consumer<Component> p_394922_, TooltipFlag p_395978_, DataComponentGetter p_392778_) {
        MapItemSavedData mapitemsaveddata = p_395024_.mapData(this);
        if (mapitemsaveddata == null) {
            p_394922_.accept(Component.translatable("filled_map.unknown").withStyle(ChatFormatting.GRAY));
        } else {
            MapPostProcessing mappostprocessing = p_392778_.get(DataComponents.MAP_POST_PROCESSING);
            if (p_392778_.get(DataComponents.CUSTOM_NAME) == null && mappostprocessing == null) {
                p_394922_.accept(Component.translatable("filled_map.id", this.id).withStyle(ChatFormatting.GRAY));
            }

            if (mapitemsaveddata.locked || mappostprocessing == MapPostProcessing.LOCK) {
                p_394922_.accept(LOCKED_TEXT);
            }

            if (p_395978_.isAdvanced()) {
                int i = mappostprocessing == MapPostProcessing.SCALE ? 1 : 0;
                int j = Math.min(mapitemsaveddata.scale + i, 4);
                p_394922_.accept(Component.translatable("filled_map.scale", 1 << j).withStyle(ChatFormatting.GRAY));
                p_394922_.accept(Component.translatable("filled_map.level", j, 4).withStyle(ChatFormatting.GRAY));
            }
        }
    }
}