feat: add metadata provider

sitkoru · Mar 26, 2024 · e4db75c · e4db75c
1 parent 4522f7d
commit e4db75c
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs b/src/Sitko.Core.Grpc.Client/GrpcClientModuleOptions.cs
@@ -29,6 +29,34 @@ internal void ConfigureClient(IServiceCollection services, IHttpClientBuilder bu
         configureAuthAction?.Invoke(services, builder);
     }
 
+    public GrpcClientModuleOptions<TClient> AddMetadataProvider<TMetadataProvider>()
+        where TMetadataProvider : class, IGrpcMetadataProvider
+    {
+        configureAuthAction = (services, builder) =>
+        {
+            services.TryAddTransient<IGrpcMetadataProvider, TMetadataProvider>();
+            builder.AddCallCredentials(async (_, metadata, serviceProvider) =>
+            {
+                var provider = serviceProvider.GetRequiredService<TMetadataProvider>();
+                var newMetadata = await provider.GetMetadataAsync();
+                if (newMetadata.Count != 0)
+                {
+                    foreach (var (key, value) in newMetadata)
+                    {
+                        var oldEntries = metadata.GetAll(key);
+                        foreach (var entry in oldEntries)
+                        {
+                            metadata.Remove(entry);
+                        }
+
+                        metadata.Add(key, value);
+                    }
+                }
+            });
+        };
+        return this;
+    }
+
     public GrpcClientModuleOptions<TClient> AddTokenAuth<TTokenProvider>()
         where TTokenProvider : class, IGrpcTokenProvider
     {

diff --git a/src/Sitko.Core.Grpc.Client/IGrpcMetadataProvider.cs b/src/Sitko.Core.Grpc.Client/IGrpcMetadataProvider.cs
@@ -0,0 +1,6 @@
+namespace Sitko.Core.Grpc.Client;
+
+public interface IGrpcMetadataProvider
+{
+    Task<Dictionary<string, string>> GetMetadataAsync();
+}