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

package noppes.npcs.controllers;

import net.minecraft.command.ICommandSender;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityList;
import net.minecraft.nbt.CompressedStreamTools;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.nbt.NBTTagString;
import net.minecraft.util.ChatComponentText;
import noppes.npcs.CustomNpcs;
import noppes.npcs.LogWriter;
import noppes.npcs.NoppesUtilServer;
import noppes.npcs.api.IPos;
import noppes.npcs.api.IWorld;
import noppes.npcs.api.entity.IEntity;
import noppes.npcs.api.handler.ICloneHandler;
import noppes.npcs.controllers.data.TagMap;
import noppes.npcs.scripted.CustomNPCsException;
import noppes.npcs.scripted.NpcAPI;
import noppes.npcs.util.NBTJsonUtil;

import java.io.File;
import java.io.FileInputStream;
import java.util.*;

public class ServerCloneController implements ICloneHandler {
	public static ServerCloneController Instance;

	public ServerCloneController(){
		loadClones();
	}

	private void loadClones(){
		try {
			File dir = new File(getDir(), "..");
			File file = new File(dir, "clonednpcs.dat");
			if(file.exists()){
				Map<Integer, Map<String, NBTTagCompound>> clones = loadOldClones(file);
				file.delete();
				file = new File(dir, "clonednpcs.dat_old");
				if(file.exists())
					file.delete();

				for(int tab : clones.keySet()){
					Map<String, NBTTagCompound> map = clones.get(tab);
					for(String name: map.keySet()){
						saveClone(tab, name, map.get(name));
					}
				}
			}
		} catch (Exception e) {
			LogWriter.except(e);
		}
	}

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

	private Map<Integer, Map<String, NBTTagCompound>> loadOldClones(File file) throws Exception{
		Map<Integer, Map<String, NBTTagCompound>> clones = new HashMap<Integer, Map<String, NBTTagCompound>>();
		NBTTagCompound nbttagcompound1 = CompressedStreamTools.readCompressed(new FileInputStream(file));
		NBTTagList list = nbttagcompound1.getTagList("Data", 10);
		if(list == null){
			return clones;
		}
		for(int i = 0; i < list.tagCount(); i++)
		{
			NBTTagCompound compound = list.getCompoundTagAt(i);
			if(!compound.hasKey("ClonedTab")){
				compound.setInteger("ClonedTab", 1);
			}

			Map<String, NBTTagCompound> tab = clones.get(compound.getInteger("ClonedTab"));
			if(tab == null)
				clones.put(compound.getInteger("ClonedTab"), tab = new HashMap<String, NBTTagCompound>());

			String name = compound.getString("ClonedName");
			int number = 1;
			while(tab.containsKey(name)){
				number++;
				name = String.format("%s%s", compound.getString("ClonedName"), number);
			}
			compound.removeTag("ClonedName");
			compound.removeTag("ClonedTab");
			compound.removeTag("ClonedDate");
			cleanTags(compound);
			tab.put(name, compound);
		}
		return clones;
	}

	public NBTTagCompound getCloneData(ICommandSender player, String name, int tab) {
		File file = new File(new File(getDir(), tab + ""), name + ".json");
		if(!file.exists()){
			if(player != null)
				player.addChatMessage(new ChatComponentText("Could not find clone file"));
			return null;
		}
		try {
			return NBTJsonUtil.LoadFile(file);
		} catch (Exception e) {
			LogWriter.error("Error loading: " + file.getAbsolutePath(), e);
			if(player != null)
				player.addChatMessage(new ChatComponentText(e.getMessage()));
		}
		return null;
	}

	public void saveClone(int tab, String name, NBTTagCompound compound){
		try {
			File dir = new File(getDir(), tab + "");
			if(!dir.exists())
				dir.mkdir();
			String filename = name + ".json";

			File file = new File(dir, filename + "_new");
			File file2 = new File(dir, filename);
			NBTJsonUtil.SaveFile(file, compound);
			addToTagMap(compound, name, tab);
			if(file2.exists()){
				file2.delete();
			}
			file.renameTo(file2);
		} catch (Exception e) {
			LogWriter.except(e);
		}
	}
	public List<String> getClones(int tab){
		List<String> list = new ArrayList<String>();
		File dir = new File(getDir(), tab + "");
		if(!dir.exists() || !dir.isDirectory())
			return list;
		for(String file : dir.list()){
			if(file.endsWith(".json"))
				list.add(file.substring(0, file.length() - 5));
		}
		return list;
	}

