Initializing

1 files changed, 226 insertions, 0 deletions

diff --git a/src/main/java/com/mavlushechka/studentdatabase/controller/AdminPanelController.java b/src/main/java/com/mavlushechka/studentdatabase/controller/AdminPanelController.java
new file mode 100755
index 0000000..82e6fa9
--- /dev/null
+++ b/src/main/java/com/mavlushechka/studentdatabase/controller/AdminPanelController.java
@@ -0,0 +1,226 @@
+package com.mavlushechka.studentdatabase.controller;
+
+import com.mavlushechka.studentdatabase.domain.*;
+import com.mavlushechka.studentdatabase.repository.*;
+import com.mavlushechka.studentdatabase.service.CustomUserDetailsService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+
+import java.util.*;
+
+@Controller
+public class AdminPanelController {
+    @Autowired
+    private CustomUserDetailsService userDetailsService;
+    @Autowired
+    private UserRepository userRepository;
+    @Autowired
+    private RoleRepository roleRepository;
+    @Autowired
+    private StudentRepository studentRepository;
+    @Autowired
+    private EducationalSystemRepository educationalSystemRepository;
+    @Autowired
+    private PassportRepository passportRepository;
+    @Autowired
+    private GraduatedInstitutionRepository graduatedInstitutionRepository;
+    @Autowired
+    private FamilyRepository familyRepository;
+    @Autowired
+    private HealthRepository healthRepository;
+    @Autowired
+    private JobRepository jobRepository;
+    @Autowired
+    private MilitaryServiceRepository militaryServiceRepository;
+
+    @GetMapping("/admin-panel/create-users")
+    public String createUsers(@RequestParam String group, @RequestParam int number, Map<String, Object> model) {
+        for (int i = 1; i <= number; i ++) {
+            Role userRole = roleRepository.findByRole("USER");
+            User user = new User(i + "student" + group, i + "student" + group, true, new HashSet<>(List.of(userRole)));
+            User userFromDb = userRepository.findByUsername(user.getUsername());
+
+            if (userFromDb != null) {
+                model.put("message", "User exists!");
+                return "/admin-panel/add-users";
+            }
+
+            userDetailsService.saveUser(user);
+        }
+        return "redirect:/admin-panel/all-users";
+    }
+
+    @GetMapping("/admin-panel/all-users")
+    public String getUsers(Map<String, Object> model) {
+        Iterable<User> users = userRepository.findAll();
+        model.put("users", users);
+        return "admin-panel/all-users";
+    }
+
+    @GetMapping("/admin-panel/information/all")
+    public String getAll(String id, String fullName, String group, Map<String, Object> model) {
+        Iterable<User> users = null;
+        Iterable<Student> students = null;
+        Iterable<Passport> passports = null;
+        Iterable<EducationalSystem> educationalSystems = null;
+        Iterable<GraduatedInstitution> graduatedInstitutions = null;
+        Iterable<Health> healths = null;
+        Iterable<Family> families = null;
+        Iterable<Job> jobs = null;
+        Iterable<MilitaryService> militaryServices = null;
+
+        List<User> userList = new ArrayList<>();
+        List<Student> studentList = new ArrayList<>();
+        List<Passport> passportList = new ArrayList<>();
+        List<EducationalSystem> educationalSystemList = new ArrayList<>();
+        List<GraduatedInstitution> graduatedInstitutionList = new ArrayList<>();
+        List<Health> healthList = new ArrayList<>();
+        List<Family> familyList = new ArrayList<>();
+        List<Job> jobList = new ArrayList<>();
+        List<MilitaryService> militaryServiceList = new ArrayList<>();
+
+        if (id != null) {
+            userList.add(userRepository.findById(id).orElse(new User()));
+            Student student = studentRepository.findById(id).orElse(new Student());
+            if (student.getPhoto() != null) {
+                student.setEncodedPhoto(Base64.getEncoder().encodeToString(student.getPhoto().getData()));
+            }
+            studentList.add(student);
+            passportList.add(passportRepository.findById(id).orElse(new Passport()));
+            educationalSystemList.add(educationalSystemRepository.findById(id).orElse(new EducationalSystem()));
+            graduatedInstitutionList.add(graduatedInstitutionRepository.findById(id).orElse(new GraduatedInstitution()));
+            healthList.add(healthRepository.findById(id).orElse(new Health()));
+            familyList.add(familyRepository.findById(id).orElse(new Family()));
+            jobList.add(jobRepository.findById(id).orElse(new Job()));
+            militaryServiceList.add(militaryServiceRepository.findById(id).orElse(new MilitaryService()));
+        } else if (fullName != null || group != null) {
+            ArrayList<String> ids = new ArrayList<>();
+            if (fullName != null) {
+                Iterable<Passport> passportIterable;
+                try {
+                    passportIterable = passportRepository.findByFirstNameAndLastName(fullName.split(" ")[0], fullName.split(" ")[1]);
+                } catch (ArrayIndexOutOfBoundsException arrayIndexOutOfBoundsException) {
+                    passportIterable = passportRepository.findByFirstName(fullName.split(" ")[0]);
+                }
+                for (Passport passport : passportIterable) {
+                    ids.add(passport.getId());
+                }
+            } else {
+                Iterable<EducationalSystem> educationalSystemIterable;
+                educationalSystemIterable = educationalSystemRepository.findByGroup(group);
+                for (EducationalSystem educationalSystem : educationalSystemIterable) {
+                    ids.add(educationalSystem.getId());
+                }
+            }
+
+            for (String eachId : ids) {
+                userList.add(userRepository.findById(eachId).orElse(new User()));
+                Student student = studentRepository.findById(eachId).orElse(new Student());
+                if (student.getPhoto() != null) {
+                    student.setEncodedPhoto(Base64.getEncoder().encodeToString(student.getPhoto().getData()));
+                }
+                studentList.add(student);
+                passportList.add(passportRepository.findById(eachId).orElse(new Passport()));
+                educationalSystemList.add(educationalSystemRepository.findById(eachId).orElse(new EducationalSystem()));
+                graduatedInstitutionList.add(graduatedInstitutionRepository.findById(eachId).orElse(new GraduatedInstitution()));
+                healthList.add(healthRepository.findById(eachId).orElse(new Health()));
+                familyList.add(familyRepository.findById(eachId).orElse(new Family()));
+                jobList.add(jobRepository.findById(eachId).orElse(new Job()));
+                militaryServiceList.add(militaryServiceRepository.findById(eachId).orElse(new MilitaryService()));
+            }
+        } else {
+            users = userRepository.findAll();
+            students = studentRepository.findAll();
+            passports = passportRepository.findAll();
+            educationalSystems = educationalSystemRepository.findAll();
+            graduatedInstitutions = graduatedInstitutionRepository.findAll();
+            healths = healthRepository.findAll();
+            families = familyRepository.findAll();
+            jobs = jobRepository.findAll();
+            militaryServices = militaryServiceRepository.findAll();
+        }
+        if (id != null || fullName != null || group != null) {
+            users = userList;
+            students = studentList;
+            passports = passportList;
+            educationalSystems = educationalSystemList;
+            graduatedInstitutions = graduatedInstitutionList;
+            healths = healthList;
+            families = familyList;
+            jobs = jobList;
+            militaryServices = militaryServiceList;
+        }
+
+        model.put("users", users);
+        model.put("students", students);
+        model.put("passports", passports);
+        model.put("educationalSystems", educationalSystems);
+        model.put("graduatedInstitutions", graduatedInstitutions);
+        model.put("healths", healths);
+        model.put("families", families);
+        model.put("jobs", jobs);
+        model.put("militaryServices", militaryServices);
+        return "admin-panel/information/all";
+    }
+
+    @GetMapping("/admin-panel/information/students")
+    public String getStudents(Map<String, Object> model) {
+        Iterable<Student> students = studentRepository.findAll();
+        for (Student student : students) {
+            student.setEncodedPhoto(Base64.getEncoder().encodeToString(student.getPhoto().getData()));
+        }
+        model.put("students", students);
+        return "admin-panel/information/students";
+    }
+
+    @GetMapping("/admin-panel/information/educational-systems")
+    public String getEducationalSystems(Map<String, Object> model) {
+        Iterable<EducationalSystem> educationalSystems = educationalSystemRepository.findAll();
+        model.put("educationalSystems", educationalSystems);
+        return "admin-panel/information/educational-systems";
+    }
+
+    @GetMapping("/admin-panel/information/passports")
+    public String getPassports(Map<String, Object> model) {
+        Iterable<Passport> passports = passportRepository.findAll();
+        model.put("passports", passports);
+        return "admin-panel/information/passports";
+    }
+
+    @GetMapping("/admin-panel/information/graduated-institutions")
+    public String getGraduatedInstitutions(Map<String, Object> model) {
+        Iterable<GraduatedInstitution> graduatedInstitutions = graduatedInstitutionRepository.findAll();
+        model.put("graduatedInstitutions", graduatedInstitutions);
+        return "admin-panel/information/graduated-institutions";
+    }
+
+    @GetMapping("/admin-panel/information/families")
+    public String getFamilies(Map<String, Object> model) {
+        Iterable<Family> families = familyRepository.findAll();
+        model.put("families", families);
+        return "admin-panel/information/families";
+    }
+
+    @GetMapping("/admin-panel/information/health")
+    public String getHealth(Map<String, Object> model) {
+        Iterable<Health> health = healthRepository.findAll();
+        model.put("health", health);
+        return "admin-panel/information/health";
+    }
+
+    @GetMapping("/admin-panel/information/jobs")
+    public String getJobs(Map<String, Object> model) {
+        Iterable<Job> jobs = jobRepository.findAll();
+        model.put("jobs", jobs);
+        return "admin-panel/information/jobs";
+    }
+
+    @GetMapping("/admin-panel/information/military-services")
+    public String getMilitaryServices(Map<String, Object> model) {
+        Iterable<MilitaryService> militaryServices = militaryServiceRepository.findAll();
+        model.put("militaryServices", militaryServices);
+        return "admin-panel/information/military-services";
+    }
+}