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

using System;
using System.IO;
using Microsoft.Build.Framework;
using BuildTask = Microsoft.Build.Utilities.Task;

namespace Microsoft.DotNet.Build.Tasks.FileCatalog
{
    /// <summary>
    /// Generates a Windows catalog (.cat) file covering a set of files, in pure managed code
    /// (no <c>makecat.exe</c> / Windows SDK required). The catalog is unsigned and ready to be
    /// Authenticode-signed by the Arcade signing infrastructure (via <c>FileExtensionSignInfo</c>).
    /// </summary>
    public class GenerateFileCatalog : BuildTask
    {
        /// <summary>The files to include in the catalog.</summary>
        [Required]
        public ITaskItem[] Files { get; set; } = Array.Empty<ITaskItem>();

        /// <summary>Full path of the .cat file to write.</summary>
        [Required]
        public string OutputPath { get; set; } = string.Empty;

        public override bool Execute()
        {
            if (string.IsNullOrEmpty(OutputPath))
            {
                Log.LogError("OutputPath must be specified.");
                return false;
            }

            if (Files is null || Files.Length == 0)
            {
                Log.LogWarning("No files were provided - skipping catalog generation for '{0}'.", OutputPath);

                // Remove any stale catalog from a previous run so incremental builds don't
                // package a catalog describing files that are no longer present.
                if (File.Exists(OutputPath))
                {
                    File.Delete(OutputPath);
                }

                return true;
            }

            var builder = new CatalogBuilder();
            foreach (ITaskItem file in Files)
            {
                // Prefer the computed FullPath, but fall back to ItemSpec for custom ITaskItem
                // implementations that don't populate FullPath metadata.
                string path = file.GetMetadata("FullPath");
                if (string.IsNullOrEmpty(path))
                {
                    path = file.ItemSpec;
                }

                if (!File.Exists(path))
                {
                    Log.LogError("File not found: '{0}'.", path);
                    return false;
                }

                builder.AddFile(path);
            }

            string? directory = Path.GetDirectoryName(OutputPath);
            if (!string.IsNullOrEmpty(directory))
            {
                Directory.CreateDirectory(directory);
            }

            builder.WriteTo(OutputPath);
            Log.LogMessage(MessageImportance.High, "Generated catalog with {0} file(s): {1}", Files.Length, OutputPath);
            return !Log.HasLoggedErrors;
        }
    }
}