Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.apache.cxf.staxutils.transform.OutTransformWriter Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.cxf.staxutils.transform;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Stack;
import javax.xml.XMLConstants;
import javax.xml.namespace.NamespaceContext;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import org.apache.cxf.helpers.XMLUtils;
import org.apache.cxf.staxutils.DelegatingXMLStreamWriter;
public class OutTransformWriter extends DelegatingXMLStreamWriter {
private String defaultNamespace;
private QNamesMap elementsMap;
private QNamesMap attributesMap;
private Map appendMap = new HashMap(5);
private Map nsMap = new HashMap(5);
private List> writtenUris = new LinkedList>();
private Set dropElements;
private Stack> pushedAheadEvents = new Stack>();
private Stack elementsStack = new Stack();
private String replaceText;
private int currentDepth;
private int dropDepth;
private boolean attributesToElements;
private DelegatingNamespaceContext namespaceContext;
public OutTransformWriter(XMLStreamWriter writer,
Map outMap,
Map append,
List dropEls,
boolean attributesToElements,
String defaultNamespace) {
this(writer, outMap, append, dropEls, null, attributesToElements, defaultNamespace);
}
public OutTransformWriter(XMLStreamWriter writer,
Map outEMap,
Map append,
List dropEls,
Map outAMap,
boolean attributesToElements,
String defaultNamespace) {
super(writer);
elementsMap = new QNamesMap(outEMap == null ? 0 : outEMap.size());
attributesMap = new QNamesMap(outAMap == null ? 0 : outAMap.size());
TransformUtils.convertToQNamesMap(outEMap, elementsMap, nsMap);
TransformUtils.convertToQNamesMap(outAMap, attributesMap, null);
TransformUtils.convertToMapOfElementProperties(append, appendMap);
dropElements = XMLUtils.convertStringsToQNames(dropEls);
this.attributesToElements = attributesToElements;
namespaceContext = new DelegatingNamespaceContext(
writer.getNamespaceContext(), nsMap);
this.defaultNamespace = defaultNamespace;
}
@Override
public void writeNamespace(String prefix, String uri) throws XMLStreamException {
if (matchesDropped(true)) {
return;
}
String value = nsMap.get(uri);
if (value != null && value.length() == 0) {
return;
}
uri = value != null ? value : uri;
if (writtenUris.get(0).contains(uri)) {
return;
}
if (defaultNamespace != null && defaultNamespace.equals(uri)) {
super.writeDefaultNamespace(uri);
} else {
if (prefix.length() == 0) {
prefix = namespaceContext.findUniquePrefix(uri);
}
super.writeNamespace(prefix, uri);
}
writtenUris.get(0).add(uri);
}
@Override
public void writeStartElement(String prefix, String local, String uri) throws XMLStreamException {
currentDepth++;
if (matchesDropped(false)) {
return;
}
Set s;
if (writtenUris.isEmpty()) {
s = new HashSet();
} else {
s = new HashSet(writtenUris.get(0));
}
writtenUris.add(0, s);
final QName theName = new QName(uri, local, prefix);
final ElementProperty appendProp = appendMap.remove(theName);
final boolean replaceContent = appendProp != null && theName.equals(appendProp.getName());
final boolean dropped = dropElements.contains(theName);
QName expected = elementsMap.get(theName);
if (expected == null) {
expected = theName;
} else {
expected = new QName(expected.getNamespaceURI(), expected.getLocalPart(), prefix);
}
List pe = null;
if (appendProp != null && !replaceContent) {
if (!appendProp.isChild()) {
// ap-pre-*
QName appendQName = appendProp.getName();
String theprefix = namespaceContext.getPrefix(appendQName.getNamespaceURI());
boolean nsadded = false;
if (theprefix == null) {
nsadded = true;
theprefix = getPrefix(appendQName.getNamespaceURI());
if (theprefix == null
&& (appendQName.getNamespaceURI().equals(expected.getNamespaceURI())
&& expected.getPrefix().length() > 0)) {
theprefix = expected.getPrefix();
} else if (theprefix == null) {
theprefix = namespaceContext.findUniquePrefix(appendQName.getNamespaceURI());
}
if (theprefix == null) {
theprefix = "";
}
}
write(new QName(appendQName.getNamespaceURI(), appendQName.getLocalPart(), theprefix));
if (nsadded && theprefix.length() > 0) {
writeNamespace(theprefix, appendQName.getNamespaceURI());
}
if (appendProp.getText() == null) {
// ap-pre-wrap
currentDepth++;
pe = new ArrayList();
pe.add(TransformUtils.createEndElementEvent(expected));
pe.add(TransformUtils.createEndElementEvent(appendProp.getName()));
pushedAheadEvents.push(null);
elementsStack.push(appendQName);
} else {
// ap-pre-incl
super.writeCharacters(appendProp.getText());
super.writeEndElement();
}
}
} else if (replaceContent) {
//
replaceText = appendProp.getText();
} else if (dropped) {
// unwrap the current element (shallow drop)
elementsStack.push(theName);
return;
} else if (TransformUtils.isEmptyQName(expected)) {
// skip the current element (deep drop));
dropDepth = currentDepth - 1;
return;
}
write(expected);
pushedAheadEvents.push(pe);
elementsStack.push(expected);
if (appendProp != null && !replaceContent && appendProp.isChild()) {
// ap-post-*
QName appendQName = appendProp.getName();
String theprefix = getPrefix(appendQName.getNamespaceURI());
if (appendProp.getText() == null) {
// ap-post-wrap
write(new QName(appendQName.getNamespaceURI(), appendQName.getLocalPart(),
theprefix == null ? "" : theprefix));
if (getNamespaceContext().getPrefix(appendQName.getNamespaceURI()) == null) {
this.writeNamespace(theprefix, uri);
}
currentDepth++;
pe = new ArrayList();
pe.add(TransformUtils.createEndElementEvent(appendProp.getName()));
pe.add(TransformUtils.createEndElementEvent(expected));
pushedAheadEvents.push(pe);
elementsStack.push(appendQName);
} else {
// ap-post-incl
pushedAheadEvents.pop();
pe = new ArrayList();
pe.add(TransformUtils.createStartElementEvent(appendProp.getName()));
pe.add(TransformUtils.createCharactersEvent(appendProp.getText()));
pe.add(TransformUtils.createEndElementEvent(appendProp.getName()));
pe.add(TransformUtils.createEndElementEvent(expected));
pushedAheadEvents.push(pe);
}
}
}
@Override
public void writeStartElement(String uri, String local) throws XMLStreamException {
pushedAheadEvents.push(null);
elementsStack.push(new QName(uri, local));
super.writeStartElement(uri, local);
}
@Override
public void writeStartElement(String local) throws XMLStreamException {
writeStartElement("", local, "");
}
@Override
public void writeEndElement() throws XMLStreamException {
--currentDepth;
if (matchesDropped(false)) {
return;
} else if (dropDepth > 0) {
dropDepth = 0;
}
if (!writtenUris.isEmpty()) {
writtenUris.remove(0);
}
QName theName = elementsStack.pop();
final boolean dropped = dropElements.contains(theName);
if (!dropped) {
List pes = pushedAheadEvents.pop();
if (null != pes) {
for (ParsingEvent pe : pes) {
switch (pe.getEvent()) {
case XMLStreamConstants.START_ELEMENT:
write(pe.getName());
break;
case XMLStreamConstants.END_ELEMENT:
super.writeEndElement();
break;
case XMLStreamConstants.CHARACTERS:
super.writeCharacters(pe.getValue());
break;
default:
}
}
} else {
super.writeEndElement();
}
}
}
@Override
public void writeCharacters(String text) throws XMLStreamException {
if (matchesDropped(false)) {
return;
}
if (replaceText != null) {
text = replaceText;
replaceText = null;
}
super.writeCharacters(text);
}
private void write(QName qname) throws XMLStreamException {
boolean writeNs = false;
String prefix = "";
if (qname.getNamespaceURI().length() > 0) {
if (qname.getPrefix().length() == 0) {
prefix = getPrefix(qname.getNamespaceURI());
if (prefix == null) {
prefix = namespaceContext.findUniquePrefix(qname.getNamespaceURI());
writeNs = true;
}
} else {
prefix = qname.getPrefix();
namespaceContext.addPrefix(prefix, qname.getNamespaceURI());
}
}
if (defaultNamespace != null && defaultNamespace.equals(qname.getNamespaceURI())) {
prefix = "";
}
super.writeStartElement(prefix, qname.getLocalPart(), qname.getNamespaceURI());
if (writeNs) {
this.writeNamespace(prefix, qname.getNamespaceURI());
}
}
private boolean matchesDropped(boolean shallow) {
if ((dropDepth > 0 && dropDepth <= currentDepth)
|| (shallow && (elementsStack.size() > 0 && dropElements.contains(elementsStack.peek())))) {
return true;
}
return false;
}
@Override
public NamespaceContext getNamespaceContext() {
return namespaceContext;
}
@Override
public void writeAttribute(String prefix, String uri, String local, String value)
throws XMLStreamException {
QName expected = attributesMap.get(new QName(uri, local, prefix));
if (expected != null) {
if (TransformUtils.isEmptyQName(expected)) {
return;
}
uri = expected.getNamespaceURI();
local = expected.getLocalPart();
}
if (!attributesToElements) {
super.writeAttribute(prefix, uri, local, value);
} else {
writeAttributeAsElement(uri, local, value);
}
}
@Override
public void writeAttribute(String local, String value) throws XMLStreamException {
String uri = XMLConstants.NULL_NS_URI;
QName expected = attributesMap.get(new QName("", local));
if (expected != null) {
if (TransformUtils.isEmptyQName(expected)) {
return;
}
uri = expected.getNamespaceURI();
local = expected.getLocalPart();
}
if (!attributesToElements) {
if (uri.length() > 0) {
super.writeAttribute(uri, local, value);
} else {
super.writeAttribute(local, value);
}
} else {
writeAttributeAsElement(uri, local, value);
}
}
private void writeAttributeAsElement(String uri, String local, String value)
throws XMLStreamException {
this.writeStartElement(uri, local);
this.writeCharacters(value);
this.writeEndElement();
}
}