1 files changed, 122 insertions, 0 deletions

diff --git a/linker/Tests/Extensions/CecilExtensions.cs b/linker/Tests/Extensions/CecilExtensions.cs
new file mode 100644
index 000000000..747abf570
--- /dev/null
+++ b/linker/Tests/Extensions/CecilExtensions.cs
@@ -0,0 +1,122 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using Mono.Cecil;
+
+namespace Mono.Linker.Tests.Extensions {
+	public static class CecilExtensions {
+		public static IEnumerable<TypeDefinition> AllDefinedTypes (this AssemblyDefinition assemblyDefinition)
+		{
+			return assemblyDefinition.Modules.SelectMany (m => m.AllDefinedTypes ());
+		}
+
+		public static IEnumerable<TypeDefinition> AllDefinedTypes (this ModuleDefinition moduleDefinition)
+		{
+			foreach (var typeDefinition in moduleDefinition.Types) {
+				yield return typeDefinition;
+
+				foreach (var definition in typeDefinition.AllDefinedTypes ())
+					yield return definition;
+			}
+		}
+
+		public static IEnumerable<TypeDefinition> AllDefinedTypes (this TypeDefinition typeDefinition)
+		{
+			foreach (var nestedType in typeDefinition.NestedTypes) {
+				yield return nestedType;
+
+				foreach (var definition in nestedType.AllDefinedTypes ())
+					yield return definition;
+			}
+		}
+
+		public static IEnumerable<IMemberDefinition> AllMembers (this ModuleDefinition module)
+		{
+			foreach (var type in module.AllDefinedTypes ()) {
+				yield return type;
+
+				foreach (var member in type.AllMembers ())
+					yield return member;
+			}
+		}
+
+		public static IEnumerable<IMemberDefinition> AllMembers (this TypeDefinition type)
+		{
+			foreach (var field in type.Fields)
+				yield return field;
+
+			foreach (var prop in type.Properties)
+				yield return prop;
+
+			foreach (var method in type.Methods)
+				yield return method;
+
+			foreach (var @event in type.Events)
+				yield return @event;
+		}
+
+		public static bool HasAttribute (this ICustomAttributeProvider provider, string name)
+		{
+			return provider.CustomAttributes.Any (ca => ca.AttributeType.Name == name);
+		}
+
+		public static bool HasAttributeDerivedFrom (this ICustomAttributeProvider provider, string name)
+		{
+			return provider.CustomAttributes.Any (ca => ca.AttributeType.Resolve ().DerivesFrom (name));
+		}
+
+		public static bool DerivesFrom (this TypeDefinition type, string baseTypeName)
+		{
+			if (type.Name == baseTypeName)
+				return true;
+
+			if (type.BaseType == null)
+				return false;
+
+			if (type.BaseType.Name == baseTypeName)
+				return true;
+
+			return type.BaseType.Resolve ().DerivesFrom (baseTypeName);
+		}
+
+		public static PropertyDefinition GetPropertyDefinition (this MethodDefinition method)
+		{
+			if (!method.IsSetter && !method.IsGetter)
+				throw new ArgumentException ();
+
+			var propertyName = method.Name.Substring (4);
+			return method.DeclaringType.Properties.First (p => p.Name == propertyName);
+		}
+
+		public static string GetSignature (this MethodDefinition method)
+		{
+			var builder = new StringBuilder ();
+			builder.Append (method.Name);
+			if (method.HasGenericParameters) {
+				builder.Append ('<');
+
+				for (int i = 0; i < method.GenericParameters.Count - 1; i++)
+					builder.Append ($"{method.GenericParameters [i]},");
+
+				builder.Append ($"{method.GenericParameters [method.GenericParameters.Count - 1]}>");
+			}
+
+			builder.Append ("(");
+
+			if (method.HasParameters) {
+				for (int i = 0; i < method.Parameters.Count - 1; i++) {
+					// TODO: modifiers
+					// TODO: default values
+					builder.Append ($"{method.Parameters [i].ParameterType},");
+				}
+
+				builder.Append (method.Parameters [method.Parameters.Count - 1].ParameterType);
+			}
+
+			builder.Append (")");
+
+			return builder.ToString ();
+		}
+	}
+}
\ No newline at end of file