dev add check tokens for monitoring

src/main/java/ru/soune/nocopy/service/monitoring/MonitoringSearchService.java

@@ -1,19 +1,30 @@
 package ru.soune.nocopy.service.monitoring;
 
+import jakarta.mail.MessagingException;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.scheduling.annotation.Async;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import ru.soune.nocopy.dto.file.YandexSearchResponse;
+import ru.soune.nocopy.dto.tarriff.TariffDTO;
 import ru.soune.nocopy.entity.file.*;
 import ru.soune.nocopy.entity.monitoring.FileMonitoringEntity;
 import ru.soune.nocopy.entity.monitoring.FoundViolationEntity;
 import ru.soune.nocopy.entity.monitoring.MonitoringType;
 import ru.soune.nocopy.entity.monitoring.SearchEngine;
+import ru.soune.nocopy.entity.tarif.TariffInfo;
+import ru.soune.nocopy.entity.tarif.TariffType;
+import ru.soune.nocopy.entity.user.User;
+import ru.soune.nocopy.exception.TariffNotFoundException;
 import ru.soune.nocopy.repository.FileMonitoringRepository;
 import ru.soune.nocopy.repository.FoundViolationRepository;
+import ru.soune.nocopy.repository.UserRepository;
+import ru.soune.nocopy.service.mail.EmailService;
 import ru.soune.nocopy.service.search.SearchImageService;
+import ru.soune.nocopy.service.tariff.TariffInfoService;
+import ru.soune.nocopy.service.tariff.TariffService;
+
 import java.io.IOException;
 import java.time.LocalDateTime;
 import java.util.List;
@@ -30,8 +41,16 @@ public class MonitoringSearchService {
 
     private final FileMonitoringRepository monitoringRepository;
 
+    private final TariffInfoService tariffInfoService;
+
+    private final TariffService tariffService;
+
+    private final EmailService emailService;
+
+    private final UserRepository userRepository;
+
     @Transactional
-    public void processFileSearch(FileMonitoringEntity monitoring) {
+    public void processFileSearch(FileMonitoringEntity monitoring) throws MessagingException, IOException {
         String searchSessionId = UUID.randomUUID().toString();
         log.info("Starting search session: {} for file: {}", searchSessionId, monitoring.getFile().getId());
 
@@ -39,7 +58,12 @@ public class MonitoringSearchService {
         monitoring.setLastRunStatus("IN_PROGRESS");
         monitoringRepository.save(monitoring);
 
+        MonitoringType monitoringType = monitoring.getMonitoringType();
+        TariffDTO tariffMonitoring = tariffService.getTariffByType(TariffType.valueOf(monitoringType.name()));
+
         try {
+            tariffInfoService.writeOffTokens(monitoring.getUserId(), tariffMonitoring.getTokens());
+
             String yandexResponse = searchImageService.searchReverseByPublicUrl(
                     monitoring.getFile(), "yandex_reverse_image", "visual_matches");
 
@@ -62,7 +86,16 @@ public class MonitoringSearchService {
             }
 
             monitoring.setLastRunStatus("SUCCESS - Found " + savedCount + " violations");
+        } catch (TariffNotFoundException e) {
+            User user = userRepository.findById(monitoring.getUserId()).orElseThrow();
+            TariffInfo activeTariffInfo = user.getActiveTariffInfo();
+            int currentTokens = activeTariffInfo.getTokens() + activeTariffInfo.getBoughtTokens();
 
+            emailService.sendTokensNotFoundEmail(user, currentTokens, tariffMonitoring.getTokens());
+            monitoring.setLastRunStatus("ERROR: " + e.getMessage());
+            updateNextRun(monitoring);
+
+            monitoringRepository.save(monitoring);
         } catch (Exception e) {
             log.error("Search failed for session: {}", searchSessionId, e);
             monitoring.setLastRunStatus("ERROR: " + e.getMessage());