aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/it/alessandroiezzi/genwebsite/Page.java
blob: b36c78faea7b6f2cc8efaf1bda181d3806061220 (plain)
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
 * Copyright (C) 2021  Alessandro Iezzi dev@alessandroiezzi.it
 *
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
package it.alessandroiezzi.genwebsite;

import com.vladsch.flexmark.html.HtmlRenderer;
import com.vladsch.flexmark.parser.Parser;
import com.vladsch.flexmark.util.ast.Node;
import com.vladsch.flexmark.util.data.MutableDataSet;
import lombok.Getter;
import lombok.Setter;

import java.io.BufferedReader;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;

public class Page {
    @Setter private String rootDir = "";
    @Getter @Setter private String out = "";
    @Getter @Setter private String content = "";
    @Getter private String in = "";
    @Getter @Setter private String template = "";
    @Setter private String title = "";
    @Getter @Setter private String id = "";
    @Getter @Setter private String dir = "";
    @Getter private boolean parsed = false;
    @Getter private String date;

    public Page() {
        date = new String("");
    }

    public String getTitle() {
        if (!parsed) {
            throw new RuntimeException("You must call parse before getTitle");
        }

        System.out.println("Page title: " + title);

        return title;
    }

    public void setIn(String in) {
        parsed = false;
        this.in = in;
    }

    public void setDate(String date) {
        this.date = date;
        System.out.println("-------------------> " + date);
    }

    public String parse() throws IOException {
        if (parsed)
            return content;

        if (in != null && !in.isEmpty() && !in.trim().isEmpty()) {
            String[] splited = in.split("\\.");
            String extension = splited[splited.length - 1];

            StringBuilder sb = new StringBuilder();
            BufferedReader br = Files.newBufferedReader(Paths.get(rootDir).resolve(in));
            String line;
            while ((line = br.readLine()) != null) {
                if (line.startsWith("# property:")) {
                    String property = line.replace("# property:", "");
                    String[] splitted = property.split("=");
                    if (splitted.length > 1) {
                        switch(splitted[0].trim()) {
                            case "title":
                                title = splitted[1].trim();
                                break;
                            case "date":
                                date = splitted[1].trim();
                                break;
                        }
                    }
                } else {
                    sb.append(line).append("\n");
                }
            }

            switch (extension) {
                case "md":
                    System.out.println("Parsing markdown");
                    MutableDataSet options = new MutableDataSet();

                    // uncomment to set optional extensions
                    //options.set(Parser.EXTENSIONS, Arrays.asList(TablesExtension.create(), StrikethroughExtension.create()));

                    // uncomment to convert soft-breaks to hard breaks
                    //options.set(HtmlRenderer.SOFT_BREAK, "<br />\n");

                    Parser parser = Parser.builder(options).build();
                    HtmlRenderer renderer = HtmlRenderer.builder(options).build();

                    // You can re-use parser and renderer instances
                    Node document = parser.parse(sb.toString());

                    content = renderer.render(document);
                    break;
                case "html":
                case "htm":
                case "xhtm":
                    content = sb.toString();
                    break;
            }
        }

        parsed = true;

        return content;
    }
}