Update middleware.ts

ResilienceWeb · Oct 1, 2023 · 23af6a9 · 23af6a9
1 parent 98736cc
commit 23af6a9
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/middleware.ts b/middleware.ts
@@ -18,13 +18,18 @@ export default function middleware(req: NextRequest) {
     })
   }
 
-  const currentHost =
-    process.env.NODE_ENV === 'production' && process.env.VERCEL === '1'
-      ? hostname
-          .replace(`.cambridgeresilienceweb.org.uk`, '')
-          .replace(`.resilienceweb.org.uk`, '')
-          .replace('.vercel.app', '')
-      : hostname.replace(`.localhost:3000`, '')
+
+  let currentHost
+  if (process.env.VERCEL_ENV === 'preview') {
+    currentHost = hostname.replace(`.${process.env.VERCEL_URL}`, '')
+  } else {
+    currentHost =
+      process.env.NODE_ENV === 'production' && process.env.VERCEL === '1'
+        ? hostname
+            .replace('.cambridgeresilienceweb.org.uk', '')
+            .replace('.resilienceweb.org.uk', '')
+        : hostname.replace(`.localhost:3000`, '')
+  }
 
   console.log('DINER2', {
     VERCEL_URL: process.env.VERCEL_URL,
@@ -82,3 +87,4 @@ export const config = {
 
 
 
+