using System;
using System.Web;
using System.Web.Handlers;
using System.Web.Hosting;
using System.Net;
using System.Web.Services.Protocols;
using System.Messaging;
using System.Text;
using System.IO;


namespace newtelligence.Services.Core
{
	/// <summary>
	/// 
	/// </summary>
	public class MessageQueueWebRequest : WebRequest
	{
		private MemoryStream _requestStream;
		private string _contentType;
		private string _connectionGroupName;
		private long _contentLength;
		private ICredentials _credentials;
		private WebHeaderCollection _headers;
		private Message _requestMessage;
		private int _timeout;
		private string _method;
		private Uri _requestUri;
		
		/// <summary>
		/// 
		/// </summary>
		public class MessageQueueWebRequestFactory : IWebRequestCreate
		{
			/// <summary>
			/// 
			/// </summary>
			/// <param name="uri"></param>
			/// <returns></returns>
			public WebRequest Create(Uri uri)
			{
				return new MessageQueueWebRequest(uri);
			}
		}

		/// <summary>
		/// This static constructor registers the message queue transport provider
		/// in the app domain when the class is being loaded.
		/// </summary>
		public static void RegisterMSMQProtocol()
		{
			WebRequest.RegisterPrefix("msmq", new MessageQueueWebRequestFactory());
		}

		/// <summary>
		/// The Message Queue URI takes the form of msmq://server/queueName
		/// </summary>
		/// <param name="uri"></param>
		public MessageQueueWebRequest(Uri uri)
		{
			_headers = new WebHeaderCollection();
			_requestStream = new MemoryStreamNoClose();
			_requestUri = uri;
			_requestMessage = new Message();
			_requestMessage.Formatter = new XmlMessageFormatter();
			_requestMessage.BodyStream = new MemoryStream();
		}

		/// <summary>
		/// 
		/// </summary>
		/// <returns></returns>
		public override Stream GetRequestStream()
		{
			return _requestStream;
		}

		/// <summary>
		/// 
		/// </summary>
		/// <param name="callback"></param>
		/// <param name="state"></param>
		/// <returns></returns>
		public override IAsyncResult BeginGetRequestStream(AsyncCallback callback, object state)
		{
			return new WebRequestAsyncResult( GetRequestStream() );
		}

		/// <summary>
		/// 
		/// </summary>
		/// <param name="asyncResult"></param>
		/// <returns></returns>
		public override Stream EndGetRequestStream(IAsyncResult asyncResult)
		{
			return asyncResult.AsyncState as Stream;
		}


		/// <summary>
		/// 
		/// </summary>
		/// <param name="callback"></param>
		/// <param name="state"></param>
		/// <returns></returns>
		public override IAsyncResult BeginGetResponse(AsyncCallback callback, object state)
		{
			return new WebRequestAsyncResult( GetResponse() );
		}
		
		/// <summary>
		/// 
		/// </summary>
		/// <param name="asyncResult"></param>
		/// <returns></returns>
		public override WebResponse EndGetResponse(IAsyncResult asyncResult)
		{
			return asyncResult.AsyncState as WebResponse;
		}


		/// <summary>
		/// 
		/// </summary>
		/// <returns></returns>
		public override WebResponse GetResponse()
		{
			using ( MessageQueue queue = OpenQueue() )
			{
				_requestMessage.BodyStream = new MemoryStream(_requestStream.ToArray());
				_requestMessage.Label = _headers["SOAPAction"];
				_requestMessage.Label = _requestMessage.Label.Trim('\"');
				_requestMessage.UseAuthentication = queue.Authenticate;
				_requestMessage.UseEncryption = queue.EncryptionRequired == EncryptionRequired.Body;;
				
				if ( queue.Transactional )
				{
					if ( System.EnterpriseServices.ContextUtil.IsInTransaction )
					{
						queue.Send( _requestMessage, MessageQueueTransactionType.Automatic );
					}
					else
					{
						queue.Send( _requestMessage, MessageQueueTransactionType.Single );
					}
				}
				else
				{
					queue.Send( _requestMessage, MessageQueueTransactionType.None );
				}
				_requestMessage.Dispose();
			}
			return new MessageQueueWebResponse();
		}

		/// <summary>
		/// 
		/// </summary>
		public override string ContentType
		{
			get
			{
				return _contentType;
			}
			set
			{
				_contentType = value;
			}
		}

		/// <summary>
		/// 
		/// </summary>
		public override void Abort()
		{
			base.Abort ();
		}

		/// <summary>
		/// 
		/// </summary>
		public override string ConnectionGroupName
		{
			get
			{
				return _connectionGroupName;
			}
			set
			{
				_connectionGroupName = value;
			}
		}

		/// <summary>
		/// 
		/// </summary>
		public override long ContentLength
		{
			get
			{
				return _contentLength;
			}
			set
			{
				_contentLength = value;
			}
		}

		/// <summary>
		/// 
		/// </summary>
		public override ICredentials Credentials
		{
			get
			{
				return _credentials;
			}
			set
			{
				_credentials = value;
			}
		}

		
        /// <summary>
        /// 
        /// </summary>
		public override WebHeaderCollection Headers
		{
			get
			{
				return _headers;
			}
			set
			{
				_headers = value;
			}
		}

		/// <summary>
		/// 
		/// </summary>
		public override string Method
		{
			get
			{
				return _method;
			}
			set
			{
				_method = value;
			}
		}

		/// <summary>
		/// 
		/// </summary>
		public override bool PreAuthenticate
		{
			get
			{
				return false;
			}
			set
			{
				// ignore that call
			}
		}

		/// <summary>
		/// 
		/// </summary>
		public override IWebProxy Proxy
		{
			get
			{
				return null;
			}
			set
			{
				// ignore
			}
		}

        /// <summary>
        /// 
        /// </summary>
		public override Uri RequestUri
		{
			get
			{
				return _requestUri;
			}
		}

		/// <summary>
		/// 
		/// </summary>
		public override int Timeout
		{
			get
			{
				return _timeout;
			}
			set
			{
				_timeout = value;
			}
		}

		/// <summary>
		/// 
		/// </summary>
		/// <returns></returns>
		private MessageQueue OpenQueue()
		{
			string queueName = _requestUri.Host + _requestUri.AbsolutePath.Replace("/","\\");
			MessageQueue queue = new MessageQueue(queueName);
			return queue;
		}

		
	}
}