feat: add middleware for custom exception

Endpoint/Common/Attributes/TokenAuthenticationAttribute.cs

@@ -0,0 +1,27 @@
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.AspNetCore.Mvc.Filters;
+using Microsoft.Extensions.DependencyInjection;
+using Mirea.Api.Endpoint.Common.Interfaces;
+using System;
+
+namespace Mirea.Api.Endpoint.Common.Attributes;
+
+[AttributeUsage(AttributeTargets.Method)]
+public class TokenAuthenticationAttribute : Attribute, IActionFilter
+{
+    public void OnActionExecuting(ActionExecutingContext context)
+    {
+        var setupToken = context.HttpContext.RequestServices.GetRequiredService<ISetupToken>();
+        if (!context.HttpContext.Request.Cookies.TryGetValue("AuthToken", out string? tokenFromCookie))
+        {
+            context.Result = new UnauthorizedResult();
+            return;
+        }
+
+        if (setupToken.MatchToken(Convert.FromBase64String(tokenFromCookie))) return;
+
+        context.Result = new UnauthorizedResult();
+    }
+
+    public void OnActionExecuted(ActionExecutedContext context) { }
+}

Endpoint/Common/Exceptions/ControllerArgumentException.cs

@@ -0,0 +1,5 @@
+using System;
+
+namespace Mirea.Api.Endpoint.Common.Exceptions;
+
+public class ControllerArgumentException(string message) : Exception(message);

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();