import { randomUUID } from "crypto"
import { NextResponse } from "next/server"
import { touchUpdatedAt } from "@/lib/content-audit"
import { Prisma } from "@prisma/client"
import { prisma } from "@/lib/prisma"
import { getCurrentUser, isAllowedRole } from "@/lib/user"
import {
  CONTACT_US_PAGE_ENTITY_KEY,
  CONTACT_US_PAGE_LOCALE_CODES,
  type ContactUsPageLocaleCode,
} from "@/lib/pages/contact-us/locales"
import {
  buildContactUsPageImageByLocale,
  normalizeContactUsPageLocaleImagesForStorage,
  resolveContactUsPageDefaultImage,
} from "@/lib/pages/contact-us/hero-fallback"
import {
  parseContactUsPageJsonForStorage,
  parseContactUsPagePutBody,
  type ContactUsPageSectionKey,
} from "@/lib/pages/contact-us/validate-payload"
import {
  createEmptyContactUsPageFormState,
  type ContactUsPageFormStateShape,
} from "@/lib/pages/contact-us/form-types"
import { deleteImage, extractMediaKeyFromUrl } from "@/lib/r2-storage"
import {
  getPersistedMediaValidationError,
  normalizePersistedMediaInPayload,
} from "@/lib/media-manager/validate-persisted-payload"

export const dynamic = "force-dynamic"
export const revalidate = 0

const NO_CACHE_HEADERS = {
  "Cache-Control": "no-store, no-cache, must-revalidate, proxy-revalidate",
  Pragma: "no-cache",
  Expires: "0",
}

async function ensureAdmin() {
  const user = await getCurrentUser()
  if (!user) {
    return { ok: false as const, res: NextResponse.json({ error: "Unauthorized" }, { status: 401, headers: NO_CACHE_HEADERS }) }
  }
  if (!isAllowedRole(user.role, ["admin"])) {
    return { ok: false as const, res: NextResponse.json({ error: "Forbidden" }, { status: 403, headers: NO_CACHE_HEADERS }) }
  }
  return { ok: true as const, user }
}

function fail(message: string, status = 400) {
  return NextResponse.json({ success: false as const, error: message }, { status, headers: NO_CACHE_HEADERS })
}

function rowToFormState(contentJson: unknown): ContactUsPageFormStateShape {
  let parsed: unknown = contentJson
  if (typeof contentJson === "string") {
    try {
      parsed = JSON.parse(contentJson)
    } catch {
      parsed = undefined
    }
  }
  return parseContactUsPageJsonForStorage(parsed).data
}

function resolveContactUsImagePublicIdForDelete(hero: ContactUsPageFormStateShape["hero"]): string | null {
  const explicit = (hero.imagePublicId ?? "").trim()
  if (explicit.startsWith("contact-us/")) return explicit
  return extractMediaKeyFromUrl(hero.image, { prefix: "contact-us" })
}

function buildMergedLocalesForSectionSave(
  existingByLocale: Partial<Record<ContactUsPageLocaleCode, ContactUsPageFormStateShape>>,
  incomingLocales: Record<string, ContactUsPageFormStateShape>,
  sectionKey: ContactUsPageSectionKey
): Record<ContactUsPageLocaleCode, ContactUsPageFormStateShape> {
  const next: Record<ContactUsPageLocaleCode, ContactUsPageFormStateShape> = {
    en: createEmptyContactUsPageFormState(),
    tr: createEmptyContactUsPageFormState(),
    ar: createEmptyContactUsPageFormState(),
  }

  for (const locale of CONTACT_US_PAGE_LOCALE_CODES) {
    const base = existingByLocale[locale] ?? createEmptyContactUsPageFormState()
    const incoming = incomingLocales[locale] ?? base
    next[locale] = {
      ...base,
      [sectionKey]: incoming[sectionKey],
    }
  }

  return next
}

