69 lines
2.8 KiB
Java
69 lines
2.8 KiB
Java
package com.ddf.vodsystem.controllers;
|
|
|
|
import com.ddf.vodsystem.services.DownloadService;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.core.io.Resource;
|
|
import org.springframework.http.HttpHeaders;
|
|
import org.springframework.http.MediaType;
|
|
import org.springframework.http.MediaTypeFactory;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.security.core.annotation.AuthenticationPrincipal;
|
|
import org.springframework.security.oauth2.core.user.OAuth2User;
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
import org.springframework.web.bind.annotation.PathVariable;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
@RestController
|
|
@RequestMapping("api/v1/download")
|
|
public class DownloadController {
|
|
private final DownloadService downloadService;
|
|
|
|
@Autowired
|
|
public DownloadController(DownloadService downloadService) {
|
|
this.downloadService = downloadService;
|
|
}
|
|
|
|
@GetMapping("/output/{uuid}")
|
|
public ResponseEntity<Resource> downloadFile(@PathVariable String uuid) {
|
|
Resource resource = downloadService.downloadOutput(uuid);
|
|
|
|
if (resource == null || !resource.exists()) {
|
|
return ResponseEntity.notFound().build();
|
|
}
|
|
|
|
return ResponseEntity.ok()
|
|
.header(HttpHeaders.CONTENT_DISPOSITION, "inline; filename=\"" + resource.getFilename() + "\"")
|
|
.contentType(MediaTypeFactory.getMediaType(resource).orElse(MediaType.APPLICATION_OCTET_STREAM))
|
|
.body(resource);
|
|
}
|
|
|
|
@GetMapping("/input/{uuid}")
|
|
public ResponseEntity<Resource> downloadInput(@PathVariable String uuid) {
|
|
Resource resource = downloadService.downloadInput(uuid);
|
|
|
|
if (resource == null || !resource.exists()) {
|
|
return ResponseEntity.notFound().build();
|
|
}
|
|
|
|
return ResponseEntity.ok()
|
|
.header(HttpHeaders.CONTENT_DISPOSITION, "inline; filename=\"" + resource.getFilename() + "\"")
|
|
.contentType(MediaTypeFactory.getMediaType(resource).orElse(MediaType.APPLICATION_OCTET_STREAM))
|
|
.body(resource);
|
|
}
|
|
|
|
@GetMapping("/clip/{id}")
|
|
public ResponseEntity<Resource> downloadClip(@AuthenticationPrincipal OAuth2User principal, @PathVariable Long id) {
|
|
Resource resource = downloadService.downloadClip(id);
|
|
|
|
if (resource == null || !resource.exists()) {
|
|
return ResponseEntity.notFound().build();
|
|
}
|
|
|
|
return ResponseEntity.ok()
|
|
.header(HttpHeaders.CONTENT_DISPOSITION, "inline; filename=\"" + resource.getFilename() + "\"")
|
|
.contentType(MediaTypeFactory.getMediaType(resource).orElse(MediaType.APPLICATION_OCTET_STREAM))
|
|
.body(resource);
|
|
}
|
|
}
|