CustomNPC-Plus/src/main/java/noppes/npcs/quests/QuestDialog.java

package noppes.npcs.quests;

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

import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import noppes.npcs.NBTTags;
import noppes.npcs.constants.EnumQuestType;
import noppes.npcs.controllers.Dialog;
import noppes.npcs.controllers.DialogController;
import noppes.npcs.controllers.PlayerData;
import noppes.npcs.controllers.PlayerDataController;
import noppes.npcs.scripted.CustomNPCsException;
import noppes.npcs.scripted.interfaces.handler.data.IQuestDialog;
import noppes.npcs.scripted.interfaces.handler.data.IQuestObjective;

public class QuestDialog extends QuestInterface implements IQuestDialog {

	public HashMap<Integer,Integer> dialogs = new HashMap<Integer,Integer>();

	@Override
	public void readEntityFromNBT(NBTTagCompound compound) {
		dialogs = NBTTags.getIntegerIntegerMap(compound.getTagList("QuestDialogs", 10));
	}

	@Override
	public void writeEntityToNBT(NBTTagCompound compound) {
		compound.setTag("QuestDialogs", NBTTags.nbtIntegerIntegerMap(dialogs));
	}

	@Override
	public boolean isCompleted(PlayerData playerData) {
		for(int dialogId : dialogs.values())
			if(!playerData.dialogData.dialogsRead.contains(dialogId))
				return false;
		return true;
	}

	@Override
	public void handleComplete(EntityPlayer player) {
		
	}

	@Override
	public Vector<String> getQuestLogStatus(EntityPlayer player) {
		Vector<String> vec = new Vector<String>();
		for(int dialogId : dialogs.values()){
			Dialog dialog = DialogController.instance.dialogs.get(dialogId);
			if(dialog == null)
				continue;
			String title = dialog.title;
			if(PlayerDataController.instance.getPlayerData(player).dialogData.dialogsRead.contains(dialogId))
				title += " (read)";
			else
				title += " (unread)";
			vec.add(title);
		}
		
		return vec;
	}

	public IQuestObjective[] getObjectives(EntityPlayer player) {
		List<IQuestObjective> list = new ArrayList();

		for(int i = 0; i < 3; ++i) {
			if (this.dialogs.containsKey(i)) {
				Dialog dialog = (Dialog)DialogController.instance.dialogs.get(this.dialogs.get(i));
				if (dialog != null) {
					list.add(new noppes.npcs.quests.QuestDialog.QuestDialogObjective(this, player, dialog));
				}
			}
		}

		return (IQuestObjective[])list.toArray(new IQuestObjective[list.size()]);
	}

	class QuestDialogObjective implements IQuestObjective {
		private final QuestDialog parent;
		private final EntityPlayer player;
		private final Dialog dialog;

		public QuestDialogObjective(QuestDialog this$0, EntityPlayer player, Dialog dialog) {
			this.parent = this$0;
			this.player = player;
			this.dialog = dialog;
		}

		public int getProgress() {
			return this.isCompleted() ? 1 : 0;
		}

		public void setProgress(int progress) {
			if (progress >= 0 && progress <= 1) {
				PlayerData data = PlayerDataController.instance.getPlayerData(player);
				boolean completed = data.dialogData.dialogsRead.contains(this.dialog.id);
				if (progress == 0 && completed) {
					data.dialogData.dialogsRead.remove(this.dialog.id);
					data.questData.checkQuestCompletion(data, EnumQuestType.values()[1]);
					data.saveNBTData(data.getNBT());
				}

				if (progress == 1 && !completed) {
					data.dialogData.dialogsRead.add(this.dialog.id);
					data.questData.checkQuestCompletion(data, EnumQuestType.values()[1]);
					data.saveNBTData(data.getNBT());
				}

			} else {
				throw new CustomNPCsException("Progress has to be 0 or 1", new Object[0]);
			}
		}

		public int getMaxProgress() {
			return 1;
		}

		public boolean isCompleted() {
			PlayerData data = PlayerDataController.instance.getPlayerData(player);
			return data.dialogData.dialogsRead.contains(this.dialog.id);
		}

		public String getText() {
			return this.dialog.title + (this.isCompleted() ? " (read)" : " (unread)");
		}
	}

}