Helius
2022-07-05 121b0c399ab5fadd9cb3a46509d68fe1ba476c95
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
package com.xcong.farmer.cms.core.template;
 
import com.xcong.farmer.cms.core.node.PartNode;
import com.xcong.farmer.cms.core.node.Template;
import org.jsoup.nodes.Document;
 
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.List;
import java.util.Map;
 
/**
 * @author wzy
 * @date 2022-07-01
 **/
public class TemplateConfiguration extends Configuration{
 
    private TemplateLoader templateLoader;
 
    public TemplateConfiguration(String templatePath, String staticPath, String outputPath) {
        super(templatePath, staticPath, outputPath);
    }
 
    public void templateLoader(TemplateLoader templateLoader) {
        this.templateLoader = templateLoader;
    }
 
    public void process(Map<String, Object> map, String templateName) {
        if (this.templateLoader == null) {
            throw new RuntimeException("TemplateLoader do not able to be null");
        }
 
        this.templateLoader.data(map);
        output(template(templateName));
    }
 
 
    public Template template(String templateName) {
        return template(new File(path(this.templatePath) + templateName));
    }
 
    public Template template(File file) {
        if (file == null) {
            throw new RuntimeException("template not exist");
        }
 
        return this.templateLoader.template(file);
    }
 
//    public void columnProcess(Map<String, Object> data, String templateName) {
//        process;
//    }
//
//
//    public void articleProcess(Map<String, Object> data, String templateName) {
//        process(data, templateName);
//    }
 
    public void output(Template template) {
        Document document = template.getDocument();
        List<PartNode> partNodes = template.getPartNodes();
        StringBuilder sb = new StringBuilder();
        for (PartNode partNode : partNodes) {
            sb.append(partNode.getHtml());
        }
        document.body().empty().html(sb.toString());
        String outPath = path(this.outputPath);
 
        String html = document.html();
        try {
            FileOutputStream outputStream = new FileOutputStream(outPath + template.getName());
            outputStream.write(html.getBytes());
            outputStream.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
 
    private String path(String path) {
        if (!path.endsWith("/")) {
            path = path + "/";
        }
        return path;
    }
}