import { randomUUID } from "crypto"
import { NextResponse } from "next/server"
import { touchUpdatedAt } from "@/lib/content-audit"
import { prisma } from "@/lib/prisma"
import { getCurrentUser, isAllowedRole } from "@/lib/user"
import { deleteImage, extractMediaKeyFromUrl } from "@/lib/r2-storage"
import {
  getPersistedMediaValidationError,
  normalizePersistedMediaInPayload,
} from "@/lib/media-manager/validate-persisted-payload"
import {
  PROCEDURES_PAGE_ENTITY_KEY,
  PROCEDURES_PAGE_LOCALE_CODES,
  type ProceduresPageLocaleCode,
} from "@/lib/pages/procedures-page/locales"
import {
  createEmptyProceduresPageFormState,
  type ProceduresPageFormStateShape,
} from "@/lib/pages/procedures-page/form-types"
import { normalizeProceduresPageLocaleImagesForStorage } from "@/lib/pages/procedures-page/shared-images"
import {
  parseProceduresPageJsonForStorage,
  parseProceduresPagePutBody,
  type ProceduresPageSectionKey,
} from "@/lib/pages/procedures-page/validate-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): ProceduresPageFormStateShape {
  let parsed: unknown = contentJson
  if (typeof contentJson === "string") {
    try {
      parsed = JSON.parse(contentJson)
    } catch {
      parsed = undefined
    }
  }
  return parseProceduresPageJsonForStorage(parsed).data
}

function buildMergedLocalesForSectionSave(
  existingByLocale: Partial<Record<ProceduresPageLocaleCode, ProceduresPageFormStateShape>>,
  incomingLocales: Record<string, ProceduresPageFormStateShape>,
  sectionKey: ProceduresPageSectionKey
): Record<ProceduresPageLocaleCode, ProceduresPageFormStateShape> {
  const next: Record<ProceduresPageLocaleCode, ProceduresPageFormStateShape> = {
    en: createEmptyProceduresPageFormState(),
    tr: createEmptyProceduresPageFormState(),
    ar: createEmptyProceduresPageFormState(),
  }
  for (const locale of PROCEDURES_PAGE_LOCALE_CODES) {
    const base = existingByLocale[locale] ?? createEmptyProceduresPageFormState()
    const incoming = incomingLocales[locale] ?? base
    next[locale] = { ...base, [sectionKey]: incoming[sectionKey] }
  }
  return next
}

function collectProceduresPageImagePublicIds(state: ProceduresPageFormStateShape): Set<string> {
  const ids = new Set<string>()
  const explicit = state.hero.imagePublicId.trim()
  if (explicit.startsWith("procedures-page/")) ids.add(explicit)
  const fromUrl = extractMediaKeyFromUrl(state.hero.image, { prefix: "procedures-page" })
  if (fromUrl?.startsWith("procedures-page/")) ids.add(fromUrl)
  return ids
}

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

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

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

    return NextResponse.json(
      { success: true as const, locales: normalizeProceduresPageLocaleImagesForStorage(locales) },
      { headers: NO_CACHE_HEADERS }
    )
  } catch (error) {
    console.error("[admin/procedures-page] GET error:", error)
    return NextResponse.json(
      { success: false as const, error: "Failed to load procedures 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 = parseProceduresPagePutBody(normalizedBody)
  if (!parsed.ok) return fail(parsed.error)

  try {
    const existingRows = await prisma.proceduresPageLocale.findMany({
      where: { entityKey: PROCEDURES_PAGE_ENTITY_KEY },
      select: { locale: true, contentJson: true },
    })

    const existingByLocale: Partial<Record<ProceduresPageLocaleCode, ProceduresPageFormStateShape>> = {}
    for (const row of existingRows) {
      if (!PROCEDURES_PAGE_LOCALE_CODES.includes(row.locale as ProceduresPageLocaleCode)) continue
      existingByLocale[row.locale as ProceduresPageLocaleCode] = rowToFormState(row.contentJson)
    }

    const localesToPersist =
      parsed.data.sectionKey
        ? buildMergedLocalesForSectionSave(existingByLocale, parsed.data.locales, parsed.data.sectionKey)
        : (parsed.data.locales as Record<ProceduresPageLocaleCode, ProceduresPageFormStateShape>)

    const normalizedLocales = normalizeProceduresPageLocaleImagesForStorage(localesToPersist)

    const previousPublicIds = new Set<string>()
    for (const row of existingRows) {
      const state = rowToFormState(row.contentJson)
      for (const id of collectProceduresPageImagePublicIds(state)) previousPublicIds.add(id)
    }
    const nextPublicIds = new Set<string>()
    for (const state of Object.values(normalizedLocales)) {
      if (!state) continue
      for (const id of collectProceduresPageImagePublicIds(state)) nextPublicIds.add(id)
    }

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

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

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