// 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.Collections.Generic;
using System.Linq;
using System.Xml.Linq;

namespace Microsoft.DotNet.Build.Manifest;

internal static class XElementParsingExtensions
{
    public static string GetRequiredAttribute(this XElement element, XName name)
    {
        string value = element.Attribute(name)?.Value;
        if (string.IsNullOrEmpty(value))
        {
            throw new ArgumentException(
                $"Required attribute '{name}' is null or empty on '{element}'");
        }
        return value;
    }

    public static Dictionary<string, string> CreateAttributeDictionary(this XElement element)
    {
        return element.Attributes().ToDictionary(a => a.Name.LocalName, a => a.Value);
    }

    public static XAttribute[] CreateXmlAttributes(
        this IDictionary<string, string> attributes,
        string[] keySortOrder)
    {
        return attributes
            .Where(pair => pair.Value != null)
            .OrderBy(pair => keySortOrder.TakeWhile(o => pair.Key != o).Count())
            .ThenBy(pair => pair.Key, StringComparer.OrdinalIgnoreCase)
            .Select(pair => new XAttribute(pair.Key, pair.Value))
            .ToArray();
    }

    public static IDictionary<string, string> ThrowIfMissingAttributes(
        this IDictionary<string, string> attributes,
        IEnumerable<string> requiredAttributes)
    {
        var missing = requiredAttributes?.Where(r => !attributes.ContainsKey(r)).ToArray();
        if (missing?.Any() == true)
        {
            throw new ArgumentException(
                $"Required attribute(s) missing: {string.Join(", ", missing)}");
        }

        return attributes;
    }
}