NCBACK-33

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

@@ -13,6 +13,7 @@ import ru.soune.nocopy.entity.file.ProtectionStatus;
 import ru.soune.nocopy.exception.DuplicateImageException;
 import ru.soune.nocopy.exception.FileEntityNotFoundException;
 import ru.soune.nocopy.repository.FileEntityRepository;
+import ru.soune.nocopy.repository.ImageHashRepository;
 import ru.soune.nocopy.repository.SimilarImageProjection;
 import ru.soune.nocopy.service.FileSimilarityService;
 import ru.soune.nocopy.service.ImageHashService;
@@ -38,6 +39,8 @@ public class FileEntityService {
 
     private final FileSimilarityService fileSimilarityService;
 
+    private final ImageHashRepository imageHashRepository;
+
     @Transactional(noRollbackFor = DuplicateImageException.class)
     public FileEntity createFromUploadSession(FileUploadSession session, String checksum) {
         log.info("Creating FileEntity for upload session: {}", session.getUploadId());
@@ -185,27 +188,45 @@ public class FileEntityService {
     }
 
     @Transactional
-    public void markAsDeleted(FileEntity fileEntity) {
+    public FileEntity markAsDeleted(FileEntity fileEntity) throws IOException {
         fileEntity.setStatus(FileStatus.DELETED);
         fileEntity.setUpdatedAt(LocalDateTime.now());
+        fileEntity.setProtectionStatus(ProtectionStatus.NOT_PROTECTED);
+
+        Path path = Paths.get(fileEntity.getProtectedFilePath());
+        Files.deleteIfExists(path);
+        fileEntity.setProtectedFilePath("");
+
+        return fileEntityRepository.save(fileEntity);
+    }
+
+    public void softDeleteFileWithHash(FileEntity fileEntity) throws IOException {
+        if (fileEntity.getImageHash() != null) {
+            fileEntity.setImageHash(null);
+        }
+
+        fileEntity.setStatus(FileStatus.DELETED);
+        fileEntity.setUpdatedAt(LocalDateTime.now());
+        fileEntity.setProtectionStatus(ProtectionStatus.NOT_PROTECTED);
+
+        Path path = Paths.get(fileEntity.getProtectedFilePath());
+        Files.deleteIfExists(path);
+        fileEntity.setProtectedFilePath("");
 
         fileEntityRepository.save(fileEntity);
     }
 
     @Transactional
-    public boolean deleteFromDisk(FileEntity fileEntity) throws IOException {
+    public void deleteFromDisk(FileEntity fileEntity) throws IOException {
         Path path = Paths.get(fileEntity.getFilePath());
 
         if (!Files.exists(path)) {
-            return true;
+            return;
         }
 
         Files.delete(path);
 
-//        markAsDeleted(fileEntity);
         fileEntityRepository.delete(fileEntity);
-
-        return true;
     }
 
     @Transactional(readOnly = true)