package com.cerbon.cerbons_api.capability.providers;

import com.cerbon.cerbons_api.api.general.data.HistoricalData;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.nbt.CompoundTag;
import net.minecraftforge.common.capabilities.*;
import net.minecraftforge.common.util.LazyOptional;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.Arrays;
import java.util.List;

@AutoRegisterCapability
public class PlayerBlockPosHistoryProvider implements ICapabilitySerializable<CompoundTag> {
    public static final Capability<HistoricalData<BlockPos>> HISTORICAL_DATA = CapabilityManager.get(new CapabilityToken<>() {});

    private HistoricalData<BlockPos> positionalHistory;
    private final LazyOptional<HistoricalData<BlockPos>> optional = LazyOptional.of(this::createHistoricalData);

    private HistoricalData<BlockPos> createHistoricalData() {
        if(this.positionalHistory == null)
            this.positionalHistory = new HistoricalData<>(new BlockPos(0, 0, 0), 10);

        return this.positionalHistory;
    }

    @Override
    public CompoundTag serializeNBT() {
        CompoundTag tag = new CompoundTag();
        if (positionalHistory == null) return tag;
        tag.putLongArray("LastBlocksPos", positionalHistory.stream().mapToLong(BlockPos::asLong).toArray());
        return tag;
    }

    @Override
    public void deserializeNBT(CompoundTag tag) {
        if (!tag.contains("LastBlocksPos")) return;
        List<BlockPos> blocksPos = Arrays.stream(tag.getLongArray("LastBlocksPos")).mapToObj(BlockPos::of).toList();

        this.positionalHistory = new HistoricalData<>(new BlockPos(0, 0, 0), 10);
        this.positionalHistory.remove(0);
        this.positionalHistory.addAll(blocksPos);
    }

    @Override
    public @NotNull <T> LazyOptional<T> getCapability(@NotNull Capability<T> cap, @Nullable Direction side) {
        return HISTORICAL_DATA.orEmpty(cap, optional);
    }
}