Add attachment download endpoint
This commit is contained in:
parent
29b9773543
commit
ded3f2f537
|
|
@ -19,6 +19,8 @@ import org.springframework.web.bind.annotation.*;
|
|||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.io.File;
|
||||
import java.net.URLEncoder;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.time.LocalDate;
|
||||
|
|
@ -313,6 +315,29 @@ public class TaskController {
|
|||
.body(resource);
|
||||
}
|
||||
|
||||
@GetMapping("/reports/{id}/attachments/{attachmentId}")
|
||||
@SaCheckRole(value = { "ADMIN", "AUDITOR", "USER" }, mode = SaMode.OR)
|
||||
public ResponseEntity<Resource> downloadAttachment(
|
||||
@PathVariable("id") String id,
|
||||
@PathVariable("attachmentId") Long attachmentId) {
|
||||
com.chinaweal.youfool.reportdetect.modules.task.entity.AuditAttachment attachment =
|
||||
taskService.getAttachment(id, attachmentId);
|
||||
if (attachment == null || attachment.getFilepath() == null) {
|
||||
return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
|
||||
}
|
||||
File file = new File(attachment.getFilepath());
|
||||
if (!file.exists()) {
|
||||
return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
|
||||
}
|
||||
String filename = attachment.getFilename() != null ? attachment.getFilename() : file.getName();
|
||||
String encoded = URLEncoder.encode(filename, StandardCharsets.UTF_8);
|
||||
Resource resource = new FileSystemResource(file);
|
||||
return ResponseEntity.ok()
|
||||
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename*=UTF-8''" + encoded)
|
||||
.contentType(MediaType.APPLICATION_OCTET_STREAM)
|
||||
.body(resource);
|
||||
}
|
||||
|
||||
@GetMapping("/statistics")
|
||||
@SaCheckRole(value = { "ADMIN", "AUDITOR", "USER" }, mode = SaMode.OR)
|
||||
public ResponseEntity<?> getStatistics() {
|
||||
|
|
|
|||
|
|
@ -671,6 +671,20 @@ public class TaskService {
|
|||
return task != null ? task.getPdfPath() : null;
|
||||
}
|
||||
|
||||
public AuditAttachment getAttachment(String approvalId, Long attachmentId) {
|
||||
if (approvalId == null || approvalId.isBlank() || attachmentId == null) {
|
||||
return null;
|
||||
}
|
||||
Task task = taskRepository.findByApprovalId(approvalId);
|
||||
if (task == null || task.getAttachments() == null) {
|
||||
return null;
|
||||
}
|
||||
return task.getAttachments().stream()
|
||||
.filter(a -> attachmentId.equals(a.getId()))
|
||||
.findFirst()
|
||||
.orElse(null);
|
||||
}
|
||||
|
||||
public Map<String, Object> getStatistics() {
|
||||
List<String> pendingStatuses = List.of("pending", "ocr_completed");
|
||||
List<String> auditedStatuses = List.of("compliant", "non-compliant");
|
||||
|
|
|
|||
Loading…
Reference in New Issue