backdev/no-copy
1
0
Fork 0
Code Issues Pull Requests 5 Actions Packages Projects Releases Wiki Activity

dev use authtoken
Test Workflow / test (push) Successful in 5s
Details

Browse Source
This commit is contained in:
vladp
2026-02-24 14:56:41 +07:00
parent 274861c300
commit 28ce99ec1c
3 changed files with 40 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/ru/soune/nocopy/service/FileSimilarityService.java
+12 -10
View File
@@ -1,32 +1,25 @@
package ru.soune.nocopy.service;
import com.vrt.NoCopyFileService;
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.phash.PHash;
import com.vrt.fileprotection.image.phash.PerceptualHashHelper;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.hibernate.validator.internal.util.stereotypes.Lazy;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.Pageable;
import org.springframework.stereotype.Service;
import ru.soune.nocopy.dto.file.SimilarFileDTO;
import ru.soune.nocopy.dto.file.SimilarityFilter;
import ru.soune.nocopy.entity.company.Company;
import ru.soune.nocopy.entity.file.FileEntity;
import ru.soune.nocopy.entity.file.ImageHashEntity;
import ru.soune.nocopy.entity.user.User;
import ru.soune.nocopy.exception.DuplicateImageException;
import ru.soune.nocopy.repository.*;
import ru.soune.nocopy.util.FileUtil;
import java.io.FileInputStream;
import java.io.InputStream;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.security.MessageDigest;
import java.util.*;
import java.util.stream.Collectors;
@@ -122,6 +115,8 @@ public class FileSimilarityService {
}
}
private final UserRepository userRepository;
public Page<SimilarFileDTO> findSimilarFiles(
String fileId,
List<String> similarityLevels,
@@ -137,7 +132,14 @@ public class FileSimilarityService {
candidates = repository.findCandidates(fileId);
} else {
Long userId = authTokenRepository.findUserIdByToken(authToken);
candidates = repository.findCandidatesFromUserFiles(fileId, userId);
User user = userRepository.findById(userId).orElseThrow(() -> new RuntimeException("User not found"));
Company company = user.getCompany();
if (company != null) {
Set<Long> userIds = company.getUsers().stream().map(User::getId).collect(Collectors.toSet());
candidates = repository.findCandidatesFromUserFiles(fileId, userIds);
} else {
candidates = repository.findCandidatesFromUserFiles(fileId, userId);
}
}
List<String> levels = (similarityLevels != null && !similarityLevels.isEmpty())