// 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;

public class ArtifactSet
{
    public List<PackageArtifactModel> Packages { get; set; } = new List<PackageArtifactModel>();

    public List<BlobArtifactModel> Blobs { get; set; } = new List<BlobArtifactModel>();

    public List<PdbArtifactModel> Pdbs { get; set; } = new List<PdbArtifactModel>();

    public void Add(ArtifactSet source)
    {
        Packages.AddRange(source.Packages);
        Blobs.AddRange(source.Blobs);
        Pdbs.AddRange(source.Pdbs);
    }

    public IEnumerable<XElement> ToXml() => Packages
            .OrderBy(p => p.Id, StringComparer.OrdinalIgnoreCase)
            .ThenBy(p => p.Version, StringComparer.OrdinalIgnoreCase)
            .Select(p => p.ToXml())
        .Concat(Blobs
            .OrderBy(b => b.Id, StringComparer.OrdinalIgnoreCase)
            .Select(b => b.ToXml()))
        .Concat(Pdbs
            .OrderBy(b => b.Id, StringComparer.OrdinalIgnoreCase)
            .Select(b => b.ToXml()));

    public static ArtifactSet Parse(XElement xml) => new ArtifactSet
    {
        Packages = xml.Elements("Package").Select(PackageArtifactModel.Parse).ToList(),
        Blobs = xml.Elements("Blob").Select(BlobArtifactModel.Parse).ToList(),
        Pdbs = xml.Elements("Pdb").Select(PdbArtifactModel.Parse).ToList(),
    };
}