fix: handle file not found error

vclemenzi · Mar 20, 2024 · 491c690 · 491c690
1 parent e01b5af
commit 491c690
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/server/create.ts b/src/server/create.ts
@@ -14,11 +14,15 @@ export const createServer = (server: ViteDevServer) => {
     if (!req.url) return;
 
     if (!/\.[a-z]+$/.test(req.url)) {
+      if (!fs.existsSync(`dist${req.url}/index.html`)) return res.end("404");
+
       return res.end(fs.readFileSync(`dist${req.url}/index.html`));
     }
 
     const ext = req.url.split(".").pop() || "";
 
+    if (!fs.existsSync(`dist${req.url}`)) return res.end("404");
+
     res.setHeader("Content-Type", mime.getType(ext) || "text/plain");
     return res.end(fs.readFileSync(`dist${req.url}`));
   });