feat: Add UseForwardedHeaders when in production

1 files changed, 8 insertions, 4 deletions

diff --git a/src/server/Startup.cs b/src/server/Startup.cs
index 58eb33a..38d754b 100644
--- a/src/server/Startup.cs
+++ b/src/server/Startup.cs
@@ -1,3 +1,5 @@
+using Microsoft.AspNetCore.HttpOverrides;
+
 namespace IOL.BookmarkThing.Server;
 
 public class Startup
@@ -69,10 +71,6 @@ public class Startup
 					options.ClientSecret = Configuration.GetValue<string>("GH_CLIENT_SECRET");
 					options.ClientId = Configuration.GetValue<string>("GH_CLIENT_ID");
 					options.SaveTokens = true;
-					options.Events.OnRedirectToAuthorizationEndpoint = context => {
-						context.RedirectUri = "https://bmt.ivarlovlie.no";
-						return Task.CompletedTask;
-					};
 					options.CorrelationCookie = new CookieBuilder {
 							Name = "gh_correlation",
 							SameSite = SameSiteMode.Lax,
@@ -138,6 +136,12 @@ public class Startup
 			});
 		}
 
+		if (WebHostEnvironment.IsProduction()) {
+			app.UseForwardedHeaders(new ForwardedHeadersOptions {
+					ForwardedHeaders = ForwardedHeaders.All
+			});
+		}
+
 		app.UseRouting();
 		app.UseSerilogRequestLogging();
 		app.UseStatusCodePages();