summaryrefslogtreecommitdiff
path: root/src/main/java/com/mavlushechka/barbershop/controller/AdminPanelController.java
blob: 84a01e77e32eb980283fe7fb720c650f5fabbb0a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package com.mavlushechka.barbershop.controller;

import com.mavlushechka.barbershop.domain.*;
import com.mavlushechka.barbershop.repository.*;
import org.bson.BsonBinarySubType;
import org.bson.types.Binary;
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.PostMapping;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.*;

@Controller
public class AdminPanelController {
    @Autowired
    private AppointmentRepository appointmentRepository;
    @Autowired
    private VisitorRepository visitorRepository;
    @Autowired
    private LevelRepository levelRepository;
    @Autowired
    private BarberRepository barberRepository;
    @Autowired
    private SlideRepository slideRepository;
    @Autowired
    private TestimonialRepository testimonialRepository;

    @GetMapping("/admin-panel")
    public String adminPanel(Map<String, Object> model) {
        HashSet<String> emails = new HashSet<>();
        List<Appointment> appointments = appointmentRepository.findAll();

        model.put("views", visitorRepository.count());
        model.put("appointments", appointments.size());
        appointments.iterator().forEachRemaining(appointment -> emails.add(appointment.getEmail()));
        model.put("clients", emails.size());
        return "/admin-panel/index";
    }

    @GetMapping("/admin-panel/tables/appointments")
    public String adminPanelTablesAppointments(Map<String, Object> model) {
        model.put("appointments", appointmentRepository.findAll());
        return "/admin-panel/pages/tables/appointments";
    }

    @GetMapping("/admin-panel/tables/visitors")
    public String adminPanelTablesVisitors(Map<String, Object> model) {
        model.put("visitors", visitorRepository.findAll());
        return "/admin-panel/pages/tables/visitors";
    }

    @GetMapping("/admin-panel/tables/barbers")
    public String adminPanelTablesBarbers(Map<String, Object> model) {
        List<Barber> barbers = barberRepository.findAll();

        for (Barber barber : barbers) {
            barber.setEncodedPhoto(Base64.getEncoder().encodeToString(barber.getPhoto().getData()));
        }
        model.put("barbers", barbers);
        return "/admin-panel/pages/tables/barbers";
    }

    @GetMapping("/admin-panel/tables/slides")
    public String adminPanelTablesSlides(Map<String, Object> model) {
        List<Slide> slides = slideRepository.findAll();

        for (Slide slide : slides) {
            slide.setEncodedPhoto(Base64.getEncoder().encodeToString(slide.getPhoto().getData()));
        }
        model.put("slides", slides);
        return "/admin-panel/pages/tables/slides";
    }

    @GetMapping("/admin-panel/tables/testimonials")
    public String adminPanelTablesTestimonial(Map<String, Object> model) {
        List<Testimonial> testimonials = testimonialRepository.findAll();

        for (Testimonial testimonial : testimonials) {
            testimonial.setEncodedPhoto(Base64.getEncoder().encodeToString(testimonial.getPhoto().getData()));
        }
        model.put("testimonials", testimonials);
        return "/admin-panel/pages/tables/testimonials";
    }

    @GetMapping("/admin-panel/pages/forms/add-barber")
    public String adminPanelPagesFormsAddBarber(Map<String, Object> model) {
        model.put("levels", levelRepository.findAll());
        return "/admin-panel/pages/forms/add-barber";
    }

    @PostMapping("/save/barber")
    public String saveBarber(String firstName, String level, MultipartFile photo) throws IOException {
        barberRepository.save(new Barber(firstName, levelRepository.findByName(level), new Binary(BsonBinarySubType.BINARY, photo.getBytes())));
        return "redirect:/admin-panel/tables/barbers";
    }

    @PostMapping("/save/slide")
    public String saveSlide(String title, String description, MultipartFile photo) throws IOException {
        slideRepository.save(new Slide(title, description, new Binary(BsonBinarySubType.BINARY, photo.getBytes())));
        return "redirect:/admin-panel/tables/slides";
    }

    @PostMapping("/save/testimonial")
    public String saveTestimonial(String name, String comment, MultipartFile photo) throws IOException {
        testimonialRepository.save(new Testimonial(name, comment, new Binary(BsonBinarySubType.BINARY, photo.getBytes())));
        return "redirect:/admin-panel/tables/testimonials";
    }
}