You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.6 KiB
84 lines
2.6 KiB
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}`)
|
|
})
|
|
|