From 1599720f10fadcaff0441072bcf02c161799f4f3 Mon Sep 17 00:00:00 2001
From: Ian Griffiths <Ian.Griffiths@endjin.com>
Date: Wed, 18 Jan 2023 11:38:37 +0000
Subject: [PATCH] Register IServiceIdentityAzureTokenCredentialSource in DI

This is required for certain V2 backwards compatibility scenarios
---
 .../Marain/Tenancy/Functions/Startup.cs               | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/Solutions/Marain.Tenancy.Host.Functions/Marain/Tenancy/Functions/Startup.cs b/Solutions/Marain.Tenancy.Host.Functions/Marain/Tenancy/Functions/Startup.cs
index 6d72f948..05e7aff0 100644
--- a/Solutions/Marain.Tenancy.Host.Functions/Marain/Tenancy/Functions/Startup.cs
+++ b/Solutions/Marain.Tenancy.Host.Functions/Marain/Tenancy/Functions/Startup.cs
@@ -8,6 +8,9 @@ namespace Marain.Tenancy.ControlHost
 {
     using System;
 
+    using Azure.Identity;
+
+    using Corvus.Identity.ClientAuthentication.Azure;
     using Corvus.Storage.Azure.BlobStorage;
 
     using Menes;
@@ -36,6 +39,14 @@ public override void Configure(IFunctionsHostBuilder builder)
                 .Get<BlobContainerConfiguration>();
             services.AddTenantStoreOnAzureBlobStorage(rootStorageConfiguration);
             services.AddTenancyApiWithOpenApiActionResultHosting(this.ConfigureOpenApiHost);
+            if (rootStorageConfiguration?.AccessKeyInKeyVault?.VaultClientIdentity is ClientIdentityConfiguration idconfig)
+            {
+                services.AddServiceIdentityAzureTokenCredentialSourceFromClientIdentityConfiguration(idconfig);
+            }
+            else
+            {
+                services.AddServiceIdentityAzureTokenCredentialSourceFromAzureCoreTokenCredential(new DefaultAzureCredential());
+            }
         }
 
         private void ConfigureOpenApiHost(IOpenApiHostConfiguration config)