From 141369cdecc046a613c06732dcfbd087fb818f44 Mon Sep 17 00:00:00 2001
From: Konstantin <vaskon0306@gmail.com>
Date: Thu, 20 Feb 2025 15:33:27 +0300
Subject: [PATCH] prohibit banning admins

---
 .../nto/service/exception/CantBanAnAdminException.java   | 7 +++++++
 .../example/nto/service/impl/EmployeeServiceImpl.java    | 9 +++++----
 2 files changed, 12 insertions(+), 4 deletions(-)
 create mode 100644 src/main/java/com/example/nto/service/exception/CantBanAnAdminException.java

diff --git a/src/main/java/com/example/nto/service/exception/CantBanAnAdminException.java b/src/main/java/com/example/nto/service/exception/CantBanAnAdminException.java
new file mode 100644
index 0000000..bb583d4
--- /dev/null
+++ b/src/main/java/com/example/nto/service/exception/CantBanAnAdminException.java
@@ -0,0 +1,7 @@
+package com.example.nto.service.exception;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.web.bind.annotation.ResponseStatus;
+
+@ResponseStatus(HttpStatus.FORBIDDEN)
+public class CantBanAnAdminException extends RuntimeException {}
diff --git a/src/main/java/com/example/nto/service/impl/EmployeeServiceImpl.java b/src/main/java/com/example/nto/service/impl/EmployeeServiceImpl.java
index 1a1cd9c..8f6ffeb 100644
--- a/src/main/java/com/example/nto/service/impl/EmployeeServiceImpl.java
+++ b/src/main/java/com/example/nto/service/impl/EmployeeServiceImpl.java
@@ -9,10 +9,7 @@ import com.example.nto.repository.CodeRepository;
 import com.example.nto.repository.EmployeeRepository;
 import com.example.nto.repository.EntryRepository;
 import com.example.nto.service.EmployeeService;
-import com.example.nto.service.exception.CodeNotFoundException;
-import com.example.nto.service.exception.EmployeeBannedException;
-import com.example.nto.service.exception.EmployeeNotFoundException;
-import com.example.nto.service.exception.NotAnAdminException;
+import com.example.nto.service.exception.*;
 import lombok.RequiredArgsConstructor;
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.stereotype.Service;
@@ -81,6 +78,10 @@ public class EmployeeServiceImpl implements EmployeeService {
     @Override
     public void banEmployee(String login) {
         employeeExists(login);
+        var e = getEmployee(login);
+        if (e.getRole() == EmployeeRoleType.ADMIN) {
+            throw new CantBanAnAdminException();
+        }
         var employee = getEmployee(login);
         employee.setIsBanned(true);
         employeeRepository.save(employee);