001    /*
002     * Licensed to the Apache Software Foundation (ASF) under one or more
003     * contributor license agreements.  See the NOTICE file distributed with
004     * this work for additional information regarding copyright ownership.
005     * The ASF licenses this file to You under the Apache License, Version 2.0
006     * (the "License"); you may not use this file except in compliance with
007     * the License.  You may obtain a copy of the License at
008     *
009     *      http://www.apache.org/licenses/LICENSE-2.0
010     *
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS,
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     * See the License for the specific language governing permissions and
015     * limitations under the License.
016     */
017    package org.apache.servicemix.soap.interceptors.xml;
018    
019    import java.io.InputStream;
020    
021    import javax.xml.stream.XMLStreamException;
022    import javax.xml.stream.XMLStreamReader;
023    
024    import org.apache.servicemix.soap.api.Fault;
025    import org.apache.servicemix.soap.api.Message;
026    import org.apache.servicemix.soap.core.AbstractInterceptor;
027    import org.apache.servicemix.soap.util.stax.DOMStreamReader;
028    import org.apache.servicemix.soap.util.stax.ExtendedXMLStreamReader;
029    import org.apache.servicemix.soap.util.stax.StaxUtil;
030    import org.w3c.dom.Document;
031    
032    /**
033     * Creates an XMLStreamReader from the InputStream on the Message.
034     * 
035     * @author <a href="mailto:gnodet [at] gmail.com">Guillaume Nodet</a>
036     */
037    public class StaxInInterceptor extends AbstractInterceptor {
038    
039        public static final String ENCODING = "Encoding";
040    
041        public StaxInInterceptor() {
042        }
043    
044        public void handleMessage(Message message) {
045            try {
046                XMLStreamReader reader;
047                Document doc = message.getContent(Document.class);
048                if (doc != null) {
049                    reader = new DOMStreamReader(doc.getDocumentElement());
050                } else {
051                    InputStream is = message.getContent(InputStream.class);
052                    if (is != null) {
053                        // TODO: where does encoding constant go?
054                        String encoding = (String) message.get(ENCODING);
055                        reader = StaxUtil.createReader(is, encoding);
056                        reader = new ExtendedXMLStreamReader(reader);
057                    } else {
058                        return;
059                    }
060                }
061                reader.nextTag();
062                message.setContent(XMLStreamReader.class, reader);
063            } catch (XMLStreamException e) {
064                throw new Fault(e);
065            }
066        }
067    
068    }