From 1ced86ff8cd082db933f4b2c216d7f6712d7e9d2 Mon Sep 17 00:00:00 2001
From: vladp <vlad.popovtsevv@gmail.com>
Date: Tue, 10 Feb 2026 18:55:21 +0700
Subject: [PATCH] dev add pdf docs protect

---
 .../soune/nocopy/service/file/impl/FileUploadServiceImpl.java | 4 +++-
 src/main/java/ru/soune/nocopy/util/FileUtil.java              | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

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 8511493..db52626 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
@@ -249,7 +249,9 @@ public class FileUploadServiceImpl implements FileUploadService {
             cleanupSessionFiles(session);
 
             if (status != FileStatus.TEMP) {
-                noCopyFileService.addFile(fileUtil.createFileInfo(fileEntity));
+                FileProtector.FileInfo fileInfo = fileUtil.createFileInfo(fileEntity);
+
+                noCopyFileService.addFile(fileInfo);
             }
 
             log.info("File processing completed for session: {}", session.getUploadId());
diff --git a/src/main/java/ru/soune/nocopy/util/FileUtil.java b/src/main/java/ru/soune/nocopy/util/FileUtil.java
index 192e1d3..7bfb07b 100644
--- a/src/main/java/ru/soune/nocopy/util/FileUtil.java
+++ b/src/main/java/ru/soune/nocopy/util/FileUtil.java
@@ -41,6 +41,8 @@ public class FileUtil {
             return FileProtector.Type.VIDEO;
         } else if (mimeType.startsWith("audio")) {
             return FileProtector.Type.AUDIO;
+        } else if (mimeType.startsWith("document")) {
+            return FileProtector.Type.DOC;
         } else {
             return FileProtector.Type.IMAGE;
         }