ApiController.java
- package edu.ucsb.cs156.happiercows.controllers;
- import edu.ucsb.cs156.happiercows.errors.EntityNotFoundException;
- import edu.ucsb.cs156.happiercows.errors.NoCowsException;
- import edu.ucsb.cs156.happiercows.errors.NotEnoughMoneyException;
- import org.springframework.beans.factory.annotation.Autowired;
- import edu.ucsb.cs156.happiercows.models.CurrentUser;
- import edu.ucsb.cs156.happiercows.services.CurrentUserService;
- import org.springframework.http.HttpStatus;
- import org.springframework.web.bind.annotation.ExceptionHandler;
- import org.springframework.web.bind.annotation.ResponseStatus;
- import java.util.Map;
- public abstract class ApiController {
- @Autowired
- private CurrentUserService currentUserService;
- protected CurrentUser getCurrentUser() {
- return currentUserService.getCurrentUser();
- }
-
- protected Object genericMessage(String message) {
- return Map.of("message", message);
- }
- @ExceptionHandler({ EntityNotFoundException.class })
- @ResponseStatus(HttpStatus.NOT_FOUND)
- public Object handleGenericException(Throwable e) {
- return Map.of(
- "type", e.getClass().getSimpleName(),
- "message", e.getMessage()
- );
- }
- @ExceptionHandler({ NoCowsException.class, NotEnoughMoneyException.class})
- @ResponseStatus(HttpStatus.BAD_REQUEST)
- public Object handleBadRequest(Throwable e) {
- return Map.of(
- "type", e.getClass().getSimpleName(),
- "message", e.getMessage()
- );
- }
- }