BCLib/src/main/java/org/betterx/bclib/blocks/BaseOreBlock.java

package org.betterx.bclib.blocks;

import org.betterx.bclib.interfaces.BlockModelProvider;
import org.betterx.bclib.interfaces.TagProvider;
import org.betterx.bclib.util.LootUtil;
import org.betterx.bclib.util.MHelper;
import org.betterx.worlds.together.tag.v3.MineableTags;

import net.minecraft.client.renderer.block.model.BlockModel;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.tags.BlockTags;
import net.minecraft.tags.TagKey;
import net.minecraft.util.Mth;
import net.minecraft.util.valueproviders.UniformInt;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.TieredItem;
import net.minecraft.world.item.Tiers;
import net.minecraft.world.item.enchantment.EnchantmentHelper;
import net.minecraft.world.item.enchantment.Enchantments;
import net.minecraft.world.level.ItemLike;
import net.minecraft.world.level.block.Block;
import net.minecraft.world.level.block.DropExperienceBlock;
import net.minecraft.world.level.block.SoundType;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.material.Material;
import net.minecraft.world.level.material.MaterialColor;
import net.minecraft.world.level.storage.loot.LootContext;
import net.minecraft.world.level.storage.loot.parameters.LootContextParams;

import net.fabricmc.fabric.api.object.builder.v1.block.FabricBlockSettings;

import java.util.Collections;
import java.util.List;
import java.util.function.Supplier;

public class BaseOreBlock extends DropExperienceBlock implements BlockModelProvider, TagProvider {
    private final Supplier<Item> dropItem;
    private final int minCount;
    private final int maxCount;
    private final int miningLevel;

    public BaseOreBlock(Supplier<Item> drop, int minCount, int maxCount, int experience) {
        this(drop, minCount, maxCount, experience, 0);
    }

    public BaseOreBlock(Supplier<Item> drop, int minCount, int maxCount, int experience, int miningLevel) {
        this(
                FabricBlockSettings
                        .of(Material.STONE, MaterialColor.SAND)
                        .requiresTool()
                        .destroyTime(3F)
                        .explosionResistance(9F)
                        .sound(SoundType.STONE),
                drop, minCount, maxCount, experience, miningLevel
        );
    }

    public BaseOreBlock(Properties properties, Supplier<Item> drop, int minCount, int maxCount, int experience) {
        this(properties, drop, minCount, maxCount, experience, 0);
    }

    public BaseOreBlock(
            Properties properties,
            Supplier<Item> drop,
            int minCount,
            int maxCount,
            int experience,
            int miningLevel
    ) {
        super(properties, UniformInt.of(experience > 0 ? 1 : 0, experience));
        this.dropItem = drop;
        this.minCount = minCount;
        this.maxCount = maxCount;
        this.miningLevel = miningLevel;
    }

    @Override
    @SuppressWarnings("deprecation")
    public List<ItemStack> getDrops(BlockState state, LootContext.Builder builder) {
        return LootUtil
                .getDrops(this, state, builder)
                .orElseGet(
                        () -> BaseOreBlock.getDroppedItems(
                                this,
                                dropItem.get(),
                                maxCount,
                                minCount,
                                miningLevel,
                                state,
                                builder
                        )
                );
    }

    public static List<ItemStack> getDroppedItems(
            ItemLike block,
            Item dropItem,
            int maxCount,
            int minCount,
            int miningLevel,
            BlockState state,
            LootContext.Builder builder
    ) {
        ItemStack tool = builder.getParameter(LootContextParams.TOOL);
        if (tool != null && tool.isCorrectToolForDrops(state)) {
            boolean canMine = miningLevel == 0;
            if (tool.getItem() instanceof TieredItem tired) {
                canMine = tired.getTier().getLevel() >= miningLevel;
            }
            if (canMine) {
                if (EnchantmentHelper.getItemEnchantmentLevel(Enchantments.SILK_TOUCH, tool) > 0) {
                    return Collections.singletonList(new ItemStack(block));
                }
                int count;
                int enchantment = EnchantmentHelper.getItemEnchantmentLevel(Enchantments.BLOCK_FORTUNE, tool);
                if (enchantment > 0) {
                    int min = Mth.clamp(minCount + enchantment, minCount, maxCount);
                    int max = maxCount + (enchantment / Enchantments.BLOCK_FORTUNE.getMaxLevel());
                    if (min == max) {
                        return Collections.singletonList(new ItemStack(dropItem, max));
                    }
                    count = MHelper.randRange(min, max, MHelper.RANDOM_SOURCE);
                } else {
                    count = MHelper.randRange(minCount, maxCount, MHelper.RANDOM_SOURCE);
                }
                return Collections.singletonList(new ItemStack(dropItem, count));
            }
        }
        return Collections.emptyList();
    }

    @Override
    public BlockModel getItemModel(ResourceLocation resourceLocation) {
        return getBlockModel(resourceLocation, defaultBlockState());
    }

    @Override
    public void addTags(List<TagKey<Block>> blockTags, List<TagKey<Item>> itemTags) {
        if (this.miningLevel == Tiers.STONE.getLevel()) {
            blockTags.add(BlockTags.NEEDS_STONE_TOOL);
        } else if (this.miningLevel == Tiers.IRON.getLevel()) {
            blockTags.add(BlockTags.NEEDS_IRON_TOOL);
        } else if (this.miningLevel == Tiers.DIAMOND.getLevel()) {
            blockTags.add(BlockTags.NEEDS_DIAMOND_TOOL);
        } else if (this.miningLevel == Tiers.NETHERITE.getLevel()) {
            blockTags.add(MineableTags.NEEDS_NETHERITE_TOOL);
        }
    }
}