ApiController.java
package edu.ucsb.cs156.organic.controllers;
import org.springframework.beans.factory.annotation.Autowired;
import edu.ucsb.cs156.organic.errors.EntityNotFoundException;
import edu.ucsb.cs156.organic.models.CurrentUser;
import edu.ucsb.cs156.organic.services.CurrentUserService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseStatus;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.introspect.AnnotatedMember;
import com.fasterxml.jackson.databind.introspect.JacksonAnnotationIntrospector;
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
import java.util.Map;
@Slf4j
public abstract class ApiController {
@Autowired
private CurrentUserService currentUserService;
protected CurrentUser getCurrentUser() {
return currentUserService.getCurrentUser();
}
@ExceptionHandler({ IllegalArgumentException.class })
@ResponseStatus(HttpStatus.BAD_REQUEST)
public Object handleIllegalArgumentException(Throwable e) {
Map<String, String> map = Map.of(
"type", e.getClass().getSimpleName(),
"message", e.getMessage());
log.error("Exception thrown: {}", map);
return map;
}
@ExceptionHandler({ EntityNotFoundException.class })
@ResponseStatus(HttpStatus.NOT_FOUND)
public Object handleGenericException(Throwable e) {
return Map.of(
"type", e.getClass().getSimpleName(),
"message", e.getMessage());
}
@ExceptionHandler({ AccessDeniedException.class })
@ResponseStatus(HttpStatus.FORBIDDEN)
public Object handleAccessDeniedException(Throwable e) {
return Map.of(
"type", e.getClass().getSimpleName(),
"message", e.getMessage());
}
private ObjectMapper mapper;
/**
* Special ObjectMapper that ignores Mockito mocks
*
* @return ObjectMapper mapper
*/
public ObjectMapper getMapper() {
return mapper;
}
public ApiController() {
mapper = mapperThatIgnoresMockitoMocks();
}
public static ObjectMapper mapperThatIgnoresMockitoMocks() {
ObjectMapper mapper = new ObjectMapper();
mapper.registerModule(new JavaTimeModule());
mapper.setAnnotationIntrospector(new JacksonAnnotationIntrospector());
return mapper;
}
protected Object genericMessage(String message) {
return Map.of("message", message);
}
}