| | |
| | | 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.utils.StringUtils;
|
| | |
|
| | | /**
|
| | | * 文件处理工具类
|
| | | *
|
| | | * @author ruoyi
|
| | | */
|
| | | public class FileUtils
|
| | | public class FileUtils extends org.apache.commons.io.FileUtils
|
| | | {
|
| | | public static String FILENAME_PATTERN = "[a-zA-Z0-9_\\-\\|\\.\\u4e00-\\u9fa5]+";
|
| | |
|
| | |
| | | }
|
| | |
|
| | | /**
|
| | | * 检查文件是否可下载
|
| | | * |
| | | * @param resource 需要下载的文件
|
| | | * @return true 正常 false 非法
|
| | | */
|
| | | public static boolean checkAllowDownload(String resource)
|
| | | {
|
| | | // 禁止目录上跳级别
|
| | | if (StringUtils.contains(resource, ".."))
|
| | | {
|
| | | return false;
|
| | | }
|
| | |
|
| | | // 检查允许下载的文件规则
|
| | | if (ArrayUtils.contains(MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION, FileTypeUtils.getFileType(resource)))
|
| | | {
|
| | | return true;
|
| | | }
|
| | |
|
| | | // 不在允许下载的文件规则
|
| | | return false;
|
| | | }
|
| | |
|
| | | /**
|
| | | * 下载文件名重新编码
|
| | | *
|
| | | * @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 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());
|
| | | }
|
| | |
|
| | | /**
|
| | | * 百分号编码工具方法
|
| | | *
|
| | | * @param s 需要百分号编码的字符串
|
| | | * @return 百分号编码后的字符串
|
| | | */
|
| | | public static String percentEncode(String s) throws UnsupportedEncodingException
|
| | | {
|
| | | String encode = URLEncoder.encode(s, StandardCharsets.UTF_8.toString());
|
| | | return encode.replaceAll("\\+", "%20");
|
| | | }
|
| | | }
|