server.js 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. // @ts-check
  2. const fs = require('fs')
  3. const path = require('path')
  4. const express = require('express')
  5. const isTest = process.env.NODE_ENV === 'test' || !!process.env.VITE_TEST_BUILD
  6. async function createServer(
  7. root = process.cwd(),
  8. isProd = process.env.NODE_ENV === 'production'
  9. ) {
  10. const resolve = (p) => path.resolve(__dirname, p)
  11. const indexProd = fs.readFileSync(resolve('dist/client/index.html'), 'utf-8')
  12. const manifest = require('./dist/client/ssr-manifest.json')
  13. const app = express()
  14. /**
  15. * @type {import('vite').ViteDevServer}
  16. */
  17. let vite
  18. if (!isProd) {
  19. vite = await require('vite').createServer({
  20. root,
  21. logLevel: isTest ? 'error' : 'info',
  22. server: {
  23. middlewareMode: 'ssr',
  24. watch: {
  25. // During tests we edit the files too fast and sometimes chokidar
  26. // misses change events, so enforce polling for consistency
  27. usePolling: true,
  28. interval: 100
  29. }
  30. }
  31. })
  32. // use vite's connect instance as middleware
  33. app.use(vite.middlewares)
  34. } else {
  35. app.use(require('compression')())
  36. app.use(
  37. require('serve-static')(resolve('dist/client'), {
  38. index: false
  39. })
  40. )
  41. }
  42. app.use('*', async (req, res) => {
  43. try {
  44. const url = req.originalUrl
  45. let template, render
  46. if (!isProd) {
  47. // always read fresh template in dev
  48. template = fs.readFileSync(resolve('index.html'), 'utf-8')
  49. template = await vite.transformIndexHtml(url, template)
  50. render = (await vite.ssrLoadModule('/src/entry-server.js')).render
  51. } else {
  52. template = indexProd
  53. // @ts-ignore
  54. render = require('./dist/server/entry-server.js').render
  55. }
  56. const [appHtml, preloadLinks] = await render(url, manifest)
  57. const html = template
  58. .replace(`<!--preload-links-->`, preloadLinks)
  59. .replace(`<!--app-html-->`, appHtml)
  60. res.status(200).set({ 'Content-Type': 'text/html' }).end(html)
  61. } catch (e) {
  62. vite && vite.ssrFixStacktrace(e)
  63. console.log(e.stack)
  64. res.status(500).end(e.stack)
  65. }
  66. })
  67. return { app, vite }
  68. }
  69. if (!isTest) {
  70. createServer().then(({ app }) =>
  71. app.listen(5173, () => {
  72. console.log('http://localhost:5173')
  73. })
  74. )
  75. }
  76. // for test use
  77. exports.createServer = createServer