[676144b] | 1 | package com.example.autopartz.controller;
|
---|
| 2 |
|
---|
[f484b14] | 3 | import com.example.autopartz.model.Deliveryman;
|
---|
[676144b] | 4 | import com.example.autopartz.model.Role;
|
---|
| 5 | import com.example.autopartz.model.User;
|
---|
| 6 | import com.example.autopartz.model.Warehouseman;
|
---|
[f484b14] | 7 | import com.example.autopartz.repository.DeliverymanRepository;
|
---|
[676144b] | 8 | import com.example.autopartz.repository.WarehousemanRepository;
|
---|
| 9 | import com.example.autopartz.service.UserService;
|
---|
| 10 | import org.springframework.stereotype.Controller;
|
---|
| 11 | import org.springframework.ui.Model;
|
---|
| 12 | import org.springframework.web.bind.annotation.GetMapping;
|
---|
| 13 | import org.springframework.web.bind.annotation.PathVariable;
|
---|
| 14 | import org.springframework.web.bind.annotation.PostMapping;
|
---|
| 15 | import org.springframework.web.bind.annotation.RequestMapping;
|
---|
| 16 |
|
---|
| 17 | import javax.servlet.http.HttpServletResponse;
|
---|
| 18 | import java.io.IOException;
|
---|
| 19 | import java.time.LocalDate;
|
---|
| 20 | import java.util.List;
|
---|
| 21 | import java.util.Objects;
|
---|
| 22 |
|
---|
| 23 | @Controller
|
---|
| 24 | @RequestMapping("/")
|
---|
| 25 | public class AdminController {
|
---|
| 26 | private final UserService userService;
|
---|
| 27 | private final WarehousemanRepository warehousemanRepository;
|
---|
[f484b14] | 28 | private final DeliverymanRepository deliverymanRepository;
|
---|
[676144b] | 29 |
|
---|
[f484b14] | 30 | public AdminController(UserService userService, WarehousemanRepository warehousemanRepository, DeliverymanRepository deliverymanRepository) {
|
---|
[676144b] | 31 | this.userService = userService;
|
---|
| 32 | this.warehousemanRepository = warehousemanRepository;
|
---|
[f484b14] | 33 | this.deliverymanRepository = deliverymanRepository;
|
---|
[676144b] | 34 | }
|
---|
| 35 |
|
---|
| 36 | @GetMapping("/viewUsers")
|
---|
| 37 | public String getAllUsers(Model model){
|
---|
| 38 | List<User> pendingList = userService.findAllUsers().stream().filter(u->u.getAuthorities().contains(Role.ROLE_PENDING_DELIVERYMAN) || u.getAuthorities().contains(Role.ROLE_PENDING_WAREHOUSEMAN)).toList();
|
---|
[cab5859] | 39 | if(pendingList.size()==0){
|
---|
| 40 | model.addAttribute("hasError",true);
|
---|
| 41 | }
|
---|
| 42 | else {
|
---|
| 43 | model.addAttribute("hasError",false);
|
---|
| 44 | model.addAttribute("users", pendingList);
|
---|
| 45 | }
|
---|
[676144b] | 46 | model.addAttribute("bodyContent", "viewUsers");
|
---|
| 47 | return "master-template";
|
---|
| 48 | }
|
---|
| 49 | @PostMapping("/approve/{id}")
|
---|
| 50 | public void approve(@PathVariable Integer id, HttpServletResponse response){
|
---|
| 51 | if(Objects.equals(userService.findById(id).getAuthorities().stream().findFirst().get(),Role.ROLE_PENDING_WAREHOUSEMAN)){
|
---|
| 52 | Warehouseman wh = (Warehouseman) userService.findById(id);
|
---|
| 53 | wh.setEmployed_from(LocalDate.now());
|
---|
| 54 | warehousemanRepository.save(wh);
|
---|
[f484b14] | 55 |
|
---|
| 56 | }
|
---|
| 57 | else {
|
---|
| 58 | Deliveryman dm = (Deliveryman) userService.findById(id);
|
---|
| 59 | dm.setEmployed_from(LocalDate.now());
|
---|
| 60 | deliverymanRepository.save(dm);
|
---|
[676144b] | 61 | try {
|
---|
| 62 | response.sendRedirect("/viewUsers");
|
---|
| 63 | } catch (IOException e) {
|
---|
| 64 | throw new RuntimeException(e);
|
---|
| 65 | }
|
---|
| 66 | }
|
---|
| 67 | }
|
---|
| 68 | }
|
---|