This commit is contained in:
@@ -1,5 +1,7 @@
|
|||||||
package ru.soune.nocopy.controller;
|
package ru.soune.nocopy.controller;
|
||||||
|
|
||||||
|
import com.vrt.NoCopyFileService;
|
||||||
|
import com.vrt.fileprotection.FileProtector;
|
||||||
import lombok.extern.slf4j.Slf4j;
|
import lombok.extern.slf4j.Slf4j;
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
import org.springframework.core.io.Resource;
|
import org.springframework.core.io.Resource;
|
||||||
@@ -15,6 +17,7 @@ import ru.soune.nocopy.repository.UserRepository;
|
|||||||
import ru.soune.nocopy.service.file.CheckCounterService;
|
import ru.soune.nocopy.service.file.CheckCounterService;
|
||||||
import ru.soune.nocopy.service.file.FileStorageService;
|
import ru.soune.nocopy.service.file.FileStorageService;
|
||||||
import ru.soune.nocopy.service.file.ImageResizeService;
|
import ru.soune.nocopy.service.file.ImageResizeService;
|
||||||
|
import ru.soune.nocopy.util.FileUtil;
|
||||||
|
|
||||||
import java.io.FileNotFoundException;
|
import java.io.FileNotFoundException;
|
||||||
import java.net.URLEncoder;
|
import java.net.URLEncoder;
|
||||||
@@ -45,6 +48,12 @@ public class FileController {
|
|||||||
@Autowired
|
@Autowired
|
||||||
private ImageResizeService imageResizeService;
|
private ImageResizeService imageResizeService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private NoCopyFileService noCopyFileService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private FileUtil fileUtil;
|
||||||
|
|
||||||
@GetMapping("/public/{fileId}")
|
@GetMapping("/public/{fileId}")
|
||||||
public ResponseEntity<Resource> getPublicFile(@PathVariable String fileId) {
|
public ResponseEntity<Resource> getPublicFile(@PathVariable String fileId) {
|
||||||
try {
|
try {
|
||||||
@@ -145,4 +154,16 @@ public class FileController {
|
|||||||
.ok()
|
.ok()
|
||||||
.body(protectedFileCheck);
|
.body(protectedFileCheck);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@PostMapping("protect/{fileId}/{convertTo}")
|
||||||
|
public ResponseEntity<FileProtector.FileInfo> convert(@PathVariable String fileId, @PathVariable String convertTo) {
|
||||||
|
FileEntity file = fileRepository.findByFileId(fileId);
|
||||||
|
FileProtector.FileInfo fileInfo = fileUtil.createFileInfo(file, convertTo);
|
||||||
|
|
||||||
|
noCopyFileService.addFile(fileInfo);
|
||||||
|
|
||||||
|
return ResponseEntity
|
||||||
|
.ok()
|
||||||
|
.body(fileInfo);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user