package org.jboss.axis.utils;
import org.w3c.dom.Attr;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.Writer;
public class DOM2Writer
{
private static String NS_URI_XMLNS = "http://www.w3.org/2000/xmlns/";
private static String NS_URI_XML = "http://www.w3.org/XML/1998/namespace";
public static String nodeToString(Node node, boolean omitXMLDecl)
{
StringWriter sw = new StringWriter();
serializeAsXML(node, sw, omitXMLDecl);
return sw.toString();
}
public static void serializeAsXML(Node node, Writer writer,
boolean omitXMLDecl)
{
serializeAsXML(node, writer, omitXMLDecl, false, 0);
}
public static void serializeAsXML(Node node, Writer writer,
boolean omitXMLDecl,
boolean pretty, int indent)
{
PrintWriter out = new PrintWriter(writer);
if (!omitXMLDecl)
out.println("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
NSStack namespaceStack = new NSStack();
print(node, namespaceStack, out, pretty, indent);
out.flush();
}
private static void print(Node node, NSStack namespaceStack,
PrintWriter out, boolean pretty,
int indent)
{
if (node == null)
{
return;
}
boolean hasChildren = false;
int type = node.getNodeType();
switch (type)
{
case Node.DOCUMENT_NODE:
{
NodeList children = node.getChildNodes();
if (children != null)
{
int numChildren = children.getLength();
for (int i = 0; i < numChildren; i++)
{
print(children.item(i), namespaceStack, out,
pretty, indent);
}
}
break;
}
case Node.ELEMENT_NODE:
{
namespaceStack.push();
if (pretty)
{
for (int i = 0; i < indent; i++)
out.print(' ');
}
out.print('<' + node.getNodeName());
NamedNodeMap attrs = node.getAttributes();
int len = (attrs != null) ? attrs.getLength() : 0;
for (int i = 0; i < len; i++)
{
Attr attr = (Attr)attrs.item(i);
String attrName = attr.getNodeName();
String attrValue = attr.getValue();
out.print(' ' + attrName + "=\"" + normalize(attrValue) + '\"');
if (attrName.startsWith("xmlns:"))
{
namespaceStack.add(attrValue, attrName.substring(6));
}
String attrPrefix = attr.getPrefix();
String attrNamespaceURI = attr.getNamespaceURI();
if (attrPrefix != null && attrNamespaceURI != null)
{
boolean prefixIsDeclared = false;
try
{
String namespaceURI = namespaceStack.getNamespaceURI(attrPrefix);
if (attrNamespaceURI.equals(namespaceURI))
{
prefixIsDeclared = true;
}
}
catch (IllegalArgumentException e)
{
}
if (!prefixIsDeclared)
{
printNamespaceDecl(attr, namespaceStack, out);
}
}
}
String elPrefix = node.getPrefix();
String elNamespaceURI = node.getNamespaceURI();
if (elPrefix != null &&
elNamespaceURI != null &&
elPrefix.length() > 0)
{
boolean prefixIsDeclared = false;
try
{
String namespaceURI = namespaceStack.getNamespaceURI(elPrefix);
if (elNamespaceURI.equals(namespaceURI))
{
prefixIsDeclared = true;
}
}
catch (IllegalArgumentException e)
{
}
if (!prefixIsDeclared)
{
printNamespaceDecl(node, namespaceStack, out);
}
}
NodeList children = node.getChildNodes();
if (children != null)
{
int numChildren = children.getLength();
hasChildren = (numChildren > 0);
if (hasChildren)
{
if (pretty)
out.println('>');
else
out.print('>');
}
for (int i = 0; i < numChildren; i++)
{
print(children.item(i), namespaceStack, out, pretty,
indent + 1);
}
}
else
{
hasChildren = false;
}
if (!hasChildren)
{
if (pretty)
out.println("/>");
else
out.print("/>");
}
namespaceStack.pop();
break;
}
case Node.ENTITY_REFERENCE_NODE:
{
out.print('&');
out.print(node.getNodeName());
out.print(';');
break;
}
case Node.CDATA_SECTION_NODE:
{
out.print("<![CDATA[");
out.print(node.getNodeValue());
out.print("]]>");
break;
}
case Node.TEXT_NODE:
{
out.print(normalize(node.getNodeValue()));
break;
}
case Node.COMMENT_NODE:
{
out.print("<!--");
out.print(node.getNodeValue());
out.print("-->");
if (pretty)
out.print(JavaUtils.LS);
break;
}
case Node.PROCESSING_INSTRUCTION_NODE:
{
out.print("<?");
out.print(node.getNodeName());
String data = node.getNodeValue();
if (data != null && data.length() > 0)
{
out.print(' ');
out.print(data);
}
out.println("?>");
if (pretty)
out.print(JavaUtils.LS);
break;
}
}
if (type == Node.ELEMENT_NODE && hasChildren == true)
{
if (pretty)
{
for (int i = 0; i < indent; i++)
out.print(' ');
}
out.print("</");
out.print(node.getNodeName());
out.print('>');
if (pretty)
out.print(JavaUtils.LS);
hasChildren = false;
}
}
private static void printNamespaceDecl(Node node,
NSStack namespaceStack,
PrintWriter out)
{
switch (node.getNodeType())
{
case Node.ATTRIBUTE_NODE:
{
printNamespaceDecl(((Attr)node).getOwnerElement(), node,
namespaceStack, out);
break;
}
case Node.ELEMENT_NODE:
{
printNamespaceDecl((Element)node, node, namespaceStack, out);
break;
}
}
}
private static void printNamespaceDecl(Element owner, Node node,
NSStack namespaceStack,
PrintWriter out)
{
String namespaceURI = node.getNamespaceURI();
String prefix = node.getPrefix();
if (!(namespaceURI.equals(NS_URI_XMLNS) && prefix.equals("xmlns")) &&
!(namespaceURI.equals(NS_URI_XML) && prefix.equals("xml")))
{
if (XMLUtils.getNamespace(prefix, owner) == null)
{
out.print(" xmlns:" + prefix + "=\"" + namespaceURI + '\"');
}
}
else
{
prefix = node.getLocalName();
namespaceURI = node.getNodeValue();
}
namespaceStack.add(namespaceURI, prefix);
}
public static String normalize(String s)
{
StringBuffer str = new StringBuffer();
int len = (s != null) ? s.length() : 0;
for (int i = 0; i < len; i++)
{
char ch = s.charAt(i);
switch (ch)
{
case '<':
{
str.append("<");
break;
}
case '>':
{
str.append(">");
break;
}
case '&':
{
str.append("&");
break;
}
case '"':
{
str.append(""");
break;
}
case JavaUtils.NL:
{
if (i > 0)
{
char lastChar = str.charAt(str.length() - 1);
if (lastChar != JavaUtils.CR)
{
str.append(JavaUtils.LS);
}
else
{
str.append(JavaUtils.NL);
}
}
else
{
str.append(JavaUtils.LS);
}
break;
}
default :
{
str.append(ch);
}
}
}
return (str.toString());
}
}