XpandPosh

1.14.0

Cmdlets/src/XpandPosh.Cmdlets/Nuget/Get-NugetPackageDownloadsCount/Get-NugetPackageDownloadsCount.cs

                                using System;

using System.Collections.Concurrent;

using System.Collections.Generic;

using System.Management.Automation;

using System.Net;

using System.Reactive.Linq;

using System.Reactive.Threading.Tasks;

using System.Threading;

using System.Threading.Tasks;

using Newtonsoft.Json;

using XpandPosh.CmdLets;

namespace XpandPosh.Cmdlets.Nuget{

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

    [CmdletBinding]

    public class GetNugetPackageDownloadsCount : XpandCmdlet{

        private ConcurrentBag<string> _packages;

        [Parameter(ValueFromPipeline = true, Mandatory = true)]

        public string Id{ get; set; }

        [Parameter(Mandatory = true)]

        public string Source{ get; set; }

        protected override Task BeginProcessingAsync(){

            _packages = new ConcurrentBag<string>();

            return base.BeginProcessingAsync();

        }

        protected override Task ProcessRecordAsync(){

            _packages.Add(Id);

            return Task.CompletedTask;

        }

        protected override async Task EndProcessingAsync(CancellationToken cancellationToken){

            var concurrentBag = _packages;

            var downloads = Downloads(concurrentBag);

            var sum = await downloads;

            WriteObject(sum);

        }

        public static IObservable<int> Downloads(IEnumerable<string> concurrentBag){

            var downloads = concurrentBag.ToObservable()

                .SelectMany(id => Observable.Using(() => new WebClient(), client => client

                    .DownloadStringTaskAsync(

                        $"https://api-v2v3search-0.nuget.org/query?q=packageid:{id}").ToObservable()))

                .Sum(json => (int) JsonConvert.DeserializeObject<dynamic>(json).data[0].totalDownloads);

            return downloads;

        }

    }

}