fix: refactor navigation of webview pages and add redirect to refresh

middlewares/webView.ts

@@ -1,19 +1,42 @@
 import { type NextMiddleware, NextResponse } from "next/server"
 
 import { findLang } from "@/constants/languages"
+import {
+  loyaltyPagesWebviews,
+  myPagesWebviews,
+  webviews,
+} from "@/constants/routes/webviews"
 import { env } from "@/env/server"
-import { badRequest, internalServerError } from "@/server/errors/next"
+import { badRequest } from "@/server/errors/next"
 
 import { decryptData } from "@/utils/aes"
 
 import type { MiddlewareMatcher } from "@/types/middleware"
 
 export const middleware: NextMiddleware = async (request) => {
+  const { nextUrl } = request
+  const lang = findLang(nextUrl.pathname)
+
+  const pathNameWithoutLang = nextUrl.pathname.replace(`/${lang}/webview`, "")
+  const searchParams = new URLSearchParams(request.nextUrl.searchParams)
+  searchParams.set("uri", pathNameWithoutLang)
+
   const webviewToken = request.cookies.get("webviewToken")
   if (webviewToken) {
     // since the token exists, this is a subsequent visit
     // we're done, allow it
-    return NextResponse.next()
+    if (myPagesWebviews.includes(nextUrl.pathname)) {
+      return NextResponse.rewrite(
+        new URL(`/${lang}/webview/my-pages?${searchParams.toString()}`, nextUrl)
+      )
+    } else if (loyaltyPagesWebviews.includes(nextUrl.pathname)) {
+      return NextResponse.rewrite(
+        new URL(
+          `/${lang}/webview/loyalty-page?${searchParams.toString()}`,
+          nextUrl
+        )
+      )
+    }
   }
 
   // Authorization header is required for webviews
@@ -37,13 +60,31 @@ export const middleware: NextMiddleware = async (request) => {
       authorization
     )
 
-    const response = NextResponse.next()
-    response.cookies.set("webviewToken", decryptedData, {
-      httpOnly: true,
-      secure: true,
-    })
-
-    return response
+    if (myPagesWebviews.includes(nextUrl.pathname)) {
+      return NextResponse.rewrite(
+        new URL(
+          `/${lang}/webview/my-pages?${searchParams.toString()}`,
+          nextUrl
+        ),
+        {
+          headers: {
+            "Set-Cookie": `webviewToken=${decryptedData}; Secure; HttpOnly; Path=/; SameSite=Strict;`,
+          },
+        }
+      )
+    } else if (loyaltyPagesWebviews.includes(nextUrl.pathname)) {
+      return NextResponse.rewrite(
+        new URL(
+          `/${lang}/webview/loyalty-page?${searchParams.toString()}`,
+          nextUrl
+        ),
+        {
+          headers: {
+            "Set-Cookie": `webviewToken=${decryptedData}; Secure; HttpOnly; Path=/; SameSite=Strict;`,
+          },
+        }
+      )
+    }
   } catch (e) {
     if (e instanceof Error) {
       console.error(`${e.name}: ${e.message}`)
@@ -55,7 +96,6 @@ export const middleware: NextMiddleware = async (request) => {
 
 export const matcher: MiddlewareMatcher = (request) => {
   const { nextUrl } = request
-  const lang = findLang(nextUrl.pathname)
-  const pathNameWithoutLang = nextUrl.pathname.replace(`/${lang}`, "")
-  return pathNameWithoutLang.startsWith("/webview/")
+
+  return webviews.includes(nextUrl.pathname)
 }