mono/mcs/class/System.ServiceModel.Web/System.ServiceModel.Description/WebScriptEnablingBehavior.cs

//
// WebScriptEnablingBehavior.cs
//
// Author:
//	Atsushi Enomoto  <[email protected]>
//
// Copyright (C) 2008 Novell, Inc (http://www.novell.com)
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
// 
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
using System;
using System.ServiceModel;
using System.ServiceModel.Channels;
using System.ServiceModel.Dispatcher;
using System.ServiceModel.Web;
using System.Web.Script.Services;

namespace System.ServiceModel.Description
{
	[ServiceContract (Namespace = "")]
	internal class InteropScriptService
	{
		Type type;
		string path;
		bool debug;

		public InteropScriptService (Type type, string path, bool debug)
		{
			this.type = type;
			this.path = path;
			this.debug = debug;
		}

		[WebGet (UriTemplate = "*")]
		[OperationContract]
		public string Get ()
		{
			return ProxyGenerator.GetClientProxyScript (type, path, debug);
		}
	}

	public sealed class WebScriptEnablingBehavior : WebHttpBehavior
	{
		public WebScriptEnablingBehavior ()
		{
			DefaultBodyStyle = WebMessageBodyStyle.WrappedRequest;
			DefaultOutgoingRequestFormat = WebMessageFormat.Json;
			DefaultOutgoingResponseFormat = WebMessageFormat.Json;
		}

		public override WebMessageBodyStyle DefaultBodyStyle { get; set; }

		public override WebMessageFormat DefaultOutgoingRequestFormat { get; set; }

		public override WebMessageFormat DefaultOutgoingResponseFormat { get; set; }

		[MonoTODO]
		protected override void AddClientErrorInspector (ServiceEndpoint endpoint, ClientRuntime clientRuntime)
		{
			base.AddClientErrorInspector (endpoint, clientRuntime);
		}

		[MonoTODO]
		protected override void AddServerErrorHandlers (ServiceEndpoint endpoint, EndpointDispatcher endpointDispatcher)
		{
			base.AddServerErrorHandlers (endpoint, endpointDispatcher);
		}

		[MonoTODO]
		public override void ApplyClientBehavior (ServiceEndpoint endpoint, ClientRuntime clientRuntime)
		{
			base.ApplyClientBehavior (endpoint, clientRuntime);
		}

		public override void ApplyDispatchBehavior (ServiceEndpoint endpoint, EndpointDispatcher endpointDispatcher)
		{
			base.ApplyDispatchBehavior (endpoint, endpointDispatcher);

			// doing similar to ServiceMetadataExtension
			BuildScriptDispatcher (endpoint, endpointDispatcher, "/js", false);
			BuildScriptDispatcher (endpoint, endpointDispatcher, "/jsdebug", true);
		}

		void BuildScriptDispatcher (ServiceEndpoint endpoint, EndpointDispatcher ed, string subPath, bool debug)
		{
			var instance = new InteropScriptService (endpoint.Contract.ContractType, endpoint.Address.Uri.ToString (), debug);

			var cdOrg = ed.ChannelDispatcher;
			var uri = new Uri (cdOrg.Listener.Uri, subPath);
			var cd = new ChannelDispatcher (endpoint.Binding.BuildChannelListener<IReplyChannel> (uri), String.Empty);
			cd.MessageVersion = MessageVersion.None;

			cd.Endpoints.Add (new EndpointDispatcher (new EndpointAddress (uri), "InteropScriptService", String.Empty));

			var dr = cd.Endpoints [0].DispatchRuntime;
			var dop = new DispatchOperation (dr, "Get", "*", "*");
			dop.DeserializeRequest = false;
			dop.SerializeReply = false;
			dop.Invoker = new DummyInvoker (instance);
			dr.UnhandledDispatchOperation = dop;
			dr.InstanceContextProvider = new SingletonInstanceContextProvider (new InstanceContext (cdOrg.Host, instance));

			var host = ed.ChannelDispatcher.Host;
			host.ChannelDispatchers.Add (cd);
		}

		class DummyInvoker : IOperationInvoker
		{
			InteropScriptService instance;

			public DummyInvoker (InteropScriptService instance)
			{
				this.instance = instance;
			}

			public object [] AllocateInputs ()
			{
				return new object [0];
			}

			public object Invoke (object instance, object [] inputs, out object [] outputs)
			{
				outputs = new object [0];
				var msg = Message.CreateMessage (MessageVersion.None, "*", (object) null);
				var hp = new HttpResponseMessageProperty ();
				hp.Headers ["Content-Type"] = "text/javascript";
				msg.Properties.Add (HttpResponseMessageProperty.Name, hp);
				msg.Properties.Add (WebMessageEncoder.ScriptPropertyName, this.instance.Get ());
				return msg;
			}

			public IAsyncResult InvokeBegin (object instance, object[] inputs, AsyncCallback callback, object state)
			{
				throw new NotSupportedException ();
			}

			public object InvokeEnd (object instance, out object [] outputs, IAsyncResult result)
			{
				throw new NotSupportedException ();
			}

			public bool IsSynchronous {
				get { return true; }
			}
		}

		[MonoTODO]
		protected override QueryStringConverter GetQueryStringConverter (OperationDescription operationDescription)
		{
			return base.GetQueryStringConverter (operationDescription);
		}

		[MonoTODO ("add non-XmlSerializer-ness check (but where?)")]
		public override void Validate (ServiceEndpoint endpoint)
		{
			if (endpoint == null)
				throw new ArgumentNullException ("endpoint");
			ValidateBinding (endpoint);

			foreach (var od in endpoint.Contract.Operations) {
				var wga = od.Behaviors.Find<WebGetAttribute> ();
				if (wga != null && wga.UriTemplate != null)
					throw new InvalidOperationException ("UriTemplate must not be used with WebScriptEnablingBehavior");
				var wia = od.Behaviors.Find<WebInvokeAttribute> ();
				if (wia != null) {
					if (wia.UriTemplate != null)
						throw new InvalidOperationException ("UriTemplate must not be used with WebScriptEnablingBehavior");
					switch (wia.Method.ToUpper ()) {
					case "GET":
					case "POST":
						break;
					default:
						throw new InvalidOperationException ("Only GET and POST HTTP methods are valid used for WebScriptEnablingBehavior");
					}
				}
			}
		}
	}
}