CustomNPC-Plus/src/main/java/noppes/npcs/DataDisplay.java

package noppes.npcs;

import com.google.common.collect.Iterables;
import com.mojang.authlib.GameProfile;
import com.mojang.authlib.properties.Property;
import cpw.mods.fml.common.FMLCommonHandler;
import cpw.mods.fml.common.Loader;
import cpw.mods.fml.common.LoaderState;
import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;
import net.minecraft.client.Minecraft;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.nbt.NBTTagString;
import net.minecraft.nbt.NBTUtil;
import net.minecraft.server.MinecraftServer;
import net.minecraft.util.StringUtils;
import noppes.npcs.config.ConfigMain;
import noppes.npcs.controllers.data.AnimationData;
import noppes.npcs.controllers.data.HitboxData;
import noppes.npcs.controllers.data.SkinOverlay;
import noppes.npcs.controllers.data.TintData;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.entity.data.DataSkinOverlays;
import noppes.npcs.util.ValueUtil;

import java.util.*;

public class DataDisplay {
    public EntityNPCInterface npc;

    public String name;
    public String title = "";

    private int markovGeneratorId = 8; //roman,japanese,slavic,welsh,sami,oldnorse,ancientgreek,aztec,classic,spanish (0 - 9 inclusively)
    private int markovGender = 0; //0:random, 1:male, 2:female

    public byte skinType = 0;    //0:normal, 1:player, 2:url, 3:url64

    public String url = "";

    public GameProfile playerProfile;
    public String texture = "customnpcs:textures/entity/humanmale/Steve.png";
    public String cloakTexture = "";

    public DataSkinOverlays skinOverlayData;
    public long overlayRenderTicks = 0;

    public AnimationData animationData;
    public HitboxData hitboxData;
    public TintData tintData;

    public String glowTexture = "";

    public int visible = 0;        //0:visible 1:Invisible 2:semi-invisible
    public int modelSize = 5;
    public int showName = 0;
    public int modelType = 0;

    public boolean disableLivingAnimation = false;
    public byte showBossBar = 0;

    public ArrayList<UUID> invisibleToList = new ArrayList<>();

    @SideOnly(Side.CLIENT)
    public boolean isInvisibleToMe;
    @SideOnly(Side.CLIENT)
    public HashSet<Integer> tempInvisIds;

    public DataDisplay(EntityNPCInterface npc) {
        this.npc = npc;
        markovGeneratorId = new Random().nextInt(CustomNpcs.MARKOV_GENERATOR.length - 1);
        skinOverlayData = new DataSkinOverlays(npc);
        name = getRandomName();
        animationData = new AnimationData(this);
        hitboxData = new HitboxData();
        tintData = new TintData();
    }

    public String getRandomName() {
        return CustomNpcs.MARKOV_GENERATOR[markovGeneratorId].fetch(markovGender);
    }

    public NBTTagCompound writeToNBT(NBTTagCompound nbttagcompound) {
        nbttagcompound.setString("Name", name);
        nbttagcompound.setInteger("MarkovGeneratorId", markovGeneratorId);
        nbttagcompound.setInteger("MarkovGender", markovGender);
        nbttagcompound.setString("Title", title);
        nbttagcompound.setString("SkinUrl", url);
        nbttagcompound.setString("Texture", texture);

        nbttagcompound.setString("CloakTexture", cloakTexture);
        nbttagcompound.setByte("UsingSkinUrl", skinType);
        nbttagcompound.setString("GlowTexture", glowTexture);

        nbttagcompound = skinOverlayData.writeToNBT(nbttagcompound);

        nbttagcompound = animationData.writeToNBT(nbttagcompound);
        nbttagcompound = hitboxData.writeToNBT(nbttagcompound);
        nbttagcompound = tintData.writeToNBT(nbttagcompound);

        if (this.playerProfile != null) {
            NBTTagCompound nbttagcompound1 = new NBTTagCompound();
            NBTUtil.func_152460_a(nbttagcompound1, this.playerProfile);
            nbttagcompound.setTag("SkinUsername", nbttagcompound1);
        }

        nbttagcompound.setInteger("Size", modelSize);
        nbttagcompound.setInteger("modelType", modelType);

        nbttagcompound.setInteger("ShowName", showName);
        nbttagcompound.setInteger("NpcVisible", visible);

        nbttagcompound.setBoolean("NoLivingAnimation", disableLivingAnimation);
        nbttagcompound.setByte("BossBar", showBossBar);

        boolean saveNonPersistendIDs = Loader.instance().isInState(LoaderState.SERVER_STARTED);
        ArrayList<Integer> nonPersistedIDs = new ArrayList<>();

        NBTTagList list = new NBTTagList();
        for (UUID uuid : invisibleToList) {
            if (saveNonPersistendIDs) {
                EntityPlayer p = NoppesUtilServer.getPlayer(uuid);
                if (p != null)
                    nonPersistedIDs.add(p.getEntityId());
            }
            list.appendTag(new NBTTagString(uuid.toString()));
        }

        if (!nonPersistedIDs.isEmpty()) {
            int[] tempIDArr = nonPersistedIDs.stream().filter(Objects::nonNull).mapToInt(i -> i).toArray();
            nbttagcompound.setIntArray("InvisibleToNonPersistentID", tempIDArr);
        }

        nbttagcompound.setTag("InvisibleToList", list);
        return nbttagcompound;
    }

