NCP-3

src/main/java/ru/soune/no_copy/controller/AuthController.java

@@ -9,9 +9,12 @@ import ru.soune.no_copy.dto.LoginRequest;
 import ru.soune.no_copy.dto.LoginResponse;
 import ru.soune.no_copy.dto.RegisterRequest;
 import ru.soune.no_copy.entity.AuthToken;
+import ru.soune.no_copy.exception.TokenNotFoundException;
+import ru.soune.no_copy.repository.AuthTokenRepository;
 import ru.soune.no_copy.service.AuthService;
 
 import java.util.Map;
+import java.util.Optional;
 
 @RestController
 @RequestMapping("/api/auth")
@@ -20,6 +23,8 @@ public class AuthController {
 
     private final AuthService authService;
 
+    private final AuthTokenRepository authTokenRepository;
+
     @PostMapping("/register")
     public ResponseEntity<AuthResponse> register(@Valid @RequestBody RegisterRequest registerRequest) {
         AuthToken authToken = authService.register(registerRequest);
@@ -40,6 +45,12 @@ public class AuthController {
     public ResponseEntity<?> logout(@RequestHeader("Authorization") String tokenHeader) {
         String token = tokenHeader.replace("Bearer ", "");
 
+        Optional<AuthToken> opToken = authTokenRepository.findByToken(token);
+
+        if (opToken.isEmpty()) {
+            throw new TokenNotFoundException("Token not found");
+        }
+
         authService.logout(token);
 
         return ResponseEntity.ok(Map.of("success", true));