package net.minecraft.client.renderer.item.properties.select;

import com.mojang.serialization.Codec;
import com.mojang.serialization.MapCodec;
import javax.annotation.Nullable;
import net.minecraft.client.multiplayer.ClientLevel;
import net.minecraft.core.component.DataComponents;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.item.CrossbowItem;
import net.minecraft.world.item.ItemDisplayContext;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;
import net.minecraft.world.item.component.ChargedProjectiles;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;

@OnlyIn(Dist.CLIENT)
public record Charge() implements SelectItemModelProperty<CrossbowItem.ChargeType> {
    public static final Codec<CrossbowItem.ChargeType> VALUE_CODEC = CrossbowItem.ChargeType.CODEC;
    public static final SelectItemModelProperty.Type<Charge, CrossbowItem.ChargeType> TYPE = SelectItemModelProperty.Type.create(
        MapCodec.unit(new Charge()), VALUE_CODEC
    );

    public CrossbowItem.ChargeType get(
        ItemStack p_378361_, @Nullable ClientLevel p_377031_, @Nullable LivingEntity p_376163_, int p_376891_, ItemDisplayContext p_378760_
    ) {
        ChargedProjectiles chargedprojectiles = p_378361_.get(DataComponents.CHARGED_PROJECTILES);
        if (chargedprojectiles == null || chargedprojectiles.isEmpty()) {
            return CrossbowItem.ChargeType.NONE;
        } else {
            return chargedprojectiles.contains(Items.FIREWORK_ROCKET) ? CrossbowItem.ChargeType.ROCKET : CrossbowItem.ChargeType.ARROW;
        }
    }

    @Override
    public SelectItemModelProperty.Type<Charge, CrossbowItem.ChargeType> type() {
        return TYPE;
    }

    @Override
    public Codec<CrossbowItem.ChargeType> valueCodec() {
        return VALUE_CODEC;
    }
}