Initial commit

src/main/java/com/mattrixwv/cipherstream/controller/ExceptionRestController.java

@@ -0,0 +1,52 @@
+package com.mattrixwv.cipherstream.controller;
+
+
+import org.springframework.http.HttpStatus;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.ResponseBody;
+import org.springframework.web.bind.annotation.ResponseStatus;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.RestControllerAdvice;
+import org.springframework.web.servlet.mvc.method.annotation.ResponseEntityExceptionHandler;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.mattrixwv.cipherstream.exception.InvalidCipherParameterException;
+
+import lombok.extern.slf4j.Slf4j;
+
+
+@Slf4j
+@RestControllerAdvice(annotations = RestController.class)
+public class ExceptionRestController extends ResponseEntityExceptionHandler{
+	private static final ObjectMapper mapper = new ObjectMapper();
+
+
+	@ExceptionHandler(InvalidCipherParameterException.class)
+	@ResponseStatus(HttpStatus.BAD_REQUEST)
+	@ResponseBody
+	public ObjectNode invalidCipherParameterHandler(InvalidCipherParameterException error){
+		log.warn("Invalid user input was supplied: {}", error.getMessage());
+
+
+		ObjectNode returnJson = mapper.createObjectNode();
+		returnJson.put("message", error.getMessage());
+
+
+		return returnJson;
+	}
+
+	@ExceptionHandler(Exception.class)
+	@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
+	@ResponseBody
+	public ObjectNode genericExceptionHandler(Exception error){
+		log.error(error.getMessage(), error);
+
+
+		ObjectNode returnJson = mapper.createObjectNode();
+		returnJson.put("message", error.getMessage());
+
+
+		return returnJson;
+	}
+}