add middleware

app/api/auth/route.ts

@@ -1,8 +1,10 @@
 import { NextRequest, NextResponse } from "next/server";
 
 export async function GET(req: NextRequest) {
-  const host = req.headers.get("host") ?? "localhost:3000";
-  console.log(req.headers);
+  const host =
+    req.headers.get("x-current-host") ??
+    req.headers.get("host") ??
+    "localhost:3000";
 
   const redirect_uri =
     `${host.includes("localhost") ? "http://" : "https://"}` +

app/auth/page.tsx

@@ -1,10 +1,18 @@
 import { redirect } from "next/navigation";
 import { Metadata } from "next";
+import { headers } from "next/headers";
+
 import { apiServer } from "@/lib/api";
 
 async function getLoginUrl() {
+  const headersList = await headers();
+  const pathname = headersList.get("x-current-host") || "/";
   try {
-    const res = await apiServer.get("/auth");
+    const res = await apiServer.get("/auth", {
+      headers: {
+        "x-current-host": pathname,
+      },
+    });
     return res.data;
   } catch (error) {
     return error;

components/editor/save-button/index.tsx

@@ -55,7 +55,7 @@ export function SaveButton({
   return (
     <>
       <Button
-        variant="sky"
+        variant="default"
         className="max-lg:hidden !px-4"
         onClick={updateSpace}
       >
@@ -64,7 +64,7 @@ export function SaveButton({
         {loading && <Loading className="ml-2 size-4 animate-spin" />}
       </Button>
       <Button
-        variant="sky"
+        variant="default"
         size="sm"
         className="lg:hidden"
         onClick={updateSpace}

middleware.ts

@@ -0,0 +1,12 @@
+import { NextResponse } from "next/server";
+import type { NextRequest } from "next/server";
+
+export function middleware(request: NextRequest) {
+  const headers = new Headers(request.headers);
+  headers.set("x-current-host", request.nextUrl.host);
+  return NextResponse.next({ headers });
+}
+
+export const config = {
+  matcher: ["/((?!api|_next/static|_next/image|favicon.ico).*)"],
+};