XpandPosh

1.9.8

Cmdlets/src/XpandPosh.Cmdlets/Nuget/GetNugetPackageSearchMetadata/Get-NugetPackageSearchMetadata.cs

                                using System;

using System.Collections.Generic;

using System.Linq;

using System.Management.Automation;

using System.Reactive.Linq;

using System.Reactive.Threading.Tasks;

using System.Threading;

using System.Threading.Tasks;

using NuGet.Common;

using NuGet.Configuration;

using NuGet.Protocol;

using NuGet.Protocol.Core.Types;

namespace XpandPosh.Cmdlets.Nuget.GetNugetPackageSearchMetadata{

    [Cmdlet(VerbsCommon.Get, "NugetPackageSearchMetadata")]

    [CmdletBinding()]

    public  class GetNugetPackageSearchMetadata : NugetCmdlet{

        [Parameter(Position = 0, ValueFromPipeline = true)]

        public string Name{ get; set; }

        [Parameter(Position = 1,Mandatory = true)]

        public string Source{ get; set; } 

        [Parameter]

        public SwitchParameter IncludePrerelease{ get; set; } = new SwitchParameter(false);

        [Parameter]

        public SwitchParameter IncludeUnlisted{ get; set; } = new SwitchParameter(false);

        [Parameter]

        public SwitchParameter AllVersions{ get; set; } = new SwitchParameter(false);

        [Parameter]

        public string[] Versions{ get; set; }

        protected override async Task ProcessRecordAsync(){

            var providers = new List<Lazy<INuGetResourceProvider>>();

            providers.AddRange(Repository.Provider.GetCoreV3());

            var metaData = ListPackages(providers)

                .SelectMany(package => SelectPackages(package, providers))

                .Where(metadata => metadata!=null)

                .HandleErrors(this)

                .Distinct(new MetadataComparer()).Replay().AutoConnect();

            var searchMetadata = await metaData.DefaultIfEmpty();

            if (searchMetadata==null)

                return;

            var packageSearchMetadatas = metaData.ToEnumerable().ToArray()

                .OrderByDescending(_ => _.Identity.Version.Version).ToArray();

            if (!AllVersions && Versions == null){

                WriteObject(packageSearchMetadatas.OrderByDescending(metadata => metadata.Identity.Version.Version).FirstOrDefault());

                return;

            }

            if (Versions==null)

                Versions=new string[0];

            foreach (var packageSearchMetadata in packageSearchMetadatas.Where(VersionMatch)){

                WriteObject(packageSearchMetadata);

            }

        }

        private bool VersionMatch(IPackageSearchMetadata metadata){

            if (Versions == null)

                return true;

            return Versions.Contains(metadata.Identity.Version.ToString());

        }

        public class MetadataComparer : IEqualityComparer<IPackageSearchMetadata>{

            public bool Equals(IPackageSearchMetadata x, IPackageSearchMetadata y){

                return x?.Identity.Id == y?.Identity.Id && x?.Identity.Version.Version==y?.Identity.Version.Version;

            }

            public int GetHashCode(IPackageSearchMetadata obj){

                return $"{obj.Identity.Id}{obj.Identity.Version.Version}".GetHashCode();

            }

        }

        private IObservable<IPackageSearchMetadata> SelectPackages(string s, List<Lazy<INuGetResourceProvider>> providers){

            return Source.Split(';').ToObservable().SelectMany(source => PackageSourceSearchMetadatas(source, s,providers));

        }

        private IObservable<string> ListPackages(List<Lazy<INuGetResourceProvider>> providers){

            if (Name != null) return Observable.Return(Name);

            return Source.Split(';').Select(source => ListPackages(providers, source)).Merge();

        } 

        private static IObservable<string> ListPackages(List<Lazy<INuGetResourceProvider>> providers, string source){

            var sourceRepository = new SourceRepository(new PackageSource(source), providers);

            return sourceRepository.GetResourceAsync<ListResource>().ToObservable()

                .Select(resource =>resource.ListAsync(null, false, false, false, NullLogger.Instance, CancellationToken.None).ToObservable()).Concat()

                .Select(async => async.GetEnumeratorAsync().ToObservable()).Concat()

                .Where(metadata => metadata != null)

                .Select(metadata => metadata.Identity.Id);

        }

        private IObservable<IPackageSearchMetadata> PackageSourceSearchMetadatas(string source,string name,

            List<Lazy<INuGetResourceProvider>> providers){

            var metadatas = new SourceRepository(new PackageSource(source), providers)

                .GetResourceAsync<PackageMetadataResource>().ToObservable()

                .SelectMany(resource => resource

                    .GetMetadataAsync(name, IncludePrerelease, IncludeUnlisted, NullLogger.Instance,CancellationToken.None).ToObservable()

                    .SelectMany(enumerable => enumerable.ToArray())

                );

            return metadatas;

        }

    }

}