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)));
            }
        }

        TagMap tagMap = ServerTagMapController.Instance.getTagMap(tab);
        if (!tagUUIDs.isEmpty()) {
            tagMap.putClone(name, tagUUIDs);
        } else {
            tagMap.removeClone(name);
        }
        ServerTagMapController.Instance.saveTagMap(tagMap);
        return true;
    }

    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);
    }
}