package com.jaydizzle.buildingblocks.datagen;

import com.jaydizzle.buildingblocks.BuildingBlocks;
import com.jaydizzle.buildingblocks.block.JDFenceGates;
import com.jaydizzle.buildingblocks.block.JDFences;
import com.jaydizzle.buildingblocks.block.JDSlabs;
import com.jaydizzle.buildingblocks.block.JDStairs;
import com.jaydizzle.buildingblocks.item.JDItems;
import net.minecraft.data.PackOutput;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.item.Item;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.Blocks;
import net.minecraftforge.client.model.generators.ItemModelBuilder;
import net.minecraftforge.client.model.generators.ItemModelProvider;
import net.minecraftforge.common.data.ExistingFileHelper;
import net.minecraftforge.registries.ForgeRegistries;
import net.minecraftforge.registries.RegistryObject;

public class JDItemModelProvider extends ItemModelProvider {
    public JDItemModelProvider(PackOutput output, ExistingFileHelper existingFileHelper) {
        super(output, BuildingBlocks.MOD_ID, existingFileHelper);
    }

    @Override
    protected void registerModels() {

        fenceItem(JDFences.SPRUCE_LOG_FENCE, Blocks.SPRUCE_LOG);
        fenceItem(JDFences.OAK_LOG_FENCE, Blocks.OAK_LOG);
        fenceItem(JDFences.DARK_OAK_LOG_FENCE, Blocks.DARK_OAK_LOG);
        fenceItem(JDFences.ACACIA_LOG_FENCE, Blocks.ACACIA_LOG);
        fenceItem(JDFences.JUNGLE_LOG_FENCE, Blocks.JUNGLE_LOG);
        fenceItem(JDFences.BIRCH_LOG_FENCE, Blocks.BIRCH_LOG);
        fenceItem(JDFences.WARPED_LOG_FENCE, Blocks.WARPED_HYPHAE);
        fenceItem(JDFences.CRIMSON_LOG_FENCE, Blocks.CRIMSON_HYPHAE);
        fenceItem(JDFences.SPRUCE_SKINNED_LOG_FENCE, Blocks.STRIPPED_SPRUCE_LOG);
        fenceItem(JDFences.OAK_SKINNED_LOG_FENCE, Blocks.STRIPPED_OAK_LOG);
        fenceItem(JDFences.DARK_SKINNED_OAK_LOG_FENCE, Blocks.STRIPPED_DARK_OAK_LOG);
        fenceItem(JDFences.ACACIA_SKINNED_LOG_FENCE, Blocks.STRIPPED_ACACIA_LOG);
        fenceItem(JDFences.JUNGLE_SKINNED_LOG_FENCE, Blocks.STRIPPED_JUNGLE_WOOD);
        fenceItem(JDFences.BIRCH_SKINNED_LOG_FENCE, Blocks.STRIPPED_BIRCH_LOG);
        fenceItem(JDFences.WARPED_SKINNED_LOG_FENCE, Blocks.STRIPPED_WARPED_HYPHAE);
        fenceItem(JDFences.CRIMSON_SKINNED_LOG_FENCE, Blocks.STRIPPED_CRIMSON_HYPHAE);

    }

    private ItemModelBuilder simpleItem(RegistryObject<Item> item) {
        return withExistingParent(item.getId().getPath(),
                new ResourceLocation("item/generated")).texture("layer0",
                new ResourceLocation(BuildingBlocks.MOD_ID,"item/" + item.getId().getPath()));
    }

    public void evenSimplerBlockItem(RegistryObject<Block> block) {
        this.withExistingParent(BuildingBlocks.MOD_ID + ":" + ForgeRegistries.BLOCKS.getKey(block.get()).getPath(),
                modLoc("block/" + ForgeRegistries.BLOCKS.getKey(block.get()).getPath()));
    }

    public void fenceItem(RegistryObject<Block> block, RegistryObject<Block> baseBlock) {
        this.withExistingParent(ForgeRegistries.BLOCKS.getKey(block.get()).getPath(), mcLoc("block/fence_inventory"))
                .texture("texture",  new ResourceLocation(BuildingBlocks.MOD_ID, "block/" + ForgeRegistries.BLOCKS.getKey(baseBlock.get()).getPath()));
    }

    private ItemModelBuilder simpleBlockItem(RegistryObject<Block> item) {
        return withExistingParent(item.getId().getPath(),
                new ResourceLocation("item/generated")).texture("layer0",
                new ResourceLocation(BuildingBlocks.MOD_ID,"item/" + item.getId().getPath()));
    }
}