新增压缩图 和 视频封面 都返回base64

zqy

14 小时以前 7c0603315e57e3765270a8ac6b310b5a32af5a40

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java

@@ -1,15 +1,36 @@
package com.ruoyi.web.controller.common;

import java.util.ArrayList;
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;
@@ -20,16 +41,32 @@
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
{
public class CommonController {
    private static final Logger log = LoggerFactory.getLogger(CommonController.class);

    @Autowired
@@ -38,43 +75,115 @@
    @Autowired
    private DownLoadFileService downLoadFileService;


    @Autowired
    private VideoProcessService videoProcessService;

    private static final String FILE_DELIMETER = ",";

    /**
     * 通用下载请求
     * 
     * @param fileName 文件名称
     * @param delete 是否删除
     */
    @GetMapping("/download")
    public void fileDownload(@PathParam("fileName") String fileName, @PathParam("delete") Boolean delete, HttpServletResponse response, HttpServletRequest request)
    {
        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;
    private static final Pattern CHINESE_PATTERN = Pattern.compile("[\u4e00-\u9fa5]");

            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);
        }
    // 缩略图配置
    @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 AjaxResult downLoadFile(@PathParam("path") String path, HttpServletResponse response) throws Exception {
        return downLoadFileService.downLoadFile(path,response);
    public void downLoadFile(@PathParam("path") String path, HttpServletResponse response) throws Exception {
        downLoadFileService.downLoadFile(path, response);
    }


@@ -82,35 +191,38 @@
     * 通用上传请求（单个）
     */
    @PostMapping("/upload")
    public AjaxResult uploadFile(@RequestParam("uploadFile") MultipartFile file) throws Exception
    {
        try
        {
    public AjaxResult uploadFile(@RequestParam("uploadFile") MultipartFile file, String fname) throws Exception {
        try {
            // 上传文件路径
            String filePath = RuoYiConfig.getUploadPath();
            // 上传并返回新文件名称
            String fileName = FileUploadUtils.upload(filePath, file);
            String fileName = FileUploadUtils.upload(filePath, file, fname);
            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());

            HashMap<String, Object> 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)
        {
        } catch (Exception e) {
            return AjaxResult.error(e.getMessage());
        }
    }


    /**
     * 通用上传请求（多个）
     */
    @PostMapping("/uploads")
    public AjaxResult uploadFiles(List<MultipartFile> files) throws Exception
    public AjaxResult uploadFiles(@RequestParam("files")  List<MultipartFile> files) throws Exception
    {
        //System.out.println("99999999999999999999999990000000000000000");
        try
        {
            // 上传文件路径
@@ -119,10 +231,17 @@
            List<String> fileNames = new ArrayList<String>();
            List<String> newFileNames = new ArrayList<String>();
            List<String> originalFilenames = new ArrayList<String>();
            // System.out.println("99999999999999999999999990000000000000000");
            //  System.out.println(files);
            //  System.out.println(files.size());

            for (MultipartFile file : files)
            {
                // 上传并返回新文件名称
                String fileName = FileUploadUtils.upload(filePath, file);
                String filename = "";
                //   System.out.println("1122123330+++++++++++++++++++++++++++++");

                String fileName = FileUploadUtils.upload(filePath, file, filename);
                String url = serverConfig.getUrl() + fileName;
                urls.add(url);
                fileNames.add(fileName);
@@ -130,10 +249,13 @@
                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)
@@ -143,16 +265,235 @@
    }

    /**
     * 通用上传请求（多个） 将中文修改为其他
     */
    @PostMapping("/noChinese/uploads")
    public AjaxResult noChineseUploadFiles(@RequestParam("files") List<MultipartFile> files) throws Exception {
        //System.out.println("99999999999999999999999990000000000000000");
        try {
            // 上传文件路径
            String filePath = RuoYiConfig.getUploadPath();
            List<String> urls = new ArrayList<String>();
            List<String> fileNames = new ArrayList<String>();
            List<String> newFileNames = new ArrayList<String>();
            List<String> originalFilenames = new ArrayList<String>();

            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<MultipartFile> files) {
        try {
            String filePath = RuoYiConfig.getUploadPath();
            List<String> urls = new ArrayList<>();
            List<String> fileNames = new ArrayList<>();
            List<String> newFileNames = new ArrayList<>();
            List<String> originalFilenames = new ArrayList<>();
            List<String> 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))
            {
            throws Exception {
        try {
            if (!FileUtils.checkAllowDownload(resource)) {
                throw new Exception(StringUtils.format("资源文件({})非法，不允许下载。 ", resource));
            }
            // 本地资源路径
@@ -164,10 +505,124 @@
            response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
            FileUtils.setAttachmentResponseHeader(response, downloadName);
            FileUtils.writeBytes(downloadPath, response.getOutputStream());
        }
        catch (Exception e)
        {
        } 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<String, Object> 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<String, Object> 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 "";
    }



}