import lombok.extern.slf4j.Slf4j;

import java.io.*;

import java.nio.charset.StandardCharsets;

import java.time.LocalDateTime;

import java.time.format.DateTimeFormatter;

/**

* 生成PDF工具类

*/

@Slf4j

public class HtmlToPdfUtils {

/**

* wkhtmltopdf在 Windows、Linux系统中的安装路径

*/

private static final String WINDOWS_HTML_TO_PDF_TOOL = "E:\\wkhtmltopdf\\bin\\wkhtmltopdf.exe";

private static final String LINUX_HTML_TO_PDF_TOOL = "/opt/wkhtmltox/bin/wkhtmltopdf";

/**

* 页眉图片html：Windows、Linux系统中的路径

*/

private static final String WINDOWS_HEAD_HTML = "D:\\head.html";

private static final String LINUX_HEAD_HTML = "/opt/head.html";

/**

* 页脚图片html：Windows、Linux系统中的路径

*/

private static final String WINDOWS_FOOT_HTML = "D:\\foot.html";

private static final String LINUX_FOOT_HTML = "/opt/foot.html";

/**

* 临时文件存放目录：Windows、Linux系统中的路径

*/

public static final String WINDOWS_FILE_URL = "D:\\pdf\\temporary\\file";

public static final String LINUX_FILE_URL = "/opt/temporary/file";

static {

String fileUrl = isWindowsSystem() ? WINDOWS_FILE_URL : LINUX_FILE_URL;

File f = new File(fileUrl);

if(!f.exists()){

f.mkdirs();

}

}

/**

* html字符串转pdf，会生成垃圾文件，需要定时清理

*

* @param data 替换好的html字符串

* @param destFileName 保存pdf的名称

* @return 返回pdf成功生成的路径

*/

public static String convertStringToHtml(String data, String destFileName) {

return convert(stringToHtml(data), destFileName);

}

/**

* 判断当前系统是否是Windows系统

* @return true：Windows系统，false：Linux系统

*/

public static boolean isWindowsSystem(){

String property = System.getProperty("os.name").toLowerCase();

return property.contains("windows");

}

/**

* html转pdf(加页眉页脚)

*

* @param srcPath html路径，可以是硬盘上的路径，也可以是网络路径

* @param destFileName 保存pdf的名称

* @return 返回pdf成功生成的路径

*/

public static String convert(String srcPath, String destFileName) {

destFileName = (isWindowsSystem() ? WINDOWS_FILE_URL + "\\" : LINUX_FILE_URL + "/") + destFileName;

File file = new File(destFileName);

// File parent = file.getParentFile();

//如果pdf保存路径不存在，则创建路径

if (!file.exists()) {

try {

file.createNewFile();

} catch (IOException e) {

e.printStackTrace();

log.error("html转pdf，创建文件失败：{}", destFileName);

}

}

StringBuilder cmd = new StringBuilder();

String toPdfTool = isWindowsSystem() ? WINDOWS_HTML_TO_PDF_TOOL : LINUX_HTML_TO_PDF_TOOL;

// 这里可以拼接页眉页脚等参数

cmd.append(toPdfTool);

cmd.append(" ");

// wkhtmltopdf默认不允许访问本地文件，需加入以下参数

cmd.append(" --enable-local-file-access");

// cmd.append(" --page-size A4");

cmd.append(" ");

cmd.append(" --disable-smart-shrinking");

// 页眉图片

cmd.append(" --header-html " + (isWindowsSystem() ? WINDOWS_HEAD_HTML : LINUX_HEAD_HTML));

cmd.append(" --header-spacing 5");

// 页脚图片

cmd.append(" --footer-html " + (isWindowsSystem() ? WINDOWS_FOOT_HTML : LINUX_FOOT_HTML));

cmd.append(" " + srcPath);

cmd.append(" ");

cmd.append(destFileName);

try {

log.info("html转pdf命令：{}", cmd.toString());

Process proc = Runtime.getRuntime().exec(cmd.toString());

HtmlToPdfInterceptor error = new HtmlToPdfInterceptor(proc.getErrorStream());

HtmlToPdfInterceptor output = new HtmlToPdfInterceptor(proc.getInputStream());

error.start();

output.start();

proc.waitFor();

log.info("html转pdf成功：html路径：{} pdf保存路径：{}", srcPath, destFileName);

} catch (Exception e) {

destFileName = "";

e.printStackTrace();

log.error("html转pdf失败：html路径：{} pdf保存路径：{}", srcPath, destFileName);

}

return destFileName;

}

/**

* html字符串转pdf，会生成临时html文件，需要定时清理

*

* @param data 替换好的html字符串

* @return 返回生成的临时文件名称

*/

public static String stringToHtml(String data) {

String srcPath = "";

OutputStreamWriter osw = null;

try {

// 生成随机名字

String code = String.format("%04d",(int) ((Math.random()*9+1) * 1000));

String nowDate = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmmss"));

// 根据系统，创建字符输出流对象，负责向文件内写入

if (!isWindowsSystem()) {

// 非Windows 系统

srcPath = LINUX_FILE_URL + "/" +String.format("%s%s", code , nowDate) + ".html";

} else {

srcPath = WINDOWS_FILE_URL + "\\" + String.format("%s%s", code , nowDate) + ".html";

}

osw = new OutputStreamWriter(new FileOutputStream(srcPath), StandardCharsets.UTF_8);

log.info("html转pdf，生成临时文件：{}", srcPath);

// 将str里面的内容读取到fw所指定的文件中

osw.write(data);

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

log.error("html转pdf，生成临时文件失败：{}", e);

}finally{

if(osw!=null){

try {

osw.close();

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

log.error("html转pdf，生成临时文件关流失败：{}", e);

}

}

}

return srcPath;

}

public static void main(String[] args) {

HtmlToPdfUtils.convert("D:\\test.html", "D:\\wkhtmltopdf.pdf");

}

}