/*
* 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.synapse.mediators.elementary;
import org.apache.axiom.om.OMAbstractFactory;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMNode;
import org.apache.axiom.om.OMText;
import org.apache.axiom.om.OMFactory;
import org.apache.axiom.om.impl.builder.StAXBuilder;
import org.apache.axiom.om.util.StAXUtils;
import org.apache.axiom.soap.SOAP12Constants;
import org.apache.axiom.soap.SOAPEnvelope;
import org.apache.axiom.soap.SOAPFactory;
import org.apache.axiom.soap.impl.builder.StAXSOAPModelBuilder;
import org.apache.synapse.MessageContext;
import org.apache.synapse.SynapseLog;
import org.apache.synapse.util.MessageHelper;
import org.apache.synapse.util.xpath.SynapseXPath;
import org.jaxen.JaxenException;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import java.io.ByteArrayInputStream;
import java.util.ArrayList;
import java.util.List;
/**
* The source of the XML node to be stored. The source can be a
* 1. Property
* 2. XPath Expression
* 3. SOAP Envelope
* 4. SOAP Body
* <p/>
* If clone is true a clone will be create and stored from the origincal content. Otherwise a
* reference will be stored.
* <p/>
* In case of property a OMElement is stored in a property and it will be fetched.
* <p/>
* In case of a XPath expression, it will be evaluated to get the OMElement
* <p/>
* In case of SOAPEnvelope entire SOAP envelope will be stored
* <p/>
* In case of Body, the first child of body will be stored
*/
public class Source {
private SynapseXPath xpath = null;
private String property = null;
private int sourceType = EnrichMediator.CUSTOM;
private boolean clone = true;
private OMNode inlineOMNode = null;
private String inlineKey = null;
public ArrayList<OMNode> evaluate(MessageContext synCtx, SynapseLog synLog)
throws JaxenException {
ArrayList<OMNode> sourceNodeList = new ArrayList<OMNode>();
if (sourceType == EnrichMediator.CUSTOM) {
assert xpath != null : "XPath should be non null in case of CUSTOM";
List selectedNodeList = xpath.selectNodes(synCtx);
if (selectedNodeList != null && selectedNodeList.size() != 0) {
for (Object o : selectedNodeList) {
if (o instanceof OMElement) {
if (clone) {
sourceNodeList.add(((OMElement) o).cloneOMElement());
} else {
sourceNodeList.add((OMElement) o);
}
} else if (o instanceof OMText) {
sourceNodeList.add((OMText) o);
} else if (o instanceof String) {
OMFactory fac = OMAbstractFactory.getOMFactory();
sourceNodeList.add(fac.createOMText(o.toString()));
}
}
} else {
synLog.error("Specified node by xpath cannot be found.");
}
} else if (sourceType == EnrichMediator.BODY) {
if (clone) {
sourceNodeList.add(synCtx.getEnvelope().getBody().getFirstElement().cloneOMElement());
} else {
sourceNodeList.add(synCtx.getEnvelope().getBody().getFirstElement());
}
} else if (sourceType == EnrichMediator.ENVELOPE) {
if (clone) {
sourceNodeList.add(MessageHelper.cloneSOAPEnvelope(synCtx.getEnvelope()));
} else {
sourceNodeList.add(synCtx.getEnvelope());
}
} else if (sourceType == EnrichMediator.PROPERTY) {
assert property != null : "property shouldn't be null when type is PROPERTY";
Object o = synCtx.getProperty(property);
if (o instanceof OMElement) {
if (clone) {
sourceNodeList.add(((OMElement) o).cloneOMElement());
}
} else if (o instanceof String) {
String soap = (String) o;
try {
XMLStreamReader xmlReader =
StAXUtils.createXMLStreamReader(new ByteArrayInputStream(soap.getBytes()));
StAXBuilder builder = new StAXSOAPModelBuilder(xmlReader);
OMElement elem = builder.getDocumentElement();
elem.build();
if (elem instanceof SOAPEnvelope) {
SOAPEnvelope soapEnvelope = (SOAPEnvelope) elem;
String soapNamespace = soapEnvelope.getNamespace().getNamespaceURI();
if (soapEnvelope.getHeader() == null) {
SOAPFactory soapFactory;
if (soapNamespace.equals(SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI)) {
soapFactory = OMAbstractFactory.getSOAP12Factory();
} else {
soapFactory = OMAbstractFactory.getSOAP11Factory();
}
soapFactory.createSOAPHeader(soapEnvelope);
}
sourceNodeList.add(soapEnvelope);
} else {
sourceNodeList.add(elem);
}
} catch (XMLStreamException e) {
synLog.error("Source Property cannot be parsed : " + e.getStackTrace().toString());
}
} else if (o instanceof ArrayList) {
ArrayList nodesList = (ArrayList) o;
for (Object node : nodesList) {
if (node instanceof OMElement) {
if (node instanceof SOAPEnvelope) {
SOAPEnvelope soapEnvelope = (SOAPEnvelope) node;
String soapNamespace = null;
if (soapEnvelope.getNamespace() != null) {
soapNamespace = soapEnvelope.getNamespace().getNamespaceURI();
}
if (soapEnvelope.getHeader() == null && soapNamespace != null) {
SOAPFactory soapFactory;
if (soapNamespace.equals(SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI)) {
soapFactory = OMAbstractFactory.getSOAP12Factory();
} else {
soapFactory = OMAbstractFactory.getSOAP11Factory();
}
soapFactory.createSOAPHeader(soapEnvelope);
}
sourceNodeList.add(soapEnvelope);
} else {
OMElement ele = (OMElement) node;
sourceNodeList.add(ele);
}
} else if (node instanceof OMText) {
sourceNodeList.add((OMText) node);
}
}
} else {
synLog.error("Invalid source property type.");
}
} else if (sourceType == EnrichMediator.INLINE) {
if (inlineOMNode instanceof OMElement) {
OMElement inlineOMElement = (OMElement) inlineOMNode;
if (inlineOMElement.getQName().getLocalPart().equals("Envelope")) {
SOAPEnvelope soapEnvelope = getSOAPEnvFromOM(inlineOMElement);
if (soapEnvelope != null) {
sourceNodeList.add(soapEnvelope);
} else {
synLog.error("Inline Source is not a valid SOAPEnvelope.");
}
} else {
sourceNodeList.add(inlineOMElement.cloneOMElement());
}
} else if (inlineOMNode instanceof OMText) {
sourceNodeList.add(inlineOMNode);
} else if (inlineKey != null) {
Object inlineObj = synCtx.getEntry(inlineKey);
if (inlineObj instanceof OMElement) {
if (((OMElement) inlineObj).getQName().getLocalPart().equals("Envelope")) {
SOAPEnvelope soapEnvelope = getSOAPEnvFromOM((OMElement) inlineObj);
if (soapEnvelope != null) {
sourceNodeList.add(soapEnvelope);
} else {
synLog.error("Specified Resource as Source is not a valid SOAPEnvelope.");
}
} else {
sourceNodeList.add((OMElement) inlineObj);
}
} else if (inlineObj instanceof OMText) {
sourceNodeList.add((OMText)inlineObj);
} else if (inlineObj instanceof String) {
sourceNodeList.add(
OMAbstractFactory.getOMFactory().createOMText(inlineObj.toString()));
} else {
synLog.error("Specified Resource as Source is not valid.");
}
} else {
synLog.error("Inline Source Content is not valid.");
}
}
return sourceNodeList;
}
private SOAPEnvelope getSOAPEnvFromOM(OMElement inlineElement) {
SOAPFactory soapFactory;
if (inlineElement.getQName().getNamespaceURI().equals(
SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI)) {
soapFactory = OMAbstractFactory.getSOAP12Factory();
} else {
soapFactory = OMAbstractFactory.getSOAP11Factory();
}
StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(inlineElement.getXMLStreamReader(),
soapFactory, inlineElement.getQName().getNamespaceURI());
return builder.getSOAPEnvelope();
}
public SynapseXPath getXpath() {
return xpath;
}
public void setXpath(SynapseXPath xpath) {
this.xpath = xpath;
}
public int getSourceType() {
return sourceType;
}
public void setSourceType(int sourceType) {
this.sourceType = sourceType;
}
public String getProperty() {
return property;
}
public void setProperty(String property) {
this.property = property;
}
public boolean isClone() {
return clone;
}
public void setClone(boolean clone) {
this.clone = clone;
}
public void setInlineOMNode(OMNode inlineOMNode) {
this.inlineOMNode = inlineOMNode;
}
public OMNode getInlineOMNode() {
return inlineOMNode;
}
public String getInlineKey() {
return inlineKey;
}
public void setInlineKey(String inlineKey) {
this.inlineKey = inlineKey;
}
}