Endpoint/Middleware/CustomExceptionHandlerMiddleware.cs

@@ -0,0 +1,60 @@
+using FluentValidation;
+using Microsoft.AspNetCore.Http;
+using Mirea.Api.DataAccess.Application.Common.Exceptions;
+using Mirea.Api.Dto.Responses;
+using Mirea.Api.Endpoint.Common.Exceptions;
+using System;
+using System.Text.Json;
+using System.Threading.Tasks;
+
+namespace Mirea.Api.Endpoint.Middleware;
+
+public class CustomExceptionHandlerMiddleware(RequestDelegate next)
+{
+    public async Task Invoke(HttpContext context)
+    {
+        try
+        {
+            await next(context);
+        }
+        catch (Exception exception)
+        {
+            await HandleExceptionAsync(context, exception);
+        }
+    }
+
+    private static Task HandleExceptionAsync(HttpContext context, Exception exception)
+    {
+        var code = StatusCodes.Status500InternalServerError;
+        var result = string.Empty;
+        switch (exception)
+        {
+            case ValidationException validationException:
+                code = StatusCodes.Status400BadRequest;
+                result = JsonSerializer.Serialize(new ErrorResponse()
+                {
+                    Error = validationException.Message,
+                    Code = code
+                });
+                break;
+            case NotFoundException:
+                code = StatusCodes.Status404NotFound;
+                break;
+            case ControllerArgumentException:
+                code = StatusCodes.Status400BadRequest;
+                break;
+        }
+
+        context.Response.ContentType = "application/json";
+        context.Response.StatusCode = code;
+
+        if (string.IsNullOrEmpty(result))
+            result = JsonSerializer.Serialize(new ErrorResponse()
+            {
+                Error = exception.Message,
+                Code = code
+            });
+
+        return context.Response.WriteAsync(result);
+    }
+}

Endpoint/Program.cs

@@ -138,6 +138,7 @@ public class Program
             });
         }
         app.UseMiddleware<MaintenanceModeMiddleware>();
+        app.UseMiddleware<CustomExceptionHandlerMiddleware>();
 
         app.UseHttpsRedirection();