Add support for URIs that contain an API key.

Author: brdeyo

VERSION

@@ -1 +1 @@
-3.1.0
+3.1.1

src/VirtualClient/VirtualClient.Core/EndpointUtility.cs

@@ -260,6 +260,24 @@ public static DependencyProfileReference CreateProfileReference(string endpoint,
             return profile;
         }
 
+        /// <summary>
+        /// Returns true/false whether the value is a URI containing an 'api-key' query string parameter.
+        /// </summary>
+        /// <param name="value">The value to evaluate.</param>
+        /// <returns>True if the value is a URI containing an 'api-key' query string parameter. False if not.</returns>
+        public static bool IsApiKeyUri(string value)
+        {
+            if (Uri.TryCreate(value, UriKind.Absolute, out Uri uri))
+            {
+                if (!string.IsNullOrWhiteSpace(uri.Query) && Regex.IsMatch(uri.Query, "api-key"))
+                {
+                    return true;
+                }
+            }
+
+            return false;
+        }
+
         /// <summary>
         /// Returns true/false whether the value is a custom Virtual Client connection string 
         /// (e.g. EndpointUrl=https://any.blob.core.windows.net;ManagedIdentityId=307591a4-abb2-4559-af59-b47177d140cf).

src/VirtualClient/VirtualClient.Main/OptionFactory.cs

@@ -1699,6 +1699,10 @@ private static DependencyStore ParseBlobStore(ArgumentResult parsedResult, strin
             {
                 store = new DependencyFileStore(storeName, Path.GetFullPath(endpoint));
             }
+            else if (EndpointUtility.IsApiKeyUri(endpoint))
+            {
+                store = new DependencyBlobStore(DependencyBlobStore.Packages, new Uri(endpoint));
+            }
             else
             {
                 store = EndpointUtility.CreateBlobStoreReference(storeName, endpoint, certificateManager);

src/VirtualClient/VirtualClient.UnitTests/OptionFactoryTests.cs

@@ -1258,6 +1258,17 @@ public void PackageStoreOptionSupportsUrisWithMicrosoftEntraIdAndCertificateRefe
             Assert.IsFalse(result.Errors.Any());
         }
 
+        [Test]
+        [TestCase("https://any.azure-api.net?api-key=12345")]
+        [TestCase("https://any.azure-api.net/packages-other?api-key=12345")]
+        public void PackageStoreOptionSupportsUrisWithApiKeys(string uri)
+        {
+            Option option = OptionFactory.CreatePackageStoreOption();
+            ParseResult result = option.Parse($"--package-store={uri}");
+            Assert.IsFalse(result.Errors.Any());
+            Assert.AreEqual(uri.ToString(), result.Tokens[1].Value);
+        }
+
         [Test]
         public void PackageStoreOptionValidatesTheConnectionTokenProvided()
         {