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

package noppes.npcs.controllers;

import net.minecraft.nbt.CompressedStreamTools;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import noppes.npcs.CustomNpcs;
import noppes.npcs.LogWriter;
import noppes.npcs.Server;
import noppes.npcs.api.handler.IFactionHandler;
import noppes.npcs.api.handler.data.IFaction;
import noppes.npcs.constants.EnumPacketClient;
import noppes.npcs.constants.SyncType;
import noppes.npcs.controllers.data.Faction;

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.zip.GZIPInputStream;

public class FactionController implements IFactionHandler {
    public HashMap<Integer,Faction> factionsSync = new HashMap<Integer,Faction>();
	public HashMap<Integer,Faction> factions;

	private static FactionController instance = new FactionController();

	private int lastUsedID = 0;

	public FactionController(){
		instance = this;
		factions = new HashMap<Integer, Faction>();
	}

	public static FactionController getInstance(){
		return instance;
	}

	public Faction getFaction(int faction) {
		return factions.get(faction);
	}

	public void load(){

		File saveDir = CustomNpcs.getWorldSaveDirectory();
		if(saveDir == null){
			return;
		}
		try {
			File file = new File(saveDir, "factions.dat");
			if(file.exists()){
				loadFactionsFile(file);
			}
		} catch (Exception e) {
			try {
				File file = new File(saveDir, "factions.dat_old");
				if(file.exists()){
					loadFactionsFile(file);
				}

			} catch (Exception ee) {
			}
		}

        if(factions.isEmpty()){
            factions.put(0,new Faction(0,"Friendly", 0x00DD00, 2000));
            factions.put(1,new Faction(1,"Neutral", 0xF2DD00, 1000));
            factions.put(2,new Faction(2,"Aggressive", 0xDD0000, 0));
        }
	}

	private void loadFactionsFile(File file) throws IOException{
		DataInputStream var1 = new DataInputStream(new BufferedInputStream(new GZIPInputStream(new FileInputStream(file))));
		loadFactions(var1);
		var1.close();
	}

	public void loadFactions(DataInputStream stream) throws IOException{
		HashMap<Integer,Faction> factions = new HashMap<Integer,Faction>();
		NBTTagCompound nbttagcompound1 = CompressedStreamTools.read(stream);
		lastUsedID = nbttagcompound1.getInteger("lastID");
		NBTTagList list = nbttagcompound1.getTagList("NPCFactions", 10);

		if(list != null){
			for(int i = 0; i < list.tagCount(); i++)
			{
				NBTTagCompound nbttagcompound = list.getCompoundTagAt(i);
				Faction faction = new Faction();
				faction.readNBT(nbttagcompound);
				factions.put(faction.id,faction);
			}
		}
		this.factions = factions;
	}
	public NBTTagCompound getNBT(){
		NBTTagList list = new NBTTagList();
		for(int slot : factions.keySet()){
			Faction faction = factions.get(slot);
			NBTTagCompound nbtfactions = new NBTTagCompound();
			faction.writeNBT(nbtfactions);
			list.appendTag(nbtfactions);
		}
		NBTTagCompound nbttagcompound = new NBTTagCompound();
		nbttagcompound.setInteger("lastID", lastUsedID);
		nbttagcompound.setTag("NPCFactions", list);
		return nbttagcompound;
	}
	public void saveFactions(){
		try {
			File saveDir = CustomNpcs.getWorldSaveDirectory();
			File file = new File(saveDir, "factions.dat_new");
			File file1 = new File(saveDir, "factions.dat_old");
			File file2 = new File(saveDir, "factions.dat");
			CompressedStreamTools.writeCompressed(getNBT(), new FileOutputStream(file));
			if(file1.exists())
			{
				file1.delete();
			}
			file2.renameTo(file1);
			if(file2.exists())
			{
				file2.delete();
			}
			file.renameTo(file2);
			if(file.exists())
			{
				file.delete();
			}
		} catch (Exception e) {
			LogWriter.except(e);
		}
	}

	public Faction get(int faction) {
		return factions.get(faction);
	}

	public List<IFaction> list() {
		return new ArrayList(this.factions.values());
	}

	public void saveFaction(Faction faction) {

		if(faction.id < 0){
			faction.id = getUnusedId();
			while(hasName(faction.name))
				faction.name += "_";
		}
		else{
			Faction existing = factions.get(faction.id);
			if(existing != null && !existing.name.equals(faction.name))
				while(hasName(faction.name))
					faction.name += "_";
		}
		factions.remove(faction.id);
		factions.put(faction.id, faction);

        NBTTagCompound facCompound = new NBTTagCompound();
        faction.writeNBT(facCompound);
        Server.sendToAll(EnumPacketClient.SYNC_UPDATE, SyncType.FACTION, facCompound);
		saveFactions();
	}

	public Faction create(Faction faction) {
		this.saveFaction(faction);
		return faction;
	}

	public Faction create(String name, int color) {
		Faction faction = new Faction();
		faction.name = name;
		faction.color = color;
		this.saveFaction(faction);
		return faction;
	}

	public int getUnusedId(){
		if(lastUsedID == 0){
			for(int catid : factions.keySet())
				if(catid > lastUsedID)
					lastUsedID = catid;
		}
		lastUsedID++;
		return lastUsedID;
	}
	public IFaction delete(int id) {
		if (id >= 0 && this.factions.size() > 1) {
			Faction faction = (Faction)this.factions.remove(id);
			saveFactions();
			if (faction == null) {
				return null;
			} else {
				this.saveFactions();
				faction.id = -1;
                Server.sendToAll(EnumPacketClient.SYNC_REMOVE, SyncType.FACTION, id);
				return faction;
			}
		} else {
			return null;
		}
	}

	public int getFirstFactionId() {
		return factions.keySet().iterator().next();
	}
	public Faction getFirstFaction() {
		return factions.values().iterator().next();
	}
	public boolean hasName(String newName) {
		if(newName.trim().isEmpty())
			return true;
		for(Faction faction : factions.values())
			if(faction.name.equals(newName))
				return true;
		return false;
	}

	public Faction getFactionFromName(String factioname){
		for (Map.Entry<Integer,Faction>entryfaction:FactionController.getInstance().factions.entrySet()){
			if (entryfaction.getValue().name.equalsIgnoreCase(factioname)){
				return entryfaction.getValue();
			}
		}
		return null;
	}

	public String[] getNames() {
		String[] names = new String[factions.size()];
		int i = 0;
		for(Faction faction : factions.values()){
			names[i] = faction.name.toLowerCase();
			i++;
		}
		return names;
	}
}