com.marklogic.client.impl.DOMWriter Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012-2015 MarkLogic Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.marklogic.client.impl;

import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;

import org.w3c.dom.Attr;
import org.w3c.dom.CDATASection;
import org.w3c.dom.Comment;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.w3c.dom.ProcessingInstruction;
import org.w3c.dom.Text;

import com.marklogic.client.MarkLogicInternalException;

public class DOMWriter {
	private XMLStreamWriter serializer;

	public DOMWriter() {
		super();
	}
	public DOMWriter(XMLStreamWriter serializer) {
		this();
		this.serializer = serializer;
	}

	public void serializeNodeList(NodeList list) throws XMLStreamException {
		for (int i=0; i < list.getLength(); i++) {
			serializeNode(list.item(i));
		}
	}
	public void serializeNode(Node node) throws XMLStreamException {
		switch (node.getNodeType()) {
		case Node.DOCUMENT_NODE:
			serializeDocument((Document) node);
			break;
		case Node.ELEMENT_NODE:
			serializeElement((Element) node);
			break;
		case Node.CDATA_SECTION_NODE:
			serializeCDATASection((CDATASection) node);
			break;
		case Node.TEXT_NODE:
			serializeText((Text) node);
			break;
		case Node.PROCESSING_INSTRUCTION_NODE:
			serializeProcessingInstruction((ProcessingInstruction) node);
			break;
		case Node.COMMENT_NODE:
			serializeComment((Comment) node);
			break;
		default:
			throw new MarkLogicInternalException(
					"Cannot process node type of: "+node.getClass().getName()
					);
		}
	}
	public void serializeDocument(Document document) throws XMLStreamException {
		String encoding = document.getInputEncoding();
		String version  = document.getXmlVersion();
		if (encoding != null) {
			serializer.writeStartDocument(encoding, version);
		} else {
			serializer.writeStartDocument(version);
		}
		if (document.hasChildNodes()) {
			serializeNodeList(document.getChildNodes());
		}
		serializer.writeEndDocument();
	}
	public void serializeElement(Element element) throws XMLStreamException {
		String namespaceURI = element.getNamespaceURI();
		String prefix       = (namespaceURI != null) ? element.getPrefix() : null;
		String localName    = (namespaceURI != null) ? element.getLocalName() : element.getTagName();
		if (element.hasChildNodes()) {
			if (prefix != null) {
				serializer.writeStartElement(prefix, localName, namespaceURI);
			} else if (namespaceURI != null) {
				serializer.writeStartElement(localName, namespaceURI);
			} else {
				serializer.writeStartElement(localName);
			}
			if (element.hasAttributes()) {
				serializeAttributes(element.getAttributes());
			}
			serializeNodeList(element.getChildNodes());
			serializer.writeEndElement();
		} else {
			if (prefix != null) {
				serializer.writeEmptyElement(prefix, localName, namespaceURI);
			} else if (namespaceURI != null) {
				serializer.writeEmptyElement(localName, namespaceURI);
			} else {
				serializer.writeEmptyElement(localName);
			}
			if (element.hasAttributes()) {
				serializeAttributes(element.getAttributes());
			}
		}
	}
	public void serializeAttributes(NamedNodeMap attributes) throws XMLStreamException {
		for (int i=0; i < attributes.getLength(); i++) {
			Attr attribute = (Attr) attributes.item(i);
			String namespaceURI = attribute.getNamespaceURI();
			String prefix       = (namespaceURI != null) ? attribute.getPrefix() : null;
			String localName    = (namespaceURI != null) ? attribute.getLocalName() : attribute.getName();
			String value        = attribute.getValue();
			if (prefix != null) {
				serializer.writeAttribute(prefix, localName, namespaceURI, value);
			} else if (namespaceURI != null) {
				serializer.writeAttribute(localName, namespaceURI, value);
			} else {
				serializer.writeAttribute(localName, value);
			}
		}
	}
	public void serializeText(Text text) throws XMLStreamException {
		serializer.writeCharacters(text.getData());
	}
	public void serializeCDATASection(CDATASection cdata) throws XMLStreamException {
		serializer.writeCData(cdata.getData());
	}
	public void serializeComment(Comment comment) throws XMLStreamException {
		serializer.writeComment(comment.getData());
	}
	public void serializeProcessingInstruction(ProcessingInstruction pi) throws XMLStreamException {
		String target = pi.getTarget();
		String data   = pi.getData();
		if (data != null)
			serializer.writeProcessingInstruction(target, data);
		else
			serializer.writeProcessingInstruction(target);
	}
}