import { NextRequest, NextResponse } from "next/server";
import {
  CLAIM_NAME,
  AUTH_USER_ID,
  PORT,
  PROXY_HOST,
  SERVER_API_URL,
  SITE_ID,
} from "./src/shared/config";

const serverApiUrl =
  process.env.NEXT_PUBLIC_SERVER_API_URL ?? "http://localhost:3000";

const linkBackendApiHost = process.env.NEXT_PUBLIC_LINK_BACKEND_API_HOST ?? "http://localhost:8080";

export function middleware(req: NextRequest) {
  const pathname = req.nextUrl.pathname;
  const searchParams = req.nextUrl.searchParams;

  const logMessages: string[] = [
    `[Middleware Request] ${new Date().toISOString()}`,
    `[Environment]\n  Mode: ${process.env.NODE_ENV}\n  Port: ${PORT}\n  Proxy Host: ${PROXY_HOST}\n  Server API URL: ${SERVER_API_URL}\n  Site ID: ${SITE_ID}`,
    `[Request]\n  Original Pathname: ${pathname}`,
  ];

  if (pathname.startsWith("/server")) {
    const queryParams = searchParams.toString();
    const newUrl = `${serverApiUrl}${pathname.replace("/server", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;

    logMessages.push(`  Rewritten URL: ${newUrl}`);
    logMessages.push("\n--------------------------------\n");
    console.log(logMessages.join("\n"));

    return NextResponse.rewrite(new URL(newUrl));
  }

  if (pathname.startsWith("/attach-service")) {
    const requestHeaders = new Headers(req.headers);
    const token = req.cookies.get(CLAIM_NAME)?.value;
    const userId = req.cookies.get(AUTH_USER_ID)?.value;
    const newUrl = `${serverApiUrl}${pathname}`;

    if (token) requestHeaders.set(CLAIM_NAME, token);
    if (userId) requestHeaders.set(AUTH_USER_ID, userId);

    logMessages.push(`  Rewritten URL: ${newUrl}`);
    logMessages.push("\n--------------------------------\n");
    console.log(logMessages.join("\n"));

    return NextResponse.rewrite(new URL(newUrl), {
      headers: requestHeaders,
    });
  }

  if (pathname.startsWith("/api/link")) {
    const requestHeaders = new Headers(req.headers);
    const token = req.cookies.get(CLAIM_NAME)?.value;
    const userId = req.cookies.get(AUTH_USER_ID)?.value;
    
    // /api/link 경로를 제거하고 백엔드 API 경로로 변환
    const apiPath = pathname.replace("/api/link", "");
    const queryParams = searchParams.toString();
    const newUrl = `${linkBackendApiHost}${apiPath}${
      queryParams ? `?${queryParams}` : ""
    }`;

    if (token) requestHeaders.set(CLAIM_NAME, token);
    if (userId) requestHeaders.set(AUTH_USER_ID, userId);

    logMessages.push(`  Link API Rewritten URL: ${newUrl}`);
    logMessages.push("\n--------------------------------\n");
    console.log(logMessages.join("\n"));

    return NextResponse.rewrite(new URL(newUrl), {
      headers: requestHeaders,
    });
  }

  if (pathname === "/") {
    return NextResponse.redirect(new URL("/dashboard", req.url));
  }

  logMessages.push("\n--------------------------------\n");
  console.log(logMessages.join("\n"));
  return NextResponse.next();
}

export const config = {
  matcher: ["/server/:path*", "/attach-service/:path*", "/api/link/:path*", "/"],
};