custom DiscoveryResponseGenerator and helpers

src/Identity/Utilities/DiscoveryResponseGenerator.cs

@@ -0,0 +1,39 @@
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using Bit.Core;
+using Bit.Core.Utilities;
+using IdentityServer4.Configuration;
+using IdentityServer4.Services;
+using IdentityServer4.Stores;
+using IdentityServer4.Validation;
+using Microsoft.Extensions.Logging;
+
+namespace Bit.Identity.Utilities
+{
+    public class DiscoveryResponseGenerator : IdentityServer4.ResponseHandling.DiscoveryResponseGenerator
+    {
+        private readonly GlobalSettings _globalSettings;
+
+        public DiscoveryResponseGenerator(
+            IdentityServerOptions options,
+            IResourceStore resourceStore,
+            IKeyMaterialService keys,
+            ExtensionGrantValidator extensionGrants,
+            ISecretsListParser secretParsers,
+            IResourceOwnerPasswordValidator resourceOwnerValidator,
+            ILogger<DiscoveryResponseGenerator> logger,
+            GlobalSettings globalSettings)
+            : base(options, resourceStore, keys, extensionGrants, secretParsers, resourceOwnerValidator, logger)
+        {
+            _globalSettings = globalSettings;
+        }
+
+        public override async Task<Dictionary<string, object>> CreateDiscoveryDocumentAsync(
+            string baseUrl, string issuerUri)
+        {
+            var dict = await base.CreateDiscoveryDocumentAsync(baseUrl, issuerUri);
+            return CoreHelpers.AdjustIdentityServerConfig(dict, _globalSettings.BaseServiceUri.Identity,
+                _globalSettings.BaseServiceUri.InternalIdentity);
+        }
+    }
+}