// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

#nullable disable

using System.Xml;
using Microsoft.Build.Framework;

namespace Microsoft.NET.Sdk.BlazorWebAssembly
{
    // Based on https://github.com/mono/linker/blob/3b329b9481e300bcf4fb88a2eebf8cb5ef8b323b/src/ILLink.Tasks/CreateRootDescriptorFile.cs
    public class CreateBlazorTrimmerRootDescriptorFile : Task
    {
        [Required]
        public ITaskItem[] Assemblies { get; set; }

        [Required]
        public ITaskItem TrimmerFile { get; set; }

        public override bool Execute()
        {
            var rootDescriptor = CreateRootDescriptorContents();
            if (File.Exists(TrimmerFile.ItemSpec))
            {
                var existing = File.ReadAllText(TrimmerFile.ItemSpec);

                if (string.Equals(rootDescriptor, existing, StringComparison.Ordinal))
                {
                    Log.LogMessage(MessageImportance.Low, "Skipping write to file {0} because contents would not change.", TrimmerFile.ItemSpec);
                    // Avoid writing if the file contents are identical. This is required for build incrementalism.
                    return !Log.HasLoggedErrors;
                }
            }

            File.WriteAllText(TrimmerFile.ItemSpec, rootDescriptor);
            return !Log.HasLoggedErrors;
        }

        internal string CreateRootDescriptorContents()
        {
            var roots = new XElement("linker");
            foreach (var assembly in Assemblies.OrderBy(a => a.ItemSpec))
            {
                // NOTE: Descriptor files don't include the file extension
                // in the assemblyName.
                var assemblyName = assembly.GetMetadata("FileName");
                var typePreserved = assembly.GetMetadata("Preserve");
                var typeRequired = assembly.GetMetadata("Required");

                var attributes = new List<XAttribute>
                {
                    new XAttribute("fullname", "*"),
                    new XAttribute("required", typeRequired),
                };

                if (!string.IsNullOrEmpty(typePreserved))
                {
                    attributes.Add(new XAttribute("preserve", typePreserved));
                }

                roots.Add(new XElement("assembly",
                    new XAttribute("fullname", assemblyName),
                    new XElement("type", attributes)));
            }

            var xmlWriterSettings = new XmlWriterSettings
            {
                Indent = true,
                OmitXmlDeclaration = true
            };

            return new XDocument(roots).Root.ToString();
        }
    }
}