package com.product.file.util;
|
|
import cn.hutool.core.lang.UUID;
|
import com.aspose.words.Document;
|
import com.aspose.words.SaveFormat;
|
import com.deepoove.poi.XWPFTemplate;
|
import com.deepoove.poi.xwpf.NiceXWPFDocument;
|
import com.google.common.collect.Maps;
|
import com.product.common.lang.StringUtils;
|
import com.product.core.config.Global;
|
import com.product.core.exception.BaseException;
|
import com.product.file.config.CmnConst;
|
import com.product.file.config.FileCode;
|
import lombok.extern.slf4j.Slf4j;
|
|
import java.io.*;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Properties;
|
import java.util.zip.ZipEntry;
|
import java.util.zip.ZipOutputStream;
|
|
/**
|
* Copyright © 6c
|
*
|
* @Date: 2021-03-05 10:59
|
* @Author: 6c
|
* @Description:
|
*/
|
@Slf4j
|
public class FileUtil {
|
|
private FileUtil() {
|
}
|
|
public static String checkDocType(File file) {
|
try (FileInputStream fis = new FileInputStream(file)) {
|
byte[] bytes = new byte[8];
|
fis.read(bytes, 0, 8);
|
|
String hex = bytesToHex(bytes);
|
|
if (hex.contains("504B0304") && (file.getName().endsWith(".doc") || file.getName().endsWith(".docx"))) {
|
return "docx";
|
} else if (hex.contains("D0CF11E0") && (file.getName().endsWith(".doc") || file.getName().endsWith(".docx"))) {
|
//因为doc文件的头部也是D0CF11E0，所以需要判断文件后缀
|
return "doc";
|
}
|
//增加xls 和 xlsx的判断
|
else if (hex.contains("504B0304") && (file.getName().endsWith(".xls") || file.getName().endsWith(".xlsx"))) {
|
return "xlsx";
|
} else if (hex.contains("D0CF11E0") && (file.getName().endsWith(".xls") || file.getName().endsWith(".xlsx"))) {
|
//因为xls文件的头部也是D0CF11E0，所以需要判断文件后缀
|
return "xls";
|
} else {
|
return "unknown";
|
}
|
} catch (
|
IOException e) {
|
e.printStackTrace();
|
return "unknown";
|
}
|
}
|
|
private static String bytesToHex(byte[] bytes) {
|
StringBuilder hex = new StringBuilder();
|
for (byte b : bytes) {
|
hex.append(String.format("%02X", b));
|
}
|
return hex.toString();
|
}
|
|
/**
|
* 合并word文档
|
*
|
* @param source 源文件
|
* @param target 目标文件
|
* @return
|
*/
|
public static File mergeFile(File source, File target) throws Exception {
|
if ("doc".equals(FileUtil.checkDocType(source))) {
|
source = toDocx(source);
|
}
|
boolean delTarget = false;
|
if ("doc".equals(FileUtil.checkDocType(target))) {
|
target = toDocx(target);
|
delTarget = true;
|
}
|
XWPFTemplate template = XWPFTemplate.compile(source);
|
NiceXWPFDocument xwpfDocument = template.getXWPFDocument();
|
try {
|
NiceXWPFDocument sub = new NiceXWPFDocument(new FileInputStream((target)));
|
NiceXWPFDocument newDoc = xwpfDocument.merge(sub);
|
String tempFile = Global.getSystemConfig("temp.dir", "./attachment/temp") + File.separator + UUID.randomUUID() + ".docx";
|
try (FileOutputStream out = new FileOutputStream(tempFile)) {
|
newDoc.write(out);
|
newDoc.close();
|
return new File(tempFile);
|
} catch (IOException e) {
|
throw e;
|
} finally {
|
if (delTarget) {
|
target.delete();
|
}
|
newDoc.close();
|
sub.close();
|
}
|
} catch (IOException e) {
|
throw e;
|
} catch (Exception e) {
|
throw e;
|
}
|
}
|
|
public static File toDocx(File file) {
|
if (!AsposeUtil.getLicense(2)) {
|
return file;
|
}
|
String docxName = UUID.randomUUID().toString() + ".docx";
|
File docxFile = new File(file.getParent() + File.separator + docxName);
|
String tempDir = Global.getSystemConfig("temp.dir", "./attachment/temp");
|
File tempFile = new File(tempDir + File.separator + docxName);
|
|
try (FileInputStream is = new FileInputStream(file);) {
|
Document document = new Document(is);
|
document.save(tempFile.getAbsolutePath(), com.aspose.words.SaveFormat.DOCX);
|
return tempFile;
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return docxFile;
|
}
|
|
public static File toDoc(File file) {
|
if (!AsposeUtil.getLicense(2)) {
|
return file;
|
}
|
String docxName = UUID.randomUUID().toString() + ".doc";
|
File docxFile = new File(file.getParent() + File.separator + docxName);
|
String tempDir = Global.getSystemConfig("temp.dir", "./attachment/temp");
|
File tempFile = new File(tempDir + File.separator + docxName);
|
try {
|
Document document = new Document(new FileInputStream(file));
|
document.save(tempFile.getAbsolutePath(), SaveFormat.DOC);
|
return tempFile;
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return docxFile;
|
}
|
|
/**
|
* 批量打包
|
*
|
* @param pathList 文件路径集合
|
* @param os 输出流
|
* @return zip文件保存绝对路径
|
*/
|
public static long createZip(List<Map<String, String>> pathList, OutputStream os) {
|
try {
|
long size = 0;
|
ZipOutputStream out = new ZipOutputStream(os);
|
|
String downloadPath;
|
int symbol;
|
Map<String, Integer> fileNameCountMap = Maps.newHashMap();
|
int showCount = 0;
|
String fileName;
|
String head;
|
String tail;
|
for (Map<String, String> pathMap : pathList) {
|
downloadPath = pathMap.get(CmnConst.SOURCE_PATH);
|
symbol = "1".equals(pathMap.get(CmnConst.ENCRPT_SIGN)) ? -1 : 0;
|
|
//获得文件名
|
fileName = pathMap.get(CmnConst.REAL_FILE_NAME);
|
head = fileName.substring(0, fileName.lastIndexOf("."));
|
tail = fileName.substring(fileName.lastIndexOf(".") + 1);
|
showCount = fileNameCountMap.get(fileName) == null ? 0 : fileNameCountMap.get(fileName);
|
if (showCount > 0) {
|
fileName = head + "(" + showCount + ")." + tail;
|
}
|
fileNameCountMap.put(fileName, ++showCount);
|
|
//以论文标题为每个文件命名
|
FileInputStream fis = new FileInputStream(downloadPath);
|
out.putNextEntry(new ZipEntry(fileName));
|
|
//写入压缩包
|
int len;
|
byte[] buffer = new byte[1024];
|
while ((len = fis.read(buffer)) > 0) {
|
if (symbol == 1) {
|
// 加密
|
out.write(encryption(buffer), 0, len);
|
} else if (symbol == -1) {
|
// 解密
|
out.write(decryption(buffer), 0, len);
|
} else {
|
// 单纯的复制
|
out.write(buffer, 0, len);
|
}
|
size += len;
|
}
|
out.closeEntry();
|
fis.close();
|
}
|
out.close();
|
out.flush();
|
return size;
|
} catch (Exception e) {
|
throw new BaseException(FileCode.LOAD_FTP_PROPERTIES_FAIL.getValue(), FileCode.LOAD_FTP_PROPERTIES_FAIL.getText(), e);
|
}
|
}
|
|
/**
|
* 拷贝文件
|
*
|
* @param sourceFile
|
* @param aimPath
|
* @param symbol 1：加密，-1：解密，0：不动
|
*/
|
public static File copyFile(File sourceFile, String aimPath, int symbol) {
|
if (StringUtils.isEmpty(aimPath)) {
|
throw new BaseException(FileCode.INVALID_FILE_PATH.getValue(), FileCode.INVALID_FILE_PATH.getText());
|
}
|
File aimFile = new File(aimPath);
|
File aimDir = aimFile.getParentFile();
|
if (!aimDir.exists()) {
|
aimDir.mkdirs();
|
}
|
try (InputStream is = new FileInputStream(sourceFile);
|
OutputStream os = new FileOutputStream(aimFile);) {
|
int len;
|
byte[] b = new byte[1024];
|
while ((len = is.read(b)) > 0) {
|
if (symbol == 1) {
|
// 加密
|
os.write(encryption(b), 0, len);
|
} else if (symbol == -1) {
|
// 解密
|
os.write(decryption(b), 0, len);
|
} else {
|
// 单纯的复制
|
os.write(b, 0, len);
|
}
|
}
|
os.flush();
|
return aimFile;
|
} catch (IOException e) {
|
throw new BaseException(FileCode.COPY_FILE_FAIL.getValue(), FileCode.COPY_FILE_FAIL.getText(), e);
|
}
|
}
|
|
public static void copyFile(File sourceFile, String aimPath) {
|
copyFile(sourceFile, aimPath, 0);
|
}
|
|
/**
|
* 加密
|
*
|
* @param bytes
|
* @return
|
*/
|
public static byte[] encryption(byte[] bytes) {
|
/**
|
* 设置一个加密算法，要求可逆，还可以通过解密得到原来的东西
|
* 我们这样做：每一个字节的值，都加上它的索引号
|
* 然后，交换第一个字节和最后一个字节的位置，即加密结束
|
*/
|
for (int i = 0; i < bytes.length; i++) {
|
bytes[i] = (byte) (bytes[i] + i);
|
}
|
byte temp = bytes[0];
|
bytes[0] = bytes[bytes.length - 1];
|
bytes[bytes.length - 1] = temp;
|
return bytes;
|
|
}
|
|
/**
|
* 解密
|
*
|
* @param bytes
|
* @return
|
*/
|
public static byte[] decryption(byte[] bytes) {
|
/**
|
* 解密算法，是加密的逆过程
|
* 先交换第一个字节和最后一个字节的位置
|
* 然后，每一个字节的值，都减去它的索引号
|
*/
|
byte temp = bytes[0];
|
bytes[0] = bytes[bytes.length - 1];
|
bytes[bytes.length - 1] = temp;
|
for (int i = 0; i < bytes.length; i++) {
|
bytes[i] = (byte) (bytes[i] - i);
|
}
|
return bytes;
|
}
|
}
|
