RunSharp - Reflection.Emit Has Never Been Easier

/*
 *  RunSharp - Runtime Code Generation Library
 *  http://www.sourceforge.net/projects/runsharp
 *  Copyright (C) 2007 Stefan Simek
 *
 *  This file is part of RunSharp.
 * 
 *  RunSharp is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  RunSharp 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with RunSharp.  If not, see <http://www.gnu.org/licenses/>.
 */

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Text;
using System.IO;
using System.Reflection;
using System.Reflection.Emit;

namespace TriAxis.RunSharp
{
	public class AssemblyGen
	{
		AssemblyBuilder asm;
		ModuleBuilder mod;
		List<TypeGen> types = new List<TypeGen>();
		string ns = null;

		internal AssemblyBuilder AssemblyBuilder { get { return asm; } }
		internal ModuleBuilder ModuleBuilder { get { return mod; } }

		internal void AddType(TypeGen tg)
		{
			types.Add(tg);
		}

		class NamespaceContext : IDisposable
		{
			AssemblyGen ag;
			string oldNs;

			public NamespaceContext(AssemblyGen ag)
			{
				this.ag = ag;
				this.oldNs = ag.ns;
			}

			public void Dispose()
			{
				ag.ns = oldNs;
			}
		}

		public IDisposable Namespace(string name)
		{
			NamespaceContext nc = new NamespaceContext(this);
			ns = Qualify(name);
			return nc;
		}

		string Qualify(string name)
		{
			if (ns == null)
				return name;
			else
				return ns + "." + name;
		}

		#region Modifiers
		TypeAttributes attrs;

		[DebuggerBrowsable(DebuggerBrowsableState.Never)]
		public AssemblyGen Public { get { attrs |= TypeAttributes.Public; return this; } }
		[DebuggerBrowsable(DebuggerBrowsableState.Never)]
		public AssemblyGen Private { get { attrs |= TypeAttributes.NotPublic; return this; } }
		[DebuggerBrowsable(DebuggerBrowsableState.Never)]
		public AssemblyGen Sealed { get { attrs |= TypeAttributes.Sealed; return this; } }
		[DebuggerBrowsable(DebuggerBrowsableState.Never)]
		public AssemblyGen Abstract { get { attrs |= TypeAttributes.Abstract; return this; } }
		[DebuggerBrowsable(DebuggerBrowsableState.Never)]
		public AssemblyGen NoBeforeFieldInit { get { attrs |= TypeAttributes.BeforeFieldInit; return this; } }
		#endregion

		#region Types
		public TypeGen Class(string name)
		{
			return Class(name, typeof(object));
		}

		public TypeGen Class(string name, Type baseType)
		{
			return Class(name, baseType, Type.EmptyTypes);
		}

		public TypeGen Class(string name, Type baseType, params Type[] interfaces)
		{
			TypeGen tg = new TypeGen(this, Qualify(name), (attrs | TypeAttributes.Class) ^ TypeAttributes.BeforeFieldInit, baseType, interfaces);
			attrs = 0;
			return tg;
		}

		public TypeGen Struct(string name)
		{
			return Struct(name, Type.EmptyTypes);
		}

		public TypeGen Struct(string name, params Type[] interfaces)
		{
			TypeGen tg = new TypeGen(this, Qualify(name), (attrs | TypeAttributes.Sealed | TypeAttributes.SequentialLayout) ^ TypeAttributes.BeforeFieldInit, typeof(ValueType), interfaces);
			attrs = 0;
			return tg;
		}

		public TypeGen Interface(string name)
		{
			return Interface(name, Type.EmptyTypes);
		}

		public TypeGen Interface(string name, params Type[] interfaces)
		{
			TypeGen tg = new TypeGen(this, Qualify(name), (attrs | TypeAttributes.Interface | TypeAttributes.Abstract) & ~(TypeAttributes.Sealed | TypeAttributes.BeforeFieldInit), null, interfaces);
			attrs = 0;
			return tg;
		}

		public TypeGen Delegate(Type returnType, string name)
		{
			return Delegate(returnType, name);
		}

		public TypeGen Delegate(Type returnType, string name, params Type[] parameterTypes)
		{
			TypeGen tg = new TypeGen(this, Qualify(name), (attrs | TypeAttributes.Sealed) & ~(TypeAttributes.Abstract | TypeAttributes.BeforeFieldInit), typeof(MulticastDelegate), Type.EmptyTypes);
			ImplementDelegate(tg, returnType, parameterTypes);
			return tg;
		}

