| | |
| | | import java.io.OutputStream;
|
| | | import java.io.UnsupportedEncodingException;
|
| | | import java.net.URLEncoder;
|
| | | import java.nio.charset.StandardCharsets;
|
| | | import javax.servlet.http.HttpServletRequest;
|
| | | import javax.servlet.http.HttpServletResponse;
|
| | | import org.apache.commons.lang3.ArrayUtils;
|
| | | import com.ruoyi.common.core.utils.StringUtils;
|
| | |
|
| | | /**
|
| | | * 文件处理工具类
|
| | | *
|
| | | * @author ruoyi
|
| | | */
|
| | | public class FileUtils extends org.apache.commons.io.FileUtils
|
| | | public class FileUtils
|
| | | {
|
| | | /** 字符常量:斜杠 {@code '/'} */
|
| | | public static final char SLASH = '/';
|
| | |
|
| | | /** 字符常量:反斜杠 {@code '\\'} */
|
| | | public static final char BACKSLASH = '\\';
|
| | |
|
| | | public static String FILENAME_PATTERN = "[a-zA-Z0-9_\\-\\|\\.\\u4e00-\\u9fa5]+";
|
| | |
|
| | | /**
|
| | |
| | | // 路径为文件且不为空则进行删除
|
| | | if (file.isFile() && file.exists())
|
| | | {
|
| | | file.delete();
|
| | | flag = true;
|
| | | flag = file.delete();
|
| | | }
|
| | | return flag;
|
| | | }
|
| | |
| | | }
|
| | |
|
| | | /**
|
| | | * 校验文件路径合法性(安全性与扩展名)
|
| | | * |
| | | * @param fileUrl 待校验的文件地址
|
| | | * @return true 正常 false 非法
|
| | | */
|
| | | public static boolean validateFilePath(String fileUrl)
|
| | | {
|
| | | // 禁止目录上跳级别
|
| | | if (StringUtils.contains(fileUrl, ".."))
|
| | | {
|
| | | return false;
|
| | | }
|
| | | // 判断是否在允许下载的文件规则内
|
| | | return ArrayUtils.contains(MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION, FileTypeUtils.getFileType(fileUrl));
|
| | | }
|
| | |
|
| | | /**
|
| | | * 下载文件名重新编码
|
| | | *
|
| | | * @param request 请求对象
|
| | | * @param fileName 文件名
|
| | | * @return 编码后的文件名
|
| | | */
|
| | | public static String setFileDownloadHeader(HttpServletRequest request, String fileName)
|
| | | throws UnsupportedEncodingException
|
| | | public static String setFileDownloadHeader(HttpServletRequest request, String fileName) throws UnsupportedEncodingException
|
| | | {
|
| | | final String agent = request.getHeader("USER-AGENT");
|
| | | String filename = fileName;
|
| | |
| | | }
|
| | | return filename;
|
| | | }
|
| | |
|
| | | /**
|
| | | * 返回文件名
|
| | | *
|
| | | * @param filePath 文件
|
| | | * @return 文件名
|
| | | */
|
| | | public static String getName(String filePath)
|
| | | {
|
| | | if (null == filePath)
|
| | | {
|
| | | return null;
|
| | | }
|
| | | int len = filePath.length();
|
| | | if (0 == len)
|
| | | {
|
| | | return filePath;
|
| | | }
|
| | | if (isFileSeparator(filePath.charAt(len - 1)))
|
| | | {
|
| | | // 以分隔符结尾的去掉结尾分隔符
|
| | | len--;
|
| | | }
|
| | |
|
| | | int begin = 0;
|
| | | char c;
|
| | | for (int i = len - 1; i > -1; i--)
|
| | | {
|
| | | c = filePath.charAt(i);
|
| | | if (isFileSeparator(c))
|
| | | {
|
| | | // 查找最后一个路径分隔符(/或者\)
|
| | | begin = i + 1;
|
| | | break;
|
| | | }
|
| | | }
|
| | |
|
| | | return filePath.substring(begin, len);
|
| | | }
|
| | |
|
| | | /**
|
| | | * 是否为Windows或者Linux(Unix)文件分隔符<br>
|
| | | * Windows平台下分隔符为\,Linux(Unix)为/
|
| | | *
|
| | | * @param c 字符
|
| | | * @return 是否为Windows或者Linux(Unix)文件分隔符
|
| | | */
|
| | | public static boolean isFileSeparator(char c)
|
| | | {
|
| | | return SLASH == c || BACKSLASH == c;
|
| | | }
|
| | |
|
| | | /**
|
| | | * 下载文件名重新编码
|
| | | *
|
| | | * @param response 响应对象
|
| | | * @param realFileName 真实文件名
|
| | | * @return
|
| | | */
|
| | | public static void setAttachmentResponseHeader(HttpServletResponse response, String realFileName) throws UnsupportedEncodingException
|
| | | {
|
| | | String percentEncodedFileName = percentEncode(realFileName);
|
| | |
|
| | | StringBuilder contentDispositionValue = new StringBuilder();
|
| | | contentDispositionValue.append("attachment; filename=")
|
| | | .append(percentEncodedFileName)
|
| | | .append(";")
|
| | | .append("filename*=")
|
| | | .append("utf-8''")
|
| | | .append(percentEncodedFileName);
|
| | |
|
| | | response.setHeader("Content-disposition", contentDispositionValue.toString());
|
| | | response.setHeader("download-filename", percentEncodedFileName);
|
| | | }
|
| | |
|
| | | /**
|
| | | * 百分号编码工具方法
|
| | | *
|
| | | * @param s 需要百分号编码的字符串
|
| | | * @return 百分号编码后的字符串
|
| | | */
|
| | | public static String percentEncode(String s) throws UnsupportedEncodingException
|
| | | {
|
| | | String encode = URLEncoder.encode(s, StandardCharsets.UTF_8.toString());
|
| | | return encode.replaceAll("\\+", "%20");
|
| | | }
|
| | | }
|