package net.lerariemann.infinity.item;

import net.lerariemann.infinity.InfinityMod;
import net.lerariemann.infinity.block.entity.InfinityPortalBlockEntity;
import net.lerariemann.infinity.options.PortalColorApplier;
import net.lerariemann.infinity.registry.core.ModComponentTypes;
import net.lerariemann.infinity.util.VersionMethods;
import net.lerariemann.infinity.util.InfinityMethods;
import net.minecraft.ChatFormatting;
import net.minecraft.core.BlockPos;
//? if >1.21 {
import net.minecraft.core.component.DataComponentPatch;
import net.minecraft.core.component.DataComponents;
//?} else {
/*import net.minecraft.nbt.CompoundTag;
import net.minecraft.world.level.Level;
*///?}
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.MutableComponent;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.entity.item.ItemEntity;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.TooltipFlag;
//? if >1.21.2 {
import net.minecraft.world.item.component.CustomModelData;
import net.minecraft.world.item.component.TooltipDisplay;
//?}
import org.jetbrains.annotations.NotNull;

//? if neoforge {
/*import javax.annotation.ParametersAreNonnullByDefault;
*///?}
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.function.Consumer;

//? if neoforge {
/*@ParametersAreNonnullByDefault
*///?}
public class TransfiniteKeyItem extends Item implements PortalDataHolder.Destinable {
    public TransfiniteKeyItem(Properties settings) {
        super(settings);
    }

    //? if >1.21 {
    @Override
    public DataComponentPatch getPortalComponents(InfinityPortalBlockEntity ipbe) {
        ResourceLocation id = ipbe.getDimension();
        int color = (ipbe.getLevel() instanceof ServerLevel w) ?
                PortalColorApplier.of(id, w.getServer()).apply(BlockPos.ZERO) :
                (int)InfinityMethods.getNumericFromId(id);
        return DataComponentPatch.builder()
                .set(ModComponentTypes.DESTINATION.get(), id)
                .set(ModComponentTypes.COLOR.get(), color & 0xFFFFFF)
                //? if >1.21.4 {
                .set(DataComponents.CUSTOM_MODEL_DATA, getDataForClientItem(id, color & 0xFFFFFF))
                //?}
                .build();
    }

    @Override
    public Optional<DataComponentPatch> getIridComponents(ItemEntity entity) {
        return Optional.of(DataComponentPatch.builder()
                .remove(ModComponentTypes.DESTINATION.get())
                .remove(ModComponentTypes.COLOR.get())
                .build());
    }
    //?} else {
    /*@Override
    public CompoundTag getPortalComponents(InfinityPortalBlockEntity ipbe) {
        ResourceLocation id = ipbe.getDimension();
        int color = (ipbe.getLevel() instanceof ServerLevel w) ?
                PortalColorApplier.of(id, w.getServer()).apply(BlockPos.ZERO) :
                (int)InfinityMethods.getNumericFromId(id);
        var c = new CompoundTag();
        c.putString(ModComponentTypes.DESTINATION, id.toString());
        c.putInt(ModComponentTypes.COLOR, color & 0xFFFFFF);
        return c;
    }

    @Override
    public Optional<CompoundTag> getIridComponents(ItemEntity entity) {
        return Optional.of(new CompoundTag());
    }
    *///?}

    @NotNull
    @Override
    public ResourceLocation getDestination(ItemStack stack) {
        return Objects.requireNonNullElse(VersionMethods.getDimensionIdentifier(stack),
                VersionMethods.id(InfinityMethods.ofRandomDim)); // no destination component -> randomize
    }

    @Override
    public ItemStack getStack() {
        return getDefaultInstance();
    }

    @Override
    public void appendHoverText(ItemStack stack,
                                //? if >1.21 {
                                TooltipContext
                                        //?} else {
                                        /*Level
                                         *///?}
                                        context,
                                //? if >1.21.2 {
                                TooltipDisplay tooltipDisplay, Consumer<Component> tooltipAdder
                                //?} else {
                                /*List<Component> tooltip
                                 *///?}
                                , TooltipFlag type) {
        //? if >1.21.2 {
        super.appendHoverText(stack, context, tooltipDisplay, tooltipAdder, type);
        //?} else {
        /*super.appendHoverText(stack, context, tooltip, type);
        Consumer<Component> tooltipAdder = tooltip::add;
        *///?}
        ResourceLocation dimension = VersionMethods.getDimensionIdentifier(stack);
        MutableComponent mutableText = (dimension != null)
                ? InfinityMethods.getDimensionNameAsText(dimension)
                : Component.translatable("tooltip.infinity.key.randomise");
        tooltipAdder.accept(mutableText.withStyle(ChatFormatting.GRAY));
    }

    //? if >1.21.4 {
    public CustomModelData getDataForClientItem(ItemStack stack) {
        return getDataForClientItem(stack.getComponents().getOrDefault(ModComponentTypes.DESTINATION.get(), InfinityMethods.getId("random")), stack.getComponents().getOrDefault(ModComponentTypes.COLOR.get(), 0));
    }

    public CustomModelData getDataForClientItem(ResourceLocation dimension, int color) {
        String data;
        if (dimension.getPath().contains("generated")) {
            data = "generated";
        } else if (dimension.equals(ResourceLocation.withDefaultNamespace("end"))) {
            data = "end";
        } else if (dimension.equals(InfinityMethods.getId("pride"))) {
            data = "pride";
        } else if (InfinityMod.provider.easterizer.isEaster(dimension.getPath()) || !dimension.getNamespace().equals("infinity")) {
            data = "golden";
        } else {
            data = "random";
        }
        return new CustomModelData(List.of(), List.of(), List.of(data), List.of(color));
    }
    //?}
}