source: Prototype Application/Paw5/src/main/java/finki/paw5/web/controllers/PostController.java@ 5f53114

package finki.paw5.web.controllers;
import finki.paw5.model.entities.Pet;
import finki.paw5.model.entities.Post;
import finki.paw5.model.entities.Employee;
import finki.paw5.model.enumerations.AgeGroup;
import finki.paw5.model.enumerations.Gender;
import finki.paw5.model.enumerations.Size;
import finki.paw5.model.enumerations.Species;
import finki.paw5.service.PersonalProfileService;
import finki.paw5.service.PetService;
import finki.paw5.service.PostService;
import finki.paw5.service.ShelterService;
import jakarta.servlet.http.HttpServletRequest;
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 org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;

import java.time.LocalDate;
import java.util.List;

@Controller
public class PostController {

    private final PostService postService;
    private final PetService petService;
    private final PersonalProfileService personalProfileService;

    public PostController(PostService postService, PetService petService, PersonalProfileService personalProfileService, ShelterService shelterService) {
        this.postService = postService;
        this.petService = petService;
        this.personalProfileService = personalProfileService;
    }

    @GetMapping("/create-post")
    public String get(Model model) {
        model.addAttribute("pets", this.petService.listpets());
        return "create-post";
    }

    @PostMapping("/submit-post")
    public String savePost(@RequestParam(required = false) Integer petId,
                           @RequestParam(required = false) boolean newPetCheckbox,
                           @RequestParam(required = false) String name,
                           @RequestParam(required = false) String gender,
                           @RequestParam(required = false) String ageGroup,
                           @RequestParam(required = false) String size,
                           @RequestParam(required = false) String species,
                           @RequestParam(required = false) String breed,
                           @RequestParam(required = false) String imageUrl,
                           @RequestParam(required = false) boolean canBeFostered,
                           HttpServletRequest request) {

        Employee employee = (Employee) request.getSession().getAttribute("user");

        if(newPetCheckbox == true){

            Pet newPet = new Pet(imageUrl, AgeGroup.valueOf(ageGroup), Size.valueOf(size), breed, name, Species.valueOf(species), Gender.valueOf(gender), canBeFostered, null, employee.getShelterId());
            this.petService.save(newPet);

            Post post = new Post(LocalDate.now(), imageUrl, newPet.getId(), null, employee.getId());
            this.postService.save(post);

        } else{

            Pet selectedPet = this.petService.findById(petId);

            Post post = new Post(LocalDate.now(), imageUrl, selectedPet.getId(), null, employee.getId());
            this.postService.save(post);

        }

        return "redirect:/home";
    }

    @GetMapping("/adoption-posts")
    public String getAdoptionPosts(Model model){

        List<Post> posts = this.postService.findAll();
        List<Pet> pets = this.petService.listpets();
        model.addAttribute("posts", posts);
        model.addAttribute("pets",pets);

        return "list-posts-adoption";
    }

    @GetMapping("/pet-details-{id}")
    public String getPostDetails(@PathVariable Integer id,
                                 Model model){

        Post post = this.postService.findById(id).get();
        Pet pet = this.petService.findById(post.getPetId());

        model.addAttribute("pet", pet);
        model.addAttribute("post", post);

        return "pet-details";
    }
}