RewardSet.java

package joshie.progression.player.nbt;

import joshie.progression.api.criteria.IRewardProvider;
import joshie.progression.handlers.APICache;
import net.minecraft.nbt.NBTBase;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;

import java.util.UUID;

public class RewardSet extends AbstractUniqueNBT {
    public static final RewardSet INSTANCE = new RewardSet();

    @Override
    public NBTBase write(Object s) {
        NBTTagCompound tag = new NBTTagCompound();
        tag.setString("Reward", ((IRewardProvider) s).getUniqueID().toString());
        return tag;
    }

    @Override
    public Object read(NBTTagList list, int i) {
        NBTTagCompound tag = list.getCompoundTagAt(i);
        return APICache.getCache(false).getRewardFromUUID(UUID.fromString(tag.getString("Reward")));
    }
}