package foundation.io;
|
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.FileOutputStream;
|
import java.io.InputStream;
|
import java.io.OutputStream;
|
import java.lang.reflect.Modifier;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
|
import com.aspose.words.Document;
|
import com.aspose.words.License;
|
import com.aspose.words.SaveFormat;
|
import com.spire.doc.FileFormat;
|
import com.spire.doc.PictureWatermark;
|
|
import foundation.token.IUser;
|
import fr.opensagres.xdocreport.document.IXDocReport;
|
import fr.opensagres.xdocreport.document.registry.XDocReportRegistry;
|
import fr.opensagres.xdocreport.template.IContext;
|
import fr.opensagres.xdocreport.template.TemplateEngineKind;
|
import fr.opensagres.xdocreport.template.freemarker.internal.XDocFreemarkerContext;
|
|
public class FileInjector {
|
|
protected static Logger logger;
|
private static int Default_WaterMark_Scaling = 140;
|
private static License aposeLic = new License();
|
|
static {
|
logger = LogManager.getLogger(FileInjector.class);
|
}
|
|
public static List<File> createFiles(IUser user, List<String> templateNames, FileTypes type, Map<String, String> data) throws Exception {
|
List<File> result = new ArrayList<File>();
|
|
for (String templateName: templateNames) {
|
File file = createOneFile(user, templateName, type, data);
|
result.add(file);
|
}
|
|
return result;
|
}
|
|
public static File createOneFile(IUser user, File template, String targetName, FileTypes type, Map<String, String> data) throws Exception {
|
//1. 创建下载临时文件
|
File result = FileRepository.createInjectTempFile(user, targetName);
|
|
//2. 注入数据
|
result = createFile(template, result, type, data);
|
|
//3. result
|
return result;
|
}
|
|
public static File createOneFile(IUser user, String templateName, FileTypes type, Map<String, String> data) throws Exception {
|
//1. 根据模板名称获取模板
|
File template = FileRepository.getTemplateFile(templateName);
|
|
//2. 创建下载临时文件
|
File result = FileRepository.createInjectTempFile(user, templateName);
|
|
//3. 注入数据
|
result = createFile(template, result, type, data);
|
|
//4.
|
return result;
|
}
|
|
public static File createFile(File template, File target, FileTypes targetType, Map<String, String> data) throws Exception {
|
return createFile(template, target, targetType, null, 0, data);
|
}
|
|
public static File createFile(File template, File target, FileTypes targetType, File waterMark, int waterMarkScaling, Map<String, String> data) throws Exception {
|
if (template == null || !template.exists()) {
|
logger.error("模板文件不存在：{}", template);
|
return null;
|
}
|
|
long begin = System.currentTimeMillis();
|
|
try {
|
if (FileTypes.Word == targetType) {
|
return createWord(template, target, waterMark, waterMarkScaling, data);
|
}
|
else if (FileTypes.PDF == targetType) {
|
createWord(template, target, waterMark, waterMarkScaling, data);
|
|
File pdfFile = translateFileName(target, FileTypes.PDF);
|
if (!pdfFile.exists()) {
|
pdfFile.createNewFile();
|
}
|
|
return createPDF(target, pdfFile);
|
}
|
else if (FileTypes.XML == targetType) {
|
return null;
|
}
|
else {
|
return null;
|
}
|
}
|
finally {
|
long end = System.currentTimeMillis();
|
System.out.println("pdf转换成功，共耗时：" + ((end - begin) / 1000.0) + "秒");
|
}
|
}
|
|
public static File createWord(File template, File target, File waterMark, int waterMarkScaling, Map<String, String> data) throws Exception {
|
if (template == null) {
|
return null;
|
}
|
|
if (!target.exists()) {
|
target.createNewFile();
|
}
|
|
FileInputStream inputStream = new FileInputStream(template);
|
try {
|
IXDocReport docReport = XDocReportRegistry.getRegistry().loadReport(inputStream, TemplateEngineKind.Freemarker);
|
IContext context = docReport.createContext();
|
((XDocFreemarkerContext)context).setDataMap(data);
|
|
//1. 注入数据，生成Word
|
OutputStream outStream = new FileOutputStream(target);
|
try {
|
docReport.process(context, outStream);
|
outStream.flush();
|
}
|
finally {
|
try {
|
outStream.close();
|
}
|
catch (Exception e) {
|
}
|
}
|
|
if (waterMark == null) {
|
return target;
|
}
|
|
//2. 注入数据，生成Word
|
com.spire.doc.Document document = new com.spire.doc.Document();
|
InputStream docStream = new FileInputStream(target);
|
try {
|
document.loadFromStream(docStream, FileFormat.Docx);
|
|
PictureWatermark picture = new PictureWatermark();
|
picture.setPicture(waterMark.getAbsolutePath());
|
picture.setScaling(Default_WaterMark_Scaling);
|
picture.isWashout(false);
|
|
document.setWatermark(picture);
|
document.saveToFile(target.getAbsolutePath(), FileFormat.Docx);
|
|
picture.close();
|
return target;
|
}
|
finally {
|
try {
|
docStream.close();
|
}
|
catch (Exception e) {
|
}
|
|
if (document != null) {
|
try {
|
document.close();
|
}
|
catch (Exception e) {
|
}
|
}
|
}
|
}
|
finally {
|
try {
|
inputStream.close();
|
}
|
catch (Exception e) {
|
}
|
}
|
}
|
|
public static File createPDF(File sourceWordFile, File target) throws Exception {
|
if (!setLicense()) {
|
System.out.println("没有导出");
|
}
|
|
FileOutputStream outStream = new FileOutputStream(target);
|
try {
|
Document doc = new Document(sourceWordFile.getAbsolutePath());
|
doc.save(outStream, SaveFormat.PDF);
|
|
return target;
|
}
|
finally {
|
try {
|
outStream.flush();
|
}
|
catch (Exception e) {
|
}
|
|
try {
|
outStream.close();
|
}
|
catch (Exception e) {
|
}
|
}
|
}
|
|
private static File translateFileName(File target, FileTypes word) {
|
if (target == null) {
|
return null;
|
}
|
|
String path = target.getAbsolutePath();
|
path = path.replace("\\", "/");
|
|
int pos_dot = path.lastIndexOf(".");
|
|
if (pos_dot <= 0) {
|
return null;
|
}
|
|
path = path.substring(0, pos_dot) + ".pdf";
|
return new File(path);
|
}
|
|
public static boolean setLicense() {
|
if (aposeLic.getIsLicensed()) {
|
return aposeLic.getIsLicensed();
|
}
|
|
try {
|
InputStream inputStream = new FileInputStream(new File("D:\\repository\\template\\license.xml"));
|
aposeLic.setLicense(inputStream);
|
|
return aposeLic.getIsLicensed();
|
}
|
catch (Exception e) {
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
public static void init() {
|
try{
|
Class<?> aClass = Class.forName("com.aspose.words.zzXyu");
|
java.lang.reflect.Field zzZXG = aClass.getDeclaredField("zzZXG");
|
zzZXG.setAccessible(true);
|
|
java.lang.reflect.Field modifiersField = zzZXG.getClass().getDeclaredField("modifiers");
|
modifiersField.setAccessible(true);
|
modifiersField.setInt(zzZXG, zzZXG.getModifiers() & ~Modifier.FINAL);
|
zzZXG.set(null,new byte[]{76, 73, 67, 69, 78, 83, 69, 68});
|
}
|
catch (Exception e){
|
e.printStackTrace();
|
}
|
}
|
|
}
|
