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

package noppes.npcs.controllers;

import kamkeel.npcs.controllers.SyncController;
import kamkeel.npcs.network.enums.EnumSyncType;
import net.minecraft.entity.player.EntityPlayer;
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.NoppesUtilServer;
import noppes.npcs.api.entity.IPlayer;
import noppes.npcs.api.handler.ICustomEffectHandler;
import noppes.npcs.api.handler.data.ICustomEffect;
import noppes.npcs.controllers.data.CustomEffect;
import noppes.npcs.controllers.data.EffectKey;
import noppes.npcs.controllers.data.EffectScript;
import noppes.npcs.controllers.data.PlayerEffect;
import noppes.npcs.util.NBTJsonUtil;

import java.io.*;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import java.util.zip.GZIPInputStream;

import static noppes.npcs.scripted.event.PlayerEvent.EffectEvent.ExpirationType;

public class CustomEffectController implements ICustomEffectHandler {

    public static CustomEffectController Instance = new CustomEffectController();
    public HashMap<Integer, HashMap<Integer, CustomEffect>> indexMapper = new HashMap<>();

    public HashMap<Integer, CustomEffect> customEffectsSync = new HashMap<>();

    public HashMap<Integer, EffectScript> customEffectScriptHandlers = new HashMap<>();
    private HashMap<Integer, String> bootOrder;

    private int lastUsedID = 0;
    public ConcurrentHashMap<UUID, ConcurrentHashMap<EffectKey, PlayerEffect>> playerEffects = new ConcurrentHashMap<>();

    public CustomEffectController() {
        HashMap<Integer, CustomEffect> customEffects = new HashMap<>();
        registerEffectMap(0, customEffects);
    }

    public static CustomEffectController getInstance() {
        return Instance;
    }

    public <T extends CustomEffect> void registerEffectMap(int index, HashMap<Integer, T> effectHashMap) {
        indexMapper.put(index, (HashMap<Integer, CustomEffect>) effectHashMap);
    }

    public HashMap<Integer, CustomEffect> getCustomEffects() {
        return indexMapper.get(0);
    }

    public void load() {
        lastUsedID = 0;
        playerEffects.clear();
        bootOrder = new HashMap<>();
        LogWriter.info("Loading custom effects...");
        readCustomEffectMap();
        loadCustomEffects();
        LogWriter.info("Done loading custom effects.");
    }

    public void runEffects(EntityPlayer player) {
        Map<EffectKey, PlayerEffect> current = getPlayerEffects(player);
        for (Map.Entry<EffectKey, PlayerEffect> entry : current.entrySet()) {
            int id = entry.getKey().getId();
            int index = entry.getKey().getIndex();
            CustomEffect effect = get(id, index);
            if (effect != null) {
                effect.runEffect(player, entry.getValue());
            }
        }
    }

