dev add monitoring basic

src/main/java/ru/soune/nocopy/handler/GetViolationsHandler.java

@@ -0,0 +1,85 @@
+package ru.soune.nocopy.handler;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+import ru.soune.nocopy.dto.BaseRequest;
+import ru.soune.nocopy.dto.BaseResponse;
+import ru.soune.nocopy.dto.MessageCode;
+import ru.soune.nocopy.dto.monitoring.GetViolationsRequest;
+import ru.soune.nocopy.dto.monitoring.ViolationResponse;
+import ru.soune.nocopy.entity.monitoring.FoundViolationEntity;
+import ru.soune.nocopy.entity.user.User;
+import ru.soune.nocopy.repository.FoundViolationRepository;
+import ru.soune.nocopy.service.register.AuthService;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class GetViolationsHandler implements RequestHandler {
+    private final ObjectMapper objectMapper;
+
+    private final FoundViolationRepository violationRepository;
+
+    private final AuthService authService;
+
+    @Override
+    public BaseResponse handle(BaseRequest request) throws Exception {
+        GetViolationsRequest violationsRequest = objectMapper.convertValue(
+                request.getMessageBody(), GetViolationsRequest.class);
+
+        String authToken = violationsRequest.getAuthToken();
+
+        //TODO add return for null token
+        if (authToken == null) {
+            return new BaseResponse(0, 0, "", new Object());
+        }
+
+        User user = authService.getAuthToken(authToken).getUser();
+        List<FoundViolationEntity> violations;
+
+        if (violationsRequest.getFileId() != null) {
+            violations = violationRepository.findByFileIdOrderByFoundAtDesc(
+                    violationsRequest.getFileId());
+        } else {
+            violations = violationRepository.findByUserIdOrderByFoundAtDesc(user.getId());
+        }
+
+        if (violationsRequest.isMarkAsRead()) {
+            violations.stream()
+                    .filter(FoundViolationEntity::isNew)
+                    .forEach(v -> v.setNew(false));
+            violationRepository.saveAll(violations);
+        }
+
+        List<ViolationResponse> response = violations.stream()
+                .map(this::mapToResponse)
+                .collect(Collectors.toList());
+
+        return new BaseResponse(
+                request.getMsgId(),
+                MessageCode.SUCCESS.getCode(),
+                "Violations retrieved",
+                response
+        );
+    }
+
+    private ViolationResponse mapToResponse(FoundViolationEntity entity) {
+        return ViolationResponse.builder()
+                .id(entity.getId())
+                .fileId(entity.getFileId())
+                .fileName(entity.getFileName())
+                .foundUrl(entity.getFoundUrl())
+                .pageUrl(entity.getPageUrl())
+                .pageTitle(entity.getPageTitle())
+                .host(entity.getHost())
+                .searchEngine(entity.getSearchEngine())
+                .foundAt(entity.getFoundAt())
+                .isNew(entity.isNew())
+                .build();
+    }
+}

src/main/java/ru/soune/nocopy/handler/MonitoringHandler.java

@@ -0,0 +1,72 @@
+package ru.soune.nocopy.handler;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+import ru.soune.nocopy.dto.BaseRequest;
+import ru.soune.nocopy.dto.BaseResponse;
+import ru.soune.nocopy.dto.MessageCode;
+import ru.soune.nocopy.dto.monitoring.MonitoringStatusResponse;
+import ru.soune.nocopy.dto.monitoring.SetMonitoringRequest;
+import ru.soune.nocopy.entity.monitoring.MonitoringType;
+import ru.soune.nocopy.entity.tarif.TariffType;
+import ru.soune.nocopy.entity.user.AuthToken;
+import ru.soune.nocopy.entity.user.User;
+import ru.soune.nocopy.repository.TariffRepository;
+import ru.soune.nocopy.service.monitoring.FileMonitoringService;
+import ru.soune.nocopy.service.register.AuthService;
+import ru.soune.nocopy.service.tariff.TariffService;
+
+import java.util.Map;
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class MonitoringHandler implements RequestHandler {
+
+    private final ObjectMapper objectMapper;
+
+    private final FileMonitoringService monitoringService;
+
+    private final AuthService authService;
+
+    private final TariffService tariffService;
+
+    private final TariffRepository tariffRepository;
+
+    @Override
+    public BaseResponse handle(BaseRequest request) throws Exception {
+        SetMonitoringRequest setRequest = objectMapper.convertValue(
+                request.getMessageBody(), SetMonitoringRequest.class);
+
+        String authToken = setRequest.getAuthToken();
+        AuthToken userAuthToken = authService.getAuthToken(authToken);
+
+        if (authToken == null || userAuthToken == null) {
+            return new BaseResponse(30007, MessageCode.AUTH_TOKEN_NOT_FOUND.getCode(),
+                    MessageCode.AUTH_TOKEN_NOT_FOUND.getDescription(),
+                    Map.of("token", authToken));
+        }
+
+        User user = userAuthToken.getUser();
+        Long userId = user.getId();
+        MonitoringType monitoringType = setRequest.getMonitoringType();
+
+        monitoringService.setMonitoring(userId, setRequest.getFileId(), monitoringType);
+
+        MonitoringStatusResponse status = monitoringService.getMonitoringStatus(
+                setRequest.getFileId(), userId);
+
+        return new BaseResponse(
+                request.getMsgId(),
+                MessageCode.SUCCESS.getCode(),
+                "Monitoring settings updated",
+                status);
+    }
+    //TODO check tokens for apply
+    private boolean checkTokensForMonitoring(User user, MonitoringType monitoringType) {
+        tariffService.getTariffByType(TariffType.valueOf(monitoringType.name()));
+        return true;
+    }
+}