package com.aionemu.gameserver.dataholders;

import java.io.File;
import java.util.*;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.bind.annotation.*;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.aionemu.gameserver.model.templates.walker.WalkerTemplate;
import com.aionemu.gameserver.utils.xml.XmlUtil;

/**
 * @author KKnD, Rolandas
 */
@XmlRootElement(name = "npc_walker")
@XmlAccessorType(XmlAccessType.FIELD)
public class WalkerData {

	private static final Logger log = LoggerFactory.getLogger(WalkerData.class);

	@XmlElement(name = "walker_template")
	private List<WalkerTemplate> walkerlist;

	@XmlTransient
	private Map<String, WalkerTemplate> walkerlistData = new LinkedHashMap<>();

	void afterUnmarshal(Unmarshaller u, Object parent) {
		for (WalkerTemplate route : walkerlist) {
			if (walkerlistData.putIfAbsent(route.getRouteId(), route) != null)
				log.warn("Duplicate route ID: " + route.getRouteId());
		}
		walkerlist.clear();
		walkerlist = null;
	}

	public int size() {
		return walkerlistData.size();
	}

	public WalkerTemplate getWalkerTemplate(String routeId) {
		return walkerlistData.get(routeId);
	}

	public void addTemplate(WalkerTemplate newTemplate) {
		if (walkerlist == null)
			walkerlist = new ArrayList<>();
		walkerlist.add(newTemplate);
	}

	public void saveData(String routeId) {
		File xml = new File("./data/static_data/npc_walker/generated_npc_walker_" + routeId + ".xml");
		try {
			JAXBContext jc = JAXBContext.newInstance(WalkerData.class);
			Marshaller marshaller = jc.createMarshaller();
			marshaller.setSchema(XmlUtil.getSchema("./data/static_data/npc_walker/npc_walker.xsd"));
			marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
			marshaller.marshal(this, xml);
		} catch (JAXBException e) {
			log.error("Error while saving data: " + e.getMessage(), e.getCause());
			return;
		} finally {
			if (walkerlist != null) {
				walkerlist.clear();
				walkerlist = null;
			}
		}
	}

	public Collection<WalkerTemplate> getTemplates() {
		return walkerlistData.values();
	}

}