From 7631ea8c8c6f5c3b42b257f412bc6007f43bd71c Mon Sep 17 00:00:00 2001
From: vladp <vlad.popovtsevv@gmail.com>
Date: Sat, 21 Feb 2026 21:41:05 +0700
Subject: [PATCH] dev comment logic for check max users

---
 .../java/ru/soune/nocopy/controller/ApiController.java   | 2 ++
 .../ru/soune/nocopy/exception/FileFormatException.java   | 7 +++++++
 .../nocopy/service/file/impl/FileUploadServiceImpl.java  | 9 ++-------
 3 files changed, 11 insertions(+), 7 deletions(-)
 create mode 100644 src/main/java/ru/soune/nocopy/exception/FileFormatException.java

diff --git a/src/main/java/ru/soune/nocopy/controller/ApiController.java b/src/main/java/ru/soune/nocopy/controller/ApiController.java
index f9b6fe9..33c4976 100644
--- a/src/main/java/ru/soune/nocopy/controller/ApiController.java
+++ b/src/main/java/ru/soune/nocopy/controller/ApiController.java
@@ -149,6 +149,8 @@ public class ApiController {
                     MessageCode.DUPLICATE_FILE_UPLOAD.getDescription(),
                     duplicateData
             ));
+        }catch (FileFormatException e){
+            return buildErrorResponse(uploadId, chunkNumber, e.getMessage());
         } catch (Exception e) {
             log.error("Error uploading chunk", e);
             return buildErrorResponse(uploadId, chunkNumber, "Failed to upload chunk: " + e.getMessage());
diff --git a/src/main/java/ru/soune/nocopy/exception/FileFormatException.java b/src/main/java/ru/soune/nocopy/exception/FileFormatException.java
new file mode 100644
index 0000000..6480320
--- /dev/null
+++ b/src/main/java/ru/soune/nocopy/exception/FileFormatException.java
@@ -0,0 +1,7 @@
+package ru.soune.nocopy.exception;
+
+public class FileFormatException extends RuntimeException {
+    public FileFormatException(String message) {
+        super(message);
+    }
+}
diff --git a/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java b/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java
index ed895cc..013f9bc 100644
--- a/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java
+++ b/src/main/java/ru/soune/nocopy/service/file/impl/FileUploadServiceImpl.java
@@ -5,8 +5,6 @@ import com.vrt.fileprotection.FileProtector;
 import com.vrt.fileprotection.NoCopyCheckResult;
 import com.vrt.fileprotection.audio.AudioCheckResult;
 import com.vrt.fileprotection.documents.DocumentCheckResult;
-import com.vrt.fileprotection.image.ImageCheckResult;
-import com.vrt.fileprotection.image.ImageLocalSearch;
 import jakarta.annotation.PostConstruct;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -19,10 +17,7 @@ import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.multipart.MultipartFile;
 import ru.soune.nocopy.dto.file.UploadProgressResponse;
 import ru.soune.nocopy.entity.file.*;
-import ru.soune.nocopy.exception.ChunkSizeExceededException;
-import ru.soune.nocopy.exception.DuplicateImageException;
-import ru.soune.nocopy.exception.FileUploadException;
-import ru.soune.nocopy.exception.UploadSessionNotFoundException;
+import ru.soune.nocopy.exception.*;
 import ru.soune.nocopy.repository.FileEntityRepository;
 import ru.soune.nocopy.repository.FileUploadSessionRepository;
 import ru.soune.nocopy.repository.SimilarImageProjection;
@@ -333,7 +328,7 @@ public class FileUploadServiceImpl implements FileUploadService {
 
                             sessionRepository.save(session);
 
-                            throw new FileUploadException("Failed to upload chunk: .wav file not have RIFF header");
+                            throw new FileFormatException("Failed to upload chunk: .wav file not have RIFF header");
                         }
 
                         FileProtector.Type type = "document".equals(session.getFileType()) ?