1 | package com.wediscussmovies.project.web.controller;
|
---|
2 |
|
---|
3 | import com.wediscussmovies.project.LoggedUser;
|
---|
4 | import com.wediscussmovies.project.model.Genre;
|
---|
5 | import com.wediscussmovies.project.model.User;
|
---|
6 | import com.wediscussmovies.project.model.relation.UserGenres;
|
---|
7 | import com.wediscussmovies.project.repository.GenreLikesRepository;
|
---|
8 | import com.wediscussmovies.project.service.GenreService;
|
---|
9 | import org.springframework.stereotype.Controller;
|
---|
10 | import org.springframework.ui.Model;
|
---|
11 | import org.springframework.web.bind.annotation.GetMapping;
|
---|
12 | import org.springframework.web.bind.annotation.RequestMapping;
|
---|
13 |
|
---|
14 | import java.util.ArrayList;
|
---|
15 | import java.util.List;
|
---|
16 |
|
---|
17 | @Controller
|
---|
18 | @RequestMapping("/genres")
|
---|
19 | public class GenreController {
|
---|
20 | private final GenreService genreService;
|
---|
21 | private final GenreLikesRepository genreLikesRepository;
|
---|
22 |
|
---|
23 | public GenreController(GenreService genreService, GenreLikesRepository genreLikesRepository) {
|
---|
24 | this.genreService = genreService;
|
---|
25 | this.genreLikesRepository = genreLikesRepository;
|
---|
26 | }
|
---|
27 | @GetMapping
|
---|
28 | public String getGenres(Model model){
|
---|
29 | model.addAttribute("genres",genreService.findAllWithLikes());
|
---|
30 | model.addAttribute("allGenres", genreService.findAll());
|
---|
31 | model.addAttribute("contentTemplate","genres");
|
---|
32 | addModelPropertiesForUser(model);
|
---|
33 | return "template";
|
---|
34 | }
|
---|
35 |
|
---|
36 | private void addModelPropertiesForUser(Model model){
|
---|
37 | User user = LoggedUser.getLoggedUser();
|
---|
38 | List<UserGenres> genreLikesList = this.genreLikesRepository.findAllByUser(user);
|
---|
39 | List<Genre> genres = new ArrayList<>();
|
---|
40 | for(UserGenres g: genreLikesList){
|
---|
41 | genres.add(genreService.findById(g.getId().getGenreId()));
|
---|
42 | }
|
---|
43 | model.addAttribute("likedGenres",genres);
|
---|
44 | model.addAttribute("user",user);
|
---|
45 | }
|
---|
46 | }
|
---|