product-inventory-management-after.git

			@@ -3,11 +3,21 @@
			import java.io.File;
			import java.io.FileInputStream;
			import java.io.FileNotFoundException;
			import java.io.FileOutputStream;
			import java.io.IOException;
			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.io.FilenameUtils;
			import org.apache.commons.io.IOUtils;
			import org.apache.commons.lang3.ArrayUtils;
			import com.ruoyi.common.utils.DateUtils;
			import com.ruoyi.common.utils.StringUtils;
			import com.ruoyi.common.utils.uuid.IdUtils;
			import com.ruoyi.framework.config.RuoYiConfig;

			/**
			* 文件处理工具类
			@@ -49,29 +59,48 @@
			}
			finally
			{
			if (os != null)
			{
			try
			{
			os.close();
			}
			catch (IOException e1)
			{
			e1.printStackTrace();
			}
			}
			if (fis != null)
			{
			try
			{
			fis.close();
			}
			catch (IOException e1)
			{
			e1.printStackTrace();
			}
			}
			IOUtils.close(os);
			IOUtils.close(fis);
			}
			}

			/**
			* 写数据到文件中
			*
			* @param data 数据
			* @return 目标文件
			* @throws IOException IO异常
			*/
			public static String writeImportBytes(byte[] data) throws IOException
			{
			return writeBytes(data, RuoYiConfig.getImportPath());
			}

			/**
			* 写数据到文件中
			*
			* @param data 数据
			* @param uploadDir 目标文件
			* @return 目标文件
			* @throws IOException IO异常
			*/
			public static String writeBytes(byte[] data, String uploadDir) throws IOException
			{
			FileOutputStream fos = null;
			String pathName = "";
			try
			{
			String extension = getFileExtendName(data);
			pathName = DateUtils.datePath() + "/" + IdUtils.fastUUID() + "." + extension;
			File file = FileUploadUtils.getAbsoluteFile(uploadDir, pathName);
			fos = new FileOutputStream(file);
			fos.write(data);
			}
			finally
			{
			IOUtils.close(fos);
			}
			return FileUploadUtils.getPathFileName(uploadDir, pathName);
			}

			/**
			@@ -87,8 +116,7 @@
			// 路径为文件且不为空则进行删除
			if (file.isFile() && file.exists())
			{
			file.delete();
			flag = true;
			flag = file.delete();
			}
			return flag;
			}
			@@ -105,14 +133,37 @@
			}

			/**
			* 检查文件是否可下载
			*
			* @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;
			@@ -139,4 +190,102 @@
			}
			return filename;
			}

			/**
			* 下载文件名重新编码
			*
			* @param response 响应对象
			* @param realFileName 真实文件名
			*/
			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.addHeader("Access-Control-Expose-Headers", "Content-Disposition,download-filename");
			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");
			}

			/**
			* 获取图像后缀
			*
			* @param photoByte 图像数据
			* @return 后缀名
			*/
			public static String getFileExtendName(byte[] photoByte)
			{
			String strFileExtendName = "jpg";
			if ((photoByte[0] == 71) && (photoByte[1] == 73) && (photoByte[2] == 70) && (photoByte[3] == 56)
			&& ((photoByte[4] == 55) \|\| (photoByte[4] == 57)) && (photoByte[5] == 97))
			{
			strFileExtendName = "gif";
			}
			else if ((photoByte[6] == 74) && (photoByte[7] == 70) && (photoByte[8] == 73) && (photoByte[9] == 70))
			{
			strFileExtendName = "jpg";
			}
			else if ((photoByte[0] == 66) && (photoByte[1] == 77))
			{
			strFileExtendName = "bmp";
			}
			else if ((photoByte[1] == 80) && (photoByte[2] == 78) && (photoByte[3] == 71))
			{
			strFileExtendName = "png";
			}
			return strFileExtendName;
			}

			/**
			* 获取文件名称 /profile/upload/2022/04/16/ruoyi.png -- ruoyi.png
			*
			* @param fileName 路径名称
			* @return 没有文件路径的名称
			*/
			public static String getName(String fileName)
			{
			if (fileName == null)
			{
			return null;
			}
			int lastUnixPos = fileName.lastIndexOf('/');
			int lastWindowsPos = fileName.lastIndexOf('\\');
			int index = Math.max(lastUnixPos, lastWindowsPos);
			return fileName.substring(index + 1);
			}

			/**
			* 获取不带后缀文件名称 /profile/upload/2022/04/16/ruoyi.png -- ruoyi
			*
			* @param fileName 路径名称
			* @return 没有文件路径和后缀的名称
			*/
			public static String getNameNotSuffix(String fileName)
			{
			if (fileName == null)
			{
			return null;
			}
			String baseName = FilenameUtils.getBaseName(fileName);
			return baseName;
			}
			}