Migrate AutoComplete

Author: Nigusu-Allehu

src/NuGet.Core/NuGet.Protocol/Model/AutoCompleteModel.cs

@@ -0,0 +1,13 @@
+// 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.Text.Json.Serialization;
+
+namespace NuGet.Protocol.Model
+{
+    internal sealed class AutoCompleteModel
+    {
+        [JsonPropertyName("data")]
+        public string[]? Data { get; set; }
+    }
+}

src/NuGet.Core/NuGet.Protocol/Resources/AutoCompleteResourceV3.cs

@@ -8,10 +8,12 @@
 using System.Globalization;
 using System.Linq;
 using System.Net;
+using System.Text.Json;
 using System.Threading;
 using System.Threading.Tasks;
-using Newtonsoft.Json.Linq;
 using NuGet.Protocol.Core.Types;
+using NuGet.Protocol.Model;
+using NuGet.Protocol.Utility;
 using NuGet.Versioning;
 
 namespace NuGet.Protocol
@@ -55,32 +57,29 @@ public override async Task<IEnumerable<string>> IdStartsWith(
             Common.ILogger logger = log ?? Common.NullLogger.Instance;
 
             var queryUri = queryUrl.Uri;
-            var results = await _client.GetJObjectAsync(
+            AutoCompleteModel results = await _client.ProcessStreamAsync(
                 new HttpSourceRequest(queryUri, logger),
+                async stream =>
+                {
+                    if (stream == null)
+                    {
+                        return null;
+                    }
+
+                    return await JsonSerializer.DeserializeAsync(stream, JsonContext.Default.AutoCompleteModel, token);
+                },
                 logger,
                 token);
+
             token.ThrowIfCancellationRequested();
-            if (results == null)
-            {
-                return Enumerable.Empty<string>();
-            }
-            var data = results.Value<JArray>("data");
-            if (data == null)
-            {
-                return Enumerable.Empty<string>();
-            }
 
-            // Resolve all the objects
-            var outputs = new List<string>();
-            foreach (var result in data)
+            if (results?.Data == null)
             {
-                if (result != null)
-                {
-                    outputs.Add(result.ToString());
-                }
+                return Enumerable.Empty<string>();
             }
 
-            return outputs.Where(item => item.StartsWith(packageIdPrefix, StringComparison.OrdinalIgnoreCase));
+            return results.Data
+                .Where(item => item != null && item.StartsWith(packageIdPrefix, StringComparison.OrdinalIgnoreCase));
         }
 
         public override async Task<IEnumerable<NuGetVersion>> VersionStartsWith(

src/NuGet.Core/NuGet.Protocol/Utility/JsonContext.cs

@@ -17,6 +17,11 @@ namespace NuGet.Protocol.Utility
     [JsonSerializable(typeof(HttpFileSystemBasedFindPackageByIdResource.FlatContainerVersionList))]
     [JsonSerializable(typeof(IReadOnlyList<V3VulnerabilityIndexEntry>), TypeInfoPropertyName = "VulnerabilityIndex")]
     [JsonSerializable(typeof(CaseInsensitiveDictionary<IReadOnlyList<PackageVulnerabilityInfo>>), TypeInfoPropertyName = "VulnerabilityPage")]
+<<<<<<< Updated upstream
+=======
+    [JsonSerializable(typeof(AutoCompleteModel))]
+    [JsonSerializable(typeof(ServiceIndexModel))]
+>>>>>>> Stashed changes
     internal partial class JsonContext : JsonSerializerContext
     {
     }