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

package noppes.npcs.controllers.data;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import noppes.npcs.api.handler.IPlayerDialogData;

import java.util.HashSet;

public class PlayerDialogData implements IPlayerDialogData {
    private final PlayerData parent;
    public HashSet<Integer> dialogsRead = new HashSet<Integer>();

    public PlayerDialogData(PlayerData parent) {
        this.parent = parent;
    }

    public void loadNBTData(NBTTagCompound compound) {
        HashSet<Integer> dialogsRead = new HashSet<Integer>();
        if (compound == null)
            return;
        NBTTagList list = compound.getTagList("DialogData", 10);
        if (list == null) {
            return;
        }

        for (int i = 0; i < list.tagCount(); i++) {
            NBTTagCompound nbttagcompound = list.getCompoundTagAt(i);
            dialogsRead.add(nbttagcompound.getInteger("Dialog"));
        }
        this.dialogsRead = dialogsRead;
    }

    public void saveNBTData(NBTTagCompound compound) {
        NBTTagList list = new NBTTagList();
        for (int dia : dialogsRead) {
            NBTTagCompound nbttagcompound = new NBTTagCompound();
            nbttagcompound.setInteger("Dialog", dia);
            list.appendTag(nbttagcompound);
        }

        compound.setTag("DialogData", list);
    }

    public boolean hasReadDialog(int id) {
        return dialogsRead.contains(id);
    }

    public void readDialog(int id) {
        dialogsRead.add(id);
    }

    public void unreadDialog(int id) {
        dialogsRead.remove(id);
    }
}