	public List<String> getClonesDate(int tab){
		List<String> list = new ArrayList<String>();
		File dir = new File(getDir(), tab + "");
		if(!dir.exists() || !dir.isDirectory())
			return list;
		File[] files = dir.listFiles();
		Arrays.sort(files, new Comparator<File>(){
			public int compare(File f1, File f2)
			{
				return Long.compare(f1.lastModified(), f2.lastModified());
			}
		});

		for(File file : files){
			String fileName = file.getName();
			if(fileName.endsWith(".json"))
				list.add(fileName.substring(0, fileName.length() - 5));
		}
		return list;
	}

	public boolean removeClone(String name, int tab){
		File file = new File(new File(getDir(), tab + ""), name + ".json");
		if(!file.exists())
			return false;
		file.delete();
		removeFromTagMap(name, tab);
		return true;
	}

	public String addClone(NBTTagCompound nbttagcompound, String name, int tab) {
		cleanTags(nbttagcompound);
		saveClone(tab, name, nbttagcompound);
		return name;
	}

	public String addClone(NBTTagCompound nbttagcompound, String name, int tab, NBTTagCompound tempTags) {
		cleanTagList(nbttagcompound, tempTags);
		cleanTags(nbttagcompound);
		saveClone(tab, name, nbttagcompound);
		return name;
	}

	public NBTTagCompound cleanTagList(NBTTagCompound nbttagcompound, NBTTagCompound tempTags){
		HashSet<UUID> tagUUIDs = new HashSet<UUID>();
		if(nbttagcompound.hasKey("TagUUIDs")){
			NBTTagList nbtTagList = nbttagcompound.getTagList("TagUUIDs",8);
			for (int i = 0; i < nbtTagList.tagCount(); i++) {
				tagUUIDs.add(UUID.fromString(nbtTagList.getStringTagAt(i)));
			}

			nbttagcompound.removeTag("TagUUIDs");
		}
		if(tempTags.hasKey("TempTagUUIDs")){
			NBTTagList nbtTagList = tempTags.getTagList("TempTagUUIDs",8);
			for (int i = 0; i < nbtTagList.tagCount(); i++) {
				tagUUIDs.add(UUID.fromString(nbtTagList.getStringTagAt(i)));
			}

			tempTags.removeTag("TempTagUUIDs");
		}

		if(tagUUIDs.size() > 0){
			NBTTagList nbtTagList = new NBTTagList();
			for (UUID uuid : tagUUIDs) {
				nbtTagList.appendTag(new NBTTagString(uuid.toString()));
			}
			nbttagcompound.setTag("TagUUIDs", nbtTagList);
		}

		return nbttagcompound;
	}

	public boolean addToTagMap(NBTTagCompound nbttagcompound, String name, int tab){
		HashSet<UUID> tagUUIDs = new HashSet<UUID>();
		if(nbttagcompound.hasKey("TagUUIDs")){
			NBTTagList nbtTagList = nbttagcompound.getTagList("TagUUIDs",8);
			for (int i = 0; i < nbtTagList.tagCount(); i++) {
				tagUUIDs.add(UUID.fromString(nbtTagList.getStringTagAt(i)));
			}
		}
		if(tagUUIDs.size() > 0){
			TagMap tagMap = ServerTagMapController.Instance.getTagMap(tab);
			tagMap.putClone(name, tagUUIDs);
			ServerTagMapController.Instance.saveTagMap(tagMap);
			return true;
		}
		return false;
	}

	public boolean removeFromTagMap(String name, int tab){
		TagMap tagMap = ServerTagMapController.Instance.getTagMap(tab);
		if(tagMap.removeClone(name)){
			ServerTagMapController.Instance.saveTagMap(tagMap);
			return true;
		}
		return false;
	}

