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

package noppes.npcs.controllers.data;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import noppes.npcs.api.handler.IPlayerTransportData;
import noppes.npcs.api.handler.data.ITransportLocation;
import noppes.npcs.controllers.TransportController;

import java.util.ArrayList;
import java.util.HashSet;

public class PlayerTransportData implements IPlayerTransportData {
	private final PlayerData parent;
	public HashSet<Integer> transports = new HashSet<Integer>();

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

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

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

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

	public boolean hasTransport(int id) {
		return transports.contains(id);
	}

	public void addTransport(int id) {
		transports.add(id);
	}

	public void addTransport(ITransportLocation location) {
		transports.add(location.getId());
	}

	public ITransportLocation getTransport(int id) {
		return TransportController.getInstance().getTransport(id);
	}

	public ITransportLocation[] getTransports() {
		ArrayList<ITransportLocation> list = new ArrayList<>();
		for (int id : transports) {
			ITransportLocation location = TransportController.getInstance().getTransport(id);
			list.add(location);
		}

		return list.toArray(new ITransportLocation[0]);
	}

	public void removeTransport(int id) {
		transports.remove(id);
	}
}