forked from vitejs/vite
-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
server.js
88 lines (77 loc) · 2.26 KB
/
server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// @ts-check
const fs = require('node:fs')
const path = require('node:path')
const express = require('express')
const isTest = process.env.NODE_ENV === 'test' || !!process.env.VITE_TEST_BUILD
async function createServer(
root = process.cwd(),
isProd = process.env.NODE_ENV === 'production'
) {
const resolve = (p) => path.resolve(__dirname, p)
const app = express()
/**
* @type {import('vite').ViteDevServer}
*/
const vite = await require('vite').createServer({
root,
logLevel: isTest ? 'error' : 'info',
server: {
middlewareMode: true,
watch: {
// During tests we edit the files too fast and sometimes chokidar
// misses change events, so enforce polling for consistency
usePolling: true,
interval: 100
}
},
appType: 'custom',
json: {
stringify: true
}
})
// use vite's connect instance as middleware
app.use(vite.middlewares)
app.use('*', async (req, res) => {
try {
let [url] = req.originalUrl.split('?')
if (url.endsWith('/')) url += 'index.ssr.html'
if (url === '/json-module') {
console.time('load module')
const json = JSON.stringify(await vite.ssrLoadModule('/test.json'))
console.timeEnd('load module')
res.status(200).end('' + json.length)
return
}
if (url === '/json-fs') {
console.time('transform module')
const source = fs.readFileSync('./test.json', { encoding: 'utf-8' })
const json = await vite.ssrTransform(
`export default ${source}`,
null,
'./output.json'
)
console.timeEnd('transform module')
res.status(200).end(String(json.code.length))
return
}
const htmlLoc = resolve(`.${url}`)
let html = fs.readFileSync(htmlLoc, 'utf8')
html = await vite.transformIndexHtml(url, html)
res.status(200).set({ 'Content-Type': 'text/html' }).end(html)
} catch (e) {
vite && vite.ssrFixStacktrace(e)
console.log(e.stack)
res.status(500).end(e.stack)
}
})
return { app, vite }
}
if (!isTest) {
createServer().then(({ app }) =>
app.listen(5173, () => {
console.log('http://localhost:5173')
})
)
}
// for test use
exports.createServer = createServer