		internal static void ImplementDelegate(TypeGen tg, Type returnType, Type[] parameterTypes)
		{
			ConstructorBuilder cb = tg.Public.RuntimeImpl.Constructor(typeof(object), typeof(IntPtr)).ConstructorBuilder;
			cb.SetImplementationFlags(MethodImplAttributes.Runtime | MethodImplAttributes.Managed);
			cb.DefineParameter(1, ParameterAttributes.None, "object");
			cb.DefineParameter(2, ParameterAttributes.None, "method");
			
			Type[] asyncParams = new Type[parameterTypes.Length + 2];
			parameterTypes.CopyTo(asyncParams, 0);
			asyncParams[parameterTypes.Length] = typeof(AsyncCallback);
			asyncParams[parameterTypes.Length + 1] = typeof(object);
			MethodBuilder mb = tg.Public.Virtual.RuntimeImpl.Method(typeof(IAsyncResult), "BeginInvoke", asyncParams).MethodBuilder;
			mb.SetImplementationFlags(MethodImplAttributes.Runtime | MethodImplAttributes.Managed);
			mb.DefineParameter(parameterTypes.Length + 1, ParameterAttributes.None, "callback");
			mb.DefineParameter(parameterTypes.Length + 2, ParameterAttributes.None, "object");

			mb = tg.Public.Virtual.RuntimeImpl.Method(returnType, "EndInvoke", typeof(IAsyncResult)).MethodBuilder;
			mb.SetImplementationFlags(MethodImplAttributes.Runtime | MethodImplAttributes.Managed);
			mb.DefineParameter(1, ParameterAttributes.None, "result");

			mb = tg.Public.Virtual.RuntimeImpl.Method(returnType, "Invoke", parameterTypes).MethodBuilder;
			mb.SetImplementationFlags(MethodImplAttributes.Runtime | MethodImplAttributes.Managed);
		}
		#endregion

		#region Construction
		public AssemblyGen(string name) : this(name, false) {}
		public AssemblyGen(string name, bool debugInfo)
		{
			if (name.IndexOfAny(new char[] { Path.DirectorySeparatorChar, Path.AltDirectorySeparatorChar, Path.VolumeSeparatorChar }) != -1 ||
				name.EndsWith(".dll", StringComparison.OrdinalIgnoreCase) ||
				name.EndsWith(".exe", StringComparison.OrdinalIgnoreCase))
			{
				// treat as filename
				Initialize(AppDomain.CurrentDomain, new AssemblyName(Path.GetFileNameWithoutExtension(name)), AssemblyBuilderAccess.RunAndSave,
					name, debugInfo);
			}
			else
			{
				Initialize(AppDomain.CurrentDomain, new AssemblyName(name), AssemblyBuilderAccess.Run, null, debugInfo);
			}
		}

		AssemblyBuilderAccess access;
		string fileName;

		public AssemblyGen(AppDomain domain, AssemblyName name, AssemblyBuilderAccess access, string fileName, bool symbolInfo)
		{
			Initialize(domain, name, access, fileName, symbolInfo);
		}

		void Initialize(AppDomain domain, AssemblyName name, AssemblyBuilderAccess access, string fileName, bool symbolInfo)
		{
			this.access = access;
			this.fileName = fileName;

			if (fileName == null && (access & AssemblyBuilderAccess.Save) != 0)
				throw new ArgumentNullException("fileName", Properties.Messages.ErrAsmMissingSaveFile);

			if (fileName == null)
				asm = domain.DefineDynamicAssembly(name, access);
			else
				asm = domain.DefineDynamicAssembly(name, access, Path.GetDirectoryName(fileName));

			if (fileName == null)
				mod = asm.DefineDynamicModule(name.Name, symbolInfo);
			else
				mod = asm.DefineDynamicModule(Path.GetFileName(fileName), Path.GetFileName(fileName), symbolInfo);
		}

		public void Save()
		{
			Complete();

			if ((access & AssemblyBuilderAccess.Save) != 0)
				asm.Save(Path.GetFileName(fileName));
		}

		public void Complete()
		{
			foreach (TypeGen tg in types)
				tg.Complete();
		}
		#endregion
	}
}