// Copyright (c) 2017-2020 Katy Coe - http://www.djkaty.com - https://github.com/djkaty
// All rights reserved

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using Il2CppInspector.Properties;
using Il2CppInspector.Reflection;
using Assembly = Il2CppInspector.Reflection.Assembly;
using CustomAttributeData = Il2CppInspector.Reflection.CustomAttributeData;
using MethodInfo = Il2CppInspector.Reflection.MethodInfo;
using TypeInfo = Il2CppInspector.Reflection.TypeInfo;

namespace Il2CppInspector.Outputs
{
    public class CSharpCodeStubs
    {
        private readonly Il2CppModel model;
        private Exception lastException;

        // Namespace prefixes whose contents should be skipped
        public List<string> ExcludedNamespaces { get; set; }

        // Make adjustments to ensure that the generated code compiles
        public bool MustCompile { get; set; }

        // Suppress binary metadata in code comments
        public bool SuppressMetadata { get; set; }

        private const string CGAttribute = "System.Runtime.CompilerServices.CompilerGeneratedAttribute";
        private const string FBAttribute = "System.Runtime.CompilerServices.FixedBufferAttribute";
        private const string ExtAttribute = "System.Runtime.CompilerServices.ExtensionAttribute";
        private const string DMAttribute = "System.Reflection.DefaultMemberAttribute";

        // Assembly attributes we have already emitted
        private HashSet<CustomAttributeData> usedAssemblyAttributes = new HashSet<CustomAttributeData>();
        private readonly object usedAssemblyAttributesLock = new object();

        public CSharpCodeStubs(Il2CppModel model) => this.model = model;

        // Get the last error that occurred and clear the error state
        public Exception GetAndClearLastException() {
            var ex = lastException;
            lastException = null;
            return ex;
        }

        public void WriteSingleFile(string outFile) => WriteSingleFile(outFile, t => t.Index);

        public void WriteSingleFile<TKey>(string outFile, Func<TypeInfo, TKey> orderBy) {
            usedAssemblyAttributes.Clear();
            writeFile(outFile, model.Assemblies.SelectMany(x => x.DefinedTypes).OrderBy(orderBy));
        }

        public void WriteFilesByNamespace<TKey>(string outPath, Func<TypeInfo, TKey> orderBy, bool flattenHierarchy) {
            usedAssemblyAttributes.Clear();
            Parallel.ForEach(model.Assemblies.SelectMany(x => x.DefinedTypes).GroupBy(t => t.Namespace), ns => {
                var relPath = !string.IsNullOrEmpty(ns.Key) ? ns.Key : "global";
                writeFile(Path.Combine(outPath, (flattenHierarchy ? relPath : Path.Combine(relPath.Split('.'))) + ".cs"),
                    ns.OrderBy(orderBy));
            });
        }

        public void WriteFilesByAssembly<TKey>(string outPath, Func<TypeInfo, TKey> orderBy, bool separateAttributes) {
            usedAssemblyAttributes.Clear();
            Parallel.ForEach(model.Assemblies, asm => {
                // Sort namespaces into alphabetical order, then sort types within the namespaces by the specified sort function
                if (writeFile(Path.Combine(outPath, Path.GetFileNameWithoutExtension(asm.ShortName) + ".cs"), asm.DefinedTypes.OrderBy(t => t.Namespace).ThenBy(orderBy), outputAssemblyAttributes: !separateAttributes)
                    && separateAttributes) {
                    File.WriteAllText(Path.Combine(outPath, $"AssemblyInfo_{Path.GetFileNameWithoutExtension(asm.ShortName)}.cs"), generateAssemblyInfo(new [] {asm}));
                }
            });
        }