export async function GET() {
  const auth = await ensureAdmin()
  if (!auth.ok) return auth.res

  try {
    const rows = await prisma.contactUsPageLocale.findMany({
      where: { entityKey: CONTACT_US_PAGE_ENTITY_KEY },
      orderBy: { locale: "asc" },
    })

    const locales: Partial<Record<ContactUsPageLocaleCode, ContactUsPageFormStateShape>> = {}
    for (const row of rows) {
      if (!CONTACT_US_PAGE_LOCALE_CODES.includes(row.locale as ContactUsPageLocaleCode)) continue
      const code = row.locale as ContactUsPageLocaleCode
      locales[code] = rowToFormState(row.contentJson)
    }

    const normalizedLocales = normalizeContactUsPageLocaleImagesForStorage(locales)
    const defaultImage = resolveContactUsPageDefaultImage(normalizedLocales)
    const imageByLocale = buildContactUsPageImageByLocale(normalizedLocales)

    return NextResponse.json(
      { success: true as const, locales: normalizedLocales, defaultImage, imageByLocale },
      { headers: NO_CACHE_HEADERS }
    )
  } catch (error) {
    console.error("[admin/contact-us] GET error:", error)
    return NextResponse.json(
      { success: false as const, error: "Failed to load contact us page" },
      { status: 500, headers: NO_CACHE_HEADERS }
    )
  }
}

export async function PUT(req: Request) {
  const auth = await ensureAdmin()
  if (!auth.ok) return auth.res

  let body: unknown
  try {
    body = await req.json()
  } catch {
    return fail("Invalid JSON body")
  }

  const normalizedBody = normalizePersistedMediaInPayload(body)
  const mediaError = getPersistedMediaValidationError(normalizedBody)
  if (mediaError) return fail(mediaError)

  const parsed = parseContactUsPagePutBody(normalizedBody)
  if (!parsed.ok) return fail(parsed.error)

  try {
    const existingRows = await prisma.contactUsPageLocale.findMany({
      where: { entityKey: CONTACT_US_PAGE_ENTITY_KEY },
      select: { locale: true, contentJson: true },
    })
    const existingByLocale: Partial<Record<ContactUsPageLocaleCode, ContactUsPageFormStateShape>> = {}
    for (const row of existingRows) {
      if (!CONTACT_US_PAGE_LOCALE_CODES.includes(row.locale as ContactUsPageLocaleCode)) continue
      existingByLocale[row.locale as ContactUsPageLocaleCode] = rowToFormState(row.contentJson)
    }
    const localesToPersist =
      parsed.data.sectionKey
        ? buildMergedLocalesForSectionSave(existingByLocale, parsed.data.locales, parsed.data.sectionKey)
        : (parsed.data.locales as Record<ContactUsPageLocaleCode, ContactUsPageFormStateShape>)
    const normalizedLocales = normalizeContactUsPageLocaleImagesForStorage(localesToPersist)

    const previousImagePublicIds = new Set<string>()
    for (const row of existingRows) {
      const state = rowToFormState(row.contentJson)
      const publicId = resolveContactUsImagePublicIdForDelete(state.hero)
      if (publicId) previousImagePublicIds.add(publicId)
    }

    const nextImagePublicIds = new Set<string>()
    for (const state of Object.values(normalizedLocales)) {
      const publicId = resolveContactUsImagePublicIdForDelete(state.hero)
      if (publicId) nextImagePublicIds.add(publicId)
    }

    const localeNow = new Date()
    const ops = (Object.entries(normalizedLocales) as [ContactUsPageLocaleCode, ContactUsPageFormStateShape][]).map(
      ([locale, state]) =>
        prisma.contactUsPageLocale.upsert({
          where: {
            entityKey_locale: {
              entityKey: CONTACT_US_PAGE_ENTITY_KEY,
              locale,
            },
          },
          create: {
            id: randomUUID(),
            updatedAt: localeNow,
            entityKey: CONTACT_US_PAGE_ENTITY_KEY,
            locale,
            contentJson: JSON.stringify(state),
          },
          update: {
            ...touchUpdatedAt(),
            contentJson: JSON.stringify(state),
          },
        })
    )
    await prisma.$transaction(ops)

    const stalePublicIds = [...previousImagePublicIds].filter(
      (publicId) => publicId.startsWith("contact-us/") && !nextImagePublicIds.has(publicId)
    )
    if (stalePublicIds.length > 0) {
      await Promise.allSettled(stalePublicIds.map((publicId) => deleteImage(publicId)))
    }
  } catch (error) {
    console.error("[admin/contact-us] PUT error:", error)
    return NextResponse.json(
      { success: false as const, error: "Failed to save contact us page" },
      { status: 500, headers: NO_CACHE_HEADERS }
    )
  }

  return NextResponse.json({ success: true as const }, { headers: NO_CACHE_HEADERS })
}