feat: redirect when missing route to web page

Asadaaaaa · Oct 6, 2023 · 71674ff · 71674ff
1 parent 122ebce
commit 71674ff
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/Main.js b/src/Main.js
@@ -57,9 +57,12 @@ class Server {
 
   run() {
     this.API = Express();
+    const middlewareHandler = new MiddlewareHandler(this);
 
-    new MiddlewareHandler(this);
+    middlewareHandler.global();
     new RouteHandler(this);
+    middlewareHandler.missingRoute();
+
 
     this.API.listen(this.env.PORT, this.env.IP, () => this.sendLogs('Server Started, Listening PORT ' + this.env.PORT));
   }

diff --git a/src/middlewares/Handler.middleware.js b/src/middlewares/Handler.middleware.js
@@ -11,8 +11,6 @@ class Handler {
       this.API = this.server.API;
 
       this.ResponsePreset = new ResponsePreset();
-
-      this.global();
     }
 
     global() {
@@ -107,6 +105,13 @@ class Handler {
         }));
       }
     }
+
+    missingRoute() {
+      this.API.get('*', (req, res) => {
+        // Redirect to the desired URL
+        res.redirect(`${this.server.env.WEB_HOST}/not-found`);
+      });
+    }
 }
 
 export default Handler;