import fs from 'node:fs/promises'
import c2k from 'koa-connect'
import type { ViteDevServer } from 'vite'
import Send from 'koa-send'
import app from "./server/app"
import { bootstrapServer } from "./server/main"

// Constants
const isProduction = process.env.NODE_ENV === 'production'
const port = process.env.PORT || 5173
const base = process.env.BASE || '/'

bootstrapServer()

// Cached production assets
const templateHtml = isProduction
    ? await fs.readFile('./dist/client/index.html', 'utf-8')
    : ''

let vite: ViteDevServer
if (!isProduction) {
    const { createServer } = await import('vite')
    vite = await createServer({
        server: { middlewareMode: true },
        appType: 'custom',
        base,
    })
    app.use(c2k(vite.middlewares))
} else {
    app.use(async (ctx, next) => {
        await Send(ctx, ctx.path, { root: './dist/client', index: false });
        if (ctx.status === 404) {
            await next()
        }
    })
}

app.use(async (ctx, next) => {
    // if (!ctx.originalUrl.startsWith(base)) return await next()
    try {
        const url = ctx.originalUrl.replace(base, '')
        let template
        let render
        if (!isProduction) {
            // Always read fresh template in development
            template = await fs.readFile('./index.html', 'utf-8')
            template = await vite.transformIndexHtml(url, template)
            render = (await vite.ssrLoadModule('/src/entry-server.ts')).render
        } else {
            template = templateHtml
            // @ts-ignore
            render = (await import('./dist/server/entry-server.js')).render
        }

        // 解析请求 Cookie 到对象（复用通用工具）
        const { parseCookieHeader } = await import('./src/compose/cookieUtils')
        const cookies = parseCookieHeader(ctx.request.headers['cookie'] as string)

        const rendered = await render(url, { cookies })

        const html = template
            .replace(`<!--app-head-->`, rendered.head ?? '')
            .replace(`<!--app-html-->`, rendered.html ?? '')
        ctx.status = 200
        ctx.set({ 'Content-Type': 'text/html' })
        ctx.body = html

        // 设置服务端渲染期间收集到的 Set-Cookie
        const setCookies: string[] = (rendered as any).setCookies || []
        if (setCookies.length > 0) {
            ctx.set('Set-Cookie', setCookies)
        }
    } catch (e: Error | any) {
        vite?.ssrFixStacktrace(e)
        ctx.status = 500
        ctx.body = e.stack
    }
    await next()
})

// Start http server
app.listen(port, () => {
    console.log(`Server started at http://localhost:${port}`)
})