package com.gpl.rpg.atcontentstudio.model.maps;

import java.awt.Image;
import java.awt.Point;
import java.io.ByteArrayOutputStream;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Result;
import javax.xml.transform.Source;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;

import com.gpl.rpg.atcontentstudio.ATContentStudio;
import com.gpl.rpg.atcontentstudio.model.GameDataElement;
import com.gpl.rpg.atcontentstudio.model.ProjectTreeNode;
import com.gpl.rpg.atcontentstudio.model.SaveEvent;
import com.gpl.rpg.atcontentstudio.model.gamedata.GameDataSet;
import com.gpl.rpg.atcontentstudio.ui.DefaultIcons;

public class WorldmapSegment extends GameDataElement {

	private static final long serialVersionUID = 2658610076889592723L;
	
	public static final String TEMP_LABEL_KEY = "ATCS_INTERNAL_TEMPORARY_KEY_FOR_LABEL";
	
	public int segmentX;
	public int segmentY;
	public Map<String, Point> mapLocations = new LinkedHashMap<String, Point>();
	public Map<String, List<String>> labelledMaps = new LinkedHashMap<String, List<String>>();
	public Map<String, NamedArea> labels = new LinkedHashMap<String, NamedArea>();
	public Element xmlNode;
	
	public WorldmapSegment(Worldmap parent, String name, Element xmlNode) {
		this.parent = parent;
		this.id = name;
		this.xmlNode = xmlNode;
	}
	
	@Override
	public GameDataSet getDataSet() {
		return parent.getDataSet();
	}

	@Override
	public String getDesc() {
		return (needsSaving() ? "*" : "")+id;
	}

	@Override
	public void parse() {
		segmentX = Integer.parseInt(xmlNode.getAttribute("x"));
		segmentY = Integer.parseInt(xmlNode.getAttribute("y"));
		NodeList mapsList = xmlNode.getElementsByTagName("map");
		for (int j = 0; j < mapsList.getLength(); j++) {
			Element mapNode = (Element) mapsList.item(j);
			mapLocations.put(mapNode.getAttribute("id"), new Point(Integer.parseInt(mapNode.getAttribute("x")) - segmentX, Integer.parseInt(mapNode.getAttribute("y")) - segmentY));
			String area;
			if ((area = mapNode.getAttribute("area")) != null && !"".equals(area)) {
				if (labelledMaps.get(area) == null) {
					labelledMaps.put(area, new ArrayList<String>());
				}
				labelledMaps.get(area).add(mapNode.getAttribute("id"));
			}
		}
		NodeList namedAreasNodeList = xmlNode.getElementsByTagName("namedarea");
		for (int j = 0; j < namedAreasNodeList.getLength(); j++) {
			Element namedAreaNode = (Element) namedAreasNodeList.item(j);
			labels.put(namedAreaNode.getAttribute("id"), new NamedArea(namedAreaNode.getAttribute("id"), namedAreaNode.getAttribute("name"), namedAreaNode.getAttribute("type")));
		}
		this.state = State.parsed;
	}

	@Override
	public void link() {
		if (this.state == State.init) {
			this.parse();
		} else if (this.state == State.linked) {
			return;
		}
		for (String mapName : mapLocations.keySet()) {
			if (getProject().getMap(mapName) != null) {
				getProject().getMap(mapName).addBacklink(this);
			}
		}
	}

	@Override
	public WorldmapSegment clone() {
		WorldmapSegment clone = new WorldmapSegment((Worldmap)parent, id, (Element) xmlNode.cloneNode(true));
		
		return clone;
	}

	@Override
	public void elementChanged(GameDataElement oldOne, GameDataElement newOne) {
		boolean modified = false;
		if (newOne == null && writable) {
			//A referenced map may have been deleted.
			if (mapLocations.containsKey(oldOne.id)) {
				mapLocations.remove(oldOne.id);
				modified = true;
			}
			List<String> deprecatedLabels = new ArrayList<String>();
			for (String label : labelledMaps.keySet()) { 
				if (labelledMaps.get(label).contains(oldOne.id)) {
					labelledMaps.get(label).remove(oldOne.id);
					modified = true;
					if (labelledMaps.get(label).isEmpty()) {
						deprecatedLabels.add(label);
					}
				}
			}
			for (String label : deprecatedLabels) {
				labelledMaps.remove(label);
				labels.remove(label);
			}
		}
		
		oldOne.removeBacklink(this);
		if(newOne != null) newOne.addBacklink(this);
		
		if (modified) {
			this.state = GameDataElement.State.modified;
			childrenChanged(new ArrayList<ProjectTreeNode>());
			ATContentStudio.frame.editorChanged(this);
		}
	}

	@Override
	public String getProjectFilename() {
		return "worldmap.xml";
	}

	@Override
	public void save() {
		((Worldmap)parent).save();
	}
	
	public String toXml() {
		Document doc;
		try {
			doc = DocumentBuilderFactory.newInstance().newDocumentBuilder().newDocument();
			doc.setXmlVersion("1.0");
			Element root = doc.createElement("worldmap");
			doc.appendChild(root);
			root.appendChild(this.toXmlElement(doc));
			Transformer transformer = TransformerFactory.newInstance().newTransformer();
			ByteArrayOutputStream baos = new ByteArrayOutputStream();
			Result output = new StreamResult(baos);
			Source input = new DOMSource(doc);
			transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
			transformer.setOutputProperty(OutputKeys.INDENT, "yes");
			transformer.setOutputProperty("{http://xml.apache.org/xalan}indent-amount", "2");
			transformer.transform(input, output);
			return baos.toString();
		} catch (ParserConfigurationException e) {
			e.printStackTrace();
		} catch (TransformerException e) {
			e.printStackTrace();
		}
		return null;
		
	}
	
	public Element toXmlElement(Document doc) {
		Element element = doc.createElement("segment");
		element.setAttribute("id", id);
		element.setAttribute("x", Integer.toString(segmentX));
		element.setAttribute("y", Integer.toString(segmentY));
		
		for (String s : mapLocations.keySet()) {
			Element map = doc.createElement("map");
			map.setAttribute("id", s);
			map.setAttribute("x", Integer.toString(mapLocations.get(s).x + segmentX));
			map.setAttribute("y", Integer.toString(mapLocations.get(s).y + segmentY));
			for (String label : labelledMaps.keySet()) {
				if (TEMP_LABEL_KEY.equals(label)) continue;
				if (labelledMaps.get(label).contains(s)) {
					map.setAttribute("area", label);
				}
			}
			element.appendChild(map);
		}
		
		for (String key : labels.keySet()) {
			if (TEMP_LABEL_KEY.equals(key)) continue;
			NamedArea area = labels.get(key);
			Element namedArea = doc.createElement("namedarea");
			namedArea.setAttribute("id", area.id);
			namedArea.setAttribute("name", area.name);
			namedArea.setAttribute("type", area.type);
			element.appendChild(namedArea);
		}
		
		return element;
	}
	

	@Override
	public List<SaveEvent> attemptSave() {
		// TODO Auto-generated method stub
		save();
		return null;
	}

	public static class NamedArea {
		public String id;
		public String name;
		public String type;
		
		public NamedArea(String id, String name, String type) {
			this.id = id;
			this.name = name;
			this.type = type;
		}
	}
	
	@Override
	public Image getIcon() {
		return DefaultIcons.getUIMapIcon();
	}
	@Override
	public Image getLeafIcon() {
		return DefaultIcons.getUIMapIcon();
	}
	
}