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

package noppes.npcs.controllers;

import kamkeel.npcs.controllers.SyncController;
import kamkeel.npcs.network.enums.EnumSyncType;
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.api.handler.IFactionHandler;
import noppes.npcs.api.handler.data.IFaction;
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() {
        factions = new HashMap<Integer, Faction>();
        lastUsedID = 0;
        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);
        SyncController.syncUpdate(EnumSyncType.FACTION, -1, 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;
                SyncController.syncRemove(EnumSyncType.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;
    }
}