package net.osmand.binary;

import net.osmand.PlatformUtil;
import net.osmand.binary.StringBundle.Item;
import net.osmand.binary.StringBundle.StringItem;
import net.osmand.binary.StringBundle.StringListItem;
import net.osmand.binary.StringBundle.StringMapItem;

import org.apache.commons.logging.Log;
import org.xmlpull.v1.XmlSerializer;

import java.io.IOException;
import java.io.StringWriter;
import java.util.Map.Entry;

public class StringBundleXmlWriter extends StringBundleWriter {

	public final static Log log = PlatformUtil.getLog(StringBundleXmlWriter.class);

	private StringWriter writer;
	private XmlSerializer serializer;

	public StringBundleXmlWriter(StringBundle bundle) {
		super(bundle);

		writer = new StringWriter();
		serializer = PlatformUtil.newSerializer();
		try {
			serializer.setOutput(writer);
			serializer.setFeature("http://xmlpull.org/v1/doc/features.html#indent-output", true);
			serializer.startDocument("UTF-8", true);
		} catch (IOException e) {
			serializer = null;
		}
	}

	public StringBundleXmlWriter(StringBundle bundle, XmlSerializer serializer) {
		super(bundle);
		this.serializer = serializer;
	}

	@Override
	protected void writeItem(String name, Item item) {
		if (serializer != null) {
			try {
				writeItemImpl(name, item);
			} catch (Exception e) {
				log.error("Error writing string bundle as xml", e);
			}
		}
	}

	@Override
	public void writeBundle() {
		if (serializer != null) {
			super.writeBundle();
			try {
				serializer.flush();
				if (writer != null) {
					result = writer.toString();
				}
			} catch (Exception e) {
				log.error("Error writing string bundle as xml", e);
			}
		}
	}

	private void writeItemImpl(String name, Item item) throws IOException {
		if (serializer != null && item != null) {
			switch (item.getType()) {
				case STRING: {
					StringItem stringItem = (StringItem) item;
					serializer.startTag(null, "s");
					serializer.attribute(null, "n", name);
					serializer.text(stringItem.getValue());
					serializer.endTag(null, "s");
					break;
				}
				case LIST: {
					StringListItem listItem = (StringListItem) item;
					serializer.startTag(null, name);
					for (Item i : listItem.getValue()) {
						writeItemImpl(i.getName(), i);
					}
					serializer.endTag(null, name);
					break;
				}
				case MAP: {
					StringMapItem mapItem = (StringMapItem) item;
					serializer.startTag(null, name);
					for (Entry<String, Item> entry : mapItem.getValue().entrySet()) {
						writeItemImpl(entry.getKey(), entry.getValue());
					}
					serializer.endTag(null, name);
					break;
				}
			}
		}
	}
}