fix: bug of relative urls

aiji42 · May 2, 2022 · 8ac6aa4 · 8ac6aa4
1 parent ecbcf5b
commit 8ac6aa4
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/handle-fallback.ts b/src/handle-fallback.ts
@@ -9,9 +9,15 @@ export const handleFallback = (
 ): ReturnType<NextMiddleware> => {
   if (typeof fallback === 'function') return fallback(request, event)
   if (request.preflight) return new NextResponse(null)
-  if (fallback.type === 'rewrite')
-    return NextResponse.rewrite(fallback.destination)
+  if (fallback.type === 'rewrite') {
+    const url = request.nextUrl.clone()
+    url.pathname = fallback.destination
+    return NextResponse.rewrite(url)
+  }
 
-  if (request.nextUrl.pathname !== fallback.destination)
-    return NextResponse.redirect(fallback.destination, fallback.statusCode)
+  if (request.nextUrl.pathname !== fallback.destination) {
+    const url = request.nextUrl.clone()
+    url.pathname = fallback.destination
+    return NextResponse.redirect(url, fallback.statusCode)
+  }
 }