no-copy/src/main/java/ru/soune/no_copy/handler/GlobalExceptionHandler.java

package ru.soune.no_copy.handler;

import lombok.AllArgsConstructor;
import org.springframework.context.support.DefaultMessageSourceResolvable;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseStatus;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import ru.soune.no_copy.exception.*;

import java.util.Map;

@RestControllerAdvice
@AllArgsConstructor
public class GlobalExceptionHandler {

    @ExceptionHandler(MethodArgumentNotValidException.class)
    public ResponseEntity<?> handleValidationException(MethodArgumentNotValidException ex) {
        String message = ex.getBindingResult().getFieldErrors().stream()
                .map(DefaultMessageSourceResolvable::getDefaultMessage)
                .findFirst()
                .orElse("");

        return ResponseEntity.badRequest()
                .body(Map.of(
                        "success", false,
                        "message" ,message));
    }

    @ExceptionHandler(NotValidFieldException.class)
    @ResponseStatus(HttpStatus.OK)
    public ResponseEntity<?> handleUserContainsException(NotValidFieldException ex) {

        return ResponseEntity
                .ok()
                .body(Map.of(
                        "msg_id", ex.getMessageId(),
                        "message_code", ex.getMessageCode(),
                        "message_desc", ex.getMessageDescription()));
    }

    @ExceptionHandler(UserNotFoundException.class)
    @ResponseStatus(HttpStatus.NOT_FOUND)
    public ResponseEntity<?> handleUserNotFoundException(UserNotFoundException ex) {
        return ResponseEntity.
                badRequest()
                .body(Map.of(
                        "success", false,
                        "message" ,ex.getMessage()
                ));
    }

    @ExceptionHandler(ContentNotFoundException.class)
    @ResponseStatus(HttpStatus.NOT_FOUND)
    public ResponseEntity<?> handleUserNotFoundException(ContentNotFoundException ex) {
        return ResponseEntity.
                badRequest()
                .body(Map.of(
                        "success", false,
                        "message" ,ex.getMessage()
                ));
    }

    @ExceptionHandler(NotValidationPasswordException.class)
    @ResponseStatus(HttpStatus.FORBIDDEN)
    public ResponseEntity<?> handleNotValidationPasswordException(NotValidationPasswordException ex) {
        return ResponseEntity.
                badRequest()
                .body(Map.of(
                        "success", false,
                        "message" ,ex.getMessage()
                ));
    }

    @ExceptionHandler(TokenNotFoundException.class)
    @ResponseStatus(HttpStatus.NOT_FOUND)
    public ResponseEntity<?> handleNotFoundTokenException(TokenNotFoundException ex) {
        return ResponseEntity.
                badRequest()
                .body(Map.of(
                        "success", false,
                        "message" ,ex.getMessage()
                ));
    }
}