src/org/apache/sandesha2/wsrm/SequenceFault.java

/*
 * Copyright  1999-2004 The Apache Software Foundation.
 *
 *  Licensed 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.sandesha2.wsrm;

import javax.xml.namespace.QName;

import org.apache.axis2.om.OMElement;
import org.apache.axis2.om.OMException;
import org.apache.axis2.om.OMNamespace;
import org.apache.axis2.soap.SOAPBody;
import org.apache.axis2.soap.SOAPFactory;
import org.apache.sandesha2.Sandesha2Constants;

/**
 * Adds the SequenceFault header block.
 * 
 * @author Chamikara Jayalath <chamikaramj@gmail.com>
 * @author Sanka Samaranayaka <ssanka@gmail.com>
 * @author Saminda Abeyruwan  <saminda@opensource.lk>
 */

public class SequenceFault implements IOMRMElement {
	private OMElement sequenceFaultElement;

	private FaultCode faultCode;
	
	SOAPFactory factory;

	OMNamespace rmNamespace = null;

	public SequenceFault(SOAPFactory factory) {
		this.factory = factory;
		rmNamespace = factory.createOMNamespace(
				Sandesha2Constants.WSRM.NS_URI_RM, Sandesha2Constants.WSRM.NS_PREFIX_RM);
		sequenceFaultElement = factory.createOMElement(
				Sandesha2Constants.WSRM.SEQUENCE_FAULT, rmNamespace);
	}

	public OMElement getOMElement() throws OMException {
		return sequenceFaultElement;
	}

	public Object fromOMElement(OMElement body) throws OMException {

		if (body == null || !(body instanceof SOAPBody))
			throw new OMException(
					"Cant get Sequence Fault part from a non-header element");

		OMElement sequenceFaultPart = body.getFirstChildWithName(new QName(
				Sandesha2Constants.WSRM.NS_URI_RM, Sandesha2Constants.WSRM.SEQUENCE_FAULT));

		if (sequenceFaultPart == null)
			throw new OMException(
					"The passed element does not contain a Sequence Fault element");

		OMElement faultCodePart = sequenceFaultPart
				.getFirstChildWithName(new QName(Sandesha2Constants.WSRM.NS_URI_RM,
						Sandesha2Constants.WSRM.FAULT_CODE));

		if (faultCodePart != null) {
			faultCode = new FaultCode(factory);
			faultCode.fromOMElement(sequenceFaultPart);
		}

		sequenceFaultElement = factory.createOMElement(
				Sandesha2Constants.WSRM.SEQUENCE_FAULT, rmNamespace);

		return this;

	}

	public OMElement toOMElement(OMElement body) throws OMException {

		if (body == null || !(body instanceof SOAPBody))
			throw new OMException(
					"Cant get Sequence Fault part from a non-header element");

		if (sequenceFaultElement == null)
			throw new OMException(
					"Cant add the sequnce fault since the internal element is null");

		if (faultCode != null)
			faultCode.toOMElement(sequenceFaultElement);

		body.addChild(sequenceFaultElement);

		sequenceFaultElement = factory.createOMElement(
				Sandesha2Constants.WSRM.SEQUENCE_FAULT, rmNamespace);

		return body;
	}

	public void setFaultCode(FaultCode faultCode) {
		this.faultCode = faultCode;
	}

	public FaultCode getFaultCode() {
		return faultCode;
	}

	public void setSequenceFaultElement(OMElement sequenceFault) {
		sequenceFaultElement = sequenceFault;
	}

	public OMElement getSequenceFaultElement() {
		return sequenceFaultElement;
	}

}