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