CustomNPC-Plus/src/main/java/noppes/npcs/controllers/data/PlayerQuestData.java

package noppes.npcs.controllers.data;

import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.entity.player.EntityPlayerMP;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import noppes.npcs.EventHooks;
import noppes.npcs.NoppesUtilPlayer;
import noppes.npcs.Server;
import noppes.npcs.api.handler.IPlayerQuestData;
import noppes.npcs.api.handler.data.IQuest;
import noppes.npcs.constants.*;
import noppes.npcs.controllers.PartyController;
import noppes.npcs.controllers.PlayerDataController;
import noppes.npcs.controllers.QuestController;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.quests.QuestInterface;
import noppes.npcs.quests.QuestItem;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

public class PlayerQuestData implements IPlayerQuestData {
	private final PlayerData parent;
	private IQuest trackedQuest = null;

	public HashMap<Integer,QuestData> activeQuests = new HashMap<Integer,QuestData>();
	public HashMap<Integer,Long> finishedQuests = new HashMap<Integer,Long>();

	public PlayerQuestData(PlayerData parent) {
		this.parent = parent;
	}

	public void loadNBTData(NBTTagCompound mainCompound) {
		if(mainCompound == null)
			return;
		NBTTagCompound compound = mainCompound.getCompoundTag("QuestData");

        NBTTagList list = compound.getTagList("CompletedQuests", 10);
        if(list != null){
        	HashMap<Integer,Long> finishedQuests = new HashMap<Integer,Long>();
            for(int i = 0; i < list.tagCount(); i++)
            {
                NBTTagCompound nbttagcompound = list.getCompoundTagAt(i);
                finishedQuests.put(nbttagcompound.getInteger("Quest"),nbttagcompound.getLong("Date"));
            }
            this.finishedQuests = finishedQuests;
        }

        NBTTagList list2 = compound.getTagList("ActiveQuests", 10);
        if(list2 != null){
        	HashMap<Integer,QuestData> activeQuests = new HashMap<Integer,QuestData>();
            for(int i = 0; i < list2.tagCount(); i++)
            {
                NBTTagCompound nbttagcompound = list2.getCompoundTagAt(i);
                int id = nbttagcompound.getInteger("Quest");
                Quest quest = QuestController.Instance.quests.get(id);
                if(quest == null)
                	continue;
                QuestData data = new QuestData(quest);
                data.readEntityFromNBT(nbttagcompound);
                activeQuests.put(id,data);
            }
            this.activeQuests = activeQuests;
        }

		this.trackedQuest = QuestController.Instance.get(mainCompound.getInteger("TrackedQuestID"));
	}

	public void saveNBTData(NBTTagCompound maincompound) {
		NBTTagCompound compound = new NBTTagCompound();
		NBTTagList list = new NBTTagList();
		for(int quest : finishedQuests.keySet()){
			NBTTagCompound nbttagcompound = new NBTTagCompound();
			nbttagcompound.setInteger("Quest", quest);
			nbttagcompound.setLong("Date", finishedQuests.get(quest));
			list.appendTag(nbttagcompound);
		}

		compound.setTag("CompletedQuests", list);

		NBTTagList list2 = new NBTTagList();
		for(int quest : activeQuests.keySet()){
			NBTTagCompound nbttagcompound = new NBTTagCompound();
			nbttagcompound.setInteger("Quest", quest);
			activeQuests.get(quest).writeEntityToNBT(nbttagcompound);
			list2.appendTag(nbttagcompound);
		}

		compound.setTag("ActiveQuests", list2);

		maincompound.setTag("QuestData", compound);

		if (this.trackedQuest != null) {
			maincompound.setInteger("TrackedQuestID", this.trackedQuest.getId());
		} else {
			maincompound.setInteger("TrackedQuestID", -1);
		}
	}

	public QuestData getQuestCompletion(EntityPlayer player,EntityNPCInterface npc) {
		PlayerData playerData = PlayerDataController.Instance.getPlayerData(player);
		for(QuestData data : activeQuests.values()){
			Quest quest = data.quest;
			if(quest != null && quest.completion == EnumQuestCompletion.Npc && quest.completerNpc.equals(npc.getCommandSenderName()) && quest.questInterface.isCompleted(playerData)){
				return data;
			}
		}
		return null;
	}

    public Party getPartyQuestCompletion(EntityPlayer player,EntityNPCInterface npc) {
        PlayerData playerData = PlayerDataController.Instance.getPlayerData(player);
        if(playerData != null){
            Party party = playerData.getPlayerParty();
            if(party != null){
                QuestData data = party.getQuestData();
                if(data != null){
                    Quest quest = data.quest;
                    if(quest != null && quest.completion == EnumQuestCompletion.Npc && quest.completerNpc.equals(npc.getCommandSenderName()) && quest.questInterface.isPartyCompleted(party)){
                        return party;
                    }
                }
            }
        }
        return null;
    }

