package ru.bclib.registry;

import net.fabricmc.fabric.api.registry.FlammableBlockRegistry;
import net.minecraft.core.Registry;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.item.BlockItem;
import net.minecraft.world.item.CreativeModeTab;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.Items;
import net.minecraft.world.level.block.Block;
import ru.bclib.api.TagAPI;
import ru.bclib.blocks.BaseLeavesBlock;
import ru.bclib.blocks.FeatureSaplingBlock;
import ru.bclib.config.PathConfig;
import ru.bclib.interfaces.CustomItemProvider;

public class BlockRegistry extends BaseRegistry<Block> {
	public BlockRegistry(CreativeModeTab creativeTab, PathConfig config) {
		super(creativeTab, config);
	}
	
	@Override
	public Block register(ResourceLocation id, Block block) {
		if (!config.getBooleanRoot(id.getNamespace(), true)) {
			return block;
		}
		
		BlockItem item = null;
		if (block instanceof CustomItemProvider) {
			item = ((CustomItemProvider) block).getCustomItem(id, makeItemSettings());
		}
		else {
			item = new BlockItem(block, makeItemSettings());
		}
		registerBlockItem(id, item);
		if (block.defaultBlockState().getMaterial().isFlammable() && FlammableBlockRegistry.getDefaultInstance().get(block).getBurnChance() == 0) {
			FlammableBlockRegistry.getDefaultInstance().add(block, 5, 5);
		}

		block = Registry.register(Registry.BLOCK, id, block);
		getModBlocks(id.getNamespace()).add(block);

		if (block instanceof BaseLeavesBlock){
			TagAPI.addTags(block, TagAPI.BLOCK_LEAVES);
			if (item != null){
				TagAPI.addTags(item, TagAPI.ITEM_LEAVES);
			}
		} else if (block instanceof FeatureSaplingBlock){
			TagAPI.addTags(block, TagAPI.BLOCK_SAPLINGS);
			if (item != null){
				TagAPI.addTags(item, TagAPI.ITEM_SAPLINGS);
			}
		}

		return block;
	}
	
	public Block registerBlockOnly(ResourceLocation id, Block block) {
		if (!config.getBooleanRoot(id.getNamespace(), true)) {
			return block;
		}
		getModBlocks(id.getNamespace()).add(block);
		return Registry.register(Registry.BLOCK, id, block);
	}
	
	private Item registerBlockItem(ResourceLocation id, Item item) {
		registerItem(id, item);
		return item;
	}
	
	@Override
	public void registerItem(ResourceLocation id, Item item) {
		if (item != null && item != Items.AIR) {
			Registry.register(Registry.ITEM, id, item);
			getModBlockItems(id.getNamespace()).add(item);
		}
	}
}