package com.ruoyi.web.controller.common; import java.io.File; import java.io.UnsupportedEncodingException; import java.net.URLDecoder; import java.net.URLEncoder; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.nio.file.StandardCopyOption; import java.util.*; import java.util.List; import java.util.regex.Matcher; import java.util.regex.Pattern; import javax.imageio.IIOImage; import javax.imageio.ImageWriteParam; import javax.imageio.ImageWriter; import javax.imageio.stream.MemoryCacheImageOutputStream; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import javax.websocket.server.PathParam; import com.ruoyi.common.annotation.Anonymous; import com.ruoyi.common.utils.RenamedMultipartFile; import com.ruoyi.common.utils.uuid.UUID; import com.ruoyi.service.DownLoadFileService; import com.ruoyi.service.impl.VideoProcessService; import lombok.Data; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.http.MediaType; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import com.ruoyi.common.config.RuoYiConfig; import com.ruoyi.common.constant.Constants; import com.ruoyi.common.core.domain.AjaxResult; import com.ruoyi.common.utils.StringUtils; import com.ruoyi.common.utils.file.FileUploadUtils; import com.ruoyi.common.utils.file.FileUtils; import com.ruoyi.framework.config.ServerConfig; import springfox.bean.validators.plugins.schema.NotNullAnnotationPlugin; import javax.imageio.ImageIO; import java.awt.*; import java.awt.image.BufferedImage; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.util.Base64; import java.util.Arrays; import java.util.HashSet; import java.util.Set; import java.util.zip.ZipEntry; import java.util.zip.ZipOutputStream; /** * 通用请求处理 * * @author ruoyi */ @RestController @RequestMapping("/common") public class CommonController { private static final Logger log = LoggerFactory.getLogger(CommonController.class); @Autowired private ServerConfig serverConfig; @Autowired private DownLoadFileService downLoadFileService; @Autowired private VideoProcessService videoProcessService; private static final String FILE_DELIMETER = ","; private static final Pattern CHINESE_PATTERN = Pattern.compile("[\u4e00-\u9fa5]"); // 缩略图配置 @Value("${thumbnail.default-width:300}") private int defaultThumbnailWidth; @Value("${thumbnail.default-height:200}") private int defaultThumbnailHeight; @Value("${thumbnail.quality:0.8}") private double thumbnailQuality; @Value("${thumbnail.cache-dir:./cache/thumbnails}") private String thumbnailCacheDir; @Value("${thumbnail.max-width:1920}") private int maxThumbnailWidth; @Value("${thumbnail.max-height:1080}") private int maxThumbnailHeight; @Value("${thumbnail.format:jpg}") private String thumbnailFormat; @Value("${thumbnail.keep-aspect-ratio:true}") private boolean keepAspectRatio; // 支持的图片格式 private static final String[] IMAGE_FORMATS = {"jpg", "jpeg", "png", "gif", "bmp", "webp"}; // 支持的视频格式 private static final String[] VIDEO_FORMATS = {"mp4", "avi", "mov", "wmv", "flv", "mkv", "webm"}; @Autowired private NotNullAnnotationPlugin notNullPlugin; @Anonymous @GetMapping("/generateThumbnail") public AjaxResult generateThumbnail(@PathParam(value = "url") String url) throws Exception { return AjaxResult.success( ); } // @GetMapping("/downloadFile") // public void fileDownload(@PathParam("path") String path, HttpServletResponse response) // { // path=path.substring(8); // try // { // if (!FileUtils.checkAllowDownload(path)) // { // throw new Exception(StringUtils.format("文件名称({})非法，不允许下载。 ", path)); // } // String realFileName = System.currentTimeMillis() + path.substring(path.indexOf("_") + 1); // String filePath = RuoYiConfig.getProfile() + path; // // response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); // FileUtils.setAttachmentResponseHeader(response, realFileName); // FileUtils.writeBytes(filePath, response.getOutputStream()); // // } // catch (Exception e) // { // log.error("下载文件失败", e); // } // } // /** // * 通用下载请求 // * // * @param fileName 文件名称 // * @param delete 是否删除 // */ // @GetMapping("/download") // public void fileDownload(@PathParam("fileName") String fileName, @PathParam("delete") Boolean delete, HttpServletResponse response) // { // try // { // if (!FileUtils.checkAllowDownload(fileName)) // { // throw new Exception(StringUtils.format("文件名称({})非法，不允许下载。 ", fileName)); // } // String realFileName = System.currentTimeMillis() + fileName.substring(fileName.indexOf("_") + 1); // String filePath = RuoYiConfig.getDownloadPath() + fileName; // // response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); // FileUtils.setAttachmentResponseHeader(response, realFileName); // FileUtils.writeBytes(filePath, response.getOutputStream()); // if (delete) // { // FileUtils.deleteFile(filePath); // } // } // catch (Exception e) // { // log.error("下载文件失败", e); // } // } @Anonymous @GetMapping("/downLoadFile") public void downLoadFile(@PathParam("path") String path, HttpServletResponse response) throws Exception { downLoadFileService.downLoadFile(path, response); } /** * 通用上传请求（单个） */ @PostMapping("/upload") public AjaxResult uploadFile(@RequestParam("uploadFile") MultipartFile file, String fname) throws Exception { try { // 上传文件路径 String filePath = RuoYiConfig.getUploadPath(); // 上传并返回新文件名称 String fileName = FileUploadUtils.upload(filePath, file, fname); String url = serverConfig.getUrl() + fileName; AjaxResult ajax = AjaxResult.success(); HashMap data = new HashMap<>(); data.put("url", url); data.put("fileName", fileName); data.put("newFileName", FileUtils.getName(fileName)); data.put("originalFilename", file.getOriginalFilename()); ajax.put("msg", "操作成功"); ajax.put("data", data); return ajax; } catch (Exception e) { return AjaxResult.error(e.getMessage()); } } /** * 通用上传请求（多个） */ @PostMapping("/uploads") public AjaxResult uploadFiles(@RequestParam("files") List files) throws Exception { //System.out.println("99999999999999999999999990000000000000000"); try { // 上传文件路径 String filePath = RuoYiConfig.getUploadPath(); List urls = new ArrayList(); List fileNames = new ArrayList(); List newFileNames = new ArrayList(); List originalFilenames = new ArrayList(); // System.out.println("99999999999999999999999990000000000000000"); // System.out.println(files); // System.out.println(files.size()); for (MultipartFile file : files) { // 上传并返回新文件名称 String filename = ""; // System.out.println("1122123330+++++++++++++++++++++++++++++"); String fileName = FileUploadUtils.upload(filePath, file, filename); String url = serverConfig.getUrl() + fileName; urls.add(url); fileNames.add(fileName); newFileNames.add(FileUtils.getName(fileName)); originalFilenames.add(file.getOriginalFilename()); } AjaxResult ajax = AjaxResult.success(); // System.out.println("99999999999999999999999990000000000000000"); 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)); // System.out.println("1122123330+++++++++++++++++++++++++++++"); return ajax; } catch (Exception e) { return AjaxResult.error(e.getMessage()); } } /** * 通用上传请求（多个） 将中文修改为其他 */ @PostMapping("/noChinese/uploads") public AjaxResult noChineseUploadFiles(@RequestParam("files") List files) throws Exception { //System.out.println("99999999999999999999999990000000000000000"); 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) { originalFilenames.add(file.getOriginalFilename()); String safeFilename = generateSafeFilename(file.getOriginalFilename()); MultipartFile renamedFile = new RenamedMultipartFile(file, safeFilename); String lastName=""; String fileName = FileUploadUtils.upload(filePath, renamedFile, lastName); String url = serverConfig.getUrl() + fileName; urls.add(url); fileNames.add(fileName); newFileNames.add(FileUtils.getName(fileName)); } 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()); } } /** * 生成安全文件名（只替换中文部分） */ private String generateSafeFilename(String originalName) { if (originalName == null) { return ""; } // 1. 获取文件扩展名 String extension = ""; int dotIndex = originalName.lastIndexOf('.'); if (dotIndex > 0) { extension = originalName.substring(dotIndex); originalName = originalName.substring(0, dotIndex); } // 2. 只替换中文部分 StringBuilder safeName = new StringBuilder(); Matcher matcher = CHINESE_PATTERN.matcher(originalName); int lastEnd = 0; while (matcher.find()) { // 添加非中文部分 safeName.append(originalName, lastEnd, matcher.start()); // 添加随机字符串替换中文 safeName.append(generateRandomString(4)); lastEnd = matcher.end(); } // 添加剩余部分 safeName.append(originalName.substring(lastEnd)); String noSpaceName = safeName.toString().replaceAll("\\s", ""); // 3. 添加扩展名 return noSpaceName + extension; } /** * 生成随机字符串（字母+数字） */ private String generateRandomString(int length) { String uuid = UUID.randomUUID().toString().replace("-", ""); return uuid.substring(0, Math.min(length, uuid.length())); } @PostMapping("/uploads1") public AjaxResult uploadFiles1(@RequestParam("files") List files) { try { String filePath = RuoYiConfig.getUploadPath(); List urls = new ArrayList<>(); List fileNames = new ArrayList<>(); List newFileNames = new ArrayList<>(); List originalFilenames = new ArrayList<>(); List httpSafePaths = new ArrayList<>(); for (MultipartFile file : files) { // 1. 上传文件 String fileName = FileUploadUtils.upload(filePath, file, ""); String originalFilename = file.getOriginalFilename(); // 2. 获取HTTP安全路径 String httpSafePath = toHttpPath(fileName); // 3. 构建完整URL（确保有斜杠分隔） String baseUrl = serverConfig.getUrl(); if (!baseUrl.endsWith("/") && !httpSafePath.startsWith("/")) { baseUrl += "/"; } String url = baseUrl + httpSafePath; urls.add(url); fileNames.add(fileName); newFileNames.add(FileUtils.getName(fileName)); originalFilenames.add(originalFilename); httpSafePaths.add(httpSafePath); } 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)); ajax.put("httpSafePaths", StringUtils.join(httpSafePaths, FILE_DELIMETER)); return ajax; } catch (Exception e) { return AjaxResult.error(e.getMessage()); } } /** * 将包含中文的文件路径转换为 HTTP 安全的 URL 路径 */ public String toHttpPath(String filePath) { try { // 1. 标准化路径 Path normalizedPath = Paths.get(filePath).normalize(); // 2. 统一使用正斜杠 String pathStr = normalizedPath.toString().replace("\\", "/"); // 3. 分割路径组件 String[] parts = pathStr.split("/"); StringBuilder encodedPath = new StringBuilder(); // 4. 对每个组件单独编码并处理空格 for (String part : parts) { if (!part.isEmpty()) { // 编码并替换空格为 %20 String encodedPart = URLEncoder.encode(part, StandardCharsets.UTF_8.name()) .replace("+", "%20"); encodedPath.append("/").append(encodedPart); } } // 5. 处理绝对路径和相对路径 return filePath.startsWith("/") || filePath.startsWith("\\") ? encodedPath.toString() : encodedPath.substring(1); } catch (UnsupportedEncodingException e) { throw new RuntimeException("UTF-8 encoding not supported", e); } } /** * 从 HTTP URL 路径还原原始中文路径 */ @PostMapping("/getFileName") public String extractFileName(@RequestBody String httpPath) { try { // 1. 处理空值 if (httpPath == null || httpPath.trim().isEmpty()) { return ""; } // 2. 移除URL协议、域名和查询参数 String pathOnly = httpPath; // 移除协议和域名 if (pathOnly.contains("://")) { pathOnly = pathOnly.substring(pathOnly.indexOf("://") + 3); pathOnly = pathOnly.substring(pathOnly.indexOf('/')); } // 移除查询参数（如 ?token=123） int queryStart = pathOnly.indexOf('?'); if (queryStart > 0) { pathOnly = pathOnly.substring(0, queryStart); } // 3. URL解码 String decodedPath = URLDecoder.decode(pathOnly, StandardCharsets.UTF_8.name()); // 4. 提取文件名（处理Windows路径） decodedPath = decodedPath.replace("\\", "/"); // 获取最后一个非空路径组件 int lastSlash = decodedPath.lastIndexOf('/'); String fileName = (lastSlash >= 0 && lastSlash < decodedPath.length() - 1) ? decodedPath.substring(lastSlash + 1) : decodedPath; // 5. 处理特殊情况（如结尾斜杠） if (fileName.isEmpty()) { // 尝试获取倒数第二个组件 int prevSlash = decodedPath.lastIndexOf('/', lastSlash - 1); if (prevSlash >= 0) { fileName = decodedPath.substring(prevSlash + 1, lastSlash); } } return fileName; } catch (UnsupportedEncodingException e) { throw new RuntimeException("UTF-8 encoding not supported", e); } } /** * 本地资源通用下载 */ @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); } } /** * 压缩图片或提取视频封面并返回Base64 * * @param file 图片或视频文件 * @param width 目标宽度，默认100 * @param height 目标高度，默认100 * @param quality 图片质量 0.1-1.0，默认0.8 * @return Map包含压缩结果和Base64 */ @Anonymous @PostMapping(value = "/zip", consumes = MediaType.MULTIPART_FORM_DATA_VALUE) public Map compressMediaToBase64( @RequestParam(value = "file", required = false) MultipartFile file, @RequestParam(value = "filePath", required = false) String filePath, @RequestParam(value = "width", defaultValue = "0") int width, @RequestParam(value = "height", defaultValue = "0") int height, @RequestParam(value = "quality", defaultValue = "0.8") float quality) { Map result = new HashMap<>(); File trueFile; boolean deleteY = true; try { // 验证文件 if ((file == null || file.isEmpty()) && filePath == null ) { result.put("success", false); result.put("message", "文件不能为空"); return result; } if ((file == null || file.isEmpty())){ deleteY = false; String fileUel = RuoYiConfig.getProfile() + filePath.replace("/profile",""); trueFile = new File(fileUel); }else { trueFile = videoProcessService.convertToFile(file); } String fileName = trueFile.getName(); if (fileName.isEmpty() && filePath == null) { result.put("success", false); result.put("message", "文件名不能为空"); return result; } System.out.println("-----------"+fileName); // 获取文件扩展名 String extension = getFileExtension(fileName).toLowerCase(); System.out.println("-----------"+extension); // 判断文件类型 if (isImageFile(extension)) { // 处理图片文件 return videoProcessService.processImage(trueFile, width, height, quality, extension,deleteY); } else if (isVideoFile(extension)) { // 处理视频文件 return videoProcessService.processVideo(trueFile,width,height,quality,extension,deleteY); } else { result.put("success", false); result.put("message", "不支持的文件格式：" + extension); return result; } } catch (Exception e) { log.error("文件处理失败", e); result.put("success", false); result.put("message", "处理失败: " + e.getMessage()); return result; } } /** * 判断是否是图片文件 */ private boolean isImageFile(String extension) { for (String format : IMAGE_FORMATS) { if (format.equalsIgnoreCase(extension)) { return true; } } return false; } /** * 判断是否是视频文件 */ private boolean isVideoFile(String extension) { for (String format : VIDEO_FORMATS) { if (format.equalsIgnoreCase(extension)) { return true; } } return false; } /** * 获取文件扩展名 */ private String getFileExtension(String fileName) { if (StringUtils.isEmpty(fileName)) { return ""; } int lastDot = fileName.lastIndexOf('.'); if (lastDot > 0 && lastDot < fileName.length() - 1) { return fileName.substring(lastDot + 1).toLowerCase(); } return ""; } }