source: src/main/java/mk/finki/ukim/mk/vehiclerent/web/HomeController.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.*;
import mk.finki.ukim.mk.vehiclerent.model.Bookings;
import mk.finki.ukim.mk.vehiclerent.model.User;
import mk.finki.ukim.mk.vehiclerent.repository.VozaciRepozitory;
import mk.finki.ukim.mk.vehiclerent.service.MenadzerService;
import mk.finki.ukim.mk.vehiclerent.service.RezervacijVozilaService;
import mk.finki.ukim.mk.vehiclerent.service.impl.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.authority.SimpleGrantedAuthority;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.web.authentication.logout.SecurityContextLogoutHandler;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.awt.print.Book;
import java.util.ArrayList;
import java.util.List;
import java.util.SimpleTimeZone;

@Controller
public class HomeController {
    @Autowired
    KlientServiceImpl klientService;
    @Autowired
    MenadzerServiceImpl menadzerService;
    @Autowired
    VraborenServiceImpl vraborenService;

    @Autowired
    VoziloServiceImpl voziloService;

    @Autowired
    RezervacijaServiceImpl rezervacijaService;

    @Autowired
    DogovorServiceImpl dogovorService;

    @Autowired
    RezervacijaVozilaServiceImpl rezervacijVozilaService;

    @Autowired
    VozaciRepozitory vozaciRepozitory;

    public HomeController() {


    }

    @RequestMapping(value = "/", method = RequestMethod.GET)
    public String index(Model model) {
        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();
        }
        if(role==ROLES.KLIENT){
            List<RezervaciiEntity> rezervacije = rezervacijaService.findByEmbgvoza(embg);
            List<Bookings> bookings = new ArrayList<>();
            for(RezervaciiEntity rez : rezervacije){
                VozilaEntity vozilo = voziloService.findById(
                        rezervacijVozilaService.findByIdrez(rez.getIdrez())
                                .getIdvozi()
                );
                Bookings book = new Bookings();
                book.setRezervacija(rez);
                book.setVozilo(vozilo);
                bookings.add(book);
            }
            model.addAttribute("bookings",bookings);
        }

        return "index.html";
    }

    @RequestMapping(value="/logout", method=RequestMethod.GET)
    public String logoutPage(HttpServletRequest request, HttpServletResponse response) {
        Authentication auth = SecurityContextHolder.getContext().getAuthentication();
        if (auth != null){
            new SecurityContextLogoutHandler().logout(request, response, auth);
        }
        return "redirect:/";
    }

    @RequestMapping(value = "/login", method = RequestMethod.GET)
    public String showLogin(Model model) {
        model.addAttribute("user",new User(null,null,null, ROLES.KLIENT,0));
        return "login.html";
    }

    @RequestMapping(value = "/login", method = RequestMethod.POST)
    public String login(User user,Model model,HttpServletRequest request) {
        boolean login = authenticateUserAndInitializeSessionByUsername(user.getEMBG(),user.getRole(),request);
        if(login){
            System.out.println("LOGIN OK");
        }else{
            System.out.println("LOGIN NOT OK");
        }
        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();
        }
        List<RezervaciiEntity> rezervacije = rezervacijaService.findByEmbgvoza(embg);
        List<Bookings> bookings = new ArrayList<>();
        for(RezervaciiEntity rez : rezervacije){
             VozilaEntity vozilo = voziloService.findById(
                            rezervacijVozilaService.findByIdrez(rez.getIdrez())
                                    .getIdvozi()
                    );
            Bookings book = new Bookings();
            book.setRezervacija(rez);
            book.setVozilo(vozilo);
            bookings.add(book);
        }
        model.addAttribute("bookings",bookings);

        return "index.html";
    }

    public boolean authenticateUserAndInitializeSessionByUsername(String EMBG, ROLES role, HttpServletRequest request)
    {
        boolean result = true;
        ArrayList<SimpleGrantedAuthority> roles = new ArrayList<>();

        try
        {
            // generate session if one doesn't exist
            request.getSession();

            // Authenticate the user
            switch (role){
                case KLIENT :
                    System.out.println("Klient login "+EMBG);
                    roles.add(new SimpleGrantedAuthority("ROLE_KLIENT"));
                    KlientiEntity user = klientService.findByEmbg(EMBG);
                    Authentication auth = new UsernamePasswordAuthenticationToken(user, null, roles);
                    SecurityContextHolder.getContext().setAuthentication(auth);
                    break;
                case MENAGER:
                    /**System.out.println("Menage login");
                    UserDetails user = userDetailsManager.loadUserByUsername(username);
                    Authentication auth = new UsernamePasswordAuthenticationToken(user, null, null);
                    SecurityContextHolder.getContext().setAuthentication(auth);*/
                    break;
                case VRABOTEN:
                    System.out.println("Vraboten login");
                    roles.add(new SimpleGrantedAuthority("ROLE_VRABOTEN"));
                    VraboteniEntity userVra = vraborenService.findByEmbg(EMBG);
                    Authentication authVra = new UsernamePasswordAuthenticationToken(userVra, null, roles);
                    SecurityContextHolder.getContext().setAuthentication(authVra);
                    break;
                default: break;
            }
        }
        catch (Exception e)
        {
            System.out.println(e.getMessage());

            result = false;
        }

        return result;
    }

    @RequestMapping(value = "/register", method = RequestMethod.GET)
    public String showRegister(Model model) {
        model.addAttribute("user",new User(null,null,null, ROLES.KLIENT,0));
        return "register.html";
    }

    @RequestMapping(value = "/register", method = RequestMethod.POST)
    public String register(User user) {
        switch (user.getRole()){
            case KLIENT :
                klientService.register(user.getEMBG(),user.getImeK(),user.getAdresaK(),false);
                vozaciRepozitory.save(new VozaciEntity(user.getEMBG(),user.getImeK(),user.getAdresaK(),"23232323"));
                break;
            case MENAGER:
                //menadzerService.register(user.getEMBG());
                break;
            case VRABOTEN:
                vraborenService.register(user.getEMBG(), user.getIdis(),user.getImeK());
                break;
            default: break;
        }
        System.out.println(user.getEMBG());
        return "index.html";
    }
}