feiyu02
2022-07-04 c93ad66797e4830ccf4de81c1e8787ab90b22791
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package cn.flightfeather.supervision.common.pdf
 
import com.itextpdf.html2pdf.ConverterProperties
import com.itextpdf.html2pdf.HtmlConverter
import com.itextpdf.layout.font.FontProvider
import com.itextpdf.text.pdf.BaseFont
import freemarker.template.Configuration
import org.xhtmlrenderer.pdf.ITextRenderer
import java.io.*
import javax.servlet.http.HttpServletResponse
 
/**
 * @Description: 生成PDF合同工具类
 */
object GeneratePdfUtil {
    private const val TEMPORARY_CONTRACT_HTML = "src/main/resources/templates/temporary.html"
 
    private const val SIMSUM_FILE = "src/main/resources/font/simsun.ttc"
    @Throws(Exception::class)
    fun generateContract(param: DynamicParam): String {
        // 生成html合同
        generateHTML(param.templatePath, param.templateName, param.param)
        // 根据html合同生成pdf合同
        generatePDF(param.outPath + param.outName)
        // 删除临时html格式合同
//        removeFile(TEMPORARY_CONTRACT_HTML)
        return param.outPath + param.outName
    }
 
    /**
     * @Description 生成html格式合同
     */
    @Throws(Exception::class)
    private fun generateHTML(templatePath: String, templateName: String, paramMap: Map<String, Any>) {
        val cfg = Configuration()
        cfg.defaultEncoding = "UTF-8"
        /**
         * 1.setClassForTemplateLoading(this.getClass(), "/HttpWeb");
         * 基于类路径,HttpWeb包下的framemaker.ftl文件
         * 2.setDirectoryForTemplateLoading(new File("/template"));
         * 基于文件系统,template目录下的文件
         * 3.setServletContextForTemplateLoading(request.getSession().getServletContext(), "/template");
         * 基于Servlet Context,指的是基于WebRoot下的template下的framemaker.ftl文件
         */
        cfg.setDirectoryForTemplateLoading(File(templatePath))
 
        // templateName.ftl为要装载的模板
        val template = cfg.getTemplate(templateName)
        val outHtmFile = File(TEMPORARY_CONTRACT_HTML)
        val out: Writer = BufferedWriter(OutputStreamWriter(FileOutputStream(outHtmFile)))
        // 将参数输出到模版,并操作到HTML上
        template.process(paramMap, out)
        out.close()
    }
 
    /**
     * @Description 根据html生成pdf格式合同
     */
    @Throws(Exception::class)
    private fun generatePDF(pdfUrl: String) {
        val htmFile = File(TEMPORARY_CONTRACT_HTML)
        val pdfFile = File(pdfUrl)
        println(pdfUrl)
//        val url = htmFile.toURI()
//        val os: OutputStream = FileOutputStream(pdfFile)
//        val renderer = ITextRenderer()
//        renderer.setDocument(url)
//        val fontResolver = renderer.fontResolver
//        // 解决中文支持问题
//        fontResolver.addFont(SIMSUM_FILE, BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED)
//        renderer.layout()
//        renderer.createPDF(os)
//        os.close()
        val p = ConverterProperties()
        p.fontProvider = FontProvider().apply {
//            this.addFont(SIMSUM_FILE)
            this.addDirectory("src/main/resources/font/")
        }
        HtmlConverter.convertToPdf(htmFile, pdfFile, p)
    }
 
    /**
     * @Description 移除文件
     */
    private fun removeFile(fileUrl: String) {
        val file = File(fileUrl)
        file.delete()
    }
 
    @Throws(Exception::class)
    fun returnPdfStream(response: HttpServletResponse, pathName: String?) {
        response.contentType = "application/pdf"
        val file = File(pathName)
        if (file.exists()) {
            val `in` = FileInputStream(file)
            val out: OutputStream = response.outputStream
            val b = ByteArray(1024 * 5)
            var n: Int
            while (`in`.read(b).also { n = it } != -1) {
                out.write(b, 0, n)
            }
            out.flush()
            `in`.close()
            out.close()
        }
    }
}