phelomi · naremloa · Mar 2, 2021
diff --git a/index.html b/index.html
@@ -2,12 +2,12 @@
 <html lang="en">
   <head>
     <meta charset="UTF-8" />
-    <link rel="icon" href="/favicon.ico" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Vite App</title>
+    <!--preload-links-->
   </head>
   <body>
-    <div id="app"></div>
-    <script type="module" src="/src/main.js"></script>
+    <div id="app"><!--app-html--></div>
+    <script type="module" src="/src/entry-client.js"></script>
   </body>
 </html>
diff --git a/package.json b/package.json
@@ -2,12 +2,19 @@
   "name": "phelomi",
   "version": "0.0.0",
   "scripts": {
-    "dev": "vite",
-    "build": "vite build",
-    "serve": "vite preview"
+    "start": "cross-env NODE_ENV=production node server",
+    "dev": "node server",
+    "build": "yarn build:client && yarn build:server",
+    "build:client": "vite build --ssrManifest --outDir dist/client",
+    "build:server": "vite build --ssr src/entry-server.js --outDir dist/server"
   },
   "dependencies": {
+    "@vue/server-renderer": "^3.0.6",
     "axios": "^0.21.1",
+    "compression": "^1.7.4",
+    "cross-env": "^7.0.3",
+    "express": "^4.17.1",
+    "serve-static": "^1.14.1",
     "swiper": "^6.4.14",
     "vue": "^3.0.5",
     "vue-router": "4"

diff --git a/server.js b/server.js
@@ -0,0 +1,81 @@
+/* eslint-disable import/no-extraneous-dependencies */
+/* eslint-disable global-require */
+/* eslint-disable import/no-unresolved */
+const fs = require('fs');
+const path = require('path');
+const express = require('express');
+
+const PORT = process.env.PORT || 8080;
+
+async function createServer(
+  root = process.cwd(),
+  isProd = process.env.NODE_ENV === 'production',
+) {
+  const resolve = (p) => path.resolve(__dirname, p);
+
+  const indexProd = isProd
+    ? fs.readFileSync(resolve('dist/client/index.html'), 'utf-8')
+    : '';
+
+  const manifest = isProd
+    ? require('./dist/client/ssr-manifest.json')
+    : {};
+
+  const app = express();
+
+  let vite;
+  if (!isProd) {
+    vite = await require('vite').createServer({
+      root,
+      logLevel: 'info',
+      server: {
+        middlewareMode: true,
+      },
+    });
+    app.use(vite.middlewares);
+  } else {
+    app.use(require('compression')());
+    app.use(
+      require('serve-static')(resolve('dist/client'), {
+        index: false,
+      }),
+    );
+  }
+
+  app.use('*', async (req, res) => {
+    try {
+      const url = req.originalUrl;
+
+      let template;
+      let render;
+      if (!isProd) {
+      // always read fresh template in dev
+        template = fs.readFileSync(resolve('./index.html'), 'utf-8');
+        template = await vite.transformIndexHtml(url, template);
+        render = (await vite.ssrLoadModule('/src/entry-server.js')).render;
+      } else {
+        template = indexProd;
+        render = require('./dist/server/entry-server.js').render;
+      }
+
+      const [appHtml, preloadLinks] = await render(url, manifest);
+
+      const html = template
+        .replace('<!--preload-links-->', preloadLinks)
+        .replace('<!--app-html-->', appHtml);
+
+      res.status(200).set({ 'Content-Type': 'text/html' }).end(html);
+    } catch (e) {
+      if (vite) vite.ssrFixStacktrace(e);
+      console.log(e.stack);
+      res.status(500).end(e.stack);
+    }
+  });
+
+  return { app, vite };
+}
+
+createServer().then(({ app }) => app.listen(PORT, () => {
+  console.log(`App listening on port ${PORT}`);
+  console.log('Press Ctrl+C to quit.');
+}));
diff --git a/src/entry-client.js b/src/entry-client.js
@@ -0,0 +1,8 @@
+import { createApp } from './main';
+
+const { app, router } = createApp();
+
+// wait until router is ready before mounting to ensure hydration match
+router.isReady().then(() => {
+  app.mount('#app');
+});
diff --git a/src/entry-server.js b/src/entry-server.js
@@ -0,0 +1,52 @@
+/* eslint-disable import/no-extraneous-dependencies */
+import { renderToString } from '@vue/server-renderer';
+import { createApp } from './main';
+
+function renderPreloadLink(file) {
+  if (file.endsWith('.js')) {
+    return `<link rel="modulepreload" crossorigin href="${file}">`;
+  }
+  if (file.endsWith('.css')) {
+    return `<link rel="stylesheet" href="${file}">`;
+  }
+  // TODO
+  return '';
+}
+
+function renderPreloadLinks(modules, manifest) {
+  let links = '';
+  const seen = new Set();
+  modules.forEach((id) => {
+    const files = manifest[id];
+    if (files) {
+      files.forEach((file) => {
+        if (!seen.has(file)) {
+          seen.add(file);
+          links += renderPreloadLink(file);
+        }
+      });
+    }
+  });
+  return links;
+}
+
+export async function render(url, manifest) {
+  const { app, router } = createApp();
+
+  // set the router to the desired URL before rendering
+  router.push(url);
+  await router.isReady();
+
+  // passing SSR context object which will be available via useSSRContext()
+  // @vitejs/plugin-vue injects code into a component's setup() that registers
+  // itself on ctx.modules. After the render, ctx.modules would contain all the
+  // components that have been instantiated during this render call.
+  const ctx = {};
+  const html = await renderToString(app, ctx);
+
+  // the SSR manifest generated by Vite contains module -> chunk/asset mapping
+  // which we can then use to determine what files need to be preloaded for this
+  // request.
+  const preloadLinks = renderPreloadLinks(ctx.modules, manifest);
+  return [html, preloadLinks];
+}
diff --git a/src/main.js b/src/main.js
@@ -1,9 +1,19 @@
-import { createApp } from 'vue';
+import { createSSRApp } from 'vue';
 import { createRouter } from './router';
 import App from './App.vue';
 import './styles/index.css';
 
-const app = createApp(App);
-const router = createRouter();
-app.use(router);
-app.mount('#app');
+// SSR requires a fresh app instance per request, therefore we export a function
+// that creates a fresh app instance. If using Vuex, we'd also be creating a
+// fresh store here.
+
+// import { worker } from './mocks/browser';
+
+// worker.start();
+
+export function createApp() {
+  const app = createSSRApp(App);
+  const router = createRouter();
+  app.use(router);
+  return { app, router };
+}