import createMiddleware from "next-intl/middleware";
import { NextResponse } from "next/server";
import type { NextRequest } from "next/server";

import { routing } from "./i18n/routing";
import {
  fromDevelopmentAdminPath,
  getAdminBaseUrl,
  getRequestHostname,
  isDevelopmentAdminPath,
  isAdminHost,
  hasDedicatedAdminHost,
  isInternalAdminPath,
  isLegacyAdminPath,
  toInternalAdminPath,
} from "./lib/admin-routing";
import { getLocalizedPathWithDefault, stripLocalePrefix } from "./lib/locale";

const intlMiddleware = createMiddleware(routing);
const ADMIN_SESSION_COOKIE = "moh_admin_session";

type SiteRuntimeState = {
  defaultLocale: (typeof routing.locales)[number];
  maintenanceEnabled: boolean;
};

function isSupportedLocale(locale: string | undefined): locale is (typeof routing.locales)[number] {
  return locale === "ar" || locale === "en" || locale === "de";
}

function getPathLocale(pathname: string, fallbackLocale: (typeof routing.locales)[number]) {
  const locale = pathname.split("/")[1];

  return isSupportedLocale(locale) ? locale : fallbackLocale;
}

function isComingSoonPath(pathname: string) {
  return stripLocalePrefix(pathname) === "/coming-soon";
}

async function getSiteRuntimeState(request: NextRequest): Promise<SiteRuntimeState> {
  try {
    const response = await fetch(new URL("/api/site/default-locale", request.url), {
      headers: {
        "x-middleware-request": "1",
      },
      cache: "no-store",
    });

    if (!response.ok) {
      return {
        defaultLocale: routing.defaultLocale,
        maintenanceEnabled: false,
      };
    }

    const data = await response.json() as {
      defaultLocale?: string;
      maintenanceEnabled?: boolean;
    };

    return {
      defaultLocale: isSupportedLocale(data.defaultLocale) ? data.defaultLocale : routing.defaultLocale,
      maintenanceEnabled: data.maintenanceEnabled === true,
    };
  } catch {
    return {
      defaultLocale: routing.defaultLocale,
      maintenanceEnabled: false,
    };
  }
}

function hasLocalePrefix(pathname: string) {
  return routing.locales.some((locale) => pathname === `/${locale}` || pathname.startsWith(`/${locale}/`));
}

function getAdminBasicAuthUser(): string {
  return process.env.ADMIN_BASIC_AUTH_USER ?? "";
}

function getAdminBasicAuthPass(): string {
  return process.env.ADMIN_BASIC_AUTH_PASS ?? "";
}

function isAdminBasicAuthConfigured(): boolean {
  return Boolean(getAdminBasicAuthUser() && getAdminBasicAuthPass());
}

function isAdminBasicAuthValid(request: NextRequest): boolean {
  if (!isAdminBasicAuthConfigured()) {
    return false;
  }

  const header = request.headers.get("authorization");
  if (!header || !header.startsWith("Basic ")) {
    return false;
  }

  try {
    const decoded = atob(header.slice(6));
    const index = decoded.indexOf(":");
    if (index === -1) {
      return false;
    }

    const user = decoded.slice(0, index);
    const pass = decoded.slice(index + 1);

    return user === getAdminBasicAuthUser() && pass === getAdminBasicAuthPass();
  } catch {
    return false;
  }
}

export default async function middleware(request: NextRequest) {
  const { pathname } = request.nextUrl;
  const isDevelopmentAdminRequest =
    process.env.NODE_ENV !== "production" && isDevelopmentAdminPath(pathname);
  const hostname = getRequestHostname(
    request.headers.get("x-forwarded-host"),
    request.headers.get("host"),
    request.nextUrl.hostname,
  );
  const isAdminRequest = isDevelopmentAdminRequest || isAdminHost(hostname);
  const hasDedicatedAdminHostname = hasDedicatedAdminHost();
  const adminRobotsHeaders = {
    "X-Robots-Tag": "noindex, nofollow, noarchive, nosnippet, noimageindex",
  };

  if (
    isDevelopmentAdminRequest &&
    hasDedicatedAdminHostname &&
    !isAdminHost(hostname)
  ) {
    const redirectUrl = new URL(getAdminBaseUrl());
    redirectUrl.pathname = fromDevelopmentAdminPath(pathname);
    redirectUrl.search = request.nextUrl.search;
    return NextResponse.redirect(redirectUrl, 308);
  }

  if (isLegacyAdminPath(pathname) && process.env.NODE_ENV === "production") {
    return new NextResponse("Not Found", {
      status: 404,
    });
  }

  if (isInternalAdminPath(pathname) && process.env.NODE_ENV === "production" && !isAdminRequest) {
    return new NextResponse("Not Found", {
      status: 404,
    });
  }

  if (isAdminRequest) {
    if (isAdminBasicAuthConfigured() && !isAdminBasicAuthValid(request)) {
      return new NextResponse("Authentication required", {
        status: 401,
        headers: {
          "WWW-Authenticate": 'Basic realm="Admin Area", charset="UTF-8"',
          ...adminRobotsHeaders,
        },
      });
    }

    const rewriteUrl = request.nextUrl.clone();
    rewriteUrl.pathname = toInternalAdminPath(
      isDevelopmentAdminRequest ? fromDevelopmentAdminPath(pathname) : pathname,
    );
    const response = NextResponse.rewrite(rewriteUrl);
    response.headers.set("X-Robots-Tag", adminRobotsHeaders["X-Robots-Tag"]);
    return response;
  }

  const siteRuntimeState = await getSiteRuntimeState(request);
  const configuredDefaultLocale = siteRuntimeState.defaultLocale;

  if (
    siteRuntimeState.maintenanceEnabled &&
    !request.cookies.has(ADMIN_SESSION_COOKIE) &&
    !isComingSoonPath(pathname)
  ) {
    const locale = getPathLocale(pathname, configuredDefaultLocale);
    const redirectUrl = request.nextUrl.clone();
    redirectUrl.pathname = getLocalizedPathWithDefault(locale, "/coming-soon", configuredDefaultLocale);
    return NextResponse.redirect(redirectUrl, 307);
  }

  if (
    configuredDefaultLocale !== routing.defaultLocale &&
    (pathname === "/de" || pathname.startsWith("/de/"))
  ) {
    return NextResponse.next();
  }

  if (
    configuredDefaultLocale !== routing.defaultLocale &&
    !hasLocalePrefix(pathname)
  ) {
    const rewriteUrl = request.nextUrl.clone();
    rewriteUrl.pathname = getLocalizedPathWithDefault(
      configuredDefaultLocale,
      pathname,
      routing.defaultLocale,
    );
    return NextResponse.rewrite(rewriteUrl);
  }

  if (
    configuredDefaultLocale === routing.defaultLocale &&
    (pathname === "/de" || pathname.startsWith("/de/"))
  ) {
    const redirectUrl = request.nextUrl.clone();
    const nextPath = pathname.slice(3) || "/";
    redirectUrl.pathname = nextPath;
    return NextResponse.redirect(redirectUrl, 308);
  }

  return intlMiddleware(request);
}

export const config = {
  matcher: ["/((?!api|trpc|_next|_vercel|.*\\..*).*)"],
};