60 lines
1.9 KiB
JavaScript
60 lines
1.9 KiB
JavaScript
|
// 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))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|