From 4653793eeac33f085744eeb4a3610eebc3b86e7e Mon Sep 17 00:00:00 2001
From: Antonio De Lucreziis <antonio.delucreziis@gmail.com>
Date: Fri, 1 Jul 2022 15:46:26 +0200
Subject: [PATCH] Fixed HMR

---
 _frontend/server.js | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/_frontend/server.js b/_frontend/server.js
index ef37a31..6d04ea0 100644
--- a/_frontend/server.js
+++ b/_frontend/server.js
@@ -2,6 +2,7 @@ import { dirname, resolve } from 'path'
 import express from 'express'
 import { createServer as createViteServer } from 'vite'
 import { fileURLToPath } from 'url'
+import { readFile } from 'fs/promises'
 
 import routes from './routes.js'
 
@@ -17,11 +18,15 @@ async function createServer(customHtmlRoutes) {
     })
 
     for (const [route, file] of Object.entries(customHtmlRoutes)) {
-        app.get(route, (req, res) => {
+        app.get(route, async (req, res) => {
             const filePath = resolve(__dirname, file)
             console.log(`Custom Route: %s`, req.url)
 
-            return res.sendFile(filePath)
+            const htmlFile = await readFile(filePath, 'utf8')
+            const htmlViteHooksFile = await vite.transformIndexHtml(req.originalUrl, htmlFile)
+
+            res.setHeader('Content-Type', 'text/html')
+            return res.send(htmlViteHooksFile)
         })
     }