source: src/main/java/project/educatum/web/TeacherController.java

package project.educatum.web;

import org.springframework.format.annotation.DateTimeFormat;
import org.springframework.http.HttpStatus;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.server.ResponseStatusException;
import project.educatum.model.*;
import project.educatum.model.Class;
import project.educatum.model.exceptions.SubjectNotFoundException;
import project.educatum.service.*;

import javax.servlet.http.HttpServletRequest;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

@Controller
@RequestMapping(path = "/teachers", method = {RequestMethod.POST, RequestMethod.DELETE, RequestMethod.GET})
public class TeacherController {

    private final TeacherService teacherService;
    private final SubjectService subjectService;
    private final TeacherSubjectService teacherSubjectService;
    private final StudentService studentService;
    private final PaymentService paymentService;
    private final TeacherStudentService teacherStudentService;
    private final ClassService classService;

    public TeacherController(TeacherService teacherService, SubjectService subjectService, TeacherSubjectService teacherSubjectService, StudentService studentService, PaymentService paymentService, TeacherStudentService teacherStudentService, ClassService classService) {
        this.teacherService = teacherService;
        this.subjectService = subjectService;
        this.teacherSubjectService = teacherSubjectService;
        this.studentService = studentService;
        this.paymentService = paymentService;
        this.teacherStudentService = teacherStudentService;
        this.classService = classService;
    }

    @GetMapping
    public String getAllTeachers(Model model) {
        model.addAttribute("teachers", teacherService.findAll());
        return "teachers";
    }

