dev add new action

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

@@ -17,6 +17,7 @@ import ru.soune.nocopy.entity.complaint.LawCasePriority;
 import ru.soune.nocopy.entity.complaint.LawCaseType;
 import ru.soune.nocopy.entity.tokenoperation.OperationType;
 import ru.soune.nocopy.entity.user.User;
+import ru.soune.nocopy.entity.violation.Violation;
 import ru.soune.nocopy.exception.NotValidFieldException;
 import ru.soune.nocopy.exception.TariffNotFoundException;
 import ru.soune.nocopy.exception.UserNotFoundException;
@@ -25,8 +26,10 @@ import ru.soune.nocopy.service.complaint.LawCaseService;
 import ru.soune.nocopy.service.register.AuthService;
 import ru.soune.nocopy.service.tariff.TariffConstants;
 import ru.soune.nocopy.service.tariff.TariffInfoService;
+import ru.soune.nocopy.service.violation.ViolationService;
 
 import java.io.IOException;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
@@ -46,6 +49,8 @@ public class LawCaseHandler implements RequestHandler {
 
     private final TariffInfoService tariffInfoService;
 
+    private final ViolationService violationService;
+
     @Override
     public BaseResponse handle(BaseRequest request) throws Exception {
 
@@ -84,9 +89,17 @@ public class LawCaseHandler implements RequestHandler {
                 lawCaseRequest.getSortBy(),
                 lawCaseRequest.getSortDir());
 
-        List<LawCaseResponse> responses = lawCases.getContent().stream()
-                .map(LawCaseResponse::fromEntity)
-                .collect(Collectors.toList());
+        List<LawCaseResponse> responses = new ArrayList<>();
+        List<LawCase> content = lawCases.getContent();
+
+        for (LawCase lawCase: content) {
+            LawCaseResponse lawCaseResponse = LawCaseResponse.fromEntity(lawCase);
+
+            Violation violation = violationService.getById(lawCase.getViolationId());
+            if (violation != null) lawCaseResponse.setFileId(violation.getFileEntity().getId());
+
+            responses.add(lawCaseResponse);
+        }
 
         PaginatedResponse<LawCaseResponse> paginatedResponse = new PaginatedResponse<>(
                 responses,