From 90e0126d4078b5fef44b085b34caf7a35161ff50 Mon Sep 17 00:00:00 2001
From: transtrike <transtrike@gmail.com>
Date: Tue, 19 Jan 2021 21:12:49 +0200
Subject: Added Custom Middleware for Exception Handling

---
 .../Models/Middleware/ExceptionMiddleware.cs       | 50 ++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 src/DevHive.Web/Models/Middleware/ExceptionMiddleware.cs

(limited to 'src/DevHive.Web/Models/Middleware/ExceptionMiddleware.cs')

diff --git a/src/DevHive.Web/Models/Middleware/ExceptionMiddleware.cs b/src/DevHive.Web/Models/Middleware/ExceptionMiddleware.cs
new file mode 100644
index 0000000..d952ff4
--- /dev/null
+++ b/src/DevHive.Web/Models/Middleware/ExceptionMiddleware.cs
@@ -0,0 +1,50 @@
+using System;
+using System.Net;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.Http;
+using Microsoft.Extensions.Logging;
+
+namespace DevHive.Web.Models.Middleware
+{
+	public class ExceptionMiddleware
+	{
+		private readonly RequestDelegate _next;
+		// private readonly ILogger _logger;
+
+		public ExceptionMiddleware(RequestDelegate next)
+		{
+			this._next = next;
+			// this._logger = logger;
+		}
+		// public ExceptionMiddleware(RequestDelegate next, ILogger logger)
+		// {
+		// 	this._logger = logger;
+		// 	this._next = next;
+		// }
+
+		public async Task InvokeAsync(HttpContext httpContext)
+		{
+			try
+			{
+				await this._next(httpContext);
+			}
+			catch (Exception ex)
+			{
+				// this._logger.LogError($"Something went wrong: {ex}");
+				await HandleExceptionAsync(httpContext, ex);
+			}
+		}
+
+		private Task HandleExceptionAsync(HttpContext context, Exception exception)
+		{
+			context.Response.ContentType = "application/json";
+			context.Response.StatusCode = (int)HttpStatusCode.BadRequest;
+
+			return context.Response.WriteAsync(new
+			{
+				StatusCode = context.Response.StatusCode,
+				Message = "Internal Server Error from the custom middleware."
+			}.ToString());
+		}
+	}
+}
-- 
cgit v1.2.3