// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Reflection.Metadata;
using System.Reflection.PortableExecutable;
 
namespace Microsoft.NET.TestFramework
{
    public static class AssemblyInfo
    {
        public static List<string> GetParameterlessAttributes(string assemblyPath)
        {
            var parameterlessAttributes = new List<string>();
 
            using (var stream = File.OpenRead(assemblyPath))
            using (var peReader = new PEReader(stream))
            {
                var metadataReader = peReader.GetMetadataReader();
                var assemblyDefinition = metadataReader.GetAssemblyDefinition();
 
                foreach (var handle in assemblyDefinition.GetCustomAttributes())
                {
                    var attribute = metadataReader.GetCustomAttribute(handle);
                    var constructor = metadataReader.GetMemberReference((MemberReferenceHandle)attribute.Constructor);
                    var type = metadataReader.GetTypeReference((TypeReferenceHandle)constructor.Parent);
                    var name = metadataReader.GetString(type.Name);
 
                    var signature = metadataReader.GetBlobReader(constructor.Signature);
                    var value = metadataReader.GetBlobReader(attribute.Value);
                    var header = signature.ReadSignatureHeader();
 
                    const ushort prolog = 1; // two-byte "prolog" defined by ECMA-335 (II.23.3) to be at the beginning of attribute value blobs
                    if (value.ReadUInt16() != prolog || header.Kind != SignatureKind.Method || header.IsGeneric)
                    {
                        throw new BadImageFormatException();
                    }
 
                    var paramCount = signature.ReadCompressedInteger();
                    if (paramCount <= 0)
                    {
                        parameterlessAttributes.Add(name);
                    }
                }
            }
 
            return parameterlessAttributes;
        }
 
        public static ISet<(string Key, string Value)> Get(string assemblyPath)
        {
            var assemblyInfo = new HashSet<(string, string)>();
 
            using (var stream = File.OpenRead(assemblyPath))
            using (var peReader = new PEReader(stream))
            {
                var metadataReader = peReader.GetMetadataReader();
                var assemblyDefinition = metadataReader.GetAssemblyDefinition();
 
                // AssemblyVersion is not actually a custom attribute
                if (assemblyDefinition.Version != new Version(0, 0, 0, 0))
                {
                    assemblyInfo.Add(("AssemblyVersionAttribute", assemblyDefinition.Version.ToString()));
                }
 
                foreach (var handle in assemblyDefinition.GetCustomAttributes())
                {
                    var attribute = metadataReader.GetCustomAttribute(handle);
                    var constructor = metadataReader.GetMemberReference((MemberReferenceHandle)attribute.Constructor);
                    var type = metadataReader.GetTypeReference((TypeReferenceHandle)constructor.Parent);
                    var name = metadataReader.GetString(type.Name);
 
                    var signature = metadataReader.GetBlobReader(constructor.Signature);
                    var value = metadataReader.GetBlobReader(attribute.Value);
                    var header = signature.ReadSignatureHeader();
 
                    const ushort prolog = 1; // two-byte "prolog" defined by ECMA-335 (II.23.3) to be at the beginning of attribute value blobs
                    if (value.ReadUInt16() != prolog || header.Kind != SignatureKind.Method || header.IsGeneric)
                    {
                        throw new BadImageFormatException();
                    }
 
                    var paramCount = signature.ReadCompressedInteger();
                    if (paramCount <= 0 || // must have at least 1 parameter
                        signature.ReadSignatureTypeCode() != SignatureTypeCode.Void) // return type must be void
                    {
                        continue;
                    }
 
                    var sb = new StringBuilder();
                    while (paramCount > 0 && sb != null)
                    {
                        switch (signature.ReadSignatureTypeCode())
                        {
                            case SignatureTypeCode.String:
                                sb.Append(value.ReadSerializedString());
                                break;
                            default:
                                sb = null;
                                break;
                        }
 
                        paramCount--;
                        if (paramCount != 0)
                        {
                            sb?.Append(':');
                        }
                    }
 
                    if (sb != null)
                    {
                        assemblyInfo.Add((name, sb.ToString()));
                    }
                }
            }
 
            return assemblyInfo;
        }
    }
}