	public boolean checkQuestCompletion(PlayerData playerData,EnumQuestType type) {
		boolean bo = false;
		EntityPlayer player = playerData.player;

        int partyQuestID = -1;
        if(playerData.partyUUID != null){
            Party party = PartyController.Instance().getParty(playerData.partyUUID);
            if(party != null){
                IQuest quest = party.getQuest();
                if(quest != null){
                    partyQuestID = quest.getId();
                }
            }
        }

		ArrayList<QuestData> activeQuestValues = new ArrayList<>(this.activeQuests.values());
		for(QuestData data : activeQuestValues){
			if(data.quest.type != type && type != null)
				continue;

            // Check if Active Quest = Party Quest
            if(partyQuestID != -1 && data.quest.id == partyQuestID && data.quest.partyOptions.isAllowParty()){
                continue;
            }

			QuestInterface inter =  data.quest.questInterface;
			if(inter.isCompleted(playerData)){
				if((!data.isCompleted && data.quest.completion == EnumQuestCompletion.Npc) || data.quest.instantComplete(player,data)){
					data.isCompleted = true;
					if (data.quest.completion == EnumQuestCompletion.Npc) {
						EventHooks.onQuestFinished(player, data.quest);
					}
					bo = true;
				}
			} else {
				data.isCompleted = false;
			}

			if (this.trackedQuest != null && data.quest.getId() == this.trackedQuest.getId()) {
				NoppesUtilPlayer.sendTrackedQuestData((EntityPlayerMP) player);
			}
		}
		QuestItem.pickedUp = null;
		return bo;
	}

	public void trackQuest(IQuest quest) {
		if (this.trackedQuest == null || quest.getId() != this.trackedQuest.getId()) {
			this.trackedQuest = quest;
			NoppesUtilPlayer.sendTrackedQuestData((EntityPlayerMP) this.parent.player);
		}
	}

	public void untrackQuest() {
		if (this.trackedQuest != null) {
			this.trackedQuest = null;
			Server.sendData((EntityPlayerMP) this.parent.player, EnumPacketClient.OVERLAY_QUEST_TRACKING);
		}
	}

	public IQuest getTrackedQuest() {
		return this.trackedQuest;
	}

	public void startQuest(int id) {
		Quest quest = QuestController.Instance.quests.get(id);
		if (quest == null)
			return;
		if(activeQuests.containsKey(id))
			return;
		QuestData questdata = new QuestData(quest);
		activeQuests.put(id, questdata);
		Server.sendData((EntityPlayerMP)parent.player, EnumPacketClient.MESSAGE, "quest.newquest", quest.title);
		Server.sendData((EntityPlayerMP)parent.player, EnumPacketClient.CHAT, "quest.newquest", ": ", quest.title);
	}

	public void finishQuest(int id) {
		Quest quest = QuestController.Instance.quests.get(id);
		if (quest == null)
			return;

		if(quest.repeat == EnumQuestRepeat.RLDAILY || quest.repeat == EnumQuestRepeat.RLWEEKLY)
			finishedQuests.put(quest.id, System.currentTimeMillis());
		else
			finishedQuests.put(quest.id, parent.player.worldObj.getTotalWorldTime());
	}

	public void stopQuest(int id) {
		Quest quest = QuestController.Instance.quests.get(id);
		if (quest == null)
			return;
		activeQuests.remove(id);
	}

	public void removeQuest(int id) {
		Quest quest = QuestController.Instance.quests.get(id);
		if (quest == null)
			return;
		activeQuests.remove(id);
		finishedQuests.remove(id);
	}

	public boolean hasFinishedQuest(int id){
		return finishedQuests.containsKey(id);
	}

	public boolean hasActiveQuest(int id){
		return activeQuests.containsKey(id);
	}

	public IQuest[] getActiveQuests() {
		List<IQuest> quests = new ArrayList<>();

		for (int id : activeQuests.keySet()) {
			IQuest quest = (IQuest) QuestController.Instance.quests.get(id);
			if (quest != null) {
				quests.add(quest);
			}
		}

		return (IQuest[])quests.toArray(new IQuest[0]);
	}

	public IQuest[] getFinishedQuests() {
		List<IQuest> quests = new ArrayList<>();

		for (int id : finishedQuests.keySet()) {
			IQuest quest = (IQuest) QuestController.Instance.quests.get(id);
			if (quest != null) {
				quests.add(quest);
			}
		}

		return (IQuest[])quests.toArray(new IQuest[0]);
	}

	public long getLastCompletedTime(int id) {
		if (this.hasFinishedQuest(id)) {
			return this.finishedQuests.get(id);
		}
		return 0;
	}

	public void setLastCompletedTime(int id,long time) {
		Quest quest = QuestController.Instance.quests.get(id);
		if (quest == null)
			return;
		this.finishedQuests.put(id,time);
	}
}