import { NextRequest, NextResponse } from "next/server";

const AUTH_USER_ID = process.env.NEXT_PUBLIC_AUTH_USER_ID || "token-id";
const CLAIM_NAME = process.env.NEXT_PUBLIC_CLAIM_NAME || "Authorization";
const SERVER_API_URL = process.env.NEXT_PUBLIC_SERVER_API_URL || "";
const GEOSERVER_URL = process.env.NEXT_PUBLIC_GEOSERVER_URL || "";
const SEWERAGE_API_URL = process.env.NEXT_PUBLIC_SEWERAGE_API_URL || "";

export function middleware(req: NextRequest) {
  console.log("Middleware 진입:", req.nextUrl.pathname);
  const pathname = req.nextUrl.pathname;
  const searchParams = req.nextUrl.searchParams;

  if (pathname.startsWith("/server")) {
    const queryParams = searchParams.toString();
    const newUrl = `${SERVER_API_URL}${pathname.replace("/server", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/server' path to: ${newUrl}`);
    return NextResponse.rewrite(new URL(newUrl));
  }

  if (pathname.startsWith("/geoserver")) {
    const queryParams = searchParams.toString();
    const newUrl = `${GEOSERVER_URL}${pathname.replace("/geoserver", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/geoserver' path to: ${newUrl}`);
    return NextResponse.rewrite(new URL(newUrl));
  }

  if (pathname.startsWith("/sewerage-service/api/v1/map/geometry")) {
    const queryParams = searchParams.toString();
    const newUrl = `${SEWERAGE_API_URL}${pathname.replace("/sewerage-service/api", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/geoserver' path to: ${newUrl}`);
    return NextResponse.rewrite(new URL(newUrl));
  }

  if (pathname.startsWith("/attach-service") || pathname.startsWith("/cms-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 = `${SERVER_API_URL}${pathname}`;

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

    console.log(
      `[Middleware] Rewriting from '/attach-service' path to: ${newUrl}`,
    );
    return NextResponse.rewrite(new URL(newUrl), {
      headers: requestHeaders,
    });
  }

  if (pathname.startsWith("/user-service")) {
    const requestHeaders = new Headers(req.headers);
    const newUrl = `${SERVER_API_URL}${pathname}`;

    console.log(
      `[Middleware] Rewriting from '/user-service' path to: ${newUrl}`,
    );
    return NextResponse.rewrite(new URL(newUrl), {
      headers: requestHeaders,
    });
  }

  return NextResponse.next();
}

export const config = {
  matcher: [
    "/server/:path*",
    "/geoserver/:path*",
    "/attach-service/:path*",
    "/user-service/:path*",
    "/cms-service/:path*",
    "/sewerage-service/api/v1/map/geometry/:path*",
  ],
};