package com.thebrokenrail.sorcerycraft.spell.util;

import com.thebrokenrail.sorcerycraft.spell.api.Spell;
import com.thebrokenrail.sorcerycraft.spell.api.registry.SpellRegistry;
import net.minecraft.item.ItemStack;
import net.minecraft.loot.condition.LootCondition;
import net.minecraft.loot.context.LootContext;
import net.minecraft.loot.function.ConditionalLootFunction;
import net.minecraft.loot.function.LootFunction;
import net.minecraft.util.Identifier;

import java.util.Map;

public class RandomSpellLootTableFunction extends ConditionalLootFunction {
    private RandomSpellLootTableFunction(LootCondition[] conditions) {
        super(conditions);
    }

    public ItemStack process(ItemStack stack, LootContext context) {
        double chance = 1.0d;
        while (!(context.getRandom().nextDouble() > chance)) {
            Spell[] spells = SpellRegistry.getSpells();
            int index = context.getRandom().nextInt(spells.length);
            Map<Identifier, Integer> spell = SpellHelper.getSpells(stack);
            spell.put(spells[index].getID(), spells[index].getLevel());
            SpellHelper.setSpells(stack, spell);
            chance = chance * 0.25d;
        }
        return stack;
    }

    public static class Builder extends ConditionalLootFunction.Builder<RandomSpellLootTableFunction.Builder> {
        @Override
        protected RandomSpellLootTableFunction.Builder getThisBuilder() {
            return this;
        }

        public LootFunction build() {
            return new RandomSpellLootTableFunction(getConditions());
        }
    }
}