dev add

src/main/java/ru/soune/nocopy/service/file/FileStatsService.java

@@ -7,15 +7,24 @@ import ru.soune.nocopy.entity.file.FileEntity;
 import ru.soune.nocopy.entity.file.FileStatus;
 import ru.soune.nocopy.entity.file.FileType;
 import ru.soune.nocopy.entity.file.ProtectionStatus;
+import ru.soune.nocopy.entity.user.User;
 import ru.soune.nocopy.repository.FileEntityRepository;
+import ru.soune.nocopy.repository.UserRepository;
+import ru.soune.nocopy.service.CompanyService;
 
+import java.util.ArrayList;
 import java.util.List;
 
 @Service
 @RequiredArgsConstructor
 public class FileStatsService {
+
     private final FileEntityRepository fileEntityRepository;
 
+    private final CompanyService companyService;
+
+    private final UserRepository userRepository;
+
     public FileInfoUserResponse getUserFileStats(Long userId) {
         List<FileEntity> userFiles = fileEntityRepository.findByUserId(userId);
         return calculateStats(userFiles);
@@ -49,6 +58,47 @@ public class FileStatsService {
                 .build();
     }
 
+    public Long calculateUserFileSize(Long userId) {
+        List<FileEntity> userFiles = fileEntityRepository.findByUserId(userId);
+
+        return calculateTotalSize(userFiles);
+    }
+
+    public Long calculateCompanyFileSize(Long userId) {
+        User user = userRepository.findById(userId).orElseThrow();
+
+        List<User> companyUsers = companyService.getCompanyUsers(user.getCompany().getId());
+
+        List<FileEntity> userFiles = new ArrayList<>();
+
+        for (User companyUser : companyUsers) {
+            userFiles.addAll(fileEntityRepository.findByUserId(companyUser.getId()));
+        }
+
+        return calculateTotalSize(userFiles);
+    }
+
+    public Integer filesUserCount(Long userId) {
+        List<FileEntity> userFiles = fileEntityRepository.findByUserId(userId);
+
+        return calculateTotalCount(userFiles);
+    }
+
+
+    public Integer filesCompanyCount(Long userId) {
+        User user = userRepository.findById(userId).orElseThrow();
+
+        List<User> companyUsers = companyService.getCompanyUsers(user.getCompany().getId());
+
+        List<FileEntity> userFiles = new ArrayList<>();
+
+        for (User companyUser : companyUsers) {
+            userFiles.addAll(fileEntityRepository.findByUserId(companyUser.getId()));
+        }
+
+        return calculateTotalCount(userFiles);
+    }
+
     private Long protectedUserFiles(List<FileEntity> files) {
         return files.stream().filter(file -> file.getProtectionStatus() == ProtectionStatus.PROTECTED).count();
     }