feat: support x-forwarded-host in ssr setup (#140)

lukeshay · Jul 18, 2024 · 18fcc04 · 18fcc04
1 parent 48e4410
commit 18fcc04
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/adapter/src/lambda/handlers/ssr.ts b/packages/adapter/src/lambda/handlers/ssr.ts
@@ -112,10 +112,11 @@ const createExports = (
 			headers.set("cookie", event.cookies.join("; "))
 		}
 
+		const domainName = headers.get("x-forwarded-host") ?? event.requestContext.domainName;
 		const qs = event.rawQueryString.length ? `?${event.rawQueryString}` : ""
 		const url = new URL(
 			`${event.rawPath.replace(/\/?index\.html$/u, "")}${qs}`,
-			`https://${event.requestContext.domainName}`,
+			`https://${domainName}`,
 		)
 
 		const request = new Request(url, {
@@ -132,7 +133,7 @@ const createExports = (
 
 		if (!routeData) {
 			const request404 = new Request(
-				new URL(`404${qs}`, `https://${event.requestContext.domainName}`),
+				new URL(`404${qs}`, `https://${domainName}`),
 				{
 					body: createRequestBody(
 						event.requestContext.http.method,