dev add method link

src/main/java/ru/soune/nocopy/controller/ApiController.java

@@ -1,7 +1,5 @@
 package ru.soune.nocopy.controller;
 
-import com.google.api.client.util.IOUtils;
-import com.google.common.io.ByteStreams;
 import com.vrt.NoCopyFileService;
 import com.vrt.fileprotection.FileProtector;
 import com.vrt.fileprotection.NoCopyCheckResult;
@@ -9,24 +7,20 @@ import com.vrt.fileprotection.audio.AudioCheckResult;
 import com.vrt.fileprotection.documents.DocumentCheckResult;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.springframework.core.io.FileSystemResource;
-import org.springframework.core.io.InputStreamResource;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.web.PageableDefault;
 import org.springframework.http.HttpHeaders;
-import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.validation.BindingResult;
 import org.springframework.validation.FieldError;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
-import org.springframework.web.server.ResponseStatusException;
-import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody;
 import ru.soune.nocopy.dto.BaseRequest;
 import ru.soune.nocopy.dto.BaseResponse;
 import ru.soune.nocopy.dto.MessageCode;
+import ru.soune.nocopy.dto.docviewer.DocViewerRequest;
 import ru.soune.nocopy.dto.file.*;
 import ru.soune.nocopy.dto.register.RegAnswer;
 import ru.soune.nocopy.entity.company.Company;
@@ -41,6 +35,7 @@ import ru.soune.nocopy.repository.AuthTokenRepository;
 import ru.soune.nocopy.repository.FileEntityRepository;
 import ru.soune.nocopy.repository.UserRepository;
 import ru.soune.nocopy.service.FileSimilarityService;
+import ru.soune.nocopy.service.dockview.DockViewService;
 import ru.soune.nocopy.service.file.*;
 import ru.soune.nocopy.service.file.CheckCounterService;
 import ru.soune.nocopy.service.file.ProtectionsLimitService;
@@ -48,14 +43,8 @@ import ru.soune.nocopy.service.file.cloud.CloudStorageService;
 import ru.soune.nocopy.service.register.AuthService;
 import ru.soune.nocopy.service.user.moderation.UserVerificationService;
 import ru.soune.nocopy.util.FileUtil;
-import software.amazon.awssdk.services.s3.model.NoSuchKeyException;
 
 import java.io.File;
-import java.io.IOException;
-import java.io.InputStream;
-import java.nio.file.Files;
-import java.nio.file.Path;
-import java.nio.file.Paths;
 import java.util.*;
 import java.util.stream.Collectors;
 
@@ -92,7 +81,7 @@ public class ApiController {
 
     private final ZipService zipService;
 
-    private final UserVerificationService userVerificationService;
+    private final DockViewService dockViewService;
 
     @PostMapping("/v{version}/data")
     public ResponseEntity<?> handlePostRequest(@RequestBody BaseRequest request,
@@ -511,27 +500,6 @@ public class ApiController {
                         MessageCode.FILE_DELETE.getDescription(),
                         errorData));
             }
-//TODO
-
-//            if (!entityResponse.isExistsOnDisk()) {
-//                Map<String, Object> errorData = new HashMap<>();
-//                errorData.put("onDisk", entityResponse.isExistsOnDisk());
-//
-//                return ResponseEntity.ok().body(new BaseResponse(20004,
-//                        MessageCode.FILE_NOT_EXIST.getCode(),
-//                        MessageCode.FILE_NOT_EXIST.getDescription(),
-//                        errorData));
-//            }
-
-//            if (!file.exists()) {
-//                Map<String, Object> errorData = new HashMap<>();
-//                errorData.put("file", file.exists());
-//
-//                return ResponseEntity.ok().body(new BaseResponse(20004,
-//                        MessageCode.FILE_DOWNLOAD_ERROR.getCode(),
-//                        MessageCode.FILE_DOWNLOAD_ERROR.getDescription(),
-//                        errorData));
-//            }
 
             byte[] bytes = cloudStorageService.readFileFromStorageBytes(entityResponse.getProtectedFilePath());
 
@@ -556,16 +524,54 @@ public class ApiController {
                     MessageCode.AUTH_TOKEN_NOT_FOUND.getCode(),
                     MessageCode.AUTH_TOKEN_NOT_FOUND.getDescription(),
                     errorData));
-//        } catch (IOException e) {
-//            Map<String, Object> errorData = new HashMap<>();
-//            errorData.put("token", tokenHeader);
-//            errorData.put("fileId", fileId);
-//            errorData.put("version", version);
-//
-//            return ResponseEntity.ok().body(new BaseResponse(20004,
-//                    MessageCode.FILE_DOWNLOAD_ERROR_NOT_CORRECT_FIELD.getCode(),
-//                    MessageCode.FILE_DOWNLOAD_ERROR_NOT_CORRECT_FIELD.getDescription(),
-//                    errorData));
+        }
+    }
+
+    @GetMapping("/file/link")
+    public ResponseEntity<?> downloadFile(@RequestBody DocViewerRequest viewerRequest) {
+        try {
+            String tokenHeader = viewerRequest.getToken();
+            Long userId = tokenHeader == null ? 0L: authService.useUserAuthToken(tokenHeader);
+
+            FileEntityResponse entityResponse = fileEntityService.getById(viewerRequest.getFileId(), 0);
+            FileStatus status = entityResponse.getStatus();
+
+            if (status.equals(FileStatus.BLOCKED) || status.equals(FileStatus.DELETED) ||
+                    status.equals(FileStatus.REMOVED)) {
+                Map<String, Object> errorData = new HashMap<>();
+                errorData.put("fileId", entityResponse.getId());
+
+                return ResponseEntity.ok().body(new BaseResponse(0,
+                        MessageCode.FILE_IS_BLOCKED.getCode(),
+                        MessageCode.FILE_IS_BLOCKED.getDescription(),
+                        errorData));
+            }
+
+            byte[] bytes = cloudStorageService.readFileFromStorageBytes(entityResponse.getProtectedFilePath());
+
+            dockViewService.createDockView(viewerRequest, userId);
+
+            return ResponseEntity.ok()
+                    .contentType(MediaType.APPLICATION_OCTET_STREAM)
+                    .header(HttpHeaders.CONTENT_DISPOSITION,
+                            "attachment; filename=\"" + entityResponse.getFileName() + "\"")
+                    .body(bytes);
+        } catch (FileEntityNotFoundException e) {
+            Map<String, Object> errorData = new HashMap<>();
+            errorData.put("fileId", viewerRequest.getFileId());
+
+            return ResponseEntity.ok().body(new BaseResponse(0,
+                    MessageCode.FILE_NOT_FOUND.getCode(),
+                    MessageCode.FILE_NOT_FOUND.getDescription(),
+                    errorData));
+        } catch (NotFoundAuthToken e) {
+            Map<String, Object> errorData = new HashMap<>();
+            errorData.put("token", viewerRequest.getFileId());
+
+            return ResponseEntity.ok().body(new BaseResponse(0,
+                    MessageCode.AUTH_TOKEN_NOT_FOUND.getCode(),
+                    MessageCode.AUTH_TOKEN_NOT_FOUND.getDescription(),
+                    errorData));
         }
     }