add AbstractRestExceptionHandler

Author: zeroleak

src/main/java/com/samourai/javaserver/exceptions/NotifiableException.java

@@ -0,0 +1,37 @@
+package com.samourai.javaserver.exceptions;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.http.HttpStatus;
+
+public class NotifiableException extends Exception {
+  private static final Logger log = LoggerFactory.getLogger(NotifiableException.class);
+  private static final HttpStatus STATUS_DEFAULT = HttpStatus.INTERNAL_SERVER_ERROR;
+
+  private HttpStatus httpStatus;
+
+  public NotifiableException(String message) {
+    this(message, STATUS_DEFAULT);
+  }
+
+  public NotifiableException(HttpStatus status) {
+    this(status.getReasonPhrase(), status);
+  }
+
+  public NotifiableException(String message, HttpStatus httpStatus) {
+    super(message);
+    this.httpStatus = httpStatus;
+  }
+
+  public HttpStatus getHttpStatus() {
+    return httpStatus;
+  }
+
+  public static NotifiableException computeNotifiableException(Exception e) {
+    if (NotifiableException.class.isAssignableFrom(e.getClass())) {
+      return (NotifiableException) e;
+    }
+    log.warn("Exception obfuscated to user", e);
+    return new NotifiableException("Error");
+  }
+}

src/main/java/com/samourai/javaserver/rest/AbstractRestExceptionHandler.java

@@ -0,0 +1,31 @@
+package com.samourai.javaserver.rest;
+
+import com.samourai.javaserver.exceptions.NotifiableException;
+import org.springframework.http.HttpHeaders;
+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.context.request.WebRequest;
+import org.springframework.web.servlet.mvc.method.annotation.ResponseEntityExceptionHandler;
+
+public abstract class AbstractRestExceptionHandler extends ResponseEntityExceptionHandler {
+  protected abstract Object handleError(NotifiableException e);
+
+  @ExceptionHandler(value = {Exception.class})
+  public ResponseEntity<Object> handleException(Exception e) {
+    NotifiableException notifiable = NotifiableException.computeNotifiableException(e);
+    Object restErrorResponse = handleError(notifiable);
+    return new ResponseEntity<>(restErrorResponse, notifiable.getHttpStatus());
+  }
+
+  @Override
+  protected ResponseEntity<Object> handleMethodArgumentNotValid(
+      MethodArgumentNotValidException e,
+      HttpHeaders headers,
+      HttpStatus status,
+      WebRequest request) {
+    return handleException(
+        new NotifiableException("Invalid parameter: " + e.getParameter().getParameterName()));
+  }
+}