/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2013 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* http://glassfish.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package com.sun.xml.ws.util.xml;
import java.io.IOException;
import javax.xml.bind.attachment.AttachmentMarshaller;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;
import javax.xml.XMLConstants;
import com.sun.xml.ws.streaming.MtomStreamWriter;
import org.jvnet.staxex.Base64Data;
import org.jvnet.staxex.XMLStreamReaderEx;
import org.jvnet.staxex.XMLStreamWriterEx;
/**
* Reads a sub-tree from {@link XMLStreamReader} and writes to {@link XMLStreamWriter}
* as-is.
*
* <p>
* This class can be sub-classed to implement a simple transformation logic.
*
* @author Kohsuke Kawaguchi
* @author Ryan Shoemaker
*/
public class XMLStreamReaderToXMLStreamWriter {
private static final int BUF_SIZE = 4096;
protected XMLStreamReader in;
protected XMLStreamWriter out;
private char[] buf;
boolean optimizeBase64Data = false;
AttachmentMarshaller mtomAttachmentMarshaller;
/**
* Reads one subtree and writes it out.
*
* <p>
* The {@link XMLStreamWriter} never receives a start/end document event.
* Those need to be written separately by the caller.
*/
public void bridge(XMLStreamReader in, XMLStreamWriter out) throws XMLStreamException {
assert in!=null && out!=null;
this.in = in;
this.out = out;
optimizeBase64Data = (in instanceof XMLStreamReaderEx);
if (out instanceof XMLStreamWriterEx && out instanceof MtomStreamWriter) {
mtomAttachmentMarshaller = ((MtomStreamWriter) out).getAttachmentMarshaller();
}
// remembers the nest level of elements to know when we are done.
int depth=0;
buf = new char[BUF_SIZE];
// if the parser is at the start tag, proceed to the first element
int event = in.getEventType();
if(event == XMLStreamConstants.START_DOCUMENT) {
// nextTag doesn't correctly handle DTDs
while( !in.isStartElement() ) {
event = in.next();
if (event == XMLStreamConstants.COMMENT)
handleComment();
}
}
if( event!=XMLStreamConstants.START_ELEMENT)
throw new IllegalStateException("The current event is not START_ELEMENT\n but " + event);
do {
// These are all of the events listed in the javadoc for
// XMLEvent.
// The spec only really describes 11 of them.
switch (event) {
case XMLStreamConstants.START_ELEMENT :
depth++;
handleStartElement();
break;
case XMLStreamConstants.END_ELEMENT :
handleEndElement();
depth--;
if(depth==0)
return;
break;
case XMLStreamConstants.CHARACTERS :
handleCharacters();
break;
case XMLStreamConstants.ENTITY_REFERENCE :
handleEntityReference();
break;
case XMLStreamConstants.PROCESSING_INSTRUCTION :
handlePI();
break;
case XMLStreamConstants.COMMENT :
handleComment();
break;
case XMLStreamConstants.DTD :
handleDTD();
break;
case XMLStreamConstants.CDATA :
handleCDATA();
break;
case XMLStreamConstants.SPACE :
handleSpace();
break;
case XMLStreamConstants.END_DOCUMENT:
throw new XMLStreamException("Malformed XML at depth="+depth+", Reached EOF. Event="+event);
default :
throw new XMLStreamException("Cannot process event: " + event);
}
event=in.next();
} while (depth!=0);
}
protected void handlePI() throws XMLStreamException {
out.writeProcessingInstruction(
in.getPITarget(),
in.getPIData());
}
protected void handleCharacters() throws XMLStreamException {
CharSequence c = null;
if (optimizeBase64Data) {
c = ((XMLStreamReaderEx)in).getPCDATA();
}
if ((c != null) && (c instanceof Base64Data)) {
if (mtomAttachmentMarshaller != null) {
Base64Data b64d = (Base64Data) c;
((XMLStreamWriterEx)out).writeBinary(b64d.getDataHandler());
} else {
try {
((Base64Data)c).writeTo(out);
} catch (IOException e) {
throw new XMLStreamException(e);
}
}
} else {
for (int start=0,read=buf.length; read == buf.length; start+=buf.length) {
read = in.getTextCharacters(start, buf, 0, buf.length);
out.writeCharacters(buf, 0, read);
}
}
}
protected void handleEndElement() throws XMLStreamException {
out.writeEndElement();
}
protected void handleStartElement() throws XMLStreamException {
String nsUri = in.getNamespaceURI();
if(nsUri==null)
out.writeStartElement(in.getLocalName());
else
out.writeStartElement(
fixNull(in.getPrefix()),
in.getLocalName(),
nsUri
);
// start namespace bindings
int nsCount = in.getNamespaceCount();
for (int i = 0; i < nsCount; i++) {
out.writeNamespace(
in.getNamespacePrefix(i),
fixNull(in.getNamespaceURI(i))); // zephyr doesn't like null, I don't know what is correct, so just fix null to "" for now
}
// write attributes
int attCount = in.getAttributeCount();
for (int i = 0; i < attCount; i++) {
handleAttribute(i);
}
}
/**
* Writes out the {@code i}-th attribute of the current element.
*
* <p>
* Used from {@link #handleStartElement()}.
*/
protected void handleAttribute(int i) throws XMLStreamException {
String nsUri = in.getAttributeNamespace(i);
String prefix = in.getAttributePrefix(i);
if (fixNull(nsUri).equals(XMLConstants.XMLNS_ATTRIBUTE_NS_URI)) {
//Its a namespace decl, ignore as it is already written.
return;
}
if(nsUri==null || prefix == null || prefix.equals("")) {
out.writeAttribute(
in.getAttributeLocalName(i),
in.getAttributeValue(i)
);
} else {
out.writeAttribute(
prefix,
nsUri,
in.getAttributeLocalName(i),
in.getAttributeValue(i)
);
}
}
protected void handleDTD() throws XMLStreamException {
out.writeDTD(in.getText());
}
protected void handleComment() throws XMLStreamException {
out.writeComment(in.getText());
}
protected void handleEntityReference() throws XMLStreamException {
out.writeEntityRef(in.getText());
}
protected void handleSpace() throws XMLStreamException {
handleCharacters();
}
protected void handleCDATA() throws XMLStreamException {
out.writeCData(in.getText());
}
private static String fixNull(String s) {
if(s==null) return "";
else return s;
}
}