dev add load and download file for moderatuion user

2026-04-04 15:31:51 +07:00
parent 6bffd5b385
commit f4fc1f26a7
13 changed files with 440 additions and 43 deletions
@@ -77,12 +77,28 @@ public class FileEntityService {
            for (Long uId : userRepository.findByCompanyId(user.getCompany().getId()).stream()
                    .map(User::getId)
                    .toList()) {
-                allFiles.addAll(fileEntityRepository.findByUserIdAndStatusIn(uId,
-                        List.of(FileStatus.ACTIVE, FileStatus.BLOCKED, FileStatus.MODERATION)));
+                allFiles.addAll(getAllUserFiles(uId, List.of(FileStatus.ACTIVE, FileStatus.BLOCKED,
+                        FileStatus.MODERATION)));
            }
        } else {
-            allFiles = fileEntityRepository.findByUserIdAndStatusIn(userId,
-                    List.of(FileStatus.ACTIVE, FileStatus.BLOCKED, FileStatus.MODERATION));
+            allFiles = getAllUserFiles(userId, List.of(FileStatus.ACTIVE, FileStatus.BLOCKED, FileStatus.MODERATION));
+        }
+
+        return allFiles;
+    }
+
+    public List<FileEntity> getAllUserFiles(Long userId, List<FileStatus> statuses) {
+        User user = userRepository.findById(userId).orElseThrow();
+        List<FileEntity> allFiles = new ArrayList<>();
+
+        if (user.getCompany() != null) {
+            for (Long uId : userRepository.findByCompanyId(user.getCompany().getId()).stream()
+                    .map(User::getId)
+                    .toList()) {
+                allFiles.addAll(fileEntityRepository.findByUserIdAndStatusIn(uId, statuses));
+            }
+        } else {
+            allFiles = fileEntityRepository.findByUserIdAndStatusIn(userId, statuses);
        }

        return allFiles;