        public void WriteFilesByClass(string outPath, bool flattenHierarchy) {
            usedAssemblyAttributes.Clear();
            Parallel.ForEach(model.Assemblies.SelectMany(x => x.DefinedTypes), type => {
                string relPath = $"{type.Namespace}{(type.Namespace.Length > 0 ? "." : "")}{Regex.Replace(type.Name, "`[0-9]", "")}";
                writeFile(Path.Combine(outPath, flattenHierarchy ? relPath : Path.Combine(relPath.Split('.')) + ".cs"), new[] {type});
            });
        }

        public HashSet<Assembly> WriteFilesByClassTree(string outPath, bool separateAttributes) {
            usedAssemblyAttributes.Clear();
            var usedAssemblies = new HashSet<Assembly>();

            // Each thread tracks its own list of used assemblies and they are merged as each thread completes
            Parallel.ForEach(model.Assemblies.SelectMany(x => x.DefinedTypes),
                () => new HashSet<Assembly>(),
                (type, _, used) => {
                    string relPath = Path.Combine($"{type.Namespace}{(type.Namespace.Length > 0 ? "." : "")}{Regex.Replace(type.Name, "`[0-9]", "")}".Split('.'));
                    if (writeFile(Path.Combine(outPath, Path.GetFileNameWithoutExtension(type.Assembly.ShortName), $"{relPath}.cs"), new[] {type}, outputAssemblyAttributes: !separateAttributes))
                        used.Add(type.Assembly);
                    return used;
                },
                usedPartition => {
                    lock (usedAssemblies) usedAssemblies.UnionWith(usedPartition);
                }
            );

            if (separateAttributes && usedAssemblies.Any() && lastException == null)
                foreach (var asm in usedAssemblies)
                    File.WriteAllText(Path.Combine(outPath, Path.GetFileNameWithoutExtension(asm.ShortName), "AssemblyInfo.cs"), generateAssemblyInfo(new [] {asm}));

            return usedAssemblies;
        }

        // Create a Visual Studio solution
        public void WriteSolution(string outPath, string unityPath, string unityAssembliesPath) {
            // Required settings
            MustCompile = true;

            // Output source files in tree format with separate assembly attributes
            var assemblies = WriteFilesByClassTree(outPath, true);

            if (lastException != null)
                return;

            // Per-project (per-assembly) solution definition and configuration
            var slnProjectDefs = new StringBuilder();
            var slnProjectConfigs = new StringBuilder();

            foreach (var asm in assemblies) {
                var guid = Guid.NewGuid();
                var name = Path.GetFileNameWithoutExtension(asm.ShortName);
                var csProjFile = Path.Combine(name, $"{name}.csproj");

                var def = Resources.SlnProjectDefinition
                    .Replace("%PROJECTGUID%", guid.ToString())
                    .Replace("%PROJECTNAME%", name)
                    .Replace("%CSPROJRELATIVEPATH%", csProjFile);

                slnProjectDefs.Append(def);

                var config = Resources.SlnProjectConfiguration
                    .Replace("%PROJECTGUID%", guid.ToString());

                slnProjectConfigs.Append(config);

                // Determine all the assemblies on which this assembly depends
                var dependencyTypes = asm.DefinedTypes.SelectMany(t => t.GetAllTypeReferences())
                    .Union(asm.CustomAttributes.SelectMany(a => a.AttributeType.GetAllTypeReferences()))
                    .Distinct();
                var dependencyAssemblies = dependencyTypes.Select(t => t.Assembly).Distinct()
                    .Except(new[] {asm});
                
                // Only create project references to those assemblies actually output in our solution
                dependencyAssemblies = dependencyAssemblies.Intersect(assemblies);

                var referenceXml = string.Concat(dependencyAssemblies.Select(
                        a => $@"    <ProjectReference Include=""..\{a.ShortName.Replace(".dll", "")}\{a.ShortName.Replace(".dll", "")}.csproj""/>" + "\n"
                ));

                // Create a .csproj file using the project Guid
                var csProj = Resources.CsProjTemplate
                    .Replace("%PROJECTGUID%", guid.ToString())
                    .Replace("%ASSEMBLYNAME%", name)
                    .Replace("%UNITYPATH%", unityPath)
                    .Replace("%SCRIPTASSEMBLIES%", unityAssembliesPath)
                    .Replace("%PROJECTREFERENCES%", referenceXml);

                File.WriteAllText(Path.Combine(outPath, csProjFile), csProj);
            }

            // Merge everything into .sln file
            var sln = Resources.SlnTemplate
                .Replace("%PROJECTDEFINITIONS%", slnProjectDefs.ToString())
                .Replace("%PROJECTCONFIGURATIONS%", slnProjectConfigs.ToString());

            var filename = Path.GetFileName(outPath);
            if (filename == "")
                filename = "Il2CppProject";
            File.WriteAllText(Path.Combine(outPath, $"{filename}.sln"), sln);
        }

