// bankai start — the dev server: Vite middleware mode for the module
// graph and HMR, per-request streaming SSR of the same entry through
// vite.ssrLoadModule so server code shares Vite's transforms and cache.

import { createServer } from 'node:http'

import bankaiPlugin, { CLIENT_URL } from './plugin.js'
import { resolveApp } from './app.js'
import { documentHead, documentTail } from './document.js'

export default async function dev ({ entry, port = 8080, title = 'choo' }) {
  const { createServer: createViteServer } = await import('vite')

  const vite = await createViteServer({
    appType: 'custom',
    server: { middlewareMode: true },
    plugins: [bankaiPlugin(entry)],
    logLevel: 'warn'
  })

  const server = createServer((req, res) => {
    vite.middlewares(req, res, async () => {
      // not an asset Vite knows: this is a page navigation — SSR it
      try {
        const mod = await vite.ssrLoadModule(entry)
        const app = resolveApp(mod, entry)
        const state = {}

        res.writeHead(200, { 'content-type': 'text/html; charset=utf-8' })
        res.write(documentHead({
          title,
          scripts: ['/@vite/client', CLIENT_URL]
        }))

        for await (const chunk of app.toStream(req.url, state)) {
          res.write(chunk)
        }
        res.end(documentTail(state))
      } catch (err) {
        vite.ssrFixStacktrace(err)
        console.error(err)
        if (!res.headersSent) res.writeHead(500, { 'content-type': 'text/plain' })
        res.end('bankai dev error:\n\n' + (err.stack || err.message))
      }
    })
  })

  await new Promise((resolve) => server.listen(port, resolve))
  return {
    server,
    vite,
    port: server.address().port,
    origin: `http://localhost:${server.address().port}`,
    close: async () => {
      await vite.close()
      await new Promise((resolve) => server.close(resolve))
    }
  }
}