UCSBDiningMenuItemsController.java

package edu.ucsb.cs156.dining.controllers;

import edu.ucsb.cs156.dining.entities.MenuItem;
import edu.ucsb.cs156.dining.errors.EntityNotFoundException;
import edu.ucsb.cs156.dining.models.Entree;
import edu.ucsb.cs156.dining.models.MenuItemDto;
import edu.ucsb.cs156.dining.repositories.MenuItemRepository;
import edu.ucsb.cs156.dining.services.UCSBDiningMenuItemsService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.transaction.Transactional;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.StreamSupport;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@Tag(name = "UCSBDiningMenuItems")
@RestController
@RequestMapping("/api/diningcommons")
@Slf4j
public class UCSBDiningMenuItemsController extends ApiController {

  @Autowired UCSBDiningMenuItemsService ucsbDiningMenuItemsService;

  @Autowired MenuItemRepository menuItemRepository;

  @Operation(summary = "Get list of entrees being served at given meal, dining common, and day")
  @GetMapping(
      value = "/{date-time}/{dining-commons-code}/{meal-code}",
      produces = "application/json")
  @Transactional
  public ResponseEntity<List<MenuItemDto>> get_menu_items(
      @Parameter(
              description =
                  "date (in iso format, e.g. YYYY-mm-dd) or date-time (in iso format e.g. YYYY-mm-ddTHH:MM:SS)")
          @PathVariable("date-time")
          String datetime,
      @PathVariable("dining-commons-code") String diningcommoncode,
      @PathVariable("meal-code") String mealcode)
      throws Exception {

    List<Entree> body = ucsbDiningMenuItemsService.get(datetime, diningcommoncode, mealcode);

    List<MenuItemDto> menuitems =
        menuItemRepository.projectExistingEntrees(diningcommoncode, mealcode, body);

    Map<String, MenuItemDto> existingMenuItemsMap =
        menuitems.stream()
            .collect(Collectors.toMap(m -> m.name() + "/" + m.station(), Function.identity()));

    List<MenuItem> newMenuItems = new ArrayList<>(50);

    for (Entree entree : body) {
      if (existingMenuItemsMap.containsKey(entree.getName() + "/" + entree.getStation())) {
        continue;
      }

      MenuItem newMenuItem =
          MenuItem.builder()
              .diningCommonsCode(diningcommoncode)
              .mealCode(mealcode)
              .name(entree.getName())
              .station(entree.getStation())
              .build();
      newMenuItems.add(newMenuItem);
    }

    Iterable<MenuItem> savedMenuItems = menuItemRepository.saveAll(newMenuItems);
    StreamSupport.stream(savedMenuItems.spliterator(), false)
        .map(
            menuItem ->
                new MenuItemDto(
                    menuItem.getId(),
                    menuItem.getDiningCommonsCode(),
                    menuItem.getMealCode(),
                    menuItem.getName(),
                    menuItem.getStation(),
                    null))
        .forEach(menuitems::add);

    return ResponseEntity.ok().body(menuitems);
  }

  @Operation(summary = "Get a single menu item by ID")
  @GetMapping(value = "/menuitem", produces = "application/json")
  public ResponseEntity<MenuItem> get_menu_item_by_id(
      @Parameter(description = "ID of the menu item") @RequestParam long id) throws Exception {
    Optional<MenuItem> menuItem = menuItemRepository.findById(id);
    if (menuItem.isEmpty()) {
      throw new EntityNotFoundException(MenuItem.class, id);
    }
    return ResponseEntity.ok().body(menuItem.get());
  }
}