diff --git a/src/main/java/ru/soune/nocopy/handler/DockViewFileHandler.java b/src/main/java/ru/soune/nocopy/handler/DockViewFileHandler.java index 7784ed7..00cb3d1 100644 --- a/src/main/java/ru/soune/nocopy/handler/DockViewFileHandler.java +++ b/src/main/java/ru/soune/nocopy/handler/DockViewFileHandler.java @@ -58,6 +58,7 @@ public class DockViewFileHandler implements RequestHandler { Long userId = authService.useUserAuthToken(docViewerRequest.getToken()); List fileIds = fileEntityService.getAllUserFiles(userId) .stream() + .filter(fileEntity -> fileEntity.getMimeType().equals("document")) .map(FileEntity::getId) .toList(); @@ -96,6 +97,7 @@ public class DockViewFileHandler implements RequestHandler { Long userId = authService.useUserAuthToken(docViewerRequest.getToken()); List fileIds = fileEntityService.getAllUserFiles(userId) .stream() + .filter(fileEntity -> fileEntity.getMimeType().equals("document")) .map(FileEntity::getId) .toList(); @@ -119,33 +121,4 @@ public class DockViewFileHandler implements RequestHandler { null); } } - -// private BaseResponse handleGetViewStats(BaseRequest request, DocViewerRequest docViewerRequest) { -// try { -// Long userId = authService.useUserAuthToken(docViewerRequest.getToken()); -// List fileIds = fileEntityService.getAllUserFiles(userId) -// .stream() -// .map(FileEntity::getId) -// .toList(); -// -// if (fileIds.isEmpty()) { -// return new BaseResponse(request.getMsgId(), -// MessageCode.FILE_NOT_FOUND.getCode(), -// "Files for search not found", -// null); -// } -// -// List statsByFileIds = dockViewService.getStatsByFileIds(fileIds); -// -// return new BaseResponse(request.getMsgId(), -// MessageCode.SUCCESS.getCode(), -// MessageCode.SUCCESS.getDescription(), -// statsByFileIds); -// } catch (NotFoundAuthToken e) { -// return new BaseResponse(request.getMsgId(), -// MessageCode.INVALID_TOKEN.getCode(), -// "Authentication required", -// null); -// } -// } }