source: src/main/java/mk/finki/ukim/mk/vehiclerent/web/VoziloController.java@ 3965aed

package mk.finki.ukim.mk.vehiclerent.web;

import mk.finki.ukim.mk.vehiclerent.enumerations.ROLES;
import mk.finki.ukim.mk.vehiclerent.generated_model.KlientiEntity;
import mk.finki.ukim.mk.vehiclerent.generated_model.VozilaEntity;
import mk.finki.ukim.mk.vehiclerent.generated_model.VraboteniEntity;
import mk.finki.ukim.mk.vehiclerent.service.VoziloService;
import mk.finki.ukim.mk.vehiclerent.service.impl.RezervacijaServiceImpl;
import mk.finki.ukim.mk.vehiclerent.service.impl.VoziloServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;

import java.util.List;

@Controller
public class VoziloController {
    @Autowired
    private final VoziloServiceImpl voziloService;

    @Autowired
    private final RezervacijaServiceImpl rezervacijaService;

    public VoziloController(VoziloServiceImpl service, RezervacijaServiceImpl rezervacijaService) {
        this.voziloService = service;
        this.rezervacijaService = rezervacijaService;
    }

    @RequestMapping(value = "/vehicles", method = RequestMethod.GET)
    public String showcars(Model model){
        List<VozilaEntity> vozila = voziloService.listAll();
        model.addAttribute("vozila",vozila);
        return "vehicles.html";
    }

    @RequestMapping(value = "/addVehicle", method = RequestMethod.GET)
    public String addVehiclea(Model model){
        model.addAttribute("vozilo",new VozilaEntity());
        return "addVehicle.html";
    }

    @RequestMapping(value = "/addVehicle", method = RequestMethod.POST)
    public String addVehiclea(Model model, VozilaEntity vozilo){
        voziloService.create(1,vozilo.getModel(),vozilo.getMarka(), vozilo.getRegtab(), vozilo.getCenavozi());
        return "redirect:/";
    }

    @RequestMapping(value = "/rentVehicle/{id}", method = RequestMethod.GET)
    public String showcars(Model model, @PathVariable(value = "id") String id){
        Object principal = SecurityContextHolder. getContext(). getAuthentication(). getPrincipal();
        String embg = "";
        ROLES role = null;
        if(principal instanceof KlientiEntity){
            role = ROLES.KLIENT;
            embg = ((KlientiEntity) principal).getEmbgk();
        }else if(principal instanceof VraboteniEntity){
            role = ROLES.VRABOTEN;
            embg = ((VraboteniEntity) principal).getEmbgv();
        }
        rezervacijaService.rentVehicle(id,embg,role);
        return "redirect:/";
    }


}