001 /**
002 *
003 * Copyright 2004 Protique Ltd
004 *
005 * Licensed under the Apache License, Version 2.0 (the "License");
006 * you may not use this file except in compliance with the License.
007 * 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 **/
018 package org.codehaus.activesoap.handler.stax;
019
020 import javax.xml.namespace.QName;
021 import javax.xml.stream.XMLStreamException;
022 import javax.xml.stream.XMLStreamWriter;
023 import javax.xml.stream.XMLStreamReader;
024 import javax.xml.stream.XMLStreamConstants;
025 import java.util.Iterator;
026 import java.util.List;
027 import java.util.Map;
028
029 /**
030 * @version $Revision: 1.4 $
031 */
032 public class StaxHelper {
033 /**
034 * Writes the given attributes to the stream
035 */
036 public static void writeAttributes(Map map, XMLStreamWriter out) throws XMLStreamException {
037 for (Iterator iter = map.entrySet().iterator(); iter.hasNext();) {
038 Map.Entry entry = (Map.Entry) iter.next();
039 Object key = entry.getKey();
040 String value = (String) entry.getValue();
041 if (key instanceof String) {
042 out.writeAttribute((String) key, value);
043 }
044 else if (key instanceof QName) {
045 QName name = (QName) key;
046 out.writeAttribute(name.getPrefix(), name.getNamespaceURI(), name.getLocalPart(), value);
047 }
048 }
049 }
050
051 /**
052 * Writes the given element to the StAX stream if it is not null
053 */
054 public static void writeElement(QName qname, ElementContent element, AnyElementMarshaler marshaler, XMLStreamWriter out) throws XMLStreamException {
055 writeElement(qname, element, marshaler, out, false);
056
057 }
058
059 public static void writeElement(QName qname, ElementContent element, AnyElementMarshaler marshaler, XMLStreamWriter out, boolean repairingMode) throws XMLStreamException {
060 if (element != null) {
061 writeStartElement(qname, out, repairingMode);
062 element.writeContent(marshaler, out);
063 out.writeEndElement();
064 }
065 }
066
067 public static void writeElementNoNamespace(QName qname, ElementContent element, AnyElementMarshaler marshaler, XMLStreamWriter out) throws XMLStreamException {
068 if (element != null) {
069 writeStartElementNoNamespace(qname, out);
070 element.writeContent(marshaler, out);
071 out.writeEndElement();
072 }
073 }
074
075 public static void writeElementNoNamespace(QName qname, Object object, AnyElementMarshaler marshaler, XMLStreamWriter out) throws XMLStreamException {
076 if (object != null) {
077 writeStartElementNoNamespace(qname, out);
078 marshaler.writeElement(object, out);
079 out.writeEndElement();
080 }
081 }
082
083 public static void writeElement(QName qname, Object object, AnyElementMarshaler marshaler, XMLStreamWriter out) throws XMLStreamException {
084 writeElement(qname, object, marshaler, out, false);
085
086 }
087 public static void writeElement(QName qname, Object object, AnyElementMarshaler marshaler, XMLStreamWriter out, boolean repairingMode) throws XMLStreamException {
088 if (object != null) {
089 writeStartElement(qname, out, repairingMode);
090 marshaler.writeElement(object, out);
091 out.writeEndElement();
092 }
093 }
094
095 public static void writeStartElementNoNamespace(QName qname, XMLStreamWriter out) throws XMLStreamException {
096 String prefix = qname.getPrefix();
097 String uri = qname.getNamespaceURI();
098 out.writeStartElement(prefix, qname.getLocalPart(), uri);
099 }
100
101 public static void writeStartElement(QName qname, XMLStreamWriter out, boolean repairingMode) throws XMLStreamException {
102 String prefix = qname.getPrefix();
103 String uri = qname.getNamespaceURI();
104
105 // before you ask - yes it really is this complicated to output QNames to StAX
106 // handling both repair namespace modes :)
107
108 boolean hasPrefix = prefix != null && prefix.length() > 0;
109 boolean hasURI = uri != null && uri.length() > 0;
110
111 if (hasPrefix) {
112 out.setPrefix(prefix, uri);
113 }
114 else if (hasURI) {
115 out.setDefaultNamespace(uri);
116 }
117 out.writeStartElement(prefix, qname.getLocalPart(), uri);
118
119 if (hasURI && !repairingMode) {
120 if (hasPrefix) {
121 out.writeNamespace(prefix, uri);
122 }
123 else {
124 out.writeDefaultNamespace(uri);
125 }
126 }
127 }
128
129
130 /**
131 * Writes the list of any objects to the stream. The list can be null to avoid firing lazy-construction logic
132 */
133 public static void writeAnyContent(List list, AnyElementMarshaler marshaller, XMLStreamWriter out) throws XMLStreamException {
134 if (list != null) {
135 for (Iterator iter = list.iterator(); iter.hasNext();) {
136 Object value = iter.next();
137 marshaller.writeElement(value, out);
138 }
139 }
140 }
141
142 /**
143 * Reads the content of the current element as a String, ignoring any embedded elements
144 */
145 public static String readTextContent(XMLStreamReader in) throws XMLStreamException {
146 StringBuffer buffer = new StringBuffer();
147
148 boolean complete = false;
149 int elements = 0;
150 int code = in.getEventType();
151 while (!complete) {
152 switch (code) {
153 case XMLStreamConstants.CHARACTERS:
154 if (elements == 1) {
155 buffer.append(in.getText());
156 }
157 break;
158
159 case XMLStreamConstants.START_ELEMENT:
160 elements++;
161 break;
162
163 case XMLStreamConstants.END_ELEMENT:
164 if (--elements <= 0) {
165 complete = true;
166 }
167 break;
168
169 case XMLStreamConstants.END_DOCUMENT:
170 complete = true;
171 break;
172 }
173 if (complete) {
174 break;
175 }
176 if (!in.hasNext()) {
177 complete = true;
178 }
179 else {
180 code = in.next();
181 }
182 }
183 return buffer.toString();
184 }
185
186 }