From beafab8f823570b5c576db1c009ca9838053bd14 Mon Sep 17 00:00:00 2001
From: ivarlovlie <git@ivarlovlie.no>
Date: Sat, 12 Feb 2022 00:07:26 +0100
Subject: feat: Add UseForwardedHeaders when in production

---
 src/server/Startup.cs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'src/server/Startup.cs')

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();
-- 
cgit v1.3