CustomNPC-Plus/src/main/java/noppes/npcs/controllers/LinkedNpcController.java

package noppes.npcs.controllers;

import net.minecraft.nbt.NBTTagCompound;
import noppes.npcs.CustomNpcs;
import noppes.npcs.LogWriter;
import noppes.npcs.NBTTags;
import noppes.npcs.entity.EntityCustomNpc;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.util.JsonException;
import noppes.npcs.util.NBTJsonUtil;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class LinkedNpcController {
	public static LinkedNpcController Instance;
	public List<LinkedData> list = new ArrayList<LinkedData>();

	public LinkedNpcController(){
		Instance = this;
		load();
	}
	private void load(){
		try {
	        loadNpcs();
		} catch (Exception e) {
			LogWriter.except(e);
		}
	}

	public File getDir(){
		File dir = new File(CustomNpcs.getWorldSaveDirectory(), "linkednpcs");
		if(!dir.exists())
			dir.mkdir();
		return dir;
	}

	private void loadNpcs(){
		LogWriter.info("Loading Linked Npcs");
		File dir = getDir();
		if(dir.exists()){
			List<LinkedData> list = new ArrayList<LinkedData>();
			for(File file : dir.listFiles()){
				if(file.getName().endsWith(".json")){
					try{
						NBTTagCompound compound = NBTJsonUtil.LoadFile(file);;
						LinkedData linked = new LinkedData();
						linked.setNBT(compound);
						list.add(linked);
					}
					catch(Exception e){
						LogWriter.error("Error loading: " + file.getAbsolutePath(), e);
					}
				}
			}

			this.list = list;
		}
		LogWriter.info("Done loading Linked Npcs");
	}

	public void save(){
		for(LinkedData npc : list){
			try {
				saveNpc(npc);
			} catch (IOException e) {
				LogWriter.except(e);
			}
		}
	}

	private void saveNpc(LinkedData npc) throws IOException {
		File file = new File(getDir(), npc.name + ".json_new");
        File file1 = new File(getDir(), npc.name + ".json");
		try{
			NBTJsonUtil.SaveFile(file, npc.getNBT());
			if(file1.exists()){
				file1.delete();
			}
			file.renameTo(file1);

		} catch (JsonException e) {
			LogWriter.except(e);
		}
	}

	public static class LinkedData{
		public String name = "LinkedNpc";
		public long time = 0;
		public NBTTagCompound data = new NBTTagCompound();

		public LinkedData(){
			time = System.currentTimeMillis();
		}

		public void setNBT(NBTTagCompound compound){
			name = compound.getString("LinkedName");
			data = compound.getCompoundTag("NPCData");
		}

		public NBTTagCompound getNBT(){
			NBTTagCompound compound = new NBTTagCompound();
			compound.setString("LinkedName", name);
			compound.setTag("NPCData", data);
			return compound;
		}
	}

	public void loadNpcData(EntityNPCInterface npc) {
		if(npc.linkedName.isEmpty())
			return;
		LinkedData data = getData(npc.linkedName);
		if(data == null){
			npc.linkedLast = 0;
			npc.linkedName = "";
			npc.linkedData = null;
		}
		else{
			npc.linkedData = data;
			if(npc.posX == 0 && npc.posY == 0 && npc.posZ == 0)
				return;
			npc.linkedLast = data.time;
			List<int[]> points = npc.ais.getMovingPath();

			NBTTagCompound compound = NBTTags.NBTMerge(readNpcData(npc), data.data);

			npc.display.readToNBT(compound);
			npc.stats.readToNBT(compound);
			npc.advanced.readToNBT(compound);
			npc.inventory.readEntityFromNBT(compound);
			if(compound.hasKey("ModelData"))
				((EntityCustomNpc)npc).modelData.readFromNBT(compound.getCompoundTag("ModelData"));

			npc.ais.readToNBT(compound);
			npc.transform.readToNBT(compound);
			npc.ais.setMovingPath(points);
			npc.updateClient = true;
		}
	}

	private void cleanTags(NBTTagCompound compound){
		compound.removeTag("MovingPathNew");
	}

	public LinkedData getData(String name){
		for(LinkedData data : list){
			if(data.name.equalsIgnoreCase(name))
				return data;
		}
		return null;
	}

	private NBTTagCompound readNpcData(EntityNPCInterface npc){
		NBTTagCompound compound = new NBTTagCompound();
		npc.display.writeToNBT(compound);
		npc.inventory.writeEntityToNBT(compound);
		npc.stats.writeToNBT(compound);
		npc.ais.writeToNBT(compound);
		npc.advanced.writeToNBT(compound);
		npc.transform.writeToNBT(compound);
		compound.setTag("ModelData", ((EntityCustomNpc)npc).modelData.writeToNBT());
		return compound;
	}

	public void saveNpcData(EntityNPCInterface npc) {
		NBTTagCompound compound = readNpcData(npc);
		cleanTags(compound);

		if(npc.linkedData.data.equals(compound))
			return;

		npc.linkedData.data = compound;
		npc.linkedData.time = System.currentTimeMillis();
		save();
	}
	public void removeData(String name) {
		Iterator<LinkedData> ita = list.iterator();
		while(ita.hasNext()){
			if(ita.next().name.equalsIgnoreCase(name))
				ita.remove();
		}
		save();
	}

	public void addData(String name){
		if(getData(name) != null || name.isEmpty())
			return;

		LinkedData data = new LinkedData();
		data.name = name;
		list.add(data);
		save();
	}
}