EpicNext-Cms/src/app/api/admin/import/furni/batch-regen/route.ts

import { existsSync, promises as fs } from 'node:fs'
import path from 'node:path'
import { withAdmin } from '@/lib/api-handler'
import { apiError } from '@/lib/api-response'
import { PERMS } from '@/lib/permissions'
import { prisma } from '@/lib/prisma'
import { getFurniAssetDirs } from '@/lib/services/furni-asset-dirs'
import { appendFurniEntriesBatch, buildFurniEntry, readFurniData } from '@/lib/services/furni-data'
import { downloadFile, ensureDirectories } from '@/lib/services/furni-import'
import { getHabboItFurnidata } from '@/lib/services/habbo-furnidata-cache'
import { rcon } from '@/lib/services/rcon'
import { convertSwfToNitro } from '@/lib/services/swf-to-nitro'

interface RegenItem {
  classname: string
  revision: number
}

/**
 * SSE batch regenerate .nitro files.
 * Streams progress events for each item.
 */
export const POST = withAdmin({ permission: PERMS.ASSETS_IMPORT }, async (request) => {
  const body = await request.json()
  const items: RegenItem[] = body.items || []
  const concurrency = Math.min(Math.max(body.concurrency || 3, 1), 5)

  if (items.length === 0) return apiError('No items to regenerate', 400)

  await ensureDirectories()
  const { swfDir, nitroDir } = await getFurniAssetDirs()
  const encoder = new TextEncoder()

  const stream = new ReadableStream({
    async start(controller) {
      const send = (data: unknown) => {
        try {
          controller.enqueue(encoder.encode(`data: ${JSON.stringify(data)}\n\n`))
        } catch {
          /* stream closed */
        }
      }

      const startTime = Date.now()
      send({ type: 'regen_start', total: items.length })

      let succeeded = 0
      let failed = 0
      let skipped = 0
      const furniDataEntries: Array<{ entry: Record<string, unknown>; itemType: string }> = []

      // Load furnidata once for all items
      const furnidata = await getHabboItFurnidata()

      // Load existing FurnitureData.json classnames for dedup
      const existingFD = (await readFurniData()) as {
        roomitemtypes?: { furnitype?: Array<{ classname?: string }> }
        wallitemtypes?: { furnitype?: Array<{ classname?: string }> }
      }
      const fdClassnames = new Set([
        ...(existingFD.roomitemtypes?.furnitype || []).map((e) => e.classname),
        ...(existingFD.wallitemtypes?.furnitype || []).map((e) => e.classname),
      ])

      for (let i = 0; i < items.length; i += concurrency) {
        const chunk = items.slice(i, i + concurrency)

        const promises = chunk.map(async (item) => {
          const starIdx = item.classname.indexOf('*')
          const baseClassname =
            starIdx !== -1 ? item.classname.substring(0, starIdx) : item.classname
          const nitroPath = path.join(/*turbopackIgnore: true*/ nitroDir, `${baseClassname}.nitro`)

          if (existsSync(/*turbopackIgnore: true*/ nitroPath)) {
            skipped++
            send({ type: 'regen_progress', classname: item.classname, status: 'skipped' })
            return
          }

          send({ type: 'regen_progress', classname: item.classname, status: 'downloading' })

          // Resolve revision
          let rev = item.revision || 0
          if (rev === 0) {
            const fd =
              furnidata.get(item.classname) ||
              furnidata.get(baseClassname) ||
              furnidata.get(`${baseClassname}*0`)
            if (fd) rev = fd.revision
          }

          // Try spriteId as fallback revision
          const revsToTry: number[] = []
          if (rev > 0) revsToTry.push(rev)
          const dbItem = await prisma.itemsBase.findFirst({
            where: { itemName: item.classname },
            select: { id: true, spriteId: true, publicName: true, type: true },
          })
          if (dbItem?.spriteId && !revsToTry.includes(dbItem.spriteId)) {
            revsToTry.push(dbItem.spriteId)
          }

          if (revsToTry.length === 0) {
            failed++
            send({
              type: 'regen_progress',
              classname: item.classname,
              status: 'failed',
              message: 'No revision found',
            })
            return
          }

          const swfPath = path.join(swfDir, `${baseClassname}.swf`)

          // Download SWF if needed
          if (!existsSync(swfPath)) {
            let downloaded = false
            for (const tryRev of revsToTry) {
              const dl = await downloadFile(
                `https://images.habbo.com/dcr/hof_furni/${tryRev}/${encodeURIComponent(baseClassname)}.swf`,
                swfPath,
                { validate: 'swf' },
              )
              if (dl.ok) {
                rev = tryRev
                downloaded = true
                break
              }
            }
            if (!downloaded) {
              failed++
              send({
                type: 'regen_progress',
                classname: item.classname,
                status: 'failed',
                message: `SWF not found (revs: ${revsToTry.join(', ')})`,
              })
              return
            }
          }

          // Convert
          send({ type: 'regen_progress', classname: item.classname, status: 'converting' })
          try {
            const swfBuffer = await fs.readFile(swfPath)
            const conversion = convertSwfToNitro(swfBuffer, baseClassname)
            await fs.writeFile(nitroPath, conversion.bundle)

            // Build FurnitureData entry if missing
            if (!fdClassnames.has(baseClassname) && !fdClassnames.has(item.classname)) {
              const fd = furnidata.get(item.classname) || furnidata.get(baseClassname)
              const itemType = dbItem?.type === 'i' ? 'i' : 's'
              const entry = buildFurniEntry({
                id: dbItem?.spriteId || 0,
                classname: baseClassname,
                revision: rev,
                category: fd?.category || '',
                name: dbItem?.publicName || baseClassname,
                description: fd?.description || '',
                spriteId: dbItem?.spriteId || 0,
                dims: conversion.dimensions || { x: 1, y: 1, z: 1 },
                metadata: conversion.metadata || null,
                habboIt: fd || null,
              })
              furniDataEntries.push({ entry, itemType })
              fdClassnames.add(baseClassname)
            }

            succeeded++
            send({ type: 'regen_progress', classname: item.classname, status: 'done' })
          } catch (err) {
            failed++
            send({
              type: 'regen_progress',
              classname: item.classname,
              status: 'failed',
              message: (err as Error).message,
            })
          }
        })

        await Promise.allSettled(promises)
      }

      // Batch write FurnitureData.json
      if (furniDataEntries.length > 0) {
        try {
          await appendFurniEntriesBatch(furniDataEntries)
          send({
            type: 'regen_info',
            message: `Added ${furniDataEntries.length} entries to FurnitureData.json`,
          })
        } catch (err) {
          send({
            type: 'regen_info',
            message: `FurnitureData.json write failed: ${(err as Error).message}`,
          })
        }
      }

      // RCON refresh
      try {
        await rcon.updateItems()
        await rcon.updateCatalog()
      } catch {
        /* non-fatal */
      }

      send({
        type: 'regen_complete',
        succeeded,
        failed,
        skipped,
        furniDataAdded: furniDataEntries.length,
        duration: Date.now() - startTime,
      })

      controller.close()
    },
  })

  return new Response(stream, {
    headers: {
      'Content-Type': 'text/event-stream',
      'Cache-Control': 'no-cache',
      Connection: 'keep-alive',
    },
  })
})