package com.thebrokenrail.energonrelics.block.entity.infuser;

import com.thebrokenrail.energonrelics.util.WeightedList;
import net.minecraft.item.ItemStack;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.World;

import java.util.ArrayList;
import java.util.List;

public class InfuserEntry {
    public final long cost;
    public final double successChance;
    private final InfuserAction[] success;
    private final InfuserAction[] fail;

    InfuserEntry(long cost, double successChance, InfuserAction[] success, InfuserAction[] fail) {
        this.cost = cost;
        this.successChance = successChance;
        this.success = success;
        this.fail = fail;
    }

    void run(World world, BlockPos pos) {
        boolean isSuccess = world.random.nextDouble() <= successChance;

        InfuserAction[] list;
        if (isSuccess) {
            list = success;
        } else {
            list = fail;
        }

        WeightedList<InfuserAction> weightedList = new WeightedList<>();
        for (InfuserAction action : list) {
            weightedList.add(1, action);
        }

        weightedList.pick(world.random).run(world, pos);
    }

    public List<ItemStack> getOutputItems(boolean isFail) {
        InfuserAction[] actionList = isFail ? fail : success;
        List<ItemStack> list = new ArrayList<>();
        for (InfuserAction action : actionList) {
            if (action instanceof InfuserAction.ItemAction) {
                list.add(((InfuserAction.ItemAction) action).stack);
            }
        }
        return list;
    }

    public double getSingleChance(boolean isFail) {
        return (1d / (double) success.length) * (isFail ? 1d - successChance : successChance);
    }
}