package net.hempcraft.hempcraft.datagen;

import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput;
import net.fabricmc.fabric.api.datagen.v1.provider.FabricBlockLootTableProvider;
import net.hempcraft.hempcraft.block.HempCropBlock;
import net.hempcraft.hempcraft.block.ModBlocks;
import net.hempcraft.hempcraft.item.ModItems;
import net.minecraft.loot.LootTable;
import net.minecraft.loot.condition.BlockStatePropertyLootCondition;
import net.minecraft.loot.condition.LootCondition;
import net.minecraft.predicate.StatePredicate;
import net.minecraft.registry.RegistryWrapper;

import java.util.concurrent.CompletableFuture;

public class ModLootTableProvider extends FabricBlockLootTableProvider {
    public ModLootTableProvider(FabricDataOutput dataOutput, CompletableFuture<RegistryWrapper.WrapperLookup> registryLookup) {
        super(dataOutput, registryLookup);
    }

    @Override
    public void generate() {
        // Condition pour vérifier si la plante est mature (âge 7)
        LootCondition.Builder matureCondition = BlockStatePropertyLootCondition.builder(ModBlocks.HEMP_CROP_BLOCK)
                .properties(StatePredicate.Builder.create().exactMatch(HempCropBlock.AGE, 7));

        // Table de butin pour le bloc de culture
        addDrop(ModBlocks.HEMP_CROP_BLOCK, cropDrops(ModBlocks.HEMP_CROP_BLOCK, ModItems.HEMP_LEAF, ModItems.HEMP_SEEDS, matureCondition));

        // Tables de butin vides pour les blocs supérieurs (pour qu'ils ne donnent rien)
        addDrop(ModBlocks.HEMP_MIDDLE_BLOCK, LootTable.builder());
        addDrop(ModBlocks.HEMP_TOP_BLOCK, LootTable.builder());
    }
}