CustomNPC-Plus/src/main/java/noppes/npcs/controllers/TransportController.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.api.handler.ITransportHandler;
import noppes.npcs.api.handler.data.ITransportCategory;
import noppes.npcs.constants.EnumRoleType;
import noppes.npcs.controllers.data.TransportCategory;
import noppes.npcs.controllers.data.TransportLocation;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.roles.RoleTransporter;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

public class TransportController implements ITransportHandler {
    private HashMap<Integer, TransportLocation> locations = new HashMap<Integer, TransportLocation>();
    public HashMap<Integer, TransportCategory> categories = new HashMap<Integer, TransportCategory>();

    private int lastUsedID = 0;

    private static TransportController instance;

    public TransportController() {
        instance = this;
        loadCategories();
        if (categories.isEmpty()) {
            TransportCategory cat = new TransportCategory();
            cat.id = 1;
            cat.title = "Default";
            categories.put(cat.id, cat);
        }
    }

    private void loadCategories() {
        File saveDir = CustomNpcs.getWorldSaveDirectory();
        if (saveDir == null)
            return;
        try {
            File file = new File(saveDir, "transport.dat");
            if (!file.exists()) {
                return;
            }
            loadCategories(file);
        } catch (IOException e) {
            try {
                File file = new File(saveDir, "transport.dat_old");
                if (!file.exists()) {
                    return;
                }
                loadCategories(file);

            } catch (IOException ee) {
            }
        }
    }

    public void loadCategories(File file) throws IOException {
        HashMap<Integer, TransportLocation> locations = new HashMap<Integer, TransportLocation>();
        HashMap<Integer, TransportCategory> categories = new HashMap<Integer, TransportCategory>();
        NBTTagCompound nbttagcompound1 = CompressedStreamTools.readCompressed(new FileInputStream(file));
        lastUsedID = nbttagcompound1.getInteger("lastID");
        NBTTagList list = nbttagcompound1.getTagList("NPCTransportCategories", 10);
        if (list == null) {
            return;
        }
        for (int i = 0; i < list.tagCount(); i++) {
            TransportCategory category = new TransportCategory();
            NBTTagCompound compound = list.getCompoundTagAt(i);
            category.readNBT(compound);

            for (TransportLocation location : category.locations.values())
                locations.put(location.id, location);

            categories.put(category.id, category);
        }
        this.locations = locations;
        this.categories = categories;
    }

    public NBTTagCompound getNBT() {
        NBTTagList list = new NBTTagList();
        for (TransportCategory category : categories.values()) {
            NBTTagCompound compound = new NBTTagCompound();
            category.writeNBT(compound);
            list.appendTag(compound);
        }
        NBTTagCompound nbttagcompound = new NBTTagCompound();
        nbttagcompound.setInteger("lastID", lastUsedID);
        nbttagcompound.setTag("NPCTransportCategories", list);
        return nbttagcompound;
    }

    public void saveCategories() {
        try {
            File saveDir = CustomNpcs.getWorldSaveDirectory();
            File file = new File(saveDir, "transport.dat_new");
            File file1 = new File(saveDir, "transport.dat_old");
            File file2 = new File(saveDir, "transport.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 TransportLocation getTransport(int transportId) {
        return locations.get(transportId);
    }

    public TransportLocation getTransport(String name) {
        for (TransportLocation loc : locations.values()) {
            if (loc.name.equals(name))
                return loc;
        }

        return null;
    }

    public int getUniqueIdLocation() {
        if (lastUsedID == 0) {
            for (int catid : locations.keySet())
                if (catid > lastUsedID)
                    lastUsedID = catid;
        }
        lastUsedID++;
        return lastUsedID;
    }

    private int getUniqueIdCategory() {
        int id = 0;
        for (int catid : categories.keySet())
            if (catid > id)
                id = catid;
        id++;
        return id;
    }

    public void setLocation(TransportLocation location) {
        if (locations.containsKey(location.id)) {
            for (TransportCategory cat : categories.values())
                cat.locations.remove(location.id);
        }
        locations.put(location.id, location);
        location.category.locations.put(location.id, location);
    }

    public TransportLocation removeLocation(int location) {
        TransportLocation loc = locations.get(location);
        if (loc == null)
            return null;
        loc.category.locations.remove(location);
        locations.remove(location);

        saveCategories();
        return loc;
    }

    private boolean containsCategoryName(String name) {
        name = name.toLowerCase();
        for (TransportCategory cat : categories.values()) {
            if (cat.title.toLowerCase().equals(name))
                return true;
        }
        return false;
    }

    public void saveCategory(String name, int id) {
        if (id < 0) {
            id = getUniqueIdCategory();
        }
        if (categories.containsKey(id)) {
            TransportCategory category = categories.get(id);
            if (!category.title.equals(name)) {
                while (containsCategoryName(name))
                    name += "_";
                categories.get(id).title = name;
            }
        } else {
            while (containsCategoryName(name))
                name += "_";
            TransportCategory category = new TransportCategory();
            category.id = id;
            category.title = name;
            categories.put(id, category);
        }
        saveCategories();
    }

    public void removeCategory(int id) {
        if (categories.size() == 1)
            return;
        TransportCategory cat = categories.get(id);
        if (cat == null)
            return;
        for (int i : cat.locations.keySet())
            locations.remove(i);
        categories.remove(id);

        saveCategories();
    }

    public boolean containsLocationName(String name) {
        name = name.toLowerCase();
        for (TransportLocation loc : locations.values()) {
            if (loc.name.toLowerCase().equals(name))
                return true;
        }
        return false;
    }

    public static TransportController getInstance() {
        return instance;
    }

    public TransportLocation saveLocation(int categoryId, TransportLocation location) {
        TransportCategory category = categories.get(categoryId);
        if (category == null)
            return null;
        location.category = category;
        if (location.id < 0 || !locations.get(location.id).name.equals(location.name)) {
            while (containsLocationName(location.name))
                location.name += "_";
        }
        if (location.id < 0)
            location.id = getUniqueIdLocation();

        category.locations.put(location.id, location);
        locations.put(location.id, location);
        saveCategories();

        return location;
    }

    public TransportLocation saveLocation(int categoryId, NBTTagCompound compound, EntityNPCInterface npc) {
        TransportCategory category = categories.get(categoryId);
        if (category == null || npc.advanced.role != EnumRoleType.Transporter)
            return null;
        RoleTransporter role = (RoleTransporter) npc.roleInterface;
        TransportLocation location = new TransportLocation();
        location.readNBT(compound);
        location.category = category;
        if (role.hasTransport())
            location.id = role.transportId;
        if (location.id < 0 || !locations.get(location.id).name.equals(location.name)) {
            while (containsLocationName(location.name))
                location.name += "_";
        }
        if (location.id < 0)
            location.id = getUniqueIdLocation();

        category.locations.put(location.id, location);
        locations.put(location.id, location);
        saveCategories();

        return location;
    }

    public ITransportCategory[] categories() {
        return (new ArrayList<>(categories.values())).toArray(new ITransportCategory[0]);
    }

    public void createCategory(String title) {
        this.saveCategory(title, -1);
    }

    public ITransportCategory getCategory(String title) {
        TransportCategory category = null;
        for (TransportCategory c : categories.values()) {
            if (c.title.equals(title)) {
                category = c;
                break;
            }
        }

        return category;
    }

    public void removeCategory(String title) {
        for (Map.Entry<Integer, TransportCategory> entry : categories.entrySet()) {
            if (entry.getValue().title.equals(title)) {
                categories.remove(entry.getKey());
                break;
            }
        }
    }
}