    public void readToNBT(NBTTagCompound nbttagcompound) {
        setName(nbttagcompound.getString("Name"));
        setMarkovGeneratorId(nbttagcompound.getInteger("MarkovGeneratorId"));
        setMarkovGender(nbttagcompound.getInteger("MarkovGender"));

        title = nbttagcompound.getString("Title");

        String prevUrl = url;
        url = nbttagcompound.getString("SkinUrl");

        int prevSkinType = skinType;
        skinType = nbttagcompound.getByte("UsingSkinUrl");

        this.playerProfile = null;
        if (skinType == 1) {
            if (nbttagcompound.hasKey("SkinUsername", 10)) {
                this.playerProfile = NBTUtil.func_152459_a(nbttagcompound.getCompoundTag("SkinUsername"));
            } else if (nbttagcompound.hasKey("SkinUsername", 8) && !StringUtils.isNullOrEmpty(nbttagcompound.getString("SkinUsername"))) {
                this.playerProfile = new GameProfile(null, nbttagcompound.getString("SkinUsername"));
            }
            this.loadProfile();
        }

        String prevTexture = texture;
        texture = nbttagcompound.getString("Texture");

        cloakTexture = nbttagcompound.getString("CloakTexture");
        glowTexture = nbttagcompound.getString("GlowTexture");

        if (!nbttagcompound.hasKey("SkinOverlayData") && !glowTexture.isEmpty()) {
            NBTTagCompound compound = new NBTTagCompound();
            compound.setInteger("SkinOverlayID", 0); //unique glow texture ID

            (new SkinOverlay(glowTexture)).writeToNBT(compound);

            if (!nbttagcompound.hasKey("SkinOverlayData")) {
                NBTTagList tagList = new NBTTagList();
                tagList.appendTag(compound);
                nbttagcompound.setTag("SkinOverlayData", tagList);
            } else if (!glowTexture.isEmpty()) {
                nbttagcompound.getTagList("SkinOverlayData", 10).appendTag(compound);
                glowTexture = "";
            }
        }

        skinOverlayData.readFromNBT(nbttagcompound);

        animationData.readFromNBT(nbttagcompound);
        hitboxData.readFromNBT(nbttagcompound);
        tintData.readFromNBT(nbttagcompound);

        modelSize = ValueUtil.clamp(nbttagcompound.getInteger("Size"), 1, Integer.MAX_VALUE);
        if (modelSize > ConfigMain.NpcSizeLimit)
            modelSize = ConfigMain.NpcSizeLimit;

        modelType = nbttagcompound.getInteger("modelType");

        showName = nbttagcompound.getInteger("ShowName");
        visible = nbttagcompound.getInteger("NpcVisible");

        disableLivingAnimation = nbttagcompound.getBoolean("NoLivingAnimation");
        showBossBar = nbttagcompound.getByte("BossBar");

        NBTTagList tagList = (NBTTagList) nbttagcompound.getTag("InvisibleToList");
        if (tagList != null) {
            invisibleToList.clear();
            for (int i = 0; i < tagList.tagCount(); i++) {
                String nbtTagString = tagList.getStringTagAt(i);
                invisibleToList.add(UUID.fromString(nbtTagString));
            }
        } else {
            invisibleToList = new ArrayList<>();
        }

        if (FMLCommonHandler.instance().getEffectiveSide() == Side.CLIENT) {

            if (tempInvisIds == null) {
                tempInvisIds = new HashSet<>();
            } else {
                tempInvisIds.clear();
            }

            int[] tempEntityIDList = nbttagcompound.getIntArray("InvisibleToNonPersistentID");
            final int playerID = Minecraft.getMinecraft().thePlayer.getEntityId();
            for (int i : tempEntityIDList) {
                if (i == playerID) {
                    isInvisibleToMe = true;
                }
                tempInvisIds.add(i);
            }
        }

        if (prevSkinType != skinType || !texture.equals(prevTexture) || !url.equals(prevUrl))
            npc.textureLocation = null;
        npc.updateHitbox();
    }

    public void loadProfile() {
        if (this.playerProfile != null && !StringUtils.isNullOrEmpty(this.playerProfile.getName()) && MinecraftServer.getServer() != null) {
            if (!this.playerProfile.isComplete() || !this.playerProfile.getProperties().containsKey("textures")) {
                GameProfile gameprofile = MinecraftServer.getServer().func_152358_ax().func_152655_a(this.playerProfile.getName());

                if (gameprofile != null) {
                    Property property = (Property) Iterables.getFirst(gameprofile.getProperties().get("textures"), (Object) null);

                    if (property == null) {
                        gameprofile = MinecraftServer.getServer().func_147130_as().fillProfileProperties(gameprofile, false);
                    }

                    this.playerProfile = gameprofile;
                }
            }
        }
    }

    public boolean showName() {
        if (npc.isKilled())
            return false;
        return showName == 0 || (showName == 2 && npc.isAttacking());
    }

    public String getSkinTexture() {
        return texture;
    }

    public void setSkinTexture(String texture) {
        if (this.texture.equals(texture))
            return;
        this.texture = texture;
        npc.textureLocation = null;
        skinType = 0;
        npc.updateClient = true;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        if (this.name.equals(name))
            return;
        this.name = name;
        npc.updateClient = true;
    }

    public int getMarkovGender() {
        return markovGender;
    }

    public void setMarkovGender(int gender) {
        if (markovGender == gender)
            return;
        this.markovGender = ValueUtil.clamp(gender, 0, 2);
    }

    public int getMarkovGeneratorId() {
        return markovGeneratorId;
    }

    public void setMarkovGeneratorId(int id) {
        if (markovGeneratorId == id)
            return;
        this.markovGeneratorId = ValueUtil.clamp(id, 0, CustomNpcs.MARKOV_GENERATOR.length - 1);
    }

    @SideOnly(Side.CLIENT)
    public boolean getTempScriptInvisible(int entityId) {
        return tempInvisIds.contains(entityId);
    }
}