1 | package it.finki.tinki.web.controller;
|
---|
2 |
|
---|
3 | import it.finki.tinki.model.Users.Account;
|
---|
4 | import it.finki.tinki.model.Users.Team;
|
---|
5 | import it.finki.tinki.model.Users.User;
|
---|
6 | import it.finki.tinki.model.dto.*;
|
---|
7 | import it.finki.tinki.model.dto.response.account.LoginResponseDTO;
|
---|
8 | import it.finki.tinki.service.AccountService;
|
---|
9 | import it.finki.tinki.service.BuilderService;
|
---|
10 | import it.finki.tinki.service.WorkService;
|
---|
11 | import org.springframework.web.bind.annotation.*;
|
---|
12 | import org.springframework.web.server.ResponseStatusException;
|
---|
13 |
|
---|
14 | import java.util.List;
|
---|
15 |
|
---|
16 | @RestController
|
---|
17 | @CrossOrigin(origins = "http://localhost:3000")
|
---|
18 | @RequestMapping("/api")
|
---|
19 | public class LoginController {
|
---|
20 |
|
---|
21 | AccountService accountService;
|
---|
22 | BuilderService builderService;
|
---|
23 |
|
---|
24 | public LoginController(AccountService accountService, BuilderService builderService) {
|
---|
25 | this.accountService = accountService;
|
---|
26 | this.builderService = builderService;
|
---|
27 | }
|
---|
28 |
|
---|
29 | @PostMapping(path = "/login")
|
---|
30 | public LoginResponseDTO loginProcess(@RequestBody AccountLoginDTO body) throws ResponseStatusException {
|
---|
31 |
|
---|
32 | Account a1 = accountService.findUser(body.getEmail(), body.getPassword(), body.getType());
|
---|
33 |
|
---|
34 | if(a1!=null){
|
---|
35 | if(a1.getClass().equals(User.class)){
|
---|
36 | return this.builderService.buildUserResponseDTO(a1);
|
---|
37 | }else if(a1.getClass().equals(Team.class)){
|
---|
38 | return this.builderService.buildTeamResponseDTO(a1);
|
---|
39 | }else{
|
---|
40 | return this.builderService.buildCompanyResponseDTO(a1);
|
---|
41 | }
|
---|
42 | }
|
---|
43 |
|
---|
44 | return new LoginResponseDTO();
|
---|
45 | }
|
---|
46 | }
|
---|