From 55e0f82139602360244656a651c709076bd6a9e8 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kspearrin@users.noreply.github.com>
Date: Tue, 8 Sep 2020 13:57:52 -0400
Subject: [PATCH] use custom DiscoveryResponseGenerator on cloud too (#921)

---
 .../src/Sso/Utilities/ServiceCollectionExtensions.cs         | 5 +----
 src/Identity/Utilities/ServiceCollectionExtensions.cs        | 5 +----
 2 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/bitwarden_license/src/Sso/Utilities/ServiceCollectionExtensions.cs b/bitwarden_license/src/Sso/Utilities/ServiceCollectionExtensions.cs
index cfb971be03..12a438bb1e 100644
--- a/bitwarden_license/src/Sso/Utilities/ServiceCollectionExtensions.cs
+++ b/bitwarden_license/src/Sso/Utilities/ServiceCollectionExtensions.cs
@@ -46,10 +46,7 @@ namespace Bit.Sso.Utilities
         public static IIdentityServerBuilder AddSsoIdentityServerServices(this IServiceCollection services,
             IWebHostEnvironment env, GlobalSettings globalSettings)
         {
-            if (globalSettings.SelfHosted)
-            {
-                services.AddTransient<IDiscoveryResponseGenerator, DiscoveryResponseGenerator>();
-            }
+            services.AddTransient<IDiscoveryResponseGenerator, DiscoveryResponseGenerator>();
 
             var issuerUri = new Uri(globalSettings.BaseServiceUri.InternalSso);
             var identityServerBuilder = services
diff --git a/src/Identity/Utilities/ServiceCollectionExtensions.cs b/src/Identity/Utilities/ServiceCollectionExtensions.cs
index 2c1bb22d8a..23e8134c4e 100644
--- a/src/Identity/Utilities/ServiceCollectionExtensions.cs
+++ b/src/Identity/Utilities/ServiceCollectionExtensions.cs
@@ -16,10 +16,7 @@ namespace Bit.Identity.Utilities
         public static IIdentityServerBuilder AddCustomIdentityServerServices(this IServiceCollection services,
             IWebHostEnvironment env, GlobalSettings globalSettings)
         {
-            if (globalSettings.SelfHosted)
-            {
-                services.AddTransient<IDiscoveryResponseGenerator, DiscoveryResponseGenerator>();
-            }
+            services.AddTransient<IDiscoveryResponseGenerator, DiscoveryResponseGenerator>();
 
             services.AddSingleton<StaticClientStore>();
             services.AddTransient<IAuthorizationCodeStore, AuthorizationCodeStore>();