import { NextRequest, NextResponse } from "next/server";
import { SERVER_API_CS_URL, SERVER_API_FO_URL, SERVER_API_WR_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.SEWERAGE_API_URL || "";
const serverApiUrl = SERVER_API_CS_URL;
const DC_URL = process.env.DC_URL || "";

/** 공통: URLSearchParams의 값을 decodeURIComponent 적용하여 querystring 생성 */
function toDecodedQueryString(params: URLSearchParams): string {
  const entries: string[][] = Array.from(params.entries()).map(([k, v]) => {
    try {
      return [k, decodeURIComponent(v)] as string[];
    } catch {
      return [k, v] as string[];
    }
  });
  return new URLSearchParams(entries).toString();
}

/** 공통: 요청 헤더에 쿠키 기반 인증 값 주입 */
function buildAuthHeaders(req: NextRequest): Headers {
  const requestHeaders = new Headers(req.headers);
  const token = req.cookies.get(CLAIM_NAME)?.value;
  const userId = req.cookies.get(AUTH_USER_ID)?.value;
  if (token) requestHeaders.set(CLAIM_NAME, token);
  if (userId) requestHeaders.set(AUTH_USER_ID, userId);
  return requestHeaders;
}

/** 공통: rewrite 응답 생성 (헤더 주입 여부 선택) */
function rewriteWith(url: string, req: NextRequest, withAuth = false): NextResponse {

  const headers = withAuth ? buildAuthHeaders(req) : new Headers(req.headers);

  // if (url.indexOf("/operation-service") !== -1) {
  //   // x-forwarded- 로 시작하는 헤더의 첫 번째 값만 남김
  //   for (const [key, value] of headers.entries()) {
  //     if (key.startsWith('x-forwarded-') && value.includes(',')) {
  //       const first = (value?.split(',')[0] || '').trim();
  //       headers.set(key, first);
  //     }
  //   }
  // }

  return NextResponse.rewrite(new URL(url), { headers });
}

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 = `${serverApiUrl}${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}/api/v1/map/geometry${queryParams ? `?${queryParams}` : ""}`;
    }

    console.log(`[Middleware] Rewriting from '/geoserver' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }

  if (pathname.startsWith("/sewerage-service")) {
    const queryParams = searchParams.toString();
    const newUrl = `${SEWERAGE_API_URL}${pathname.replace("/sewerage-service", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/sewerage-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }

  if (pathname.startsWith("/attach-service") || pathname.startsWith("/cms-service")) {
    const newUrl = `${serverApiUrl}${pathname}`;
    console.log(`[Middleware] Rewriting from '/attach-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }
  
  if (pathname.startsWith("/reuse-service")) {
    const queryParams = searchParams.toString();
    const newUrl = `${SERVER_API_WR_URL}${pathname.replace("/reuse-service", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/reuse-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }

  if (pathname.startsWith("/operation-service")) {
    const queryParams = searchParams.toString();
    const newUrl = `${SERVER_API_FO_URL}${pathname.replace("/operation-service", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/operation-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }

  if (pathname.startsWith("/dc-service")) {
    const queryParams = searchParams.toString();
    const newUrl = `${DC_URL}${pathname.replace("/dc-service", "")}${
      queryParams ? `?${queryParams}` : ""
    }`;
    console.log(`[Middleware] Rewriting from '/dc-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }
 
  if (pathname.startsWith("/user-service")) {
    const newUrl = `${SERVER_API_CS_URL}${pathname}`;
    console.log(`[Middleware] Rewriting from '/user-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }
  
  if (pathname.startsWith("/ngii-server")) {
    const decodedQuery = toDecodedQueryString(searchParams);
    const newUrl = `${process.env.NEXT_PUBLIC_NGII_URL}${pathname.replace("/ngii-server", "")}${decodedQuery ? `?${decodedQuery}` : ""}`;
    console.log(`[Middleware] Rewriting from '${pathname}' path to: ${newUrl}`);
    return NextResponse.rewrite(new URL(newUrl));
  }

  return NextResponse.next();
}

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