/*
* 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.axis2.util;
import org.apache.neethi.Assertion;
import org.apache.neethi.Constants;
import org.apache.neethi.Policy;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLOutputFactory;
import javax.xml.stream.XMLStreamWriter;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
public class ExternalPolicySerializer {
private List assertions2Filter = new ArrayList();
public void addAssertionToFilter(QName name) {
assertions2Filter.add(name);
}
public void setAssertionsToFilter(List assertions2Filter) {
this.assertions2Filter = assertions2Filter;
}
public List getAssertionsToFilter() {
return assertions2Filter;
}
public void serialize(Policy policy, OutputStream os) {
try {
XMLStreamWriter writer = XMLOutputFactory.newInstance()
.createXMLStreamWriter(os);
policy = (Policy) policy.normalize(false);
String wspPrefix = writer.getPrefix(Constants.URI_POLICY_NS);
if (wspPrefix == null) {
wspPrefix = Constants.ATTR_WSP;
writer.setPrefix(wspPrefix, Constants.URI_POLICY_NS);
}
String wsuPrefix = writer.getPrefix(Constants.URI_WSU_NS);
if (wsuPrefix == null) {
wsuPrefix = Constants.ATTR_WSU;
writer.setPrefix(wsuPrefix, Constants.URI_WSU_NS);
}
writer.writeStartElement(wspPrefix, Constants.ELEM_POLICY,
Constants.URI_POLICY_NS);
QName key;
String prefix = null;
String namespaceURI = null;
String localName = null;
HashMap prefix2ns = new HashMap();
for (Iterator iterator = policy.getAttributes().keySet().iterator(); iterator
.hasNext();) {
key = (QName) iterator.next();
localName = key.getLocalPart();
namespaceURI = key.getNamespaceURI();
namespaceURI = (namespaceURI == null || namespaceURI.length() == 0) ? null
: namespaceURI;
if (namespaceURI != null) {
String writerPrefix = writer.getPrefix(namespaceURI);
writerPrefix = (writerPrefix == null || writerPrefix
.length() == 0) ? null : writerPrefix;
if (writerPrefix == null) {
prefix = key.getPrefix();
prefix = (prefix == null || prefix.length() == 0) ? null
: prefix;
} else {
prefix = writerPrefix;
}
if (prefix != null) {
writer.writeAttribute(prefix, namespaceURI, localName,
policy.getAttribute(key));
prefix2ns.put(prefix, key.getNamespaceURI());
} else {
writer.writeAttribute(namespaceURI, localName, policy
.getAttribute(key));
}
} else {
writer.writeAttribute(localName, policy.getAttribute(key));
}
}
// writes xmlns:wsp=".."
writer.writeNamespace(wspPrefix, Constants.URI_POLICY_NS);
String prefiX;
for (Iterator iterator = prefix2ns.keySet().iterator(); iterator
.hasNext();) {
prefiX = (String) iterator.next();
writer.writeNamespace(prefiX, (String) prefix2ns.get(prefiX));
}
writer.writeStartElement(Constants.ATTR_WSP,
Constants.ELEM_EXACTLYONE, Constants.URI_POLICY_NS);
// write <wsp:ExactlyOne>
List assertionList;
for (Iterator iterator = policy.getAlternatives(); iterator
.hasNext();) {
assertionList = (List) iterator.next();
// write <wsp:All>
writer.writeStartElement(Constants.ATTR_WSP, Constants.ELEM_ALL,
Constants.URI_POLICY_NS);
Assertion assertion;
for (Iterator assertions = assertionList.iterator(); assertions
.hasNext();) {
assertion = (Assertion) assertions.next();
if (assertions2Filter.contains(assertion.getName())) {
// since this is an assertion to filter, we will not serialize this
continue;
}
assertion.serialize(writer);
}
// write </wsp:All>
writer.writeEndElement();
}
// write </wsp:ExactlyOne>
writer.writeEndElement();
// write </wsp:Policy>
writer.writeEndElement();
writer.flush();
} catch (Exception ex) {
throw new RuntimeException(ex);
}
}
}