diff --git a/src/main/java/catering/order/OrderController.java b/src/main/java/catering/order/OrderController.java index a1b21c7..b854d93 100644 --- a/src/main/java/catering/order/OrderController.java +++ b/src/main/java/catering/order/OrderController.java @@ -173,6 +173,11 @@ public class OrderController { @PostMapping("/event/addProduct") @PreAuthorize("hasRole('CUSTOMER')") public String addProduct(@RequestParam("pid") Product product, @RequestParam("number") int number, @ModelAttribute("event") CustomCart cart) { + // check if product is suitable + if (product.getCategories().stream().noneMatch(c -> c.equals(cart.getOrderType().toString()))) { + return "redirect:/event"; + } + Quantity amount = Quantity.of(number > 0 ? number : 1); Quantity cartQuantity = cart.getQuantity(product); Quantity available;