package com.ruoyi.project.common; import com.ruoyi.basic.service.StorageBlobService; import com.ruoyi.framework.aspectj.lang.annotation.Anonymous; import com.ruoyi.framework.config.ServerConfig; import com.ruoyi.framework.web.domain.R; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import lombok.AllArgsConstructor; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import jakarta.servlet.http.HttpServletResponse; import java.io.File; import java.net.URLEncoder; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.util.List; /** * 通用请求处理 * * @author ruoyi */ @Api(tags = "通用接口") @AllArgsConstructor @RestController @RequestMapping("/common") public class CommonController { private static final Logger log = LoggerFactory.getLogger(CommonController.class); private ServerConfig serverConfig; private static final String FILE_DELIMETER = ","; private final StorageBlobService storageBlobService; // /** // * 通用下载请求 // * // * @param fileName 文件名称 // * @param delete 是否删除 // */ // @GetMapping("/download") // public void fileDownload(String fileName, Boolean delete, HttpServletResponse response, HttpServletRequest request) // { // try // { // if (!FileUtils.checkAllowDownload(fileName)) // { // throw new Exception(StringUtils.format("文件名称({})非法,不允许下载。 ", fileName)); // } // String realFileName = fileName.substring(fileName.indexOf("_") + 1); // // response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); // FileUtils.setAttachmentResponseHeader(response, realFileName); // FileUtils.writeBytes(fileName, response.getOutputStream()); /// / if (delete) /// / { /// / FileUtils.deleteFile(fileName); /// / } // } // catch (Exception e) // { // log.error("下载文件失败", e); // } // } // // /** // * minio通用上传请求(多个) // */ // @PostMapping("/minioUploads") // @ApiOperation(value = "minio通用上传请求") // public AjaxResult minioUploadFiles(List files, String bucketName, Long type) throws Exception // { // return AjaxResult.success(); // } // // /** // * 通用上传请求(单个) // */ // @PostMapping("/upload") // public AjaxResult uploadFile(MultipartFile file) throws Exception // { // try // { // // 上传文件路径 // String filePath = RuoYiConfig.getUploadPath(); // // 上传并返回新文件名称 // String fileName = FileUploadUtils.upload(filePath, file); // String url = serverConfig.getUrl() + fileName; // AjaxResult ajax = AjaxResult.success(); // ajax.put("url", url); // ajax.put("fileName", fileName); // ajax.put("newFileName", FileUtils.getName(fileName)); // ajax.put("originalFilename", file.getOriginalFilename()); // return ajax; // } // catch (Exception e) // { // return AjaxResult.error(e.getMessage()); // } // } // // /** // * 通用上传请求(多个) // */ // @PostMapping("/uploads") // public AjaxResult uploadFiles(List files) throws Exception // { // try // { // // 上传文件路径 // String filePath = RuoYiConfig.getUploadPath(); // List urls = new ArrayList(); // List fileNames = new ArrayList(); // List newFileNames = new ArrayList(); // List originalFilenames = new ArrayList(); // for (MultipartFile file : files) // { // // 上传并返回新文件名称 // String fileName = FileUploadUtils.upload(filePath, file); // String url = serverConfig.getUrl() + fileName; // urls.add(url); // fileNames.add(fileName); // newFileNames.add(FileUtils.getName(fileName)); // originalFilenames.add(file.getOriginalFilename()); // } // AjaxResult ajax = AjaxResult.success(); // ajax.put("urls", StringUtils.join(urls, FILE_DELIMETER)); // ajax.put("fileNames", StringUtils.join(fileNames, FILE_DELIMETER)); // ajax.put("newFileNames", StringUtils.join(newFileNames, FILE_DELIMETER)); // ajax.put("originalFilenames", StringUtils.join(originalFilenames, FILE_DELIMETER)); // return ajax; // } // catch (Exception e) // { // return AjaxResult.error(e.getMessage()); // } // } // // /** // * 本地资源通用下载 // */ // @GetMapping("/download/resource") // public void resourceDownload(String resource, HttpServletRequest request, HttpServletResponse response) // throws Exception // { // try // { // if (!FileUtils.checkAllowDownload(resource)) // { // throw new Exception(StringUtils.format("资源文件({})非法,不允许下载。 ", resource)); // } // // 本地资源路径 // String localPath = RuoYiConfig.getProfile(); // // 数据库资源地址 // String downloadPath = localPath + StringUtils.substringAfter(resource, Constants.RESOURCE_PREFIX); // // 下载名称 // String downloadName = StringUtils.substringAfterLast(downloadPath, "/"); // response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); // FileUtils.setAttachmentResponseHeader(response, downloadName); // FileUtils.writeBytes(downloadPath, response.getOutputStream()); // } // catch (Exception e) // { // log.error("下载文件失败", e); // } // } @PostMapping({"/upload"}) @ApiOperation(value = "文件上传") public R upload(@RequestParam("files") List files) throws Exception { return R.ok(storageBlobService.upload(files)); } @GetMapping("/download/{fileName}") @Anonymous public void download(@PathVariable String fileName, @RequestParam("token") String token, HttpServletResponse response) throws Exception { File file = storageBlobService.getFileByToken(fileName, token); String originalFileName = storageBlobService.getDownloadFileName(fileName); String encodedFileName = URLEncoder.encode(originalFileName, StandardCharsets.UTF_8.name()).replace("+", "%20"); response.setHeader("Content-Disposition", "attachment;filename*=UTF-8''" + encodedFileName); Files.copy(file.toPath(), response.getOutputStream()); } @GetMapping("/preview/{fileName}") @Anonymous public void preview(@PathVariable String fileName, @RequestParam("token") String token, HttpServletResponse response) throws Exception { File file = storageBlobService.getFileByToken(fileName, token); String contentType = Files.probeContentType(file.toPath()); response.setContentType(contentType == null ? "application/octet-stream" : contentType); String encodedFileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.name()).replace("+", "%20"); response.setHeader("Content-Disposition", "inline;filename*=UTF-8''" + encodedFileName); Files.copy(file.toPath(), response.getOutputStream()); } }