diff --git a/src/main/java/ru/soune/nocopy/controller/ApiController.java b/src/main/java/ru/soune/nocopy/controller/ApiController.java index 719f967..f4ba07e 100644 --- a/src/main/java/ru/soune/nocopy/controller/ApiController.java +++ b/src/main/java/ru/soune/nocopy/controller/ApiController.java @@ -772,14 +772,10 @@ public class ApiController { } @GetMapping("/check/file_stats") - public ResponseEntity checkFileProtectStats(@RequestHeader("Authorization") String tokenHeader) { - String token = tokenHeader.replace("Bearer ", ""); + public ResponseEntity checkFileProtectStats() { + Long userId = (Long) httpServletRequest.getAttribute("userId"); - Optional tokenOptional = authTokenRepository.findByToken(token); - - AuthToken authToken = tokenOptional.orElseThrow(() -> new NotFoundAuthToken("Token: " + token + "not found")); - - Map fileTypeStats = protectionsLimitService.getFileTypeStats(authToken.getUser().getId()); + Map fileTypeStats = protectionsLimitService.getFileTypeStats(userId); return ResponseEntity.ok().body(fileTypeStats); }