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

package noppes.npcs;

import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import noppes.npcs.api.handler.data.IQuest;
import noppes.npcs.constants.EnumQuestCompletion;
import noppes.npcs.controllers.PlayerDataController;
import noppes.npcs.controllers.PlayerQuestController;
import noppes.npcs.controllers.data.Party;
import noppes.npcs.controllers.data.PlayerData;
import noppes.npcs.controllers.data.Quest;

import java.util.HashMap;
import java.util.Vector;

public class QuestLogData {
	public String trackedQuestKey = "";
	public HashMap<String,Vector<String>> categories = new HashMap<String,Vector<String>>();
	public String selectedQuest = "";
	public String selectedCategory = "";
	public HashMap<String,String> questText = new HashMap<>();
	public HashMap<String,String> questAlerts = new HashMap<>();
	public HashMap<String,Vector<String>> questStatus = new HashMap<>();
	public HashMap<String,String> finish = new HashMap<>();
	public HashMap<String, Integer> partyQuests = new HashMap<>();
    public HashMap<String,Vector<String>> partyOptions = new HashMap<>();

	public NBTTagCompound writeNBT(){
		NBTTagCompound compound = new NBTTagCompound();
		compound.setTag("Categories", NBTTags.nbtVectorMap(categories));
		compound.setTag("Logs", NBTTags.nbtStringStringMap(questText));
		compound.setTag("Alerts", NBTTags.nbtStringStringMap(questAlerts));
		compound.setTag("Status", NBTTags.nbtVectorMap(questStatus));
		compound.setTag("QuestFinisher", NBTTags.nbtStringStringMap(finish));
		compound.setString("TrackedQuestID", trackedQuestKey);
		compound.setTag("PartyQuests", NBTTags.nbtStringIntegerMap(partyQuests));
        compound.setTag("PartyOptions", NBTTags.nbtVectorMap(partyOptions));
		return compound;
	}

	public void readNBT(NBTTagCompound compound){
		categories = NBTTags.getVectorMap(compound.getTagList("Categories", 10));
		questText = NBTTags.getStringStringMap(compound.getTagList("Logs", 10));
		questAlerts = NBTTags.getStringStringMap(compound.getTagList("Alerts", 10));
		questStatus = NBTTags.getVectorMap(compound.getTagList("Status", 10));
		finish = NBTTags.getStringStringMap(compound.getTagList("QuestFinisher", 10));
		trackedQuestKey = compound.getString("TrackedQuestID");
		partyQuests = NBTTags.getStringIntegerMap(compound.getTagList("PartyQuests", 10));
        partyOptions = NBTTags.getVectorMap(compound.getTagList("PartyOptions", 10));
	}

    public NBTTagCompound writeTrackedQuest(){
        NBTTagCompound compound = new NBTTagCompound();
        compound.setString("TrackedQuestID", trackedQuestKey);
        return compound;
    }

    public void readTrackedQuest(NBTTagCompound compound){
        trackedQuestKey = compound.getString("TrackedQuestID");
    }

	public void setData(EntityPlayer player){
		PlayerData playerData = PlayerDataController.Instance.getPlayerData(player);

		for(Quest quest : PlayerQuestController.getActiveQuests(player))
        {
    		String category = quest.category.title;
    		if(!categories.containsKey(category))
    			categories.put(category, new Vector<String>());
    		Vector<String> list = categories.get(category);
    		list.add(quest.title);

            String key = category + ":" + quest.title;
    		questText.put(key, quest.logText);
			questAlerts.put(key, String.valueOf(playerData.questData.activeQuests.get(quest.id).sendAlerts));
    		questStatus.put(key, quest.questInterface.getQuestLogStatus(player));
    		if(quest.completion == EnumQuestCompletion.Npc && quest.questInterface.isCompleted(playerData))
    			finish.put(key, quest.completerNpc);

			if (playerData.questData.getTrackedQuest() != null) {
				if (quest.id == playerData.questData.getTrackedQuest().getId()) {
					trackedQuestKey = key;
				}
			}

			if (quest.partyOptions.allowParty) {
				partyQuests.put(key, quest.id);
                partyOptions.put(key, quest.partyOptions.getPartyOptionsList());
			}
        }

        Party party = playerData.getPlayerParty();
        if(party != null && party.getQuest() != null){
            if (playerData.questData.getTrackedQuest() != null) {
                if (party.getQuest().getId() == playerData.questData.getTrackedQuest().getId()) {
                    trackedQuestKey = "P" + ":" + party.getQuest().getCategory().getName() + ":" + party.getQuest().getName();
                }
            }
        }
	}

    public void setTrackedQuestKey(EntityPlayer player){
        PlayerData playerData = PlayerDataController.Instance.getPlayerData(player);

        Party party = playerData.getPlayerParty();
        if(party != null && party.getQuest() != null){
            if (playerData.questData.getTrackedQuest() != null) {
                if (party.getQuest().getId() == playerData.questData.getTrackedQuest().getId()) {
                    trackedQuestKey = "P" + ":" + party.getQuest().getCategory().getName() + ":" + party.getQuest().getName();
                }
            }
        } else if(playerData.questData != null && playerData.questData.getTrackedQuest() != null){
            IQuest quest = playerData.questData.getTrackedQuest();
            trackedQuestKey = quest.getCategory().getName() + ":" + quest.getName();
        }
        else {
            trackedQuestKey = "";
        }
    }

	public boolean hasSelectedQuest() {
		return !selectedQuest.isEmpty();
	}

	public String getQuestText() {
		return questText.get(selectedCategory + ":" + selectedQuest);
	}

	public Boolean getQuestAlerts() {
		return Boolean.valueOf(questAlerts.get(selectedCategory+":"+selectedQuest));
	}

	public void toggleQuestAlerts() {
		questAlerts.put(selectedCategory+":"+selectedQuest,String.valueOf(!getQuestAlerts()));
	}

	public Vector<String> getQuestStatus() {
		return questStatus.get(selectedCategory + ":" + selectedQuest);
	}

	public String getComplete() {
		return finish.get(selectedCategory + ":" + selectedQuest);
	}

    public Vector<String> getPartyOptions() {
        return partyOptions.get(selectedCategory + ":" + selectedQuest);
    }
}