From 32aa6cf273dcbb2e9ab179e2adc73fe1f0ce75ed Mon Sep 17 00:00:00 2001
From: Michael Zetterberg <michael.zetterberg@scandichotels.com>
Date: Tue, 20 Aug 2024 15:02:02 +0200
Subject: [PATCH] fix(SW-236): add returnUrl to refresh

---
 app/[lang]/webview/[contentType]/[uid]/page.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/[lang]/webview/[contentType]/[uid]/page.tsx b/app/[lang]/webview/[contentType]/[uid]/page.tsx
index 8c6de3a0d..0efdea462 100644
--- a/app/[lang]/webview/[contentType]/[uid]/page.tsx
+++ b/app/[lang]/webview/[contentType]/[uid]/page.tsx
@@ -1,3 +1,4 @@
+import { headers } from "next/headers"
 import { notFound, redirect } from "next/navigation"
 
 import { serverClient } from "@/lib/trpc/server"
@@ -28,7 +29,11 @@ export default async function ContentTypePage({
       case "unauthorized": // fall through
       case "forbidden": // fall through
       case "token_expired":
-        redirect(`/${getLang()}/webview/refresh`)
+        const h = headers()
+        const returnURL = `/${getLang()}/webview${h.get("x-pathname")!}`
+        redirect(
+          `/${getLang()}/webview/refresh?returnUrl=${encodeURIComponent(returnURL)}`
+        )
     }
   }