mono/mcs/class/System.XML/System.Xml.Query/XQueryCliFunction.cs

//
// XQueryCliFunction.cs
//
// Author:
//	Atsushi Enomoto <[email protected]>
//
//
// Copyright (C) 2004 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.
//

//
// Runtime type method wrapper for XPath2 function.
//
#if NET_2_0
using System;
using System.Collections;
using System.Reflection;
using System.Security;
using System.Security.Policy;
using System.Xml;
using System.Xml.Schema;
using System.Xml.Query;
using System.Xml.XPath;
using System.Xml.Xsl;

namespace Mono.Xml.XPath2
{

	// Ideas:
	// declare function namespace cli = "http://mono-project.com/xquery/function/cli"
	// declare variable v = cli:invoke (cli:new (Microsoft.CSharp:CSharpCodeProvider), CreateCompiler);
	// declare variable v2 = System.Math:Abs (0.5);
	//

	public class XQueryCliFunction : XQueryFunction
	{
		internal static XQueryCliFunction CreateFromMethodInfo (MethodInfo mi)
		{
			return CreateFromMethodInfo (null, mi);
		}

		internal static XQueryCliFunction CreateFromMethodInfo (XmlQualifiedName name, MethodInfo mi)
		{
			return CreateFromMethodInfo (name, new MethodInfo [] {mi});
		}

		internal static XQueryCliFunction CreateFromMethodInfo (MethodInfo [] methods)
		{
			return CreateFromMethodInfo (null, methods);
		}

		internal static XQueryCliFunction CreateFromMethodInfo (XmlQualifiedName name, MethodInfo [] methodList)
		{
			if (methodList == null || methodList.Length == 0)
				throw new ArgumentException (String.Format ("Argument methods is missing or zero-length array. Name is {0}", name));

			Type cliReturnType = null;
			ArrayList arguments = new ArrayList ();

			if (name == null || name == XmlQualifiedName.Empty)
				name = new XmlQualifiedName (methodList [0].Name, methodList [0].DeclaringType.FullName);

			int maxArgs = 0;
			int minArgs = -1;
			Hashtable methods = new Hashtable ();

			foreach (MethodInfo mi in methodList) {
				if (cliReturnType == null)
					cliReturnType = mi.ReturnType;
				else if (mi.ReturnType != cliReturnType)
					throw new ArgumentException (String.Format ("Every XQuery functions which share the same name must have the same return type. Method name is {0}.", mi.Name));
				ParameterInfo [] prms = mi.GetParameters ();

				int args = prms.Length;

				// Whether it takes "current context" or not.
				Type t = args > 0 ? prms [0].ParameterType : null;
				bool ctxSeq = mi.GetCustomAttributes (typeof (XQueryFunctionContextAttribute), false).Length > 0;
				bool hasContextArg = ctxSeq || t == typeof (XQueryContext);
				if (ctxSeq || hasContextArg)
					args--;
				if (methods [args] != null)
					throw new ArgumentException (String.Format ("XQuery does not allow functions that accepts such methods that have the same number of parameters in different types. Method name is {0}", mi.Name));
				methods.Add ((int) args, mi);
				if (args < minArgs || minArgs < 0)
					minArgs = args;
				if (args > maxArgs)
					maxArgs = args;
			}

			MethodInfo m = (MethodInfo) methods [(int) maxArgs];
			if (m == null)
				throw new SystemException ("Should not happen: maxArgs is " + maxArgs);
			ParameterInfo [] pl = m.GetParameters ();
			for (int i = 0; i < pl.Length; i++) {
				Type t = pl [i].ParameterType;
				if (t != typeof (XQueryContext))
					arguments.Add (
						new XQueryFunctionArgument (new XmlQualifiedName (pl [i].Name), SequenceType.Create (pl [i].ParameterType)));
			}

			return new XQueryCliFunction (name,
				arguments.ToArray (typeof (XQueryFunctionArgument)) as XQueryFunctionArgument [],
				SequenceType.Create (cliReturnType),
				methods,
				minArgs,
				maxArgs);
		}

		private XQueryCliFunction (XmlQualifiedName name, 
			XQueryFunctionArgument [] args,
			SequenceType returnType,
			Hashtable methods,
			int minArgs,
			int maxArgs)
			: base (name, args, returnType)
		{
			this.methods = methods;
			this.maxArgs = maxArgs;
			this.minArgs = minArgs;
		}

		// instance members

		// [int argsize] -> MethodInfo (according to the spec 1.1,
		// there should be no overloads that accepts the same parameter
		// count in different types).
		Hashtable methods = new Hashtable ();
		int maxArgs = 0;
		int minArgs = -1;
		SequenceType returnType;

		public override int MinArgs {
			get { return minArgs; }
		}

		public override int MaxArgs {
			get { return maxArgs; }
		}

		public override object Invoke (XPathSequence current, object [] args)
		{
			MethodInfo mi = methods [args.Length] as MethodInfo;
			if (mi == null)
				throw new ArgumentException ("The number of custom function parameter does not match with the registered method's signature.");
			ParameterInfo [] prms = mi.GetParameters ();

			// Use Evidence and PermissionSet.Demand() here
			// before invoking external function.
			Evidence e = current.Context.StaticContext.Evidence;
			if (e != null)
				SecurityManager.ResolvePolicy (e).Demand ();

			Type t = prms.Length > 0 ? prms [0].ParameterType : null;
			bool ctxSeq = mi.GetCustomAttributes (
				typeof (XQueryFunctionContextAttribute),
				false).Length > 0;
			if (t == typeof (XQueryContext)) {
				ArrayList pl = new ArrayList (args);
				pl.Insert (0, current.Context);
				args = pl.ToArray ();
			}
			else if (ctxSeq) {
				ArrayList pl = new ArrayList (args);
				pl.Insert (0, current);
				args = pl.ToArray ();
			}

			if (args.Length != prms.Length)
				throw new XmlQueryException (String.Format ("Argument numbers were different for function {0}. Signature requires {1} while actual call was {2}.", mi.Name, prms.Length, args.Length));

			// If native parameter type is XPathSequence and the actual values are not, adjust them
			for (int i = 0; i < args.Length; i++) {
				if (prms [i].ParameterType == typeof (XPathSequence) && !(args [i] is XPathSequence)) {
					XPathItem item = args [i] as XPathItem;
					if (item == null)
						item = args [i] == null ? null : new XPathAtomicValue (args [i], XmlSchemaType.GetBuiltInType (XPathAtomicValue.XmlTypeCodeFromRuntimeType (prms [i].ParameterType, true)));
					if (item == null)
						args [i] = new XPathEmptySequence (current.Context);
					else
						args [i] = new SingleItemIterator (item, current.Context);
				}
			}

			return mi.Invoke (null, args);
		}
	}

}
#endif