        private bool writeFile(string outFile, IEnumerable<TypeInfo> types, bool useNamespaceSyntax = true, bool outputAssemblyAttributes = true) {

            var nsRefs = new HashSet<string>();
            var code = new StringBuilder();
            var nsContext = "";
            var usedTypes = new List<TypeInfo>();

            // Determine all namespace references (note: this may include some that aren't actually used due to output suppression in generateType()
            // We have to do this first so that we can generate properly scoped type references in the code
            foreach (var type in types) {
                var refs = type.GetAllTypeReferences();
                var ns = refs.Where(r => !string.IsNullOrEmpty(r.Namespace) && r.Namespace != type.Namespace).Select(r => r.Namespace);
                nsRefs.UnionWith(ns);
            }

            // Determine assemblies used in this file
            var assemblies = types.Select(t => t.Assembly).Distinct();

            // Add assembly attribute namespaces to reference list
            if (outputAssemblyAttributes)
                nsRefs.UnionWith(assemblies.SelectMany(a => a.CustomAttributes).Select(a => a.AttributeType.Namespace));

            // Generate each type
            foreach (var type in types) {

                // Skip namespace and any children if requested
                if (ExcludedNamespaces?.Any(x => x == type.Namespace || type.Namespace.StartsWith(x + ".")) ?? false)
                    continue;

                // Don't output global::Locale if desired
                if (MustCompile
                    && type.Name == "Locale" && type.Namespace == string.Empty
                    && type.BaseType.FullName == "System.Object"
                    && type.IsClass && type.IsSealed && type.IsNotPublic && !type.ContainsGenericParameters
                    && type.DeclaredMembers.Count == type.DeclaredMethods.Count
                    && type.GetMethods("GetText").Length == type.DeclaredMethods.Count)
                    continue;

                // Assembly.DefinedTypes returns nested types in the assembly by design - ignore them
                if (type.IsNested)
                    continue;

                // Get code
                var text = generateType(type, nsRefs);
                if (string.IsNullOrEmpty(text))
                    continue;

                // Determine if we need to change namespace (after we have established the code block is not empty)
                if (useNamespaceSyntax) {
                    if (type.Namespace != nsContext) {
                        if (!string.IsNullOrEmpty(nsContext))
                            code.Remove(code.Length - 1, 1).Append("}\n\n");

                        if (!string.IsNullOrEmpty(type.Namespace))
                            code.Append("namespace " + type.Namespace + "\n{\n");

                        nsContext = type.Namespace;
                    }

                    if (!string.IsNullOrEmpty(nsContext)) {
                        text = "\t" + text.Replace("\n", "\n\t");
                        text = text.Remove(text.Length - 1);
                    }
                }

                // Append namespace
                if (!useNamespaceSyntax)
                    code.Append($"// Namespace: {(!string.IsNullOrEmpty(type.Namespace) ? type.Namespace : "<global namespace>")}\n");
                
                // Append type definition
                code.Append(text + "\n");

                // Add to list of used types
                usedTypes.Add(type);
            }

            // Stop if nothing to output
            if (!usedTypes.Any())
                return false;

            // Close namespace
            if (useNamespaceSyntax && !string.IsNullOrEmpty(nsContext))
                code.Remove(code.Length - 1, 1).Append("}\n");
            
            // Determine using directives (put System namespaces first)
            nsRefs.Clear();
            foreach (var type in usedTypes) {
                var refs = type.GetAllTypeReferences();
                var ns = refs.Where(r => !string.IsNullOrEmpty(r.Namespace) && r.Namespace != type.Namespace).Select(r => r.Namespace);
                nsRefs.UnionWith(ns);
            }
            nsRefs.UnionWith(assemblies.SelectMany(a => a.CustomAttributes).Select(a => a.AttributeType.Namespace));

            var usings = nsRefs.OrderBy(n => (n.StartsWith("System.") || n == "System") ? "0" + n : "1" + n);

            // Ensure output directory exists and is not a file
            var dir = Path.GetDirectoryName(outFile);
            if (!string.IsNullOrEmpty(dir)) {
                try {
                    Directory.CreateDirectory(dir);
                }
                catch (IOException ex) {
                    lastException = ex;
                    return false;
                }
            }

            // Sanitize leafname (might be class name with invalid characters)
            var leafname = Regex.Replace(Path.GetFileName(outFile), @"[<>:""\|\?\*]", "_");
            outFile = Path.Combine(dir, leafname);

            // Create output file
            bool fileWritten = false;
            do {
                try {
                    using StreamWriter writer = new StreamWriter(new FileStream(outFile, FileMode.Create), Encoding.UTF8);

                    // Write preamble
                    writer.Write(@"/*
 * Generated code file by Il2CppInspector - http://www.djkaty.com - https://github.com/djkaty
 */

");

                    // Output using directives
                    writer.Write(string.Concat(usings.Select(n => $"using {n};\n")));
                    if (nsRefs.Any())
                        writer.Write("\n");

                    // Output assembly information and attributes
                    writer.Write(generateAssemblyInfo(assemblies, nsRefs, outputAssemblyAttributes) + "\n\n");

                    // Output type definitions
                    writer.Write(code);

                    fileWritten = true;
                }
                catch (IOException ex) {
                    // If we get "file is in use by another process", we are probably writing a duplicate class in another thread
                    // Wait a bit and try again
                    if ((uint) ex.HResult != 0x80070020)
                        throw;

                    System.Threading.Thread.Sleep(100);
                }
            } while (!fileWritten);

            return true;
        }

        private string generateAssemblyInfo(IEnumerable<Reflection.Assembly> assemblies, IEnumerable<string> namespaces = null, bool outputAssemblyAttributes = true) {
            var text = new StringBuilder();

            foreach (var asm in assemblies) {
                text.Append($"// Image {asm.Index}: {asm.ShortName} - Assembly: {asm.FullName} - Types {asm.ImageDefinition.typeStart}-{asm.ImageDefinition.typeStart + asm.ImageDefinition.typeCount - 1}\n");

                // Assembly-level attributes
                if (outputAssemblyAttributes)
                    lock (usedAssemblyAttributesLock) {
                        text.Append(asm.CustomAttributes.Where(a => a.AttributeType.FullName != ExtAttribute)
                            .Except(usedAssemblyAttributes ?? new HashSet<CustomAttributeData>())
                            .OrderBy(a => a.AttributeType.Name)
                            .ToString(new Scope { Current = null, Namespaces = namespaces ?? new List<string>() }, attributePrefix: "assembly: ", emitPointer: !SuppressMetadata, mustCompile: MustCompile));
                        if (asm.CustomAttributes.Any())
                            text.Append("\n");

                        usedAssemblyAttributes.UnionWith(asm.CustomAttributes);
                    }
            }
            return text.ToString().TrimEnd();
        }

        private string generateType(TypeInfo type, IEnumerable<string> namespaces, string prefix = "") {
            // Don't output compiler-generated types if desired
            if (MustCompile && type.GetCustomAttributes(CGAttribute).Any())
                return string.Empty;

            var codeBlocks = new Dictionary<string, string>();
            var usedMethods = new List<MethodInfo>();
            var sb = new StringBuilder();

            var scope = new Scope {
                Current = type,
                Namespaces = namespaces
            };

            // Fields
            sb.Clear();
            if (!type.IsEnum) {
                foreach (var field in type.DeclaredFields) {
                    if (MustCompile && field.GetCustomAttributes(CGAttribute).Any())
                        continue;

                    if (field.IsNotSerialized)
                        sb.Append(prefix + "\t[NonSerialized]\n");

                    // Attributes
                    sb.Append(field.CustomAttributes.Where(a => a.AttributeType.FullName != FBAttribute).OrderBy(a => a.AttributeType.Name)
                        .ToString(scope, prefix + "\t", emitPointer: !SuppressMetadata, mustCompile: MustCompile));
                    sb.Append(prefix + "\t");
                    sb.Append(field.GetModifierString());

                    // Fixed buffers
                    if (field.GetCustomAttributes(FBAttribute).Any()) {
                        if (!SuppressMetadata)
                            sb.Append($"/* {field.GetCustomAttributes(FBAttribute)[0].VirtualAddress.ToAddressString()} */ ");
                        sb.Append($"{field.FieldType.DeclaredFields[0].FieldType.GetScopedCSharpName(scope)} {field.CSharpSafeName}[0]"); // FixedElementField
                    }
                    // Regular fields
                    else
                        sb.Append($"{field.FieldType.GetScopedCSharpName(scope)} {field.CSharpSafeName}");
                    if (field.HasDefaultValue)
                        sb.Append($" = {field.DefaultValueString}");
                    sb.Append(";");
                    // Don't output field indices for const fields (they don't have any storage)
                    // or open generic types (they aren't known until runtime)
                    if (!field.IsLiteral && !SuppressMetadata && !type.ContainsGenericParameters)
                        sb.Append($" // 0x{(uint) field.Offset:X2}");
                    // Output metadata file offset for const fields
                    if (field.IsLiteral && !SuppressMetadata)
                        sb.Append($" // Metadata: 0x{(uint) field.DefaultValueMetadataAddress:X8}");
                    sb.Append("\n");
                }
                codeBlocks.Add("Fields", sb.ToString());
            }

            // Properties
            sb.Clear();
            foreach (var prop in type.DeclaredProperties) {
                // Attributes
                sb.Append(prop.CustomAttributes.OrderBy(a => a.AttributeType.Name)
                    .ToString(scope, prefix + "\t", emitPointer: !SuppressMetadata, mustCompile: MustCompile));

                // The access mask enum values go from 1 (private) to 6 (public) in order from most to least restrictive
                var getAccess = (prop.GetMethod?.Attributes ?? 0) & MethodAttributes.MemberAccessMask;
                var setAccess = (prop.SetMethod?.Attributes ?? 0) & MethodAttributes.MemberAccessMask;

                // In case the access level of both is the same and the selected method is null, pick the other one (rare edge case)
                var primary = (getAccess >= setAccess ? prop.GetMethod : prop.SetMethod) ?? prop.GetMethod ?? prop.SetMethod;
                sb.Append($"{prefix}\t{primary.GetModifierString()}{prop.PropertyType.GetScopedCSharpName(scope)} ");

                // Non-indexer; non-auto-properties should have a body
                var needsBody = MustCompile && !type.IsInterface && !type.IsAbstract && !prop.IsAutoProperty;

                var getBody = needsBody? " => default;" : ";";
                var setBody = needsBody? " {}" : ";";
                if ((!prop.CanRead || !prop.GetMethod.DeclaredParameters.Any()) && (!prop.CanWrite || prop.SetMethod.DeclaredParameters.Count == 1))
                    sb.Append($"{prop.CSharpName} {{ ");
                
                // Indexer
                else {
                    // Replace "Item" with "this" - preserves explicit interface implementations
                    sb.Append(prop.CSharpName[..^4] + "this");
                    sb.Append("[" + string.Join(", ", primary.DeclaredParameters.SkipLast(getAccess >= setAccess ? 0 : 1)
                                  .Select(p => p.GetParameterString(scope, !SuppressMetadata, MustCompile))) + "] { ");
                    getBody = " => default;";
                    setBody = " {}";
                }

                sb.Append((prop.CanRead? prop.GetMethod.CustomAttributes.Where(a => !MustCompile || a.AttributeType.FullName != CGAttribute)
                                             .ToString(scope, inline: true, emitPointer: !SuppressMetadata, mustCompile: MustCompile) 
                                               + (getAccess < setAccess? prop.GetMethod.GetAccessModifierString() : "") + $"get{getBody} " : "")
                             // Auto-properties must have get accessors (exclude indexers)
                             + (MustCompile && !prop.CanRead && setBody == ";"? "get; " : "")
                             + (prop.CanWrite? prop.SetMethod.CustomAttributes.Where(a => !MustCompile || a.AttributeType.FullName != CGAttribute)
                                                   .ToString(scope, inline: true, emitPointer: !SuppressMetadata, mustCompile: MustCompile) 
                                               + (setAccess < getAccess? prop.SetMethod.GetAccessModifierString() : "") + $"set{setBody} " : "") + "}");
                if (!SuppressMetadata) {
                    if ((prop.CanRead && prop.GetMethod.VirtualAddress != null) || (prop.CanWrite && prop.SetMethod.VirtualAddress != null))
                        sb.Append(" // ");
                    sb.Append((prop.CanRead && prop.GetMethod.VirtualAddress != null ? prop.GetMethod.VirtualAddress.ToAddressString() + " " : "")
                                + (prop.CanWrite && prop.SetMethod.VirtualAddress != null ? prop.SetMethod.VirtualAddress.ToAddressString() : ""));
                }
                sb.Append("\n");

                usedMethods.Add(prop.GetMethod);
                usedMethods.Add(prop.SetMethod);
            }
            codeBlocks.Add("Properties", sb.ToString());

            // Events
            sb.Clear();
            foreach (var evt in type.DeclaredEvents) {
                // Attributes
                sb.Append(evt.CustomAttributes.OrderBy(a => a.AttributeType.Name)
                    .ToString(scope, prefix + "\t", emitPointer: !SuppressMetadata, mustCompile: MustCompile));

                string modifiers = evt.AddMethod?.GetModifierString();
                sb.Append($"{prefix}\t{modifiers}event {evt.EventHandlerType.GetScopedCSharpName(scope)} {evt.CSharpSafeName}");
                
                if (!MustCompile) {
                    sb.Append(" {\n");
                    var m = new Dictionary<string, (ulong, ulong)?>();
                    if (evt.AddMethod != null) m.Add("add", evt.AddMethod.VirtualAddress);
                    if (evt.RemoveMethod != null) m.Add("remove", evt.RemoveMethod.VirtualAddress);
                    if (evt.RaiseMethod != null) m.Add("raise", evt.RaiseMethod.VirtualAddress);
                    sb.Append(string.Join("\n", m.Select(x => $"{prefix}\t\t{x.Key};{(SuppressMetadata? "" : " // " + x.Value.ToAddressString())}")) + "\n" + prefix + "\t}\n");
                } else
                    sb.Append(";\n");

                usedMethods.Add(evt.AddMethod);
                usedMethods.Add(evt.RemoveMethod);
                usedMethods.Add(evt.RaiseMethod);
            }
            codeBlocks.Add("Events", sb.ToString());

            // Nested types
            codeBlocks.Add("Nested types", string.Join("\n", type.DeclaredNestedTypes
                .Select(n => generateType(n, namespaces, prefix + "\t")).Where(c => !string.IsNullOrEmpty(c))));

            // Constructors
            var fields = type.DeclaredFields.Where(f => !f.GetCustomAttributes(CGAttribute).Any());

            sb.Clear();

            // Crete a parameterless constructor for every relevant type when making code that compiles to mitigate CS1729 and CS7036
            if (MustCompile && !type.IsInterface && !(type.IsAbstract && type.IsSealed) && !type.IsValueType
                && type.DeclaredConstructors.All(c => c.IsStatic || c.DeclaredParameters.Any()))
                sb.Append($"{prefix}\t{(type.IsAbstract? "protected" : "public")} {type.UnmangledBaseName}() {{}} // Dummy constructor\n");

            foreach (var method in type.DeclaredConstructors) {
                // Attributes
                sb.Append(method.CustomAttributes.OrderBy(a => a.AttributeType.Name)
                    .ToString(scope, prefix + "\t", emitPointer: !SuppressMetadata, mustCompile: MustCompile));

                sb.Append($"{prefix}\t{method.GetModifierString()}{method.DeclaringType.UnmangledBaseName}{method.GetTypeParametersString(scope)}");
                sb.Append($"({method.GetParametersString(scope, !SuppressMetadata)})");

                if (MustCompile) {
                    // Class constructor
                    if (method.IsAbstract)
                        sb.Append(";");
                    else if (!type.IsValueType)
                        sb.Append(" {}");

                    // Struct constructor
                    else {
                        // Parameterized struct constructors must call the parameterless constructor to create the object
                        // if the object has any auto-implemented properties
                        if (type.DeclaredProperties.Any() && method.DeclaredParameters.Any())
                            sb.Append(" : this()");

                        // Struct construvctors must initialize all fields in the struct
                        if (fields.Any()) {
                            var paramNames = method.DeclaredParameters.Select(p => p.Name);
                            sb.Append(" {\n" + string.Join("\n", fields
                                            .Where(f => !f.IsLiteral && f.IsStatic == method.IsStatic)
                                            .Select(f => $"{prefix}\t\t{(paramNames.Contains(f.Name) ? "this." : "")}{f.Name} = default;"))
                                        + $"\n{prefix}\t}}");
                        } else
                            sb.Append(" {}");
                    }
                } else
                    sb.Append(";");

                sb.Append((!SuppressMetadata && method.VirtualAddress != null ? $" // {method.VirtualAddress.ToAddressString()}" : "") + "\n");
            }
            codeBlocks.Add("Constructors", sb.ToString());

            // Methods
            // Don't re-output methods for constructors, properties, events etc.
            var methods = type.DeclaredMethods.Except(usedMethods).Where(m => m.CustomAttributes.All(a => a.AttributeType.FullName != ExtAttribute));
            codeBlocks.Add("Methods", string.Concat(methods.Select(m => generateMethod(m, scope, prefix))));
            usedMethods.AddRange(methods);

            // Extension methods 
            codeBlocks.Add("Extension methods", string.Concat(type.DeclaredMethods.Except(usedMethods).Select(m => generateMethod(m, scope, prefix))));

            // Type declaration
            sb.Clear();

            if (type.IsImport)
                sb.Append(prefix + "[ComImport]\n");
            if (type.IsSerializable)
                sb.Append(prefix + "[Serializable]\n");

            // TODO: DefaultMemberAttribute should be output if it is present and the type does not have an indexer, otherwise suppressed
            // See https://docs.microsoft.com/en-us/dotnet/api/system.reflection.defaultmemberattribute?view=netframework-4.8
            sb.Append(type.CustomAttributes.Where(a => a.AttributeType.FullName != DMAttribute && a.AttributeType.FullName != ExtAttribute)
                                            .OrderBy(a => a.AttributeType.Name).ToString(scope, prefix, emitPointer: !SuppressMetadata, mustCompile: MustCompile));

            // Roll-up multicast delegates to use the 'delegate' syntactic sugar
            if (type.IsClass && type.IsSealed && type.BaseType?.FullName == "System.MulticastDelegate") {
                sb.Append(prefix + type.GetAccessModifierString());

                var del = type.GetMethod("Invoke");
                // IL2CPP doesn't seem to retain return type attributes
                //sb.Append(del.ReturnType.CustomAttributes.ToString(prefix, "return: ", emitPointer: !SuppressMetadata, mustCompile: MustCompile));
                if (del.RequiresUnsafeContext)
                    sb.Append("unsafe ");
                sb.Append($"delegate {del.ReturnType.GetScopedCSharpName(scope)} {type.GetCSharpTypeDeclarationName()}(");
                sb.Append(del.GetParametersString(scope, !SuppressMetadata) + ");");
                if (!SuppressMetadata)
                    sb.Append($" // TypeDefIndex: {type.Index}; {del.VirtualAddress.ToAddressString()}");
                sb.Append("\n");
                return sb.ToString();
            }

            sb.Append(prefix + type.GetModifierString());

            var @base = type.ImplementedInterfaces.Select(x => x.GetScopedCSharpName(scope, isPartOfTypeDeclaration: true)).ToList();
            if (type.BaseType != null && type.BaseType.FullName != "System.Object" && type.BaseType.FullName != "System.ValueType" && !type.IsEnum)
                @base.Insert(0, type.BaseType.GetScopedCSharpName(scope, isPartOfTypeDeclaration: true));
            if (type.IsEnum && type.GetEnumUnderlyingType().FullName != "System.Int32") // enums derive from int by default
                @base.Insert(0, type.GetEnumUnderlyingType().GetScopedCSharpName(scope));
            var baseText = @base.Count > 0 ? " : " + string.Join(", ", @base) : string.Empty;

            sb.Append($"{type.GetCSharpTypeDeclarationName()}{baseText}");
            if (!SuppressMetadata)
                sb.Append($" // TypeDefIndex: {type.Index}");
            sb.Append("\n");

            foreach (var gp in type.GetGenericArguments()) {
                var constraint = gp.GetTypeConstraintsString(scope);
                if (constraint != string.Empty)
                    sb.Append($"{prefix}\t{constraint}\n");
            }

            sb.Append(prefix + "{\n");

            // Enumeration
            if (type.IsEnum) {
                sb.Append(string.Join(",\n", type.GetEnumNames().Zip(type.GetEnumValues().OfType<object>(),
                              (k, v) => new { k, v }).OrderBy(x => x.v).Select(x => $"{prefix}\t{x.k} = {x.v}")) + "\n");
            }

            // Type definition
            else
                sb.Append(string.Join("\n", codeBlocks.Where(b => b.Value != string.Empty).Select(b => prefix + "\t// " + b.Key + "\n" + b.Value)));

            sb.Append(prefix + "}\n");
            return sb.ToString();
        }

        private string generateMethod(MethodInfo method, Scope scope, string prefix) {
            if (MustCompile && method.GetCustomAttributes(CGAttribute).Any())
                return string.Empty;

            var writer = new StringBuilder();

            // Attributes
            writer.Append(method.CustomAttributes.Where(a => a.AttributeType.FullName != ExtAttribute).OrderBy(a => a.AttributeType.Name)
                .ToString(scope, prefix + "\t", emitPointer: !SuppressMetadata, mustCompile: MustCompile));

            // IL2CPP doesn't seem to retain return type attributes
            //writer.Append(method.ReturnType.CustomAttributes.ToString(prefix + "\t", "return: ", emitPointer: !SuppressMetadata));
            writer.Append($"{prefix}\t{method.GetModifierString()}");

            // Finalizers become destructors
            if (method.Name == "Finalize" && method.IsVirtual && method.ReturnType.FullName == "System.Void" && method.IsFamily)
                writer.Append("~" + method.DeclaringType.UnmangledBaseName);
                
            // Regular method or operator overload
            else if (method.Name != "op_Implicit" && method.Name != "op_Explicit")
                writer.Append($"{method.ReturnParameter.GetReturnParameterString(scope)} {method.CSharpName}{method.GetTypeParametersString(scope)}");

            // User-defined conversion operator
            else
                writer.Append($"{method.CSharpName}{method.ReturnType.GetScopedCSharpName(scope)}");

            // Parameters
            writer.Append("(" + method.GetParametersString(scope, !SuppressMetadata) + ")");

            // Generic type constraints
            foreach (var gp in method.GetGenericArguments()) {
                var constraint = gp.GetTypeConstraintsString(scope);
                if (constraint != string.Empty)
                    writer.Append($"\n{prefix}\t\t{constraint}");
            }

            // Body
            var methodBody = MustCompile? method switch {
                    // Abstract method
                    { IsAbstract: true } => ";",

                    // Extern method
                    { Attributes: var a } when (a & MethodAttributes.PinvokeImpl) == MethodAttributes.PinvokeImpl => ";",

                    // Method with out parameters
                    { DeclaredParameters: var d } when d.Any(p => p.IsOut) =>
                        " {\n" + string.Join("\n", d.Where(p => p.IsOut).Select(p => $"{prefix}\t\t{p.Name} = default;"))
                        + (method.ReturnType.FullName != "System.Void"? $"\n{prefix}\t\treturn default;" : "")
                        + $"\n{prefix}\t}}",

                    // No return type
                    { ReturnType: var retType } when retType.FullName == "System.Void" => " {}",

                    // Ref return type
                    { ReturnType: var retType } when retType.IsByRef => " => ref _refReturnTypeFor" + method.CSharpName + ";",

                    // Regular return type
                    _ => " => default;"
                }

                // Only make a method body if we are trying to compile the output
                : ";";

            writer.Append(methodBody + (!SuppressMetadata && method.VirtualAddress != null ? $" // {method.VirtualAddress.ToAddressString()}" : "") + "\n");

            // Ref return type requires us to invent a field
            if (MustCompile && method.ReturnType.IsByRef)
                writer.Append($"{prefix}\tprivate {method.ReturnType.GetScopedCSharpName(scope)} _refReturnTypeFor{method.CSharpName}; // Dummy field\n");

            return writer.ToString();
        }
    }
}