Changeset 7d43957 for src/main/java/com/example/autopartz/controller/PartController.java

Timestamp:

01/05/23 01:31:58 (23 months ago)

Author:

andrejtodorovski <82031894+andrejtodorovski@…>

Branches:

main

Children:

676144b

Parents:

ab952ab

Message:

Added functionalities

File:

• src/main/java/com/example/autopartz/controller/PartController.java (modified) (4 diffs)

src/main/java/com/example/autopartz/controller/PartController.java

@@ -1 +1 @@
 package com.example.autopartz.controller;
 
+import com.example.autopartz.model.Client;
+import com.example.autopartz.model.Order;
 import com.example.autopartz.model.Part;
-import com.example.autopartz.service.PartService;
-import com.example.autopartz.service.PriceService;
-import com.example.autopartz.service.RepairShopService;
+import com.example.autopartz.model.User;
+import com.example.autopartz.model.manytomany.OrderContainsPart;
+import com.example.autopartz.repository.OrderContainsPartRepository;
+import com.example.autopartz.service.*;
 import org.springframework.stereotype.Controller;
 import org.springframework.ui.Model;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import javax.servlet.http.HttpSession;
 import java.io.IOException;
 
@@ -18 +23 @@
     private final RepairShopService repairShopService;
     private final PriceService priceService;
-    public PartController(PartService partService, RepairShopService repairShopService, PriceService priceService) {
+    private final OrderService orderService;
+    private final UserService userService;
+    private final OrderContainsPartRepository orderContainsPartRepository;
+    public PartController(PartService partService, RepairShopService repairShopService, PriceService priceService, OrderService orderService, UserService userService, OrderContainsPartRepository orderContainsPartRepository) {
         this.partService = partService;
         this.repairShopService = repairShopService;
         this.priceService = priceService;
+        this.orderService = orderService;
+        this.userService = userService;
+        this.orderContainsPartRepository = orderContainsPartRepository;
     }
     @GetMapping("/{id}")
@@ -32 +43 @@
         return "master-template";
     }
-    @GetMapping("/delivery/{id}")
-    public String getDeliveryPage(@PathVariable Integer id, Model model){
+    @GetMapping("/delivery")
+    public String getDeliveryPage(Model model){
         model.addAttribute("repairShops",repairShopService.findAll());
-        model.addAttribute("partId",id);
         model.addAttribute("bodyContent","deliveryForPart");
         return "master-template";
@@ -57 +67 @@
         }
     }
+    @PostMapping("/addToOrder/{id}")
+    public void addToOrder(@PathVariable Integer id,@RequestParam Integer quantity, HttpSession session, HttpServletResponse response, HttpServletRequest request){
+        if(session.getAttribute("order")==null){
+            User u = userService.findByUsername(request.getRemoteUser());
+            Order newOrder = orderService.create((Client) u);
+            session.setAttribute("order",newOrder);
+        }
+        Order order = (Order) session.getAttribute("order");
+        orderContainsPartRepository.save(new OrderContainsPart(id,order.getID_order(),quantity));
+        try {
+            response.sendRedirect("/products");
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
 }