/*
    *  soapUI, copyright (C) 2004-2010 eviware.com 
    *
    *  soapUI is free software; you can redistribute it and/or modify it under the 
    *  terms of version 2.1 of the GNU Lesser General Public License as published by 
    *  the Free Software Foundation.
    *
    *  soapUI is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without 
    *  even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
   *  See the GNU Lesser General Public License for more details at gnu.org.
   */
  package com.eviware.soapui.impl.wsdl.submit.transports.jms;
  
  import java.net.URL;
  import java.util.ArrayList;
  import java.util.Calendar;
  import java.util.Enumeration;
  import java.util.List;
  import java.util.Vector;
  
  import javax.jms.JMSException;
  import javax.jms.Message;
  import javax.jms.TextMessage;
  
  import com.eviware.soapui.SoapUI;
  import com.eviware.soapui.impl.rest.RestRequestInterface.RequestMethod;
  import com.eviware.soapui.impl.wsdl.WsdlRequest;
  import com.eviware.soapui.impl.wsdl.submit.transports.http.SSLInfo;
  import com.eviware.soapui.impl.wsdl.submit.transports.http.WsdlResponse;
  import com.eviware.soapui.model.iface.Attachment;
  import com.eviware.soapui.model.iface.Request;
  import com.eviware.soapui.support.types.StringToStringsMap;
  
  public class JMSResponse implements WsdlResponse
  {
  
  	private String payload;
  	private Message messageReceive;
  	private Message messageSend;
  	private Attachment[] attachments = new Attachment[0];
  	private Request request;
  	private long requestStartedTime;
  	private String endpoint;
  
  	public JMSResponse( String payload, Message messageSend, Message messageReceive, Request request,
  			long requestStartedTime )
  	{
  		this.payload = payload;
  		this.messageReceive = messageReceive;
  		this.messageSend = messageSend;
  		this.request = request;
  		this.requestStartedTime = requestStartedTime;
  		this.endpoint = request.getEndpoint();
  	}
  
  	public Attachment[] getAttachments()
  	{
  		return attachments;
  	}
  
  	public void setAttachments( Attachment[] attachments )
  	{
  		this.attachments = attachments;
  	}
  
  	public Attachment[] getAttachmentsForPart( String partName )
  	{
  		return attachments;
  	}
  
  	public String getContentAsString()
  	{
  		return payload;
  	}
  
  	public long getContentLength()
  	{
  		return payload.length();
  	}
  
  	public String getContentType()
  	{
  		if( messageReceive != null )
  			try
  			{
  				return messageReceive.getJMSType();
  			}
  			catch( JMSException e )
  			{
  				SoapUI.logError( e );
  			}
  		return null;
  	}
  
  	public String getProperty( String name )
  	{
  		if( messageReceive != null )
  			try
  			{
 				return messageReceive.getStringProperty( name );
 			}
 			catch( JMSException e )
 			{
 				SoapUI.logError( e );
 			}
 		return null;
 	}
 
 	public String[] getPropertyNames()
 	{
 		List<String> propertyNames = new ArrayList<String>();
 		Enumeration<?> temp;
 		try
 		{
 			if( messageReceive != null )
 			{
 				temp = messageReceive.getPropertyNames();
 				while( temp.hasMoreElements() )
 				{
 					propertyNames.add( ( String )temp.nextElement() );
 				}
 				return propertyNames.toArray( new String[propertyNames.size()] );
 			}
 			else
 			{
 				return new String[0];
 			}
 		}
 		catch( JMSException e )
 		{
 			SoapUI.logError( e );
 		}
 		return null;
 	}
 
 	public byte[] getRawRequestData()
 	{
 		if( messageSend != null )
 			return messageSend.toString().getBytes();
 		else
 			return "".getBytes();
 	}
 
 	public byte[] getRawResponseData()
 	{
 		if( messageReceive != null )
 			return messageReceive.toString().getBytes();
 		else
 			return "".getBytes();
 	}
 
 	public String getRequestContent()
 	{
 		if( messageSend != null )
 		{
 			try
 			{
 				if( messageSend instanceof TextMessage )
 				{
 					return ( ( TextMessage )messageSend ).getText();
 				}
 			}
 			catch( JMSException e )
 			{
 				SoapUI.logError( e );
 			}
 			return messageSend.toString();
 		}
 		return "";
 	}
 
 	public StringToStringsMap getRequestHeaders()
 	{
 		if( messageSend != null )
 		{
 			return JMSHeader.getMessageHeadersAndProperties( messageSend );
 		}
 		else
 			return new StringToStringsMap();
 
 	}
 
 	public StringToStringsMap getResponseHeaders()
 	{
 		if( messageReceive != null )
 		{
 			return JMSHeader.getMessageHeadersAndProperties( messageReceive );
 		}
 		else
 			return new StringToStringsMap();
 	}
 
 	public long getTimeTaken()
 	{
 		return Calendar.getInstance().getTimeInMillis() - requestStartedTime;
 	}
 
 	public long getTimestamp()
 	{
 		try
 		{
 			if( messageReceive != null )
 				return messageReceive.getJMSTimestamp();
 			else
 				return 0;
 		}
 		catch( JMSException e )
 		{
 			SoapUI.logError( e );
 		}
 		return 0;
 	}
 
 	public void setProperty( String name, String value )
 	{
 		try
 		{
 			messageReceive.setStringProperty( name, value );
 		}
 		catch( JMSException e )
 		{
 			SoapUI.logError( e );
 		}
 
 	}
 
 	public String getContentAsXml()
 	{
 		if( payload != null && !"".equals( payload ) )
 			return payload;
 		else
 			return "<xml/>";
 	}
 
 	public String getHttpVersion()
 	{
 		return null;
 	}
 
 	public RequestMethod getMethod()
 	{
 		return null;
 	}
 
 	public SSLInfo getSSLInfo()
 	{
 		return null;
 	}
 
 	public int getStatusCode()
 	{
 		return 0;
 	}
 
 	public URL getURL()
 	{
 		return null;
 	}
 
 	public void setResponseContent( String responseContent )
 	{
 		this.payload = responseContent;
 	}
 
 	public Vector<?> getWssResult()
 	{
 		return null;
 	}
 
 	public WsdlRequest getRequest()
 	{
 		return ( WsdlRequest )request;
 	}
 
 	public Message getMessageReceive()
 	{
 		return messageReceive;
 	}
 
 	public Message getMessageSend()
 	{
 		return messageSend;
 	}
 
 	public String getEndpoint()
 	{
 		return endpoint;
 	}
 
 }