package com.yumingzhu.wxweb.util;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

/**
 * @Description 
 * @Author yumigzhu
 * @Date 2020/7/22 20:12
 */

public class CustomWKHtmlToPdfUtil {
	private static String tempPath = "C:/apps/tmpFile";// 图片保存目录

	public static String getCommand(String htmlToImage, String sourceFilePath, String targetFilePath) {
		//--quality 设置为50 是比较合适的， 默认的94 可能会导致图片文件过大
		ProcessBuilder pb = new ProcessBuilder(htmlToImage, "--crop-w", "800", "--width", "800","--quality", "50",
				sourceFilePath, targetFilePath);
		Process process;
		try {
			process = pb.start();
			//注意，调用process.getErrorStream()而不是process.getInputStream()
			BufferedReader errStreamReader = new BufferedReader(new InputStreamReader(process.getErrorStream()));
			System.out.println("read errstreamreader");
			String line = null;
			line = errStreamReader.readLine();
			while (line != null) {
				System.out.println(line);
				line = errStreamReader.readLine();
			}
			process.destroy();
			System.out.println("destroyed process");
		} catch (IOException e) {
			e.printStackTrace();
		}
		return targetFilePath;
	}

	public static void main(String[] args) throws Exception {
		String imagePath = tempPath + "/" + System.currentTimeMillis() + ".png";//图片路径
		System.out.println(imagePath);
		String htmlToImage = "E:\softwareAPP\wkhtmltopdf\bin\wkhtmltoimage.exe";

		CustomWKHtmlToPdfUtil.getCommand(htmlToImage,
				"file:///G:/share/text_none_title_share/index.html",
				imagePath);

		System.out.println("执行完成");
	}
}