// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System;
using System.Collections.Generic;
using System.Linq;
using Roslyn.Utilities;
using MSB = Microsoft.Build;

namespace Microsoft.CodeAnalysis.MSBuild;

internal static class Extensions
{
    public static IEnumerable<MSB.Framework.ITaskItem> GetAdditionalFiles(this MSB.Execution.ProjectInstance executedProject)
        => executedProject.GetItems(ItemNames.AdditionalFiles);

    public static IEnumerable<MSB.Framework.ITaskItem> GetAnalyzers(this MSB.Execution.ProjectInstance executedProject)
        => executedProject.GetItems(ItemNames.Analyzer);

    public static IEnumerable<MSB.Framework.ITaskItem> GetDocuments(this MSB.Execution.ProjectInstance executedProject)
        => executedProject.GetItems(ItemNames.Compile);

    public static IEnumerable<MSB.Framework.ITaskItem> GetEditorConfigFiles(this MSB.Execution.ProjectInstance executedProject)
        => executedProject.GetItems(ItemNames.EditorConfigFiles);

    public static IEnumerable<MetadataReferenceItem> GetMetadataReferences(this MSB.Execution.ProjectInstance project)
    {
        foreach (var item in project.GetItems(ItemNames.ReferencePath))
        {
            if (item.ReferenceOutputAssemblyIsTrue())
            {
                yield return new(project.GetAbsolutePath(item.EvaluatedInclude), item.GetAliases());
            }
        }
    }

    public static string GetAbsolutePath(this MSB.Execution.ProjectInstance project, string path)
    {
        var baseDirectory = PathUtilities.GetDirectoryName(project.FullPath);
        var absolutePath = FileUtilities.ResolveRelativePath(path, baseDirectory) ?? path;
        return FileUtilities.TryNormalizeAbsolutePath(absolutePath) ?? absolutePath;
    }

    public static IEnumerable<ProjectFileReference> GetProjectReferences(this MSB.Execution.ProjectInstance executedProject)
        => executedProject
            .GetItems(ItemNames.ProjectReference)
            .Select(CreateProjectFileReference);

    public static PackageReferenceItem[] GetPackageReferences(this MSB.Execution.ProjectInstance executedProject)
    {
        var packageReferenceItems = executedProject.GetItems(ItemNames.PackageReference);
        var references = new HashSet<PackageReferenceItem>(capacity: packageReferenceItems.Count);

        foreach (var item in packageReferenceItems)
        {
            var name = item.EvaluatedInclude;
            var versionRangeValue = item.GetMetadataValue(MetadataNames.Version);
            var packageReference = new PackageReferenceItem(name, versionRangeValue);
            references.Add(packageReference);
        }

        return [.. references];
    }

    /// <summary>
    /// Create a <see cref="ProjectFileReference"/> from a ProjectReference node in the MSBuild file.
    /// </summary>
    private static ProjectFileReference CreateProjectFileReference(MSB.Execution.ProjectItemInstance reference)
        => new(reference.EvaluatedInclude, reference.GetAliases(), reference.ReferenceOutputAssemblyIsTrue());

    public static string[] GetAliases(this MSB.Framework.ITaskItem item)
    {
        var aliasesText = item.GetMetadata(MetadataNames.Aliases);

        return !string.IsNullOrWhiteSpace(aliasesText)
            ? [.. aliasesText.Split([','], StringSplitOptions.RemoveEmptyEntries).Select(a => a.Trim())]
            : [];
    }

    public static bool ReferenceOutputAssemblyIsTrue(this MSB.Framework.ITaskItem item)
    {
        var referenceOutputAssemblyText = item.GetMetadata(MetadataNames.ReferenceOutputAssembly);

        return string.IsNullOrWhiteSpace(referenceOutputAssemblyText) ||
            !string.Equals(referenceOutputAssemblyText, bool.FalseString, StringComparison.OrdinalIgnoreCase);
    }

    public static string? ReadPropertyString(this MSB.Execution.ProjectInstance executedProject, string propertyName)
        => executedProject.GetProperty(propertyName)?.EvaluatedValue;

    public static bool ReadPropertyBool(this MSB.Execution.ProjectInstance executedProject, string propertyName)
        => Conversions.ToBool(executedProject.ReadPropertyString(propertyName));

    public static int ReadPropertyInt(this MSB.Execution.ProjectInstance executedProject, string propertyName)
        => Conversions.ToInt(executedProject.ReadPropertyString(propertyName));

    public static int ReadCodePage(this MSB.Execution.ProjectInstance executedProject)
        => executedProject.ReadPropertyInt(PropertyNames.CodePage) is >= 0 and var codePage ? codePage : 0;

    public static ulong ReadPropertyULong(this MSB.Execution.ProjectInstance executedProject, string propertyName)
        => Conversions.ToULong(executedProject.ReadPropertyString(propertyName));

    public static TEnum? ReadPropertyEnum<TEnum>(this MSB.Execution.ProjectInstance executedProject, string propertyName, bool ignoreCase)
        where TEnum : struct
        => Conversions.ToEnum<TEnum>(executedProject.ReadPropertyString(propertyName), ignoreCase);

    public static string ReadItemsAsString(this MSB.Execution.ProjectInstance executedProject, string itemType)
    {
        var items = executedProject.GetItems(itemType);

        if (items.Count == 0)
            return "";

        if (items.Count == 1)
            return items.Single().EvaluatedInclude;

        return string.Join(" ", items.Select(static i => i.EvaluatedInclude));
    }

    public static IEnumerable<MSB.Framework.ITaskItem> GetTaskItems(this MSB.Execution.ProjectInstance executedProject, string itemType)
        => executedProject.GetItems(itemType);
}