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

using System.Collections.Generic;
using System.IO;
using System.Linq;
using Microsoft.Build.Framework;
using Microsoft.Build.Utilities;
using Microsoft.DotNet.Build.Tasks.Installers;

namespace Microsoft.DotNet.Build.Tasks.Workloads.Wix
{
    /// <summary>
    /// Tool task for invoking the WiX CLI (version 5 and above). This commands is responsible for compiling and linking.
    /// </summary>
    public class WixToolTask : WixToolTaskBase
    {
        private List<string> _sourceFiles = new();
        private List<string> _preprocessorDefinitions = new();

        private WixToolsetConfiguration _wixToolsetConfig;

        /// <summary>
        /// The default architecture used for packages, components, etc.
        /// </summary>
        public string Architecture
        {
            get;
            set;
        } = "x86";

        /// <summary>
        /// The file system path of the MSI to generate.
        /// </summary>
        public string OutputPath
        {
            get;
            set;
        }

        /// <summary>
        /// All the source files to compile and link the MSI.
        /// </summary>
        public string[] SourceFiles => _sourceFiles.ToArray();

        /// <summary>
        /// Creates a new <see cref="WixToolTask"/> instance that can be used to invoke the WiX CLI to build an MSI.
        /// </summary>
        /// <param name="engine">The build engine interface to use.</param>
        /// <param name="wixToolsetConfig">Configuration information related to the WiX toolset.</param>
        public WixToolTask(IBuildEngine engine, WixToolsetConfiguration wixToolsetConfig) :
            base(engine, wixToolsetConfig.CliPath)
        {
            _wixToolsetConfig = wixToolsetConfig;
        }

        /// <summary>
        /// Adds one or more source file to compile.
        /// </summary>
        /// <param name="sourceFiles">The set of source files to compile.</param>
        public void AddSourceFiles(params string[] sourceFiles)
        {
            foreach (string sourceFile in sourceFiles)
            {
                _sourceFiles.Add(sourceFile);
            }
        }

        /// <summary>
        /// Adds a list of source files to compile.
        /// </summary>
        /// <param name="sourceFiles">The source files to compile.</param>
        public void AddSourceFiles(IEnumerable<string> sourceFiles) =>
            _sourceFiles.AddRange(sourceFiles);

        /// <summary>
        /// Adds a preprocessor variable definition that will be passed to WiX.
        /// </summary>
        /// <param name="name">The name of the preprocessor variable.</param>
        /// <param name="value">The value of the preprocessor variable.</param>
        public void AddPreprocessorDefinition(string name, string value) =>
            _preprocessorDefinitions.Add($"{name}={value}");

        protected override string GenerateCommandLineCommands()
        {
            // Primary command to execute.
            CommandLineBuilder.AppendSwitch("build");

            CommandLineBuilder.AppendSwitchIfNotNull("-arch ", Architecture);
            // Ensure v3 backward compatible GUIDs. v5 introduced a breaking change when generating
            // GUIDs for components where a registry key is the keypath. 
            CommandLineBuilder.AppendSwitch("-bcgg");
            CommandLineBuilder.AppendArrayIfNotNull("-d ", _preprocessorDefinitions.ToArray());
            CommandLineBuilder.AppendArrayIfNotNull("-ext ", _wixToolsetConfig.Extensions.ToArray());
            CommandLineBuilder.AppendFileNamesIfNotNull(_sourceFiles.ToArray(), " ");
            CommandLineBuilder.AppendSwitchIfNotNull("-out ", OutputPath);

            return CommandLineBuilder.ToString();
        }

        public string GetWixCommandLine() => Path.Combine(ToolPath, ToolExe) + " " +
            CommandLineBuilder.ToString();

        /// <summary>
        /// Gets a new instance of the <see cref="CreateWixBuildWixpack"/> task that can be used to create a wixpack to sign
        /// the installer.
        /// </summary>
        /// <returns>A new instance of the <see cref="CreateWixBuildWixpack"/> task.</returns>
        public CreateWixBuildWixpack GetCreateWixBuildWixpackTask(IBuildEngine buildEngine, string intermediateOutputPath,
            string outputPath, string installerFile, string workingDirectory) =>
            new CreateWixBuildWixpack
            {
                AdditionalOptions = "-bcgg",
                BuildEngine = buildEngine,
                Cultures = ["en-us"],
                DefineConstants = [.. _preprocessorDefinitions.ToArray()],
                Extensions = [.. _wixToolsetConfig.Extensions.Select(e => new TaskItem(e))],
                InstallerFile = installerFile,
                InstallerPlatform = Architecture,
                IntermediateDirectory = new TaskItem(intermediateOutputPath),
                OutputFolder = outputPath,
                SourceFiles = [.. _sourceFiles.Select(s => new TaskItem(s))],
                WixpackWorkingDir = workingDirectory
            };
    }
}