	public void cleanTags(NBTTagCompound nbttagcompound){
		if(nbttagcompound.hasKey("ItemGiverId"))
			nbttagcompound.setInteger("ItemGiverId", 0);
		if(nbttagcompound.hasKey("TransporterId"))
			nbttagcompound.setInteger("TransporterId", -1);

		nbttagcompound.removeTag("StartPosNew");
		nbttagcompound.removeTag("StartPos");
		nbttagcompound.removeTag("MovingPathNew");
		nbttagcompound.removeTag("Pos");
		nbttagcompound.removeTag("Riding");

		if(!nbttagcompound.hasKey("ModRev"))
			nbttagcompound.setInteger("ModRev", 1);

		if(nbttagcompound.hasKey("TransformRole")){
			NBTTagCompound adv = nbttagcompound.getCompoundTag("TransformRole");
			adv.setInteger("TransporterId", -1);
			nbttagcompound.setTag("TransformRole", adv);
		}

		if(nbttagcompound.hasKey("TransformJob")){
			NBTTagCompound adv = nbttagcompound.getCompoundTag("TransformJob");
			adv.setInteger("ItemGiverId", 0);
			nbttagcompound.setTag("TransformJob", adv);
		}

		if(nbttagcompound.hasKey("TransformAI")){
			NBTTagCompound adv = nbttagcompound.getCompoundTag("TransformAI");
			adv.removeTag("StartPosNew");
			adv.removeTag("StartPos");
			adv.removeTag("MovingPathNew");
			nbttagcompound.setTag("TransformAI", adv);
		}
	}

	public IEntity spawn(double x, double y, double z, int tab, String name, IWorld world, boolean ignoreProtection) {
		NBTTagCompound compound = this.getCloneData((ICommandSender)null, name, tab);
		if (compound == null) {
			throw new CustomNPCsException("Unknown clone tab:" + tab + " name:" + name, new Object[0]);
		} else {
			Entity entity;
			if (!ignoreProtection) {
				entity = NoppesUtilServer.spawnCloneWithProtection(compound, (int) x, (int) y, (int) z, world.getMCWorld());
			} else {
				entity = NoppesUtilServer.spawnClone(compound, (int) x, (int) y, (int) z, world.getMCWorld());
			}
			return entity == null ? null : NpcAPI.Instance().getIEntity(entity);
		}
	}

	public IEntity spawn(IPos pos, int tab, String name, IWorld world, boolean ignoreProtection) {
		return this.spawn(pos.getX(),pos.getY(),pos.getZ(),tab,name,world,ignoreProtection);
	}

	public IEntity spawn(double x, double y, double z, int tab, String name, IWorld world) {
		return spawn(x, y, z, tab, name, world, true);
	}

	public IEntity spawn(IPos pos, int tab, String name, IWorld world) {
		return this.spawn(pos.getX(),pos.getY(),pos.getZ(),tab,name,world);
	}

	public IEntity[] getTab(int tab, IWorld world) {
		File dir = new File(getDir(), tab + "");
		if (!dir.exists() || !dir.isDirectory() || dir.listFiles() == null) {
			return new IEntity[]{};
		}

		ArrayList<IEntity> arrayList = new ArrayList<>();

		try {
			for (File file : dir.listFiles()) {
				if (file.getName().endsWith(".json")) {
					NBTTagCompound compound = NBTJsonUtil.LoadFile(file);
					Instance.cleanTags(compound);
					Entity entity = EntityList.createEntityFromNBT(compound, world.getMCWorld());
					arrayList.add(entity == null ? null : NpcAPI.Instance().getIEntity(entity));
				}
			}
		} catch (Exception ignored) {}

		return arrayList.toArray(new IEntity[]{});
	}

	public IEntity get(int tab, String name, IWorld world) {
		NBTTagCompound compound = this.getCloneData((ICommandSender)null, name, tab);
		if (compound == null) {
			throw new CustomNPCsException("Unknown clone tab:" + tab + " name:" + name, new Object[0]);
		} else {
			Instance.cleanTags(compound);
			Entity entity = EntityList.createEntityFromNBT(compound, world.getMCWorld());
			return entity == null ? null : NpcAPI.Instance().getIEntity(entity);
		}
	}

    public boolean has(int tab, String name) {
        NBTTagCompound compound = this.getCloneData((ICommandSender)null, name, tab);
        return compound != null;
    }


    public void set(int tab, String name, IEntity entity) {
		NBTTagCompound compound = new NBTTagCompound();
		if (!entity.getMCEntity().writeMountToNBT(compound))
			throw new CustomNPCsException("Cannot save dead entities", new Object[0]);

        this.cleanTags(compound);
        this.saveClone(tab, name, compound);
	}

	public void remove(int tab, String name) {
		this.removeClone(name, tab);
	}
}