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

using Microsoft.Build.Framework;
using Microsoft.Build.Utilities;
using NuGet.Packaging.Core;
using NuGet.Versioning;

namespace Microsoft.NET.Build.Tasks
{
    /// <summary>
    /// Removes Duplicate Package References.
    /// </summary>
    public class RemoveDuplicatePackageReferences : TaskBase
    {
        private readonly List<ITaskItem> _packageList = new List<ITaskItem>();

        [Required]
        public ITaskItem[] InputPackageReferences { get; set; }

        /// <summary>
        /// Unique package references
        /// </summary>

        [Output]
        public ITaskItem[] UniquePackageReferences
        {
            get { return _packageList.ToArray(); }
        }

        protected override void ExecuteCore()
        {
            var packageSet = new HashSet<PackageIdentity>();

            foreach(var pkg in InputPackageReferences)
            {
                var pkgName = pkg.ItemSpec;
                var pkgVersion = NuGetVersion.Parse(pkg.GetMetadata("Version"));
                packageSet.Add(new PackageIdentity(pkgName, pkgVersion));
            }

            foreach (var pkg in packageSet)
            {
                TaskItem item = new TaskItem(pkg.Id);
                item.SetMetadata("Version", pkg.Version.ToString());
                _packageList.Add(item);
            }
        }
    }
}