    @GetMapping("/allStudents")
    public String getAllStudentsByTeacher(@RequestParam(required = false) String ime, Model model, HttpServletRequest request) {
        List<Student> students = new ArrayList<>();
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher teacher = teacherService.findByEmail(user.getUsername());
            if (ime == null) {
                students = teacherService.getStudentsByTeacher(teacher.getId());
            } else {
                students = this.studentService.findAllByNameLike(ime, teacherService.getStudentsByTeacher(teacher.getId()));
            }
            model.addAttribute("teacher", teacherService.findById(teacher.getId()));
            model.addAttribute("students", students);
            return "evidencija";
        } else return "redirect:/home";
    }

    @PostMapping("/delete/{id}")
    public String deleteTeacher(@PathVariable String id) {
        teacherService.delete(Integer.parseInt(id));
        return "redirect:/admin/allTeachers";
    }


    @PostMapping("/payments/{id}")
    public String paymentForStudent(@PathVariable String id, Model model, HttpServletRequest request) {
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher teacher = teacherService.findByEmail(user.getUsername());
            Integer owes = paymentService.studentTeacherLoan(Integer.valueOf(id), teacher.getId());
            model.addAttribute("owes", owes);
            Integer numScheduledClasses = teacherStudentService.find(teacher.getId(), Integer.valueOf(id)).getNumScheduledClasses();
            model.addAttribute("numScheduledClasses", numScheduledClasses);
            Integer priceByClass = teacherStudentService.find(teacher.getId(), Integer.valueOf(id)).getPriceByClass();
            model.addAttribute("priceByClass", priceByClass);
            model.addAttribute("classes", teacherService.getClassesByTeacher(teacher.getId()));

            Integer numListenedClasses = paymentService.numListenedClasses(Integer.valueOf(id), teacher.getId());
            model.addAttribute("numListenedClasses", numListenedClasses);
            model.addAttribute("student", studentService.findById(Integer.valueOf(id)));
            return "payment";
        } else return "redirect:/home";
    }

    @PostMapping("/updatePayment")
    public String updatePaymentForStudent(@RequestParam String studentID, Model model, HttpServletRequest request,
                                          @RequestParam(required = false) String price, @RequestParam String classID) {

        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher teacher = teacherService.findByEmail(user.getUsername());


            if (Integer.parseInt(price) != 0) {
                paymentService.addPayment(teacher.getId(), Integer.valueOf(price), Integer.valueOf(classID), Integer.valueOf(studentID));
            }

            Integer owes = paymentService.studentTeacherLoan(Integer.valueOf(studentID), teacher.getId());
            model.addAttribute("owes", owes);
            Integer numScheduledClasses = teacherStudentService.find(teacher.getId(), Integer.valueOf(studentID)).getNumScheduledClasses();
            model.addAttribute("numScheduledClasses", numScheduledClasses);
            Integer priceByClass = teacherStudentService.find(teacher.getId(), Integer.valueOf(studentID)).getPriceByClass();
            model.addAttribute("priceByClass", priceByClass);
            Integer numListenedClasses = paymentService.numListenedClasses(Integer.valueOf(studentID), teacher.getId());
            model.addAttribute("numListenedClasses", numListenedClasses);
            model.addAttribute("classes", teacherService.getClassesByTeacher(teacher.getId()));
            model.addAttribute("student", studentService.findById(Integer.valueOf(studentID)));
            return "payment";
        } else return "redirect:/home";
    }


    @GetMapping("/allClasses")
    public String timetable(Model model, HttpServletRequest request) {
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher teacher = teacherService.findByEmail(user.getUsername());
            List<Class> classes = teacherService.getClassesByTeacher(teacher.getId());
            model.addAttribute("teacher", teacherService.findById(teacher.getId()));
            model.addAttribute("classes", classes);
            return "raspored";
        } else return "redirect:/home";
    }


    @GetMapping("/allSubjects")
    public String getAllSubjectsByTeacher(@RequestParam(required = false) String ime, Model model, HttpServletRequest request) {
        List<Subject> subjects = new ArrayList<>();
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher teacher = teacherService.findByEmail(user.getUsername());
            if (ime == null) {
                subjects = teacherService.getSubjectsByTeacher(teacher.getId());
            } else {
                subjects = this.subjectService.findAllByNameAndTeacherLike(ime, teacherService.getSubjectsByTeacher(teacher.getId()));
            }
            model.addAttribute("teacher", teacherService.findById(teacher.getId()));
            model.addAttribute("subjects", subjects);
            model.addAttribute("allSubjects", subjectService.findAll());
            return "subjectsByTeacher";
        } else return "redirect:/home";
    }


    @PostMapping("/addStudentForm")
    public String addStudentForm() {
        return "addNewStudent";
    }

    @PostMapping("/addSubjectForm")
    public String addSubjectForm(Model model) {
        model.addAttribute("subjects", subjectService.findAll());
        return "addNewSubject";
    }

    @PostMapping("/addStudent")
    public String addStudent(@RequestParam String price, @RequestParam String numClasses,
                             @RequestParam String email, HttpServletRequest request) {
        Student ucenik = studentService.findByEmail(email);
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher teacher = teacherService.findByEmail(user.getUsername());

            teacherService.addStudent(teacher.getId(), ucenik.getId(), Integer.valueOf(price), Integer.valueOf(numClasses));
            return "redirect:/teachers/allStudents";
        } else return "redirect:/home";
    }


    @PostMapping("/addSubject")
    public String addSubject(Model model, HttpServletRequest request,
                             @RequestParam String subjectId,
                             @RequestParam String desc) {

        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher teacher = teacherService.findByEmail(user.getUsername());
            teacherService.addSubject(teacher.getId(), Integer.valueOf(subjectId), desc);

            return "redirect:/teachers/allSubjects";
        } else return "redirect:/home";
    }

    @PostMapping("/teachesSubject")
    public String teachesSubject(@RequestParam String tema, @RequestParam String subjectID, HttpServletRequest request) {
        Optional<Subject> p = subjectService.findById(Integer.valueOf(subjectID));
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            String username = user.getUsername();
            Teacher n = teacherService.findByEmail(username);
            teacherSubjectService.addSubject(n.getId(), Integer.valueOf(subjectID), tema);
            return "redirect:/home/document";
        } else return "redirect:/home";
    }


    @PostMapping("/addClass")
    public String addClass(
            @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime date,
            @RequestParam String desc,
            @RequestParam String ime,
            HttpServletRequest request) {
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if (user != null) {
            Teacher t = teacherService.findByEmail(user.getUsername());
            Subject s = subjectService.findByName(ime);
            if (s != null) {
                classService.addClass(date, desc, t.getId(), s.getId());
                return "redirect:/teachers/allClasses";
            } else
                return "redirect:/teachers/allClasses";
        } else
            return "redirect:/home";

    }

    @PostMapping("/showProfileTeacher/{id}")
    public String showProfileTeacher(@PathVariable String id, Model model) {
        Teacher teacher = teacherService.findById(Integer.valueOf(id));
        model.addAttribute("teacher", teacher);
        model.addAttribute("teacherRating",teacherService.getRatingForTeacher(Long.valueOf(teacher.getId())));

        return "showProfileTeacher.html";
    }

    @PostMapping("/showProfileTeacher2/{id}")
    public String showProfileTeacher2(@PathVariable String id, Model model) {
        Teacher teacher = teacherService.findById(Integer.valueOf(id));
        model.addAttribute("teacher", teacher);
        model.addAttribute("teacherRating",teacherService.getRatingForTeacher(Long.valueOf(teacher.getId())));
        return "showProfileTeacher2.html";
    }

    @PostMapping("/showProfile")
    public String showProfile(HttpServletRequest request, Model model) {
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        Teacher teacher = teacherService.findByEmail(user.getUsername());
        model.addAttribute("teacher", teacher);
        return "showProfile.html";
    }

    @GetMapping("/{id}/edit")
    public String showEdit(@PathVariable Integer id, Model model) {
        Teacher teacher = this.teacherService.findById(id);
        model.addAttribute("teacher", teacher);
        return "showProfile";
    }

    @PostMapping("/{id}")
    public String editTeacher(@PathVariable Integer id,
                         @RequestParam String name,
                         @RequestParam String surname,
                         @RequestParam String description,
                         @RequestParam String email,
                         @RequestParam String telephoneNumber,
                         HttpServletRequest request) {
        this.teacherService.edit(id, name, surname,description,email,telephoneNumber);
        UserDetails user = (UserDetails) request.getSession().getAttribute("user");
        if(!Objects.equals(email, user.getUsername())){
            return "redirect:/login";
        }
        return "redirect:/teachers/showProfile";
    }
}