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.SERVER_API_URL || "";
const SEWERAGE_API_URL = process.env.SEWERAGE_API_URL || "";
const GEOSERVER_URL = process.env.NEXT_PUBLIC_GEOSERVER_URL || "";
const LX_GEOSERVER_URL = process.env.NEXT_PUBLIC_LX_GEOSERVER_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 {
  return withAuth
    ? NextResponse.rewrite(new URL(url), { headers: buildAuthHeaders(req) })
    : NextResponse.rewrite(new URL(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 decodedQuery = toDecodedQueryString(searchParams);
    const replacedPath = pathname.replace("/server", "");
    // '/server//cms-service/..' 같은 케이스에서 '//' 정리
    const normalizedPath = replacedPath.replace(/^\/+/g, "/");
    const newUrl = `${SERVER_API_URL}${normalizedPath}${decodedQuery ? `?${decodedQuery}` : ""}`;
    console.log(`[Middleware] Rewriting from '/server' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }

  if (pathname.startsWith("/api/proxy")) {
    const decodedQuery = toDecodedQueryString(searchParams);
    const newUrl = `${SERVER_API_URL}${pathname}${decodedQuery ? `?${decodedQuery}` : ""
      }`;
    console.log(`[Middleware] Rewriting from '/server' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }

  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_fldng_wtlv_map"  //침수흔적도
      || typename == "hasudo:tb_ms_fldng_dpth_map"  //침수흔적도-깊이
      || 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 == "hasudo:tb_ms_rain_obsvtr_pstn_info"  //강우관측소
      || typename == "hasudo:tb_ms_rain_obsvtr_impct_rgn"  //티센망
      || typename == "hasudo:tb_ms_shtm_frcst_latc_info"  //초단기예보격자망
    ) {
      const url = req.nextUrl.clone();
      url.searchParams.set('TYPENAME', typename.replace("hasudo:", ""));
      queryParams = url.searchParams.toString();
      newUrl = `${SEWERAGE_API_URL}/sewerage-service/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("/attach-service")
    || pathname.startsWith("/cms-service")
    || pathname.startsWith("/user-service")
  ) {
    const newUrl = `${SERVER_API_URL}${pathname}`;
    console.log(`[Middleware] Rewriting from '/attach-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }
  
  if (pathname.startsWith("/dc-service")) {
    const decodedParams = Array.from(searchParams.entries())
    .map(([key, value]) => `${key}=${decodeURIComponent(value)}`)
    .join("&");
    
    const newUrl = `${DC_URL}${pathname}${decodedParams ? `?${decodedParams}` : ""}`;    
    console.log(`[Middleware] Rewriting from '/dc-service' path to: ${newUrl}`);
    const rewriteRequest = rewriteWith(newUrl, req, true);
    return rewriteRequest;
  }

  if (pathname.startsWith("/sewerage-service")) {
    const decodedQuery = toDecodedQueryString(searchParams);
    const newUrl = `${SEWERAGE_API_URL}${pathname}${decodedQuery ? `?${decodedQuery}` : ""}`;
    console.log(`[Middleware] Rewriting from '${pathname}' 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*",
    "/attach-service/:path*",
    "/user-service/:path*",
    "/cms-service/:path*",
    "/dc-service/:path*",
    "/sewerage-service/:path*",
    "/ngii-server/:path*",
  ],

};