    public void killEffects(EntityPlayer player) {
        Map<EffectKey, PlayerEffect> current = getPlayerEffects(player);
        Iterator<Map.Entry<EffectKey, PlayerEffect>> iterator = current.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<EffectKey, PlayerEffect> entry = iterator.next();
            int id = entry.getKey().getId();
            int index = entry.getKey().getIndex();
            CustomEffect effect = get(id, index);
            if (effect != null) {
                if (effect.lossOnDeath) {
                    effect.onRemoved(player, entry.getValue(), ExpirationType.DEATH);
                    iterator.remove();
                }
            } else {
                iterator.remove();
            }
        }
    }

    public void delete(int id) {
        ICustomEffect effect = get(id);
        if (effect != null) {
            CustomEffect foundEffect = getCustomEffects().remove(effect.getID());
            customEffectScriptHandlers.remove(effect.getID());
            if (foundEffect != null && foundEffect.name != null) {
                File dir = this.getDir();
                for (File file : dir.listFiles()) {
                    if (!file.isFile() || !file.getName().endsWith(".json"))
                        continue;
                    if (file.getName().equalsIgnoreCase(foundEffect.name + ".json")) {
                        file.delete();
                        SyncController.syncRemove(EnumSyncType.CUSTOM_EFFECTS, foundEffect.getID());
                        break;
                    }
                }
                saveEffectLoadMap();
            }
        }
    }

    public int getUnusedId() {
        for (int catid : getCustomEffects().keySet()) {
            if (catid > lastUsedID)
                lastUsedID = catid;
        }
        lastUsedID++;
        return lastUsedID;
    }


    public CustomEffect get(int id, int index) {
        HashMap<Integer, CustomEffect> effectMap = indexMapper.get(index);
        return effectMap != null ? effectMap.get(id) : null;
    }

    public CustomEffect get(int id) {
        return get(id, 0);
    }

    public boolean has(int id, int index) {
        HashMap<Integer, CustomEffect> effectMap = indexMapper.get(index);
        return effectMap != null && effectMap.containsKey(id);
    }

    public boolean has(int id) {
        return has(id, 0);
    }

    public CustomEffect get(String name, int index) {
        HashMap<Integer, CustomEffect> effectMap = indexMapper.get(index);
        if (effectMap != null) {
            for (CustomEffect effect : effectMap.values()) {
                if (effect.getName().equalsIgnoreCase(name)) {
                    return effect;
                }
            }
        }
        return null;
    }

    public CustomEffect get(String name) {
        return get(name, 0);
    }

    public boolean has(String name, int index) {
        return get(name, index) != null;
    }

    public boolean has(String name) {
        return has(name, 0);
    }

    public ConcurrentHashMap<EffectKey, PlayerEffect> getPlayerEffects(EntityPlayer player) {
        UUID playerId = NoppesUtilServer.getUUID(player);
        ConcurrentHashMap<EffectKey, PlayerEffect> effects = playerEffects.get(playerId);
        if (effects == null) {
            effects = new ConcurrentHashMap<>();
            playerEffects.put(playerId, effects);
        }
        return effects;
    }

    public void removeEffect(EntityPlayer player, PlayerEffect effect, ExpirationType type) {
        if (effect == null)
            return;

        Map<EffectKey, PlayerEffect> currentEffects = getPlayerEffects(player);
        EffectKey key = new EffectKey(effect.id, effect.index);
        if (currentEffects.containsKey(key)) {
            CustomEffect parent = get(effect.id, effect.index);
            if (parent != null) {
                parent.onRemoved(player, effect, type);
            }
            currentEffects.remove(key);
        }
    }

    private boolean hasOther(String name, int id) {
        for (CustomEffect effect : getCustomEffects().values()) {
            if (effect.getID() != id && effect.getName().equalsIgnoreCase(name))
                return true;
        }
        return false;
    }

    public void clearEffects(EntityPlayer player) {
        Map<EffectKey, PlayerEffect> effects = getPlayerEffects(player);
        if (effects != null) {
            effects.clear();
        }
    }

    public void clearEffect(EntityPlayer player, int id) {
        clearEffect(player, id, 0);
    }

    public void clearEffect(EntityPlayer player, int id, int index) {
        Map<EffectKey, PlayerEffect> effects = getPlayerEffects(player);
        if (effects != null) {
            effects.remove(new EffectKey(id, index));
        }
    }


    public boolean hasEffect(EntityPlayer player, int id) {
        return hasEffect(player, id, 0);
    }

    public boolean hasEffect(EntityPlayer player, int id, int index) {
        return getPlayerEffects(player).containsKey(new EffectKey(id, index));
    }

    public int getEffectDuration(EntityPlayer player, int id) {
        return getEffectDuration(player, id, 0);
    }

    public int getEffectDuration(EntityPlayer player, int id, int index) {
        PlayerEffect effect = getPlayerEffects(player).get(new EffectKey(id, index));
        return effect != null ? effect.duration : -1;
    }

    public void applyEffect(EntityPlayer player, int id, int duration, byte level) {
        applyEffect(player, id, duration, level, 0);
    }

    public void applyEffect(EntityPlayer player, int id, int duration, byte level, int index) {
        if (player == null || id <= 0) return;
        Map<EffectKey, PlayerEffect> currentEffects = getPlayerEffects(player);
        CustomEffect parent = get(id, index);
        if (parent != null) {
            PlayerEffect playerEffect = new PlayerEffect(id, duration, level, index);
            currentEffects.put(new EffectKey(id, index), playerEffect);
            parent.onAdded(player, playerEffect);
        }
    }

    public void removeEffect(EntityPlayer player, int id) {
        removeEffect(player, id, 0, ExpirationType.REMOVED);
    }

    public void removeEffect(EntityPlayer player, int id, ExpirationType type) {
        removeEffect(player, id, 0, type);
    }

    public void removeEffect(EntityPlayer player, int id, int index) {
        removeEffect(player, id, index, ExpirationType.REMOVED);
    }

    public void removeEffect(EntityPlayer player, int id, int index, ExpirationType type) {
        if (player == null || id <= 0) return;
        Map<EffectKey, PlayerEffect> currentEffects = getPlayerEffects(player);
        EffectKey key = new EffectKey(id, index);
        PlayerEffect effect = currentEffects.get(key);
        if (effect != null) {
            this.removeEffect(player, effect, type);
        }
    }

    public void decrementEffects(EntityPlayer player) {
        Iterator<PlayerEffect> iterator = getPlayerEffects(player).values().iterator();
        IPlayer iPlayer = NoppesUtilServer.getIPlayer(player);
        while (iterator.hasNext()) {
            PlayerEffect effect = iterator.next();
            if (effect == null) {
                iterator.remove();
                continue;
            }
            if (effect.duration == -100)
                continue;
            if (effect.duration <= 0) {
                CustomEffect parent = CustomEffectController.Instance.get(effect.id, effect.index);
                if (parent != null) {
                    parent.onRemoved(player, effect, ExpirationType.RUN_OUT);
                }
                iterator.remove();
                continue;
            }
            effect.duration--;
        }
    }

    public File getMapDir() {
        File dir = CustomNpcs.getWorldSaveDirectory();
        if (!dir.exists())
            dir.mkdir();
        return dir;
    }

    private void loadCustomEffects() {
        getCustomEffects().clear();

        File dir = getDir();
        if (!dir.exists()) {
            dir.mkdir();
        } else {
            for (File file : dir.listFiles()) {
                if (!file.isFile() || !file.getName().endsWith(".json"))
                    continue;
                try {
                    CustomEffect effect = new CustomEffect();
                    effect.readFromNBT(NBTJsonUtil.LoadFile(file));
                    effect.name = file.getName().substring(0, file.getName().length() - 5);
                    if (effect.id == -1) {
                        effect.id = getUnusedId();
                    }
                    int originalID = effect.id;
                    int setID = effect.id;
                    while (bootOrder.containsKey(setID) || getCustomEffects().containsKey(setID)) {
                        if (bootOrder.containsKey(setID))
                            if (bootOrder.get(setID).equalsIgnoreCase(effect.name))
                                break;
                        setID++;
                    }
                    effect.id = setID;
                    if (originalID != setID) {
                        LogWriter.info("Found Custom Effect ID Mismatch: " + effect.name + ", New ID: " + setID);
                        effect.save();
                    }
                    getCustomEffects().put(effect.id, effect);
                } catch (Exception e) {
                    LogWriter.error("Error loading: " + file.getAbsolutePath(), e);
                }
            }
        }
        this.registerEffectMap(0, getCustomEffects());
        saveEffectLoadMap();
    }

    public HashMap<Integer, CustomEffect> getEffectMap(int index) {
        if (indexMapper.containsKey(index)) {
            return getEffectMap(index);
        }
        return null;
    }

    private File getDir() {
        return new File(CustomNpcs.getWorldSaveDirectory(), "customeffects");
    }

    private void saveEffectLoadMap() {
        try {
            File saveDir = getMapDir();
            File file = new File(saveDir, "customeffects.dat_new");
            File file1 = new File(saveDir, "customeffects.dat_old");
            File file2 = new File(saveDir, "customeffects.dat");
            CompressedStreamTools.writeCompressed(this.writeMapNBT(), 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);
        }
    }

    private NBTTagCompound writeMapNBT() {
        NBTTagCompound nbt = new NBTTagCompound();
        NBTTagList customEffectsList = new NBTTagList();
        for (Integer key : this.getCustomEffects().keySet()) {
            CustomEffect customEffect = this.getCustomEffects().get(key);
            if (!customEffect.getName().isEmpty()) {
                NBTTagCompound effectCompound = new NBTTagCompound();
                effectCompound.setString("Name", customEffect.getName());
                effectCompound.setInteger("ID", key);
                customEffectsList.appendTag(effectCompound);
            }
        }
        nbt.setTag("CustomEffects", customEffectsList);
        nbt.setInteger("lastID", lastUsedID);
        return nbt;
    }

    private void readCustomEffectMap() {
        bootOrder.clear();
        try {
            File file = new File(getMapDir(), "customeffects.dat");
            if (file.exists()) {
                loadCustomEffectMap(file);
            }
        } catch (Exception e) {
            try {
                File file = new File(getMapDir(), "customeffects.dat_old");
                if (file.exists()) {
                    loadCustomEffectMap(file);
                }
            } catch (Exception ignored) {
            }
        }
    }

    private void loadCustomEffectMap(File file) throws IOException {
        DataInputStream dis = new DataInputStream(new BufferedInputStream(new GZIPInputStream(new FileInputStream(file))));
        readCustomEffectMap(dis);
        dis.close();
    }

    private void readCustomEffectMap(DataInputStream stream) throws IOException {
        NBTTagCompound nbtCompound = CompressedStreamTools.read(stream);
        this.readMapNBT(nbtCompound);
    }

    private void readMapNBT(NBTTagCompound compound) {
        lastUsedID = compound.getInteger("lastID");
        NBTTagList list = compound.getTagList("CustomEffects", 10);
        if (list != null) {
            for (int i = 0; i < list.tagCount(); i++) {
                NBTTagCompound nbttagcompound = list.getCompoundTagAt(i);
                String effectName = nbttagcompound.getString("Name");
                Integer key = nbttagcompound.getInteger("ID");
                bootOrder.put(key, effectName);
            }
        }
    }

    public void deleteEffectFile(String prevName) {
        File dir = this.getDir();
        if (!dir.exists())
            dir.mkdirs();
        File file2 = new File(dir, prevName + ".json");
        if (file2.exists())
            file2.delete();
    }

    // API Versions
    @Override
    public ICustomEffect createEffect(String name) {
        if (has(name))
            return get(name);

        CustomEffect effect = new CustomEffect();
        effect.name = name;

        if (effect.id == -1) {
            int id = getUnusedId();
            while (getCustomEffects().containsKey(id)) {
                id = getUnusedId();
            }
            effect.id = id;
        }
        getCustomEffects().put(effect.id, effect);
        return effect;
    }

    @Override
    public ICustomEffect getEffect(String name) {
        return get(name);
    }

    @Override
    public void deleteEffect(String name) {
        ICustomEffect effect = getEffect(name);
        if (effect != null) {
            CustomEffect foundEffect = getCustomEffects().remove(effect.getID());
            customEffectScriptHandlers.remove(effect.getID());
            if (foundEffect != null && foundEffect.name != null) {
                File dir = this.getDir();
                for (File file : dir.listFiles()) {
                    if (!file.isFile() || !file.getName().endsWith(".json"))
                        continue;
                    if (file.getName().equalsIgnoreCase(foundEffect.name + ".json")) {
                        file.delete();
                        SyncController.syncRemove(EnumSyncType.CUSTOM_EFFECTS, foundEffect.getID());
                        break;
                    }
                }
                saveEffectLoadMap();
            }
        }
    }

    @Override
    public boolean hasEffect(IPlayer player, int id) {
        if (player == null || player.getMCEntity() == null)
            return false;
        return hasEffect((EntityPlayer) player.getMCEntity(), id);
    }

    @Override
    public boolean hasEffect(IPlayer player, ICustomEffect effect) {
        if (player == null || player.getMCEntity() == null)
            return false;
        return hasEffect((EntityPlayer) player.getMCEntity(), effect.getID(), effect.getIndex()));
    }

    @Override
    public int getEffectDuration(IPlayer player, int id) {
        if (player == null || player.getMCEntity() == null)
            return -1;
        return getEffectDuration((EntityPlayer) player.getMCEntity(), id);
    }

    @Override
    public int getEffectDuration(IPlayer player, ICustomEffect effect) {
        if(effect == null)
            return -1;
        return getEffectDuration(player, effect.getID(), effect.getIndex());
    }

    @Override
    public void applyEffect(IPlayer player, int id, int duration, byte level) {
        if (player == null || player.getMCEntity() == null)
            return;
        applyEffect((EntityPlayer) player.getMCEntity(), id, duration, level);
    }

    @Override
    public void applyEffect(IPlayer player, ICustomEffect effect, int duration, byte level) {
        applyEffect(player, effect.getID(), duration, level);
    }

    @Override
    public void removeEffect(IPlayer player, int id) {
        if (player == null || player.getMCEntity() == null)
            return;
        removeEffect((EntityPlayer) player.getMCEntity(), id, ExpirationType.REMOVED);
    }

    @Override
    public void removeEffect(IPlayer player, ICustomEffect effect) {
        removeEffect((EntityPlayer) player.getMCEntity(), (PlayerEffect) effect, ExpirationType.REMOVED);
    }

    @Override
    public void clearEffects(IPlayer player) {
        if (player == null || player.getMCEntity() == null)
            return;
        clearEffects((EntityPlayer) player);
    }

    @Override
    public void applyEffect(IPlayer player, int id, int duration, byte level, int index) {
        if (player == null || player.getMCEntity() == null)
            return;
        applyEffect((EntityPlayer) player.getMCEntity(), id, duration, level, index);
    }

    @Override
    public void applyEffect(IPlayer player, ICustomEffect effect, int duration, byte level, int index) {
        applyEffect(player, effect.getID(), duration, level, index);
    }

    @Override
    public void removeEffect(IPlayer player, int id, int index) {
        if (player == null || player.getMCEntity() == null)
            return;
        removeEffect((EntityPlayer) player.getMCEntity(), id, index, ExpirationType.REMOVED);
    }

    @Override
    public void removeEffect(IPlayer player, ICustomEffect effect, int index) {
        removeEffect(player, effect.getID(), index);
    }

    @Override
    public void clearEffects(IPlayer player, int index) {
        if (player == null || player.getMCEntity() == null)
            return;
        EntityPlayer entity = (EntityPlayer) player.getMCEntity();
        ConcurrentHashMap<EffectKey, PlayerEffect> effects = getPlayerEffects(entity);
        Iterator<Map.Entry<EffectKey, PlayerEffect>> iterator = effects.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<EffectKey, PlayerEffect> entry = iterator.next();
            if (entry.getKey().getIndex() == index) {
                CustomEffect parent = get(entry.getKey().getId(), index);
                if (parent != null) {
                    parent.onRemoved(entity, entry.getValue(), ExpirationType.REMOVED);
                }
                iterator.remove();
            }
        }
    }

    @Override
    public int getEffectDuration(IPlayer player, int id, int index) {
        if (player == null || player.getMCEntity() == null)
            return -1;
        return getEffectDuration((EntityPlayer) player.getMCEntity(), id, index);
    }

    @Override
    public int getEffectDuration(IPlayer player, ICustomEffect effect, int index) {
        return getEffectDuration(player, effect.getID(), index);
    }

    @Override
    public ICustomEffect getEffect(String name, int index) {
        return get(name, index);
    }

    @Override
    public ICustomEffect getEffect(int id, int index) {
        return get(id, index);
    }

    @Override
    public ICustomEffect saveEffect(ICustomEffect customEffect) {
        if (customEffect.getID() < 0) {
            customEffect.setID(getUnusedId());
            while (has(customEffect.getName()))
                customEffect.setName(customEffect.getName() + "_");
        }

        while (hasOther(customEffect.getName(), customEffect.getID()))
            customEffect.setName(customEffect.getName() + "_");

        getCustomEffects().remove(customEffect.getID());
        getCustomEffects().put(customEffect.getID(), (CustomEffect) customEffect);

        saveEffectLoadMap();

        File dir = this.getDir();
        if (!dir.exists())
            dir.mkdirs();

        File file = new File(dir, customEffect.getName() + ".json_new");
        File file2 = new File(dir, customEffect.getName() + ".json");

        try {
            NBTTagCompound nbtTagCompound = ((CustomEffect) customEffect).writeToNBT(true);
            NBTJsonUtil.SaveFile(file, nbtTagCompound);
            if (file2.exists())
                file2.delete();
            file.renameTo(file2);
            nbtTagCompound.removeTag("ScriptData");
            SyncController.syncUpdate(EnumSyncType.CUSTOM_EFFECTS, -1, nbtTagCompound);
        } catch (Exception e) {
            LogWriter.except(e);
        }
        return getCustomEffects().get(customEffect.getID());
    }

}