From 9a0b65a3ab6247517de9ad545ba9fd9ec20b18f1 Mon Sep 17 00:00:00 2001
From: hamster1963 <1410514192@qq.com>
Date: Mon, 21 Oct 2024 10:57:13 +0800
Subject: [PATCH] fix: docker callback url

---
 app/api/auth/[...nextauth]/route.ts | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/app/api/auth/[...nextauth]/route.ts b/app/api/auth/[...nextauth]/route.ts
index 87f6488..4ddc1da 100644
--- a/app/api/auth/[...nextauth]/route.ts
+++ b/app/api/auth/[...nextauth]/route.ts
@@ -1,4 +1,22 @@
 import { handlers } from "@/auth";
+import { NextRequest } from "next/server";
 
-// Referring to the auth.ts we just created
-export const { GET, POST } = handlers;
+const reqWithTrustedOrigin = (req: NextRequest): NextRequest => {
+  const proto = req.headers.get("x-forwarded-proto");
+  const host = req.headers.get("x-forwarded-host");
+  if (!proto || !host) {
+    console.warn("Missing x-forwarded-proto or x-forwarded-host headers.");
+    return req;
+  }
+  const envOrigin = `${proto}://${host}`;
+  const { href, origin } = req.nextUrl;
+  return new NextRequest(href.replace(origin, envOrigin), req);
+};
+
+export const GET = (req: NextRequest) => {
+  return handlers.GET(reqWithTrustedOrigin(req));
+};
+
+export const POST = (req: NextRequest) => {
+  return handlers.POST(reqWithTrustedOrigin(req));
+};