BloodMagic/src/main/java/WayofTime/bloodmagic/altar/AltarUpgrade.java

package WayofTime.bloodmagic.altar;

import WayofTime.bloodmagic.block.enums.BloodRuneType;
import com.google.common.collect.Maps;

import java.util.EnumMap;

public class AltarUpgrade {

    private final EnumMap<BloodRuneType, Integer> upgradeLevels;

    public AltarUpgrade() {
        this.upgradeLevels = Maps.newEnumMap(BloodRuneType.class);
    }

    public AltarUpgrade upgrade(BloodRuneType rune) {
        upgradeLevels.compute(rune, (r, l) -> l == null ? 1 : l + 1);
        return this;
    }

    public int getLevel(BloodRuneType rune) {
        return upgradeLevels.getOrDefault(rune, 0);
    }
}