// 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.Linq;
using NuGet.Client;
using NuGet.ContentModel;
using NuGet.Frameworks;

namespace Microsoft.DotNet.PackageTesting;

public class Package
{
    public IEnumerable<NuGetFramework> FrameworksInPackage { get; }
    public string PackageId { get; }
    public string Version { get; }

    public Package(string packageId, string version, IEnumerable<string> packageAssetPaths, IEnumerable<NuGetFramework> dependencyFrameworks)
    {
        PackageId = packageId;
        Version = version;

        ContentItemCollection packageAssets = new();
        packageAssets.Load(packageAssetPaths);
        ManagedCodeConventions conventions = new(null);

        IEnumerable<ContentItem> RefAssets = packageAssets.FindItems(conventions.Patterns.CompileRefAssemblies);
        IEnumerable<ContentItem> LibAssets = packageAssets.FindItems(conventions.Patterns.CompileLibAssemblies);
        IEnumerable<ContentItem> CompileAssets = RefAssets.Any() ? RefAssets : LibAssets;
        List<NuGetFramework> FrameworksInPackageList = CompileAssets.Select(t => (NuGetFramework)t.Properties["tfm"]).ToList();

        IEnumerable<ContentItem> RuntimeAssets = packageAssets.FindItems(conventions.Patterns.RuntimeAssemblies);
        FrameworksInPackageList.AddRange(RuntimeAssets.Select(t => (NuGetFramework)t.Properties["tfm"]).Distinct());
        FrameworksInPackageList.AddRange(dependencyFrameworks);
        FrameworksInPackage = FrameworksInPackageList.Distinct();
    }
}