1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using NuGet.Packaging.Core;
using NuGet.Protocol.Core.Types;
namespace NuGet.PackageManagement.UI
{
/// <summary>
/// Package feed facilitating iteration over installed packages needing version consolidation
/// </summary>
internal class ConsolidatePackageFeed : PlainPackageFeedBase
{
IEnumerable<PackageIdentity> _installedPackages;
readonly IPackageMetadataProvider _metadataProvider;
PackageLoadContext _context;
public ConsolidatePackageFeed (
PackageLoadContext context,
IPackageMetadataProvider metadataProvider,
Common.ILogger logger)
: this (new PackageIdentity[0], metadataProvider, logger)
{
_context = context;
}
public ConsolidatePackageFeed (
IEnumerable<PackageIdentity> installedPackages,
IPackageMetadataProvider metadataProvider,
Common.ILogger logger)
{
if (installedPackages == null) {
throw new ArgumentNullException (nameof (installedPackages));
}
_installedPackages = installedPackages;
if (metadataProvider == null) {
throw new ArgumentNullException (nameof (metadataProvider));
}
_metadataProvider = metadataProvider;
if (logger == null) {
throw new ArgumentNullException (nameof (logger));
}
PageSize = 25;
}
public override async Task<SearchResult<IPackageSearchMetadata>> ContinueSearchAsync (ContinuationToken continuationToken, CancellationToken cancellationToken)
{
var searchToken = continuationToken as FeedSearchContinuationToken;
if (searchToken == null) {
throw new InvalidOperationException ("Invalid token");
}
if (_context != null) {
_installedPackages = await _context.GetInstalledPackagesAsync ();
_context = null;
}
var packagesNeedingConsolidation = _installedPackages
.GroupById ()
.Where (g => g.Count () > 1)
.Select (g => new PackageIdentity (g.Key, g.Max ()))
.ToArray ();
var packages = packagesNeedingConsolidation
.Where (p => p.Id.IndexOf (searchToken.SearchString, StringComparison.OrdinalIgnoreCase) != -1)
.OrderBy (p => p.Id)
.Skip (searchToken.StartIndex)
.Take (PageSize + 1)
.ToArray ();
var hasMoreItems = packages.Length > PageSize;
if (hasMoreItems) {
packages = packages.Take (packages.Length - 1).ToArray ();
}
var items = await TaskCombinators.ThrottledAsync (
packages,
(p, t) => _metadataProvider.GetPackageMetadataAsync (p, searchToken.SearchFilter.IncludePrerelease, t),
cancellationToken);
var result = SearchResult.FromItems (items.ToArray ());
var loadingStatus = hasMoreItems
? LoadingStatus.Ready
: packages.Length == 0
? LoadingStatus.NoItemsFound
: LoadingStatus.NoMoreItems;
result.SourceSearchStatus = new Dictionary<string, LoadingStatus> {
{ "Consolidate", loadingStatus }
};
if (hasMoreItems) {
result.NextToken = new FeedSearchContinuationToken {
SearchString = searchToken.SearchString,
SearchFilter = searchToken.SearchFilter,
StartIndex = searchToken.StartIndex + packages.Length
};
}
return result;
}
}
}
|