auth/src/main/java/org/waterproofingdata/wpdauth/exception/GlobalExceptionHandlerController.java

package org.waterproofingdata.wpdauth.exception;

import java.io.IOException;

import javax.servlet.http.HttpServletResponse;

import org.waterproofingdata.wpdauth.exception.CustomErrorAttributes;
import org.waterproofingdata.wpdauth.exception.CustomException;
import org.springframework.boot.web.servlet.error.ErrorAttributes;
import org.springframework.context.annotation.Bean;
import org.springframework.http.HttpStatus;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseEntityExceptionHandler;

@RestControllerAdvice
public class GlobalExceptionHandlerController extends ResponseEntityExceptionHandler {
	  @Bean
	  public ErrorAttributes errorAttributes() {
	    // Hide exception field in the return object
	    return new CustomErrorAttributes();
	  }

	  @ExceptionHandler(CustomException.class)
	  public void handleCustomException(HttpServletResponse res, CustomException ex) throws IOException {
	    res.sendError(ex.getHttpStatus().value(), ex.getMessage());
	  }

	  @ExceptionHandler(AccessDeniedException.class)
	  public void handleAccessDeniedException(HttpServletResponse res) throws IOException {
	    res.sendError(HttpStatus.FORBIDDEN.value(), "Access denied");
	  }

	  @ExceptionHandler(Exception.class)
	  public void handleException(HttpServletResponse res) throws IOException {
	    res.sendError(HttpStatus.BAD_REQUEST.value(), "Something went wrong");
	  }
}