source: src/main/java/com/example/cookbook/controller/HomeController.java@ aea04dd

package com.example.cookbook.controller;


import com.example.cookbook.model.Komentar;
import com.example.cookbook.model.Recept;
import com.example.cookbook.model.Slika;
import com.example.cookbook.model.Sostojka;
import com.example.cookbook.model.exception.ReceptNeEPronajdenException;
import com.example.cookbook.service.KomentariService;
import com.example.cookbook.service.ReceptService;
import com.example.cookbook.service.SlikiService;
import com.example.cookbook.service.SostojkiService;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;

import java.sql.SQLException;
import java.util.List;

@Controller
public class HomeController {

    private final ReceptService receptService;
    private final SlikiService slikiService;

    private final SostojkiService sostojkiService;

    private final KomentariService komentariService;

    public HomeController(ReceptService receptService,
                          SlikiService slikiService,
                          SostojkiService sostojkiService, KomentariService komentariService) {

        this.receptService = receptService;
        this.slikiService = slikiService;
        this.sostojkiService = sostojkiService;
        this.komentariService = komentariService;
    }

    @GetMapping({"/", "/recepti"})
    public String getHomePage(Model model){

        List<Recept> recepti = null;
        try {
            recepti = receptService.listAll();
        } catch (SQLException e) {
            return "redirect:/error-page/SQL%20Exception";
        }
        model.addAttribute("recepti", recepti);
        return "home";
    }

    @GetMapping("/recept/{id}")
    public String getReceptPage(@PathVariable Long id, Model model){

        Recept recept = null;
        List<Slika> sliki;
        List<Sostojka> sostojki;
        List<Komentar> komentari;
        try {
            recept = receptService.findById(id);
            sliki = slikiService.findAllPicById(id);
            sostojki = sostojkiService.findAllById(id);
            komentari = komentariService.findAllById(id);
        } catch (SQLException e) {
            return "redirect:/error-page/SQL%20Exception";
        }catch (ReceptNeEPronajdenException e){
            return "redirect:/error-page/" + e.getMessage();
        }
        model.addAttribute("recept", recept);
        model.addAttribute("sliki", sliki);
        model.addAttribute("sostojki", sostojki);
        model.addAttribute("komentari", komentari);
        return "recept";
    }


    @GetMapping("/error-page/{error}")
    public String getErrorPage(@PathVariable(required = false) String error, Model model){

        if (error != null){
            model.addAttribute("error", error);
        }
        return "error-page";
    }
}