import { NextRequest, NextResponse } from "next/server";
import { SERVER_API_CS_URL } from "./src/shared/config/env";

const AUTH_USER_ID = process.env.NEXT_PUBLIC_AUTH_USER_ID || "token-id";
const CLAIM_NAME = process.env.NEXT_PUBLIC_CLAIM_NAME || "Authorization";
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 || "";

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_CS_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")) {
    let queryParams = searchParams.toString();
    const typename = searchParams.get("TYPENAME");

    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"  //잠재수요처
    ){
      const url = req.nextUrl.clone();
      url.searchParams.set('TYPENAME', typename.replace("hasudo:",""));
      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/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_CS_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_CS_URL}${pathname}`;

    console.log(
      `[Middleware] Rewriting from '/user-service' path to: ${newUrl}`,
    );
    return NextResponse.rewrite(new URL(newUrl), {
      headers: requestHeaders,
    });
  }
  if (pathname === "/") {
    return NextResponse.redirect(new URL("/pages/dashboard", req.url));
  }
  return NextResponse.next();
}

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