From b80cd78e943b9d11dc6e0bab67bdb236246329ac Mon Sep 17 00:00:00 2001
From: fpasquet <fab.pasquet@gmail.com>
Date: Tue, 12 Mar 2024 12:01:11 +0100
Subject: [PATCH] fix: 404 page (#1071)

---
 src/helpers/prerenderHelper/generateHtmlFiles.ts | 2 +-
 src/helpers/prerenderHelper/index.ts             | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/helpers/prerenderHelper/generateHtmlFiles.ts b/src/helpers/prerenderHelper/generateHtmlFiles.ts
index c93feee03..bc0d9f4d7 100644
--- a/src/helpers/prerenderHelper/generateHtmlFiles.ts
+++ b/src/helpers/prerenderHelper/generateHtmlFiles.ts
@@ -36,7 +36,7 @@ export const generateHtmlFiles = async (options: {
 
     const urlWithoutBaseUrl = url.replace(options.baseUrl, '');
     let fileName = 'index.html';
-    if (urlWithoutBaseUrl === '404') {
+    if (url.match(/\/404\/$/)) {
       fileName = '404.html';
     } else if (urlWithoutBaseUrl) {
       fileName = `${urlWithoutBaseUrl}/index.html`;
diff --git a/src/helpers/prerenderHelper/index.ts b/src/helpers/prerenderHelper/index.ts
index a8d67412f..b98c372dc 100644
--- a/src/helpers/prerenderHelper/index.ts
+++ b/src/helpers/prerenderHelper/index.ts
@@ -16,6 +16,11 @@ export const generateFiles = async (options: { rootDir: string; baseUrl: string
       url: param.url.replace(/^\//, options.baseUrl || '/'),
     }));
 
+  urls.push({
+    lang: 'fr',
+    url: `${options.baseUrl || '/'}404/`,
+  });
+
   await Promise.all([
     generateHtmlFiles({
       rootDir: __dirname,