CustomNPC-Plus/src/main/java/noppes/npcs/controllers/data/ProfileOptions.java

package noppes.npcs.controllers.data;

import net.minecraft.nbt.NBTTagCompound;
import noppes.npcs.api.handler.data.IProfileOptions;
import noppes.npcs.constants.EnumProfileSync;

public class ProfileOptions implements IProfileOptions {
    public boolean enableOptions = false;

    public EnumProfileSync cooldownControl = EnumProfileSync.Individual;
    public EnumProfileSync completeControl = EnumProfileSync.Individual;

    public void readFromNBT(NBTTagCompound compound) {
        // Party Management
        enableOptions = compound.getBoolean("EnableProfiles");
        if (enableOptions) {
            cooldownControl = EnumProfileSync.values()[compound.getInteger("CooldownControl")];
            completeControl = EnumProfileSync.values()[compound.getInteger("CompleteControl")];
        } else {
            if (compound.hasKey("CooldownControl")) {
                compound.removeTag("CooldownControl");
            }
            if (compound.hasKey("CompleteControl")) {
                compound.removeTag("CompleteControl");
            }

            cooldownControl = EnumProfileSync.Individual;
            completeControl = EnumProfileSync.Individual;
        }
    }

    public NBTTagCompound writeToNBT() {
        NBTTagCompound compound = new NBTTagCompound();
        compound.setBoolean("EnableProfiles", enableOptions);
        if (enableOptions) {
            compound.setInteger("CooldownControl", cooldownControl.ordinal());
            compound.setInteger("CompleteControl", completeControl.ordinal());
        } else {
            if (compound.hasKey("CooldownControl")) {
                compound.removeTag("CooldownControl");
            }
            if (compound.hasKey("CompleteControl")) {
                compound.removeTag("CompleteControl");
            }
        }

        return compound;
    }

    @Override
    public boolean hasProfileOptions() {
        return enableOptions;
    }

    @Override
    public void setProfileOptions(boolean enable) {
        this.enableOptions = enable;
    }

    @Override
    public void setCooldownControl(int profileType) {
        if (profileType < 0 || profileType >= EnumProfileSync.values().length) {
            return;
        }
        this.cooldownControl = EnumProfileSync.values()[profileType];
    }

    @Override
    public int getCooldownControl() {
        return this.cooldownControl.ordinal();
    }

    @Override
    public void setCompleteControl(int profileType) {
        if (profileType < 0 || profileType >= EnumProfileSync.values().length) {
            return;
        }
        this.completeControl = EnumProfileSync.values()[profileType];
    }

    @Override
    public int getCompleteControl() {
        return this.completeControl.ordinal();
    }
}