[0f3491d] | 1 | package com.example.salonbella.service.reservation;
|
---|
| 2 |
|
---|
| 3 | import com.example.salonbella.entity.ReservationEntity;
|
---|
| 4 | import com.example.salonbella.entity.UserEntity;
|
---|
| 5 | import com.example.salonbella.repository.ReservationRepository;
|
---|
| 6 | import com.example.salonbella.repository.UserRepository;
|
---|
| 7 | import org.springframework.beans.factory.annotation.Autowired;
|
---|
| 8 | import org.springframework.security.core.Authentication;
|
---|
| 9 | import org.springframework.security.core.context.SecurityContextHolder;
|
---|
| 10 | import org.springframework.stereotype.Service;
|
---|
| 11 |
|
---|
| 12 | import javax.transaction.Transactional;
|
---|
| 13 | import java.time.LocalDate;
|
---|
| 14 | import java.time.LocalDateTime;
|
---|
| 15 | import java.time.LocalTime;
|
---|
| 16 | import java.util.ArrayList;
|
---|
| 17 | import java.util.List;
|
---|
| 18 | import java.util.Optional;
|
---|
| 19 |
|
---|
| 20 | @Service
|
---|
| 21 | public class ReservationService {
|
---|
| 22 |
|
---|
| 23 | private final ReservationRepository reservationRepository;
|
---|
| 24 | private final UserRepository userRepository;
|
---|
| 25 |
|
---|
| 26 | @Autowired
|
---|
| 27 | public ReservationService(ReservationRepository reservationRepository, UserRepository userRepository) {
|
---|
| 28 | this.reservationRepository = reservationRepository;
|
---|
| 29 | this.userRepository = userRepository;
|
---|
| 30 | }
|
---|
| 31 |
|
---|
| 32 | // get all reservations for specified date and type
|
---|
| 33 | private List<ReservationEntity> getReservations(String date, String type) {
|
---|
| 34 | LocalDateTime start = stringToDateAndTime(date, "00:00");
|
---|
| 35 | LocalDateTime end = stringToDateAndTime(date, "23:59");
|
---|
| 36 | return reservationRepository.findByTypeAndDateTimeBetween(type, start, end);
|
---|
| 37 | }
|
---|
| 38 |
|
---|
| 39 | // make reservation
|
---|
| 40 | public void makeReservation(String date, String time, String type) {
|
---|
| 41 | Authentication auth = SecurityContextHolder.getContext().getAuthentication();
|
---|
| 42 | String username = auth.getName();
|
---|
| 43 | UserEntity user = userRepository.findByUsername(username);
|
---|
| 44 | ReservationEntity reservationEntity = new ReservationEntity();
|
---|
| 45 | reservationEntity.setDateTime(stringToDateAndTime(date, time));
|
---|
| 46 | reservationEntity.setType(type);
|
---|
| 47 | reservationEntity.setUser(user);
|
---|
| 48 | reservationRepository.save(reservationEntity);
|
---|
| 49 | }
|
---|
| 50 |
|
---|
| 51 | // convert string to localdatetime
|
---|
| 52 | private LocalDateTime stringToDateAndTime(String date, String time) {
|
---|
| 53 | int year = Integer.parseInt(date.split("-")[0]);
|
---|
| 54 | int month = Integer.parseInt(date.split("-")[1]);
|
---|
| 55 | int day = Integer.parseInt(date.split("-")[2]);
|
---|
| 56 | int hour = Integer.parseInt(time.split(":")[0]);
|
---|
| 57 | int minutes = Integer.parseInt(time.split(":")[1]);
|
---|
| 58 | LocalDate tempDate = LocalDate.of(year, month, day);
|
---|
| 59 | LocalTime tempTime = LocalTime.of(hour, minutes);
|
---|
| 60 | return LocalDateTime.of(tempDate, tempTime);
|
---|
| 61 | }
|
---|
| 62 |
|
---|
| 63 | // get free reservations
|
---|
| 64 | public List<LocalTime> getFreeReservations(String date, String type) {
|
---|
| 65 |
|
---|
| 66 | List<ReservationEntity> reservations = getReservations(date, type);
|
---|
| 67 | List<LocalTime> freeReservations = getWorkingHours();
|
---|
| 68 | for (ReservationEntity r : reservations) {
|
---|
| 69 | LocalTime temp = r.getDateTime().toLocalTime();
|
---|
| 70 | for (LocalTime l : freeReservations) {
|
---|
| 71 | if (temp.equals(l)) {
|
---|
| 72 | freeReservations.remove(l);
|
---|
| 73 | break;
|
---|
| 74 | }
|
---|
| 75 | }
|
---|
| 76 | }
|
---|
| 77 | return freeReservations;
|
---|
| 78 | }
|
---|
| 79 |
|
---|
| 80 | // get all reservations for specified date and type
|
---|
| 81 | public List<ReservationResponse> getScheduledReservations(String date, String type) {
|
---|
| 82 | List<ReservationEntity> reservations = getReservations(date, type);
|
---|
| 83 | List<ReservationResponse> result = new ArrayList<>();
|
---|
| 84 | for (ReservationEntity r : reservations) {
|
---|
| 85 | ReservationResponse reservationResponse = new ReservationResponse(r.getId(), r.getDateTime(), r.getType(), r.isValid());
|
---|
| 86 | reservationResponse.setName(r.getUser().getName());
|
---|
| 87 | reservationResponse.setSurname(r.getUser().getSurname());
|
---|
| 88 | result.add(reservationResponse);
|
---|
| 89 | }
|
---|
| 90 | return result;
|
---|
| 91 | }
|
---|
| 92 |
|
---|
| 93 | // get all user reservations, return another class for better security
|
---|
| 94 | public List<ReservationResponse> getUserReservations() {
|
---|
| 95 | UserEntity user = getUser();
|
---|
| 96 | List<ReservationEntity> reservationEntities = reservationRepository.findAllByUser(user);
|
---|
| 97 | List<ReservationResponse> result = new ArrayList<>();
|
---|
| 98 | for (ReservationEntity r : reservationEntities) {
|
---|
| 99 | result.add(new ReservationResponse(r.getId(), r.getDateTime(), r.getType(), r.isValid()));
|
---|
| 100 | }
|
---|
| 101 | return result;
|
---|
| 102 | }
|
---|
| 103 |
|
---|
| 104 |
|
---|
| 105 | // get current user
|
---|
| 106 | public UserEntity getUser() {
|
---|
| 107 | Authentication auth = SecurityContextHolder.getContext().getAuthentication();
|
---|
| 108 | String username = auth.getName();
|
---|
| 109 | UserEntity user = userRepository.findByUsername(username);
|
---|
| 110 | return user;
|
---|
| 111 | }
|
---|
| 112 |
|
---|
| 113 | // cancel reservation - user
|
---|
| 114 | @Transactional
|
---|
| 115 | public void cancelReservation(Long id) {
|
---|
| 116 | Optional<ReservationEntity> reservationEntity = reservationRepository.findById(id);
|
---|
| 117 | if (reservationEntity.isPresent()) {
|
---|
| 118 | if (getUser().getId().equals(reservationEntity.get().getUser().getId())) {
|
---|
| 119 | reservationRepository.delete(reservationEntity.get());
|
---|
| 120 | }
|
---|
| 121 | }
|
---|
| 122 | }
|
---|
| 123 |
|
---|
| 124 | // cancel reservation - user
|
---|
| 125 | @Transactional
|
---|
| 126 | public void cancelReservationAdmin(Long id) {
|
---|
| 127 | Optional<ReservationEntity> reservationEntity = reservationRepository.findById(id);
|
---|
| 128 | reservationEntity.ifPresent(reservationRepository::delete);
|
---|
| 129 | }
|
---|
| 130 |
|
---|
| 131 | // block reservation
|
---|
| 132 |
|
---|
| 133 | public void blockReservation(String date, String time, String type) {
|
---|
| 134 | Optional<ReservationEntity> reservationEntity = Optional.ofNullable(reservationRepository.findReservationEntityByDateTimeAndType(stringToDateAndTime(date, time), type));
|
---|
| 135 | if (reservationEntity.isPresent()) {
|
---|
| 136 | reservationEntity.get().setValid(false);
|
---|
| 137 | reservationRepository.save(reservationEntity.get());
|
---|
| 138 | } else {
|
---|
| 139 | ReservationEntity reservation = new ReservationEntity();
|
---|
| 140 | reservation.setDateTime(stringToDateAndTime(date, time));
|
---|
| 141 | reservation.setType(type);
|
---|
| 142 | reservation.setValid(false);
|
---|
| 143 | reservation.setUser(getUser());
|
---|
| 144 | reservationRepository.save(reservation);
|
---|
| 145 | }
|
---|
| 146 | }
|
---|
| 147 |
|
---|
| 148 | // get blocked reservations
|
---|
| 149 | public List<ReservationResponse> getBlockedReservations() {
|
---|
| 150 | List<ReservationEntity> reservationEntities = reservationRepository.findAllByValid(false);
|
---|
| 151 | List<ReservationResponse> result = new ArrayList<>();
|
---|
| 152 | for (ReservationEntity r : reservationEntities) {
|
---|
| 153 | result.add(new ReservationResponse(r.getId(), r.getDateTime(), r.getType(), r.isValid()));
|
---|
| 154 | }
|
---|
| 155 | return result;
|
---|
| 156 | }
|
---|
| 157 |
|
---|
| 158 | // unblock reservation
|
---|
| 159 | @Transactional
|
---|
| 160 | public void unblockReservation(Long id) {
|
---|
| 161 | Optional<ReservationEntity> reservationEntity = reservationRepository.findById(id);
|
---|
| 162 | if (reservationEntity.isPresent()) {
|
---|
| 163 | if (reservationEntity.get().getUser().getName().equals("admin")) {
|
---|
| 164 | reservationRepository.delete(reservationEntity.get());
|
---|
| 165 | } else {
|
---|
| 166 | reservationEntity.get().setValid(true);
|
---|
| 167 | }
|
---|
| 168 | }
|
---|
| 169 | }
|
---|
| 170 |
|
---|
| 171 | // get working hours for salon
|
---|
| 172 | public List<LocalTime> getWorkingHours() {
|
---|
| 173 | List<LocalTime> temp = new ArrayList<>();
|
---|
| 174 | for (int i = 9; i <= 21; i++) {
|
---|
| 175 | temp.add(LocalTime.of(i, 0));
|
---|
| 176 | }
|
---|
| 177 | return temp;
|
---|
| 178 | }
|
---|
| 179 |
|
---|
| 180 |
|
---|
| 181 | }
|
---|
| 182 |
|
---|