Merge pull request #4 from tsvx/patch-2

Fix swagger to work correctly behind the proxy
drwatson1 · Aug 14, 2020 · 606a406 · 606a406
2 parents 529f296 + 92aff30
commit 606a406
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/ProjectTemplates/ReferenceProject/Configuration/MiddlewareConfig.cs b/ProjectTemplates/ReferenceProject/Configuration/MiddlewareConfig.cs
@@ -1,4 +1,6 @@
 using Microsoft.AspNetCore.Builder;
+using Microsoft.OpenApi.Models;
+using System.Collections.Generic;
 
 namespace ReferenceProject
 {
@@ -13,7 +15,22 @@ public static class MiddlewareConfig
         public static IApplicationBuilder UseSwaggerWithOptions(this IApplicationBuilder app)
         {
             // Enable middleware to serve generated Swagger as a JSON endpoint.
-            SwaggerBuilderExtensions.UseSwagger(app);
+            app.UseSwagger(c =>
+            {
+                c.PreSerializeFilters.Add((swaggerDoc, httpRequest) =>
+                {
+                    if (!httpRequest.Headers.ContainsKey("X-Forwarded-Host")) return;
+
+                    var serverUrl = $"{httpRequest.Headers["X-Forwarded-Proto"]}://" +
+                        $"{httpRequest.Headers["X-Forwarded-Host"]}" +
+                        $"{httpRequest.Headers["X-Forwarded-Prefix"]}";
+
+                    swaggerDoc.Servers = new List<OpenApiServer>()
+                    {
+                        new OpenApiServer { Url = serverUrl }
+                    };
+                });
+            });
 
             // Enable middleware to serve swagger-ui (HTML, JS, CSS, etc.), 
             // specifying the Swagger JSON endpoint.