From 9769e2cf6368837a6dd69f6233bf9687c72f9fc3 Mon Sep 17 00:00:00 2001
From: Mr Swift <amirshayan1381@yahoo.com>
Date: Sun, 11 Aug 2024 18:26:18 +0330
Subject: [PATCH] set error handler

---
 server/constants.js    | 10 ++++++
 server/errorHandler.js | 73 ++++++++++++++++++++++++++++++++++++++++++
 server/index.js        |  2 ++
 3 files changed, 85 insertions(+)
 create mode 100644 server/constants.js
 create mode 100644 server/errorHandler.js

diff --git a/server/constants.js b/server/constants.js
new file mode 100644
index 0000000..6ba4bcb
--- /dev/null
+++ b/server/constants.js
@@ -0,0 +1,10 @@
+exports.constants = {
+  VALIDATION_ERROR: 400,
+  UNAUTHORIZED: 401,
+  FORBIDOEN: 403,
+  NOT_FOUND: 404,
+  SERVER_ERROR: 500,
+  SITE_UNAUTHORIZED: 422,
+  TOO_MANY: 429,
+  Method_Not_Allowed:405
+};
diff --git a/server/errorHandler.js b/server/errorHandler.js
new file mode 100644
index 0000000..5d52dd0
--- /dev/null
+++ b/server/errorHandler.js
@@ -0,0 +1,73 @@
+const { constants } = require("./constants");
+const errorHandler = (err, req, res, next) => {
+  const statusCode = res.statusCode ? res.statusCode : 500;
+
+  switch (statusCode) {
+    case constants.VALIDATION_ERROR:
+      res.status(statusCode).json({
+        title: "Validation Failed ",
+        msg: err.message,
+        stackTrace: err.stack,
+      });
+      break;
+      case constants.Method_Not_Allowed:
+        res.status(statusCode).json({
+          title: "Method Not Allowed ",
+          msg: err.message,
+          stackTrace: err.stack,
+        });
+        break;
+    case constants.NOT_FOUND:
+      res.status(statusCode).json({
+        title: "Not found ",
+        msg: err.message,
+        stackTrace: err.stack,
+      });
+      break;
+    case constants.FORBIDOEN:
+      res.status(statusCode).json({
+        title: "Forbidoen ",
+        msg: err.message,
+        stackTrace: err.stack,
+      });
+      break;
+    case constants.SERVER_ERROR:
+      res.status(statusCode).json({
+        title: "Server Error ",
+        msg: err.message,
+        stackTrace: err.stack,
+      });
+      break;
+    case constants.UNAUTHORIZED:
+      res.status(statusCode).json({
+        title: "Unauthorized ",
+        msg: err.message,
+        stackTrace: err.stack,
+      });
+      break;
+    case constants.SITE_UNAUTHORIZED:
+      res.status(statusCode).json({
+        title: "Site Unauthorized ",
+        msg: err.message,
+        stackTrace: err.stack,
+      });
+      break;
+    case constants.TOO_MANY:
+      res.status(statusCode).json({
+        title: "Too many request",
+        msg: err.message,
+        stackTrace: err.stack,
+      });
+      break;
+    default:
+      console.log(err);
+      res.status(500).json({ 
+       title: "Unhandled error",
+       msg: err.message,
+       stackTrace: err.stack 
+      });
+      break;
+  }
+};
+
+module.exports = { name: "errorHandler", run: errorHandler };
diff --git a/server/index.js b/server/index.js
index 11edb15..6e0dd63 100644
--- a/server/index.js
+++ b/server/index.js
@@ -1,5 +1,6 @@
 /// / develop environment config
 const { isDev, isProduction, DevServerPort } = require('./_env')
+const errorHandler = require('./errorHandler')
 const wss = require('./WebSocket/wss')
 /// / init express app (wrapping all functions into init() to prevent running in development mode by nuxt dev mode)
 function init() {
@@ -53,6 +54,7 @@ function init() {
 
   app.use(isDev ? '/api' : '', router)
 
+  app.use(errorHandler.run)
 
   return app
 }