1 /*
   2  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 /*
  26  * Copyright (C) 2004-2011
  27  *
  28  * Permission is hereby granted, free of charge, to any person obtaining a copy
  29  * of this software and associated documentation files (the "Software"), to deal
  30  * in the Software without restriction, including without limitation the rights
  31  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  32  * copies of the Software, and to permit persons to whom the Software is
  33  * furnished to do so, subject to the following conditions:
  34  *
  35  * The above copyright notice and this permission notice shall be included in
  36  * all copies or substantial portions of the Software.
  37  *
  38  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  39  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  40  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  41  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  42  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  43  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  44  * THE SOFTWARE.
  45  */
  46 package com.sun.xml.internal.rngom.digested;
  47 
  48 import org.w3c.dom.Attr;
  49 import org.w3c.dom.CDATASection;
  50 import org.w3c.dom.Comment;
  51 import org.w3c.dom.Document;
  52 import org.w3c.dom.DocumentFragment;
  53 import org.w3c.dom.Element;
  54 import org.w3c.dom.EntityReference;
  55 import org.w3c.dom.NamedNodeMap;
  56 import org.w3c.dom.Node;
  57 import org.w3c.dom.NodeList;
  58 import org.w3c.dom.ProcessingInstruction;
  59 import org.w3c.dom.Text;
  60 
  61 import javax.xml.stream.XMLStreamException;
  62 import javax.xml.stream.XMLStreamWriter;
  63 
  64 /**
  65  * Printer of DOM to XML using StAX {@link XMLStreamWriter}.
  66  *
  67  * @author <A href="mailto:demakov@ispras.ru">Alexey Demakov</A>
  68  */
  69 class DOMPrinter {
  70     protected XMLStreamWriter out;
  71 
  72     public DOMPrinter(XMLStreamWriter out) {
  73         this.out = out;
  74     }
  75 
  76     public void print(Node node) throws XMLStreamException {
  77         switch (node.getNodeType()) {
  78         case Node.DOCUMENT_NODE:
  79             visitDocument((Document) node);
  80             break;
  81         case Node.DOCUMENT_FRAGMENT_NODE:
  82             visitDocumentFragment((DocumentFragment) node);
  83             break;
  84         case Node.ELEMENT_NODE:
  85             visitElement((Element) node);
  86             break;
  87         case Node.TEXT_NODE:
  88             visitText((Text) node);
  89             break;
  90         case Node.CDATA_SECTION_NODE:
  91             visitCDATASection((CDATASection) node);
  92             break;
  93         case Node.PROCESSING_INSTRUCTION_NODE:
  94             visitProcessingInstruction((ProcessingInstruction) node);
  95             break;
  96         case Node.ENTITY_REFERENCE_NODE:
  97             visitReference((EntityReference) node);
  98             break;
  99         case Node.COMMENT_NODE:
 100             visitComment((Comment) node);
 101             break;
 102         case Node.DOCUMENT_TYPE_NODE:
 103             break;
 104         case Node.ATTRIBUTE_NODE:
 105         case Node.ENTITY_NODE:
 106         default:
 107             throw new XMLStreamException("Unexpected DOM Node Type "
 108                 + node.getNodeType()
 109             );
 110         }
 111     }
 112 
 113     protected void visitChildren(Node node)
 114         throws XMLStreamException {
 115         NodeList nodeList = node.getChildNodes();
 116         if (nodeList != null) {
 117             for (int i = 0; i < nodeList.getLength(); i++) {
 118                 print(nodeList.item(i));
 119             }
 120         }
 121     }
 122 
 123     protected void visitDocument(Document document)
 124         throws XMLStreamException {
 125         out.writeStartDocument();
 126         print(document.getDocumentElement());
 127         out.writeEndDocument();
 128     }
 129 
 130     protected void visitDocumentFragment(DocumentFragment documentFragment)
 131         throws XMLStreamException {
 132         visitChildren(documentFragment);
 133     }
 134 
 135     protected void visitElement(Element node)
 136         throws XMLStreamException {
 137         out.writeStartElement(node.getPrefix()
 138             , node.getLocalName()
 139             , node.getNamespaceURI()
 140         );
 141         NamedNodeMap attrs = node.getAttributes();
 142         for (int i = 0; i < attrs.getLength(); i++) {
 143             visitAttr((Attr) attrs.item(i));
 144         }
 145         visitChildren(node);
 146         out.writeEndElement();
 147     }
 148 
 149     protected void visitAttr(Attr node)
 150         throws XMLStreamException {
 151         String name = node.getLocalName();
 152         if (name.equals("xmlns")) {
 153             out.writeDefaultNamespace(node.getNamespaceURI());
 154         } else {
 155             String prefix = node.getPrefix();
 156             if (prefix != null && prefix.equals("xmlns")) {
 157                 out.writeNamespace(prefix, node.getNamespaceURI());
 158             } else if (prefix != null) {
 159                 out.writeAttribute(prefix
 160                     , node.getNamespaceURI()
 161                     , name
 162                     , node.getNodeValue()
 163                 );
 164             } else {
 165                 out.writeAttribute(node.getNamespaceURI()
 166                         , name
 167                         , node.getNodeValue());
 168             }
 169         }
 170     }
 171 
 172     protected void visitComment(Comment comment) throws XMLStreamException {
 173         out.writeComment(comment.getData());
 174     }
 175 
 176     protected void visitText(Text node) throws XMLStreamException {
 177         out.writeCharacters(node.getNodeValue());
 178     }
 179 
 180     protected void visitCDATASection(CDATASection cdata) throws XMLStreamException {
 181         out.writeCData(cdata.getNodeValue());
 182     }
 183 
 184     protected void visitProcessingInstruction(ProcessingInstruction processingInstruction)
 185         throws XMLStreamException {
 186         out.writeProcessingInstruction(processingInstruction.getNodeName()
 187             , processingInstruction.getData()
 188         );
 189     }
 190 
 191     protected void visitReference(EntityReference entityReference)
 192         throws XMLStreamException {
 193         visitChildren(entityReference);
 194     }
 195 }