From 0f25f9295c9b777c34fa2ae272972e9ee227758d Mon Sep 17 00:00:00 2001
From: ibelick <julien.thibeaut@gmail.com>
Date: Wed, 23 Apr 2025 22:49:34 +0200
Subject: [PATCH] feat: manaage preview url in auth

---
 app/auth/callback/route.ts | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/app/auth/callback/route.ts b/app/auth/callback/route.ts
index 784fac9..45614d1 100644
--- a/app/auth/callback/route.ts
+++ b/app/auth/callback/route.ts
@@ -52,12 +52,16 @@ export async function GET(request: Request) {
 
   const forwardedHost = request.headers.get("x-forwarded-host")
   const isLocal = process.env.NODE_ENV === "development"
+  const host = forwardedHost || new URL(request.url).hostname
 
-  const redirectUrl = isLocal
-    ? `${origin}${next}`
-    : forwardedHost
-      ? `https://${forwardedHost}${next}`
-      : `${origin}${next}`
+  let baseUrl = ""
+  if (isLocal) {
+    baseUrl = origin
+  } else {
+    baseUrl = `https://${host}`
+  }
+
+  const redirectUrl = `${baseUrl}${next}`
 
   return NextResponse.redirect(redirectUrl)
 }