CleanArchitecture/CleanArchitecture/CleanArchitecture.API/Middlewares/ExceptionMiddleware.cs

using CleanArchitecture.API.Errors;
using CleanArchitecture.Application.Exceptions;
using System.Text.Json;

namespace CleanArchitecture.API.Middlewares
{
    public class ExceptionMiddleware
    {
        private readonly RequestDelegate _next;
        private readonly ILogger<ExceptionMiddleware> _logger;
        private readonly IWebHostEnvironment _env;

        public ExceptionMiddleware(RequestDelegate next, ILogger<ExceptionMiddleware> logger, IWebHostEnvironment env)
        {
            _next = next;
            _logger = logger;
            _env = env;
        }

        public async Task InvokeAsync(HttpContext context)
        {
            try
            {
                await _next(context);
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, ex.Message);
                context.Response.ContentType = "application/json";
                var statusCode = StatusCodes.Status500InternalServerError;
                var result = string.Empty;

                switch (ex)
                {
                    case NotFoundException notFoundException:
                        statusCode = StatusCodes.Status404NotFound;
                        break;
                    case ValidationException validationException:
                        statusCode = StatusCodes.Status400BadRequest;
                        var validationJson = JsonSerializer.Serialize(validationException.Errors);
                        result = JsonSerializer.Serialize(new CodeErrorException(statusCode, ex.Message, validationJson));
                        break;
                    case BadRequestException badRequestException:
                        statusCode = StatusCodes.Status400BadRequest;
                        break;
                    default:                        
                        break;
                }

                if(string.IsNullOrEmpty(result))
                {
                   result = JsonSerializer.Serialize(new CodeErrorException(statusCode, ex.Message, ex.StackTrace));
                }

                context.Response.StatusCode = statusCode;

                await context.Response.WriteAsync(result);

            }
        }
    }
}