add HttpClient utility to set user agent

Signed-off-by: Sajay Antony <sajaya@microsoft.com>
oras-project · Dec 12, 2023 · f1634b8 · f1634b8
1 parent 4c473a9
commit f1634b8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/Oras/Remote/Auth/HttpClientWithBasicAuth.cs b/Oras/Remote/Auth/HttpClientWithBasicAuth.cs
@@ -12,14 +12,14 @@ public class HttpClientWithBasicAuth : HttpClient
     {
         public HttpClientWithBasicAuth(string username, string password)
         {
-            DefaultRequestHeaders.Add("User-Agent", new string[] { "oras-dotnet" });
+            this.AddUserAgent();
             DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Basic",
                 Convert.ToBase64String(Encoding.UTF8.GetBytes($"{username}:{password}")));
         }
 
         public HttpClientWithBasicAuth(string username, string password, HttpMessageHandler handler) : base(handler)
         {
-            DefaultRequestHeaders.Add("User-Agent", new string[] { "oras-dotnet" });
+            this.AddUserAgent();
             DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Basic",
                 Convert.ToBase64String(Encoding.UTF8.GetBytes($"{username}:{password}")));
         }

diff --git a/Oras/Remote/HttpClientExtensions.cs b/Oras/Remote/HttpClientExtensions.cs
@@ -0,0 +1,12 @@
+using System.Net.Http;
+
+namespace Oras.Remote
+{
+    internal static class HttpClientExtensions
+    {
+        public static void AddUserAgent(this HttpClient client)
+        {
+            client.DefaultRequestHeaders.Add("User-Agent", new string[] { "oras-dotnet" });
+        }
+    }
+}
diff --git a/Oras/Remote/Registry.cs b/Oras/Remote/Registry.cs
@@ -28,7 +28,7 @@ public Registry(string name)
             reference.ValidateRegistry();
             RemoteReference = reference;
             HttpClient = new HttpClient();
-            HttpClient.DefaultRequestHeaders.Add("User-Agent", new string[] { "oras-dotnet" });
+            HttpClient.AddUserAgent();
         }
 
         public Registry(string name, HttpClient httpClient)