HexCasting/Forge/src/main/java/at/petrak/hexcasting/forge/cap/adimpl/CapStaticMediaHolder.java

package at.petrak.hexcasting.forge.cap.adimpl;

import at.petrak.hexcasting.api.addldata.ADMediaHolder;
import net.minecraft.world.item.ItemStack;

import java.util.function.Supplier;

/**
 * Things that always hold a constant amount of media, like amethyst
 */
public record CapStaticMediaHolder(Supplier<Integer> baseWorth,
                                   int consumptionPriority,
                                   ItemStack stack) implements ADMediaHolder {
    @Override
    public long getMedia() {
        return (long) baseWorth.get() * stack.getCount();
    }

    @Override
    public long getMaxMedia() {
        return getMedia();
    }

    @Override
    public void setMedia(long media) {
        // NO-OP
    }

    @Override
    public boolean canRecharge() {
        return false;
    }

    @Override
    public boolean canProvide() {
        return true;
    }

    @Override
    public int getConsumptionPriority() {
        return consumptionPriority;
    }

    @Override
    public boolean canConstructBattery() {
        return true;
    }

    @Override
    public long withdrawMedia(long cost, boolean simulate) {
        long worth = baseWorth.get();
        if (cost < 0) {
            cost = worth * stack.getCount();
        }
        double itemsRequired = cost / (double) worth;
        int itemsUsed = Math.min((int) Math.ceil(itemsRequired), stack.getCount());
        if (!simulate) {
            stack.shrink(itemsUsed);
        }
        return itemsUsed * worth;
    }
}