frontend/apps/wr/middleware.ts

import { NextRequest, NextResponse } from "next/server";
import { CS_URL } from "./components/lib/config";
import { CLAIM_NAME, AUTH_USER_ID } from "./src/shared/config";

// const CLAIM_NAME = "Authorization";
// const AUTH_USER_ID = "token-id";
const serverApiUrl = CS_URL;
const GEOSERVER_URL = process.env.NEXT_PUBLIC_GEOSERVER_URL || "";
const LX_GEOSERVER_URL = process.env.NEXT_PUBLIC_LX_GEOSERVER_URL || "";
const SEWERAGE_API_URL = process.env.NEXT_PUBLIC_SEWERAGE_API_URL || "";
const REUSE_API_URL = process.env.NEXT_PUBLIC_WR_SERVER_API_URL || "";
const OPERATION_API_URL = process.env.NEXT_PUBLIC_FO_SERVER_API_URL || "";

export function middleware(req: NextRequest) {
  const pathname = req.nextUrl.pathname;
  console.log(`[Middleware] Request URL: ${req.url}`);
  const searchParams = req.nextUrl.searchParams;

  if (pathname.startsWith("/server")) {
    const decodedParams = Array.from(searchParams.entries())
      .map(([key, value]) => `${key}=${decodeURIComponent(value)}`)
      .join("&");

    const newUrl = `${serverApiUrl}${pathname.replace("/server", "")}${
      decodedParams ? `?${decodedParams}` : ""
    }`;
    return NextResponse.rewrite(new URL(newUrl));
  }

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

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

    if (Authorization) requestHeaders.set(CLAIM_NAME, Authorization);
    //if (userId) requestHeaders.set(AUTH_USER_ID, userId);

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

  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);

    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 = `${serverApiUrl}${pathname}`;

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

  if (pathname.startsWith("/geoserver")) {
    let queryParams = searchParams.toString();
    const typename = searchParams.get("TYPENAME");
    const startYmd = searchParams.get("START_YMD") ?? null;
    const endYmd = searchParams.get("END_YMD") ?? null;

    let newUrl = `${GEOSERVER_URL}${pathname.replace("/geoserver", "")}${queryParams ? `?${queryParams}` : ""}`;

    if (
      typename == "hasudo:tb_il_swage_prcs_fclt" || //하수처리시설
      typename == "hasudo:tb_ms_swage_prcs_zone" || //하수처리구역
      typename == "hasudo:tb_ms_sbmrg_dngr_map" || //침수위험지도
      typename == "hasudo:tb_ms_flood_dngr_map" //도시침수지도
    ) {
      newUrl = `${GEOSERVER_URL}${pathname.replace("/geoserver", "")}${queryParams ? `?${queryParams}` : ""}`;
    } else if (
      typename == "hasudo:tb_ms_swage_ppl" || //관로
      typename == "hasudo:tb_ms_swage_mhl" || //맨홀
      typename == "hasudo:tb_ms_swage_pmstn" || //펌프장
      typename == "hasudo:tb_ms_stofcbr" //우수토실
    ) {
      newUrl = `${LX_GEOSERVER_URL}${pathname.replace("/geoserver", "")}${queryParams ? `?${queryParams}` : ""}`;
    } else if (
      typename == "hasudo:om_dt_untrt_swg_rain" || //중점우수토실
      typename == "hasudo:om_wqlt_station" || //관측소
      typename == "hasudo:tb_wr_bdst_ruse_fclt" || //재이용시설
      typename == "hasudo:tb_wr_rnwtr_utztn_fclt" || //빗물이용시설
      typename == "hasudo:tb_wr_rswtr_fclt" || //중수도시설
      typename == "hasudo:tb_wr_htdfl_fclt" || //온배수시설
      typename == "hasudo:tb_wr_ruse_fclt_dmd_co" || //수요처
      typename == "hasudo:tb_wr_tmpr_dmd_co" || //잠재수요처
      typename?.startsWith("hasudo:drought") //가뭄 지도
    ) {
      const url = req.nextUrl.clone();
      url.searchParams.set("TYPENAME", typename.replace("hasudo:", ""));
      if (startYmd) url.searchParams.set("startYmd", startYmd);
      if (endYmd) url.searchParams.set("endYmd", endYmd);
      queryParams = url.searchParams.toString();
      newUrl = `${SEWERAGE_API_URL}/v1/map/geometry${queryParams ? `?${queryParams}` : ""}`;
    }

    //console.log(`[Middleware] Rewriting from '/geoserver' path to: ${newUrl}`);
    return NextResponse.rewrite(new URL(newUrl));
  }

  if (pathname.startsWith("/sewerage-service")) {
    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("/reuse-service")) {
    const queryParams = searchParams.toString();
    const newUrl = `${REUSE_API_URL}${pathname.replace("/reuse-service", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/reuse-service' path to: ${newUrl}`);
    return NextResponse.rewrite(new URL(newUrl));
  }

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

  return NextResponse.next();
}

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