package yarfraw.utils;

 import java.io.File;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.io.OutputStreamWriter;

 import java.io.Writer;

   // DOM imports

 import org.w3c.dom.Document;

 import org.w3c.dom.DocumentType;

 import org.w3c.dom.NamedNodeMap;

 import org.w3c.dom.Node;

 import org.w3c.dom.NodeList;

 public class DOMSerializer {

       /** Indentation to use (default is no indentation) */

   private String indent = "";

   /** Line separator to use (default is for Windows) */

   private String lineSeparator = "\n";

   /** Encoding for output (default is UTF-Cool */

   private String encoding = "UTF8";

   public void setLineSeparator(String lineSeparator) {

       this.lineSeparator = lineSeparator;

   }

   public void setEncoding(String encoding) {

       this.encoding = encoding;

   }

   public void setIndent(int numSpaces) {

       StringBuffer buffer = new StringBuffer();

       for (int i = 0; i < numSpaces; i++)

           buffer.append(" ");

       this.indent = buffer.toString();

   }

   public void serialize(Document doc, OutputStream out) throws IOException {

       Writer writer = new OutputStreamWriter(out, encoding);

       serialize(doc, writer);

   }

   public void serialize(Document doc, File file) throws IOException {

       Writer writer = new FileWriter(file);

       serialize(doc, writer);

   }

   public void serialize(Document doc, Writer writer) throws IOException {

       // Start serialization recursion with no indenting

     serializeNode(doc, writer, "");

       writer.flush();

   }

   public void serializeNode(Node node, Writer writer, String indentLevel)

           throws IOException {

       // Determine action based on node type

     switch (node.getNodeType()) {

     case Node.DOCUMENT_NODE:

         Document doc = (Document) node;

         /**

          * DOM Level 2 code writer.write("<?xml version=\"1.0\"

          * encoding=\"UTF-8\"?>");

          */

         writer.write("<?xml version=\"");

         writer.write(doc.getXmlVersion());

         writer.write("\" encoding=\"UTF-8\" standalone=\"");

         if (doc.getXmlStandalone())

             writer.write("yes");

         else

             writer.write("no");

         writer.write("\"");

         writer.write("?>");

         writer.write(lineSeparator);

         // recurse on each top-level node

         NodeList nodes = node.getChildNodes();

         if (nodes != null)

             for (int i = 0; i < nodes.getLength(); i++)

                 serializeNode(nodes.item(i), writer, "");

         break;

     case Node.ELEMENT_NODE:

         String name = node.getNodeName();

         writer.write(indentLevel + "<" + name);

         NamedNodeMap attributes = node.getAttributes();

         for (int i = 0; i < attributes.getLength(); i++) {

             Node current = attributes.item(i);

             writer.write(" " + current.getNodeName() + "=\"");

             print(writer, current.getNodeValue());

             writer.write("\"");

         }

         writer.write(">");

         // recurse on each child

         NodeList children = node.getChildNodes();

         if (children != null) {

             if ((children.item(0) != null)

                     && (children.item(0).getNodeType() == Node.ELEMENT_NODE))

                 writer.write(lineSeparator);

             for (int i = 0; i < children.getLength(); i++)

                 serializeNode(children.item(i), writer, indentLevel

                         + indent);

             if ((children.item(0) != null)

                     && (children.item(children.getLength() - 1)

                             .getNodeType() == Node.ELEMENT_NODE))

                 writer.write(indentLevel);

         }

         writer.write("</" + name + ">");

         writer.write(lineSeparator);

         break;

     case Node.TEXT_NODE:

         print(writer, node.getNodeValue());

         break;

     case Node.CDATA_SECTION_NODE:

         writer.write("<![CDATA[");

         print(writer, node.getNodeValue());

         writer.write("]]>");

         break;

     case Node.COMMENT_NODE:

         writer.write(indentLevel + "<!-- " + node.getNodeValue() + " -->");

         writer.write(lineSeparator);

         break;

     case Node.PROCESSING_INSTRUCTION_NODE:

         writer.write("<?" + node.getNodeName() + " " + node.getNodeValue()

                 + "?>");

         writer.write(lineSeparator);

         break;

     case Node.ENTITY_REFERENCE_NODE:

         writer.write("&" + node.getNodeName( ) + ";");

         break;

     case Node.DOCUMENT_TYPE_NODE:

         DocumentType docType = (DocumentType) node;

         String publicId = docType.getPublicId();

         String systemId = docType.getSystemId();

         String internalSubset = docType.getInternalSubset();

         writer.write("<!DOCTYPE " + docType.getName());

         if (publicId != null)

             writer.write(" PUBLIC \"" + publicId + "\" ");

         else

             writer.write(" SYSTEM ");

         writer.write("\"" + systemId + "\"");

         if (internalSubset != null)

             writer.write(" [" + internalSubset + "]");

         writer.write(">");

           writer.write(lineSeparator);

           break;

       }

   }

   private void print(Writer writer, String s) throws IOException {

       if (s == null)

           return;

       for (int i = 0, len = s.length(); i < len; i++) {

           char c = s.charAt(i);

           switch (c) {

           case '<':

             writer.write("<");

             break;

         case '>':

             writer.write(">");

             break;

         case '&':

             writer.write("&");

             break;

         case '\r':

             writer.write("
");

                   break;

               default:

                   writer.write(c);

               }

           }

       }

 }

 /* From the book: Hardcore Java

 * Title: Java and XML

 * Third Edition: December 2006

 * ISBN 10: 0-596-10149-X

 * ISBN 13: 9780596101497

 * http://www.oreilly.com/

 * Java Code Example at http://www.JavaFAQ.nu

 */