modules/core/src/main/java/org/apache/savan/atom/AtomMessageReceiverDeligater.java - savan - Git at Google

 /*
  * 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.savan.atom;

 import java.util.Calendar;
 import java.util.Date;

 import javax.xml.namespace.QName;

 import org.apache.axiom.om.OMAbstractFactory;
 import org.apache.axiom.om.OMElement;
 import org.apache.axiom.om.OMException;
 import org.apache.axiom.om.OMNamespace;
 import org.apache.axiom.soap.SOAPEnvelope;
 import org.apache.axiom.soap.SOAPFactory;
 import org.apache.axiom.soap.SOAPProcessingException;
 import org.apache.axis2.AxisFault;
 import org.apache.axis2.context.MessageContext;
 import org.apache.axis2.databinding.utils.ConverterUtil;
 import org.apache.savan.SavanConstants;
 import org.apache.savan.SavanException;
 import org.apache.savan.SavanMessageContext;
 import org.apache.savan.messagereceiver.MessageReceiverDeligater;
 import org.apache.savan.storage.SubscriberStore;
 import org.apache.savan.subscribers.Subscriber;
 import org.apache.savan.util.CommonUtil;
 import org.apache.xmlbeans.XmlCursor;
 import org.apache.xmlbeans.XmlObject;
 import org.w3.x2005.x08.addressing.EndpointReferenceType;

 import com.wso2.eventing.atom.CreateFeedResponseDocument;
 import com.wso2.eventing.atom.RenewFeedResponseDocument;
 import com.wso2.eventing.atom.CreateFeedResponseDocument.CreateFeedResponse;


 public class AtomMessageReceiverDeligater extends MessageReceiverDeligater {
 	private SOAPEnvelope findOrCreateSoapEnvelope(SavanMessageContext subscriptionMessage, MessageContext outMessage) throws AxisFault{
 		MessageContext subscriptionMsgCtx = subscriptionMessage.getMessageContext();

 		SOAPEnvelope outMessageEnvelope = outMessage.getEnvelope();
 		SOAPFactory factory = null;

 		if (outMessageEnvelope!=null) {
 			factory = (SOAPFactory) outMessageEnvelope.getOMFactory();
 		} else {
 			factory = (SOAPFactory) subscriptionMsgCtx.getEnvelope().getOMFactory();
 			outMessageEnvelope = factory.getDefaultEnvelope();
 			outMessage.setEnvelope(outMessageEnvelope);
 		}
 		return outMessageEnvelope;
 	}


 	public void handleSubscriptionRequest(SavanMessageContext subscriptionMessage, MessageContext outMessage) throws SavanException {
 		try {
 			if (outMessage != null){
 				MessageContext subscriptionMsgCtx = subscriptionMessage.getMessageContext();

 				SOAPEnvelope outMessageEnvelope = findOrCreateSoapEnvelope(subscriptionMessage, outMessage);
 				//setting the action
 				outMessage.getOptions().setAction(AtomConstants.Actions.SubscribeResponse);

 				CreateFeedResponseDocument createFeedResponseDocument = CreateFeedResponseDocument.Factory.newInstance();
 				CreateFeedResponse createFeedResponse = createFeedResponseDocument.addNewCreateFeedResponse();
 				EndpointReferenceType savenEpr = createFeedResponse.addNewSubscriptionManager();
 				savenEpr.addNewAddress().setStringValue(subscriptionMsgCtx.getOptions().getTo().getAddress());

 				String id = (String) subscriptionMessage.getProperty(AtomConstants.TransferedProperties.SUBSCRIBER_UUID);
 				if (id != null){
 					XmlCursor c = savenEpr.addNewReferenceParameters().newCursor();
 					c.toNextToken();
 					addNameValuePair(c,new QName(AtomConstants.ATOM_NAMESPACE,AtomConstants.ElementNames.Identifier), id);
 				}else{
 					throw new SavanException ("Subscription UUID is not set");
 				}

 //				HttpServletRequest request = (HttpServletRequest) subscriptionMessage.getProperty(HTTPConstants.MC_HTTP_SERVLETREQUEST);
 //				request.getServerPort()


 				createFeedResponse.setFeedUrl((String)subscriptionMessage.getProperty(AtomConstants.Properties.feedUrl));
 				outMessageEnvelope.getBody().addChild(CommonUtil.toOM(createFeedResponseDocument));
 			}else{
 				throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");
 			}
 			//setting the message type
 			outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.SUBSCRIPTION_RESPONSE_MESSAGE));
 		} catch (SOAPProcessingException e) {
 			throw new SavanException(e);
 		} catch (AxisFault e) {
 			throw new SavanException(e);
 		} catch (OMException e) {
 			throw new SavanException(e);
 		}
 	}

 	public void handleRenewRequest(SavanMessageContext renewMessage, MessageContext outMessage) throws SavanException {

 		try {
 			if (outMessage==null)
 				throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");

 			SOAPEnvelope outMessageEnvelope = findOrCreateSoapEnvelope(renewMessage, outMessage);
 			RenewFeedResponseDocument renewFeedResponseDocument = RenewFeedResponseDocument.Factory.newInstance();


 			//setting the action
 			outMessage.getOptions().setAction(AtomConstants.Actions.RenewResponse);
 			String subscriberID = (String) renewMessage.getProperty(AtomConstants.TransferedProperties.SUBSCRIBER_UUID);
 			if (subscriberID==null) {
 				String message = "SubscriberID TransferedProperty is not set";
 				throw new SavanException (message);
 			}

 			SubscriberStore store = CommonUtil.getSubscriberStore(renewMessage.getMessageContext().getAxisService());
 			Subscriber subscriber = store.retrieve(subscriberID);
 			AtomSubscriber atomSubscriber = (AtomSubscriber) subscriber;
 			if (atomSubscriber==null) {
 				String message = "Cannot find the AbstractSubscriber with the given ID";
 				throw new SavanException (message);
 			}

 			Date expiration = atomSubscriber.getSubscriptionEndingTime();
 			Calendar calendar = Calendar.getInstance();
 			calendar.setTime(expiration);
 			renewFeedResponseDocument.addNewRenewFeedResponse().setExpires(calendar);
 			outMessageEnvelope.getBody().addChild(CommonUtil.toOM(renewFeedResponseDocument));
 			//setting the message type
 			outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.RENEW_RESPONSE_MESSAGE));
 		} catch (AxisFault e) {
 			throw new SavanException(e);
 		}
 	}

 	public void handleEndSubscriptionRequest(SavanMessageContext renewMessage, MessageContext outMessage) throws SavanException {
 		try {
 			if (outMessage==null)
 				throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");
 			//setting the action
 			outMessage.getOptions().setAction(AtomConstants.Actions.UnsubscribeResponse);
 			SOAPEnvelope outMessageEnvelope = findOrCreateSoapEnvelope(renewMessage, outMessage);
 			outMessageEnvelope.getBody().addChild(OMAbstractFactory.getOMFactory().createOMElement(new QName(AtomConstants.ATOM_MSG_NAMESPACE,AtomConstants.ElementNames.deleteFeedResponse)));
 			outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.UNSUBSCRIPTION_RESPONSE_MESSAGE));
 		} catch (AxisFault e) {
 			throw new SavanException(e);
 		} catch (OMException e) {
 			throw new SavanException(e);
 		}
 	}

 	public void handleGetStatusRequest(SavanMessageContext getStatusMessage, MessageContext outMessage) throws SavanException {

 		if (outMessage==null)
 			throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");

 		MessageContext subscriptionMsgCtx = getStatusMessage.getMessageContext();

 		String id = (String) getStatusMessage.getProperty(AtomConstants.TransferedProperties.SUBSCRIBER_UUID);
 		if (id==null)
 			throw new SavanException ("Cannot fulfil request. AbstractSubscriber ID not found");

 		//setting the action
 		outMessage.getOptions().setAction(AtomConstants.Actions.UnsubscribeResponse);

 		SOAPEnvelope outMessageEnvelope = outMessage.getEnvelope();
 		SOAPFactory factory = null;

 		if (outMessageEnvelope!=null) {
 			factory = (SOAPFactory) outMessageEnvelope.getOMFactory();
 		} else {
 			factory = (SOAPFactory) subscriptionMsgCtx.getEnvelope().getOMFactory();
 			outMessageEnvelope = factory.getDefaultEnvelope();

 			try {
 				outMessage.setEnvelope(outMessageEnvelope);
 			} catch (AxisFault e) {
 				throw new SavanException (e);
 			}
 		}

 		SubscriberStore store = CommonUtil.getSubscriberStore(getStatusMessage.getMessageContext().getAxisService());


 		if (store==null) {
 			throw new SavanException ("AbstractSubscriber Store was not found");
 		}

 		AtomSubscriber subscriber = (AtomSubscriber) store.retrieve(id);
 		if (subscriber==null) {
 			throw new SavanException ("AbstractSubscriber not found");
 		}

 		OMNamespace ens = factory.createOMNamespace(AtomConstants.ATOM_NAMESPACE,AtomConstants.ATOM_PREFIX);
 		OMElement getStatusResponseElement = factory.createOMElement(AtomConstants.ElementNames.GetStatusResponse,ens);

 		Date expires = subscriber.getSubscriptionEndingTime();
 		if (expires!=null) {
 			OMElement expiresElement = factory.createOMElement(AtomConstants.ElementNames.Expires,ens);
 			Calendar calendar = Calendar.getInstance();
 			calendar.setTime(expires);
 			String expirationString = ConverterUtil.convertToString(calendar);
 			expiresElement.setText(expirationString);
 			getStatusResponseElement.addChild(expiresElement);
 		}

 		outMessageEnvelope.getBody().addChild(getStatusResponseElement);

 		//setting the message type
 		outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.GET_STATUS_RESPONSE_MESSAGE));
 	}
 	public static void addNameValuePair(XmlCursor c,QName name,Object value){
         c.beginElement(name);

         if(value instanceof String){
             c.insertChars((String)value);
         }else {
             //Make sure this works, code is taken from
             //http://xmlbeans.apache.org/docs/2.0.0/guide/conHandlingAny.html
             XmlCursor cc = ((XmlObject)value).newCursor();
             cc.toFirstContentToken();
             cc.copyXml(c);
             cc.dispose();
         }
         c.toParent();
     }


 	public void doProtocolSpecificProcessing(SavanMessageContext inSavanMessage, MessageContext outMessage) throws SavanException {
 		int msgtype = ((Integer)inSavanMessage.getProperty(SavanConstants.MESSAGE_TYPE)).intValue();
 		switch(msgtype){
 			case SavanConstants.MessageTypes.SUBSCRIPTION_MESSAGE:
 				handleSubscriptionRequest(inSavanMessage, outMessage);
 				break;
 			case SavanConstants.MessageTypes.RENEW_MESSAGE:
 				handleRenewRequest(inSavanMessage, outMessage);
 				break;
 			case SavanConstants.MessageTypes.UNSUBSCRIPTION_MESSAGE:
 				handleEndSubscriptionRequest(inSavanMessage, outMessage);
 				break;
 			default:
 				throw new SavanException("Unknow Message type ["+msgtype+ "]");
 		}
 	}


 	public void doProtocolSpecificProcessing(SavanMessageContext inSavanMessage) throws SavanException {
 		// TODO Auto-generated method stub

 	}


 }
	/*
	* 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.savan.atom;

	import java.util.Calendar;
	import java.util.Date;

	import javax.xml.namespace.QName;

	import org.apache.axiom.om.OMAbstractFactory;
	import org.apache.axiom.om.OMElement;
	import org.apache.axiom.om.OMException;
	import org.apache.axiom.om.OMNamespace;
	import org.apache.axiom.soap.SOAPEnvelope;
	import org.apache.axiom.soap.SOAPFactory;
	import org.apache.axiom.soap.SOAPProcessingException;
	import org.apache.axis2.AxisFault;
	import org.apache.axis2.context.MessageContext;
	import org.apache.axis2.databinding.utils.ConverterUtil;
	import org.apache.savan.SavanConstants;
	import org.apache.savan.SavanException;
	import org.apache.savan.SavanMessageContext;
	import org.apache.savan.messagereceiver.MessageReceiverDeligater;
	import org.apache.savan.storage.SubscriberStore;
	import org.apache.savan.subscribers.Subscriber;
	import org.apache.savan.util.CommonUtil;
	import org.apache.xmlbeans.XmlCursor;
	import org.apache.xmlbeans.XmlObject;
	import org.w3.x2005.x08.addressing.EndpointReferenceType;

	import com.wso2.eventing.atom.CreateFeedResponseDocument;
	import com.wso2.eventing.atom.RenewFeedResponseDocument;
	import com.wso2.eventing.atom.CreateFeedResponseDocument.CreateFeedResponse;


	public class AtomMessageReceiverDeligater extends MessageReceiverDeligater {
	private SOAPEnvelope findOrCreateSoapEnvelope(SavanMessageContext subscriptionMessage, MessageContext outMessage) throws AxisFault{
	MessageContext subscriptionMsgCtx = subscriptionMessage.getMessageContext();

	SOAPEnvelope outMessageEnvelope = outMessage.getEnvelope();
	SOAPFactory factory = null;

	if (outMessageEnvelope!=null) {
	factory = (SOAPFactory) outMessageEnvelope.getOMFactory();
	} else {
	factory = (SOAPFactory) subscriptionMsgCtx.getEnvelope().getOMFactory();
	outMessageEnvelope = factory.getDefaultEnvelope();
	outMessage.setEnvelope(outMessageEnvelope);
	}
	return outMessageEnvelope;
	}



	public void handleSubscriptionRequest(SavanMessageContext subscriptionMessage, MessageContext outMessage) throws SavanException {
	try {
	if (outMessage != null){
	MessageContext subscriptionMsgCtx = subscriptionMessage.getMessageContext();

	SOAPEnvelope outMessageEnvelope = findOrCreateSoapEnvelope(subscriptionMessage, outMessage);
	//setting the action
	outMessage.getOptions().setAction(AtomConstants.Actions.SubscribeResponse);

	CreateFeedResponseDocument createFeedResponseDocument = CreateFeedResponseDocument.Factory.newInstance();
	CreateFeedResponse createFeedResponse = createFeedResponseDocument.addNewCreateFeedResponse();
	EndpointReferenceType savenEpr = createFeedResponse.addNewSubscriptionManager();
	savenEpr.addNewAddress().setStringValue(subscriptionMsgCtx.getOptions().getTo().getAddress());

	String id = (String) subscriptionMessage.getProperty(AtomConstants.TransferedProperties.SUBSCRIBER_UUID);
	if (id != null){
	XmlCursor c = savenEpr.addNewReferenceParameters().newCursor();
	c.toNextToken();
	addNameValuePair(c,new QName(AtomConstants.ATOM_NAMESPACE,AtomConstants.ElementNames.Identifier), id);
	}else{
	throw new SavanException ("Subscription UUID is not set");
	}

	// HttpServletRequest request = (HttpServletRequest) subscriptionMessage.getProperty(HTTPConstants.MC_HTTP_SERVLETREQUEST);
	// request.getServerPort()



	createFeedResponse.setFeedUrl((String)subscriptionMessage.getProperty(AtomConstants.Properties.feedUrl));
	outMessageEnvelope.getBody().addChild(CommonUtil.toOM(createFeedResponseDocument));
	}else{
	throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");
	}
	//setting the message type
	outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.SUBSCRIPTION_RESPONSE_MESSAGE));
	} catch (SOAPProcessingException e) {
	throw new SavanException(e);
	} catch (AxisFault e) {
	throw new SavanException(e);
	} catch (OMException e) {
	throw new SavanException(e);
	}
	}

	public void handleRenewRequest(SavanMessageContext renewMessage, MessageContext outMessage) throws SavanException {

	try {
	if (outMessage==null)
	throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");

	SOAPEnvelope outMessageEnvelope = findOrCreateSoapEnvelope(renewMessage, outMessage);
	RenewFeedResponseDocument renewFeedResponseDocument = RenewFeedResponseDocument.Factory.newInstance();


	//setting the action
	outMessage.getOptions().setAction(AtomConstants.Actions.RenewResponse);
	String subscriberID = (String) renewMessage.getProperty(AtomConstants.TransferedProperties.SUBSCRIBER_UUID);
	if (subscriberID==null) {
	String message = "SubscriberID TransferedProperty is not set";
	throw new SavanException (message);
	}

	SubscriberStore store = CommonUtil.getSubscriberStore(renewMessage.getMessageContext().getAxisService());
	Subscriber subscriber = store.retrieve(subscriberID);
	AtomSubscriber atomSubscriber = (AtomSubscriber) subscriber;
	if (atomSubscriber==null) {
	String message = "Cannot find the AbstractSubscriber with the given ID";
	throw new SavanException (message);
	}

	Date expiration = atomSubscriber.getSubscriptionEndingTime();
	Calendar calendar = Calendar.getInstance();
	calendar.setTime(expiration);
	renewFeedResponseDocument.addNewRenewFeedResponse().setExpires(calendar);
	outMessageEnvelope.getBody().addChild(CommonUtil.toOM(renewFeedResponseDocument));
	//setting the message type
	outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.RENEW_RESPONSE_MESSAGE));
	} catch (AxisFault e) {
	throw new SavanException(e);
	}
	}

	public void handleEndSubscriptionRequest(SavanMessageContext renewMessage, MessageContext outMessage) throws SavanException {
	try {
	if (outMessage==null)
	throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");
	//setting the action
	outMessage.getOptions().setAction(AtomConstants.Actions.UnsubscribeResponse);
	SOAPEnvelope outMessageEnvelope = findOrCreateSoapEnvelope(renewMessage, outMessage);
	outMessageEnvelope.getBody().addChild(OMAbstractFactory.getOMFactory().createOMElement(new QName(AtomConstants.ATOM_MSG_NAMESPACE,AtomConstants.ElementNames.deleteFeedResponse)));
	outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.UNSUBSCRIPTION_RESPONSE_MESSAGE));
	} catch (AxisFault e) {
	throw new SavanException(e);
	} catch (OMException e) {
	throw new SavanException(e);
	}
	}

	public void handleGetStatusRequest(SavanMessageContext getStatusMessage, MessageContext outMessage) throws SavanException {

	if (outMessage==null)
	throw new SavanException ("Eventing protocol need to sent the SubscriptionResponseMessage. But the outMessage is null");

	MessageContext subscriptionMsgCtx = getStatusMessage.getMessageContext();

	String id = (String) getStatusMessage.getProperty(AtomConstants.TransferedProperties.SUBSCRIBER_UUID);
	if (id==null)
	throw new SavanException ("Cannot fulfil request. AbstractSubscriber ID not found");

	//setting the action
	outMessage.getOptions().setAction(AtomConstants.Actions.UnsubscribeResponse);

	SOAPEnvelope outMessageEnvelope = outMessage.getEnvelope();
	SOAPFactory factory = null;

	if (outMessageEnvelope!=null) {
	factory = (SOAPFactory) outMessageEnvelope.getOMFactory();
	} else {
	factory = (SOAPFactory) subscriptionMsgCtx.getEnvelope().getOMFactory();
	outMessageEnvelope = factory.getDefaultEnvelope();

	try {
	outMessage.setEnvelope(outMessageEnvelope);
	} catch (AxisFault e) {
	throw new SavanException (e);
	}
	}

	SubscriberStore store = CommonUtil.getSubscriberStore(getStatusMessage.getMessageContext().getAxisService());


	if (store==null) {
	throw new SavanException ("AbstractSubscriber Store was not found");
	}

	AtomSubscriber subscriber = (AtomSubscriber) store.retrieve(id);
	if (subscriber==null) {
	throw new SavanException ("AbstractSubscriber not found");
	}

	OMNamespace ens = factory.createOMNamespace(AtomConstants.ATOM_NAMESPACE,AtomConstants.ATOM_PREFIX);
	OMElement getStatusResponseElement = factory.createOMElement(AtomConstants.ElementNames.GetStatusResponse,ens);

	Date expires = subscriber.getSubscriptionEndingTime();
	if (expires!=null) {
	OMElement expiresElement = factory.createOMElement(AtomConstants.ElementNames.Expires,ens);
	Calendar calendar = Calendar.getInstance();
	calendar.setTime(expires);
	String expirationString = ConverterUtil.convertToString(calendar);
	expiresElement.setText(expirationString);
	getStatusResponseElement.addChild(expiresElement);
	}

	outMessageEnvelope.getBody().addChild(getStatusResponseElement);

	//setting the message type
	outMessage.setProperty(SavanConstants.MESSAGE_TYPE,new Integer (SavanConstants.MessageTypes.GET_STATUS_RESPONSE_MESSAGE));
	}
	public static void addNameValuePair(XmlCursor c,QName name,Object value){
	c.beginElement(name);

	if(value instanceof String){
	c.insertChars((String)value);
	}else {
	//Make sure this works, code is taken from
	//http://xmlbeans.apache.org/docs/2.0.0/guide/conHandlingAny.html
	XmlCursor cc = ((XmlObject)value).newCursor();
	cc.toFirstContentToken();
	cc.copyXml(c);
	cc.dispose();
	}
	c.toParent();
	}





	public void doProtocolSpecificProcessing(SavanMessageContext inSavanMessage, MessageContext outMessage) throws SavanException {
	int msgtype = ((Integer)inSavanMessage.getProperty(SavanConstants.MESSAGE_TYPE)).intValue();
	switch(msgtype){
	case SavanConstants.MessageTypes.SUBSCRIPTION_MESSAGE:
	handleSubscriptionRequest(inSavanMessage, outMessage);
	break;
	case SavanConstants.MessageTypes.RENEW_MESSAGE:
	handleRenewRequest(inSavanMessage, outMessage);
	break;
	case SavanConstants.MessageTypes.UNSUBSCRIPTION_MESSAGE:
	handleEndSubscriptionRequest(inSavanMessage, outMessage);
	break;
	default:
	throw new SavanException("Unknow Message type ["+msgtype+ "]");
	}
	}



	public void doProtocolSpecificProcessing(SavanMessageContext inSavanMessage) throws SavanException {
	// TODO Auto-generated method stub

	}



	}