Allow option customization for keyvault configuration provider

gsoft-inc · Aug 19, 2022 · acf57e3 · acf57e3
1 parent f775567
commit acf57e3
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/ShareGate.Extensions.Configuration.Secrets/ServiceCollectionExtensions.cs b/src/ShareGate.Extensions.Configuration.Secrets/ServiceCollectionExtensions.cs
@@ -17,21 +17,25 @@ public static IServiceCollection AddKeyVaultSecrets(this IServiceCollection serv
         return services;
     }
 
-    public static IConfigurationBuilder AddKeyVaultSecrets(this IConfigurationBuilder builder, IHostEnvironment environment, KeyVaultKind keyVaultKind = KeyVaultKind.ApplicationConfiguration)
+    public static IConfigurationBuilder AddKeyVaultSecrets(
+        this IConfigurationBuilder builder,
+        IHostEnvironment environment,
+        AzureKeyVaultConfigurationOptions? options = null,
+        KeyVaultKind keyVaultKind = KeyVaultKind.ApplicationConfiguration)
     {
         var secretClient = new SecretClientProvider(builder, environment).GetSecretClient(keyVaultKind);
-        return builder.AddAzureKeyVault(secretClient, new KeyVaultSecretManager());
+        return builder.AddAzureKeyVault(secretClient, options ?? new AzureKeyVaultConfigurationOptions());
     }
 
-    public static IConfigurationBuilder AddKeyVaultSecrets(this IConfigurationBuilder builder, IHostEnvironment environment, Uri keyVaultUri)
+    public static IConfigurationBuilder AddKeyVaultSecrets(this IConfigurationBuilder builder, IHostEnvironment environment, Uri keyVaultUri, AzureKeyVaultConfigurationOptions? options = null)
     {
         var secretClient = new SecretClientProvider(builder, environment).GetSecretClient(keyVaultUri);
-        return builder.AddAzureKeyVault(secretClient, new KeyVaultSecretManager());
+        return builder.AddAzureKeyVault(secretClient, options ?? new AzureKeyVaultConfigurationOptions());
     }
 
-    public static IConfigurationBuilder AddKeyVaultSecrets(this IConfigurationBuilder builder, IHostEnvironment environment, string configurationKey)
+    public static IConfigurationBuilder AddKeyVaultSecrets(this IConfigurationBuilder builder, IHostEnvironment environment, string configurationKey, AzureKeyVaultConfigurationOptions? options = null)
     {
         var secretClient = new SecretClientProvider(builder, environment).GetSecretClient(configurationKey);
-        return builder.AddAzureKeyVault(secretClient, new KeyVaultSecretManager());
+        return builder.AddAzureKeyVault(secretClient, options ?? new AzureKeyVaultConfigurationOptions());
     }
 }