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.Path;
|
import java.nio.file.Paths;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
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 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;
|
|
/**
|
* 通用请求处理
|
*
|
* @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;
|
|
private static final String FILE_DELIMETER = ",";
|
|
private static final Pattern CHINESE_PATTERN = Pattern.compile("[\u4e00-\u9fa5]");
|
|
|
// @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<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) {
|
return AjaxResult.error(e.getMessage());
|
}
|
}
|
|
|
/**
|
* 通用上传请求（多个）
|
*/
|
@PostMapping("/uploads")
|
public AjaxResult uploadFiles(@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>();
|
// 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<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)) {
|
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);
|
}
|
|
}
|
}
|
