db9f9df642d61a506d249cca9e66a05c40b9735c
[sixth.git] / src / main / java / eu / svjatoslav / sixth / core / document / Heading.java
1 package eu.svjatoslav.sixth.core.document;
2
3 import eu.svjatoslav.commons.string.String2;
4 import eu.svjatoslav.commons.string.tokenizer.Tokenizer;
5 import eu.svjatoslav.commons.string.tokenizer.TokenizerMatch;
6 import eu.svjatoslav.sixth.core.document.content.ListElement;
7 import eu.svjatoslav.sixth.core.document.text.FormattedText;
8
9 import java.util.ArrayList;
10 import java.util.List;
11
12 import static eu.svjatoslav.sixth.core.document.Helper.TG_LIST;
13
14 public class Heading {
15     public final FormattedText name;
16     public final int level;
17     public final Heading parent;
18     private final List<Heading> children = new ArrayList<>();
19
20     public final ListElement rootListElement = new ListElement(null, -2, null, "");
21     private ListElement currentListElement = rootListElement;
22
23     public Heading(FormattedText name, int level, Heading parent){
24         this.level = level;
25         this.name = name;
26         this.parent = parent;
27     }
28
29     public void addChild(Heading heading){
30         children.add(heading);
31     }
32
33     public List<Heading> getChildren(){
34         return children;
35     }
36
37     public String toMD () {
38         StringBuilder sb = new StringBuilder();
39
40         if (level > 0) sb.append(enlistTitleInMD());
41
42         rootListElement.toMD(sb, -2);
43
44         children.stream().map(Heading::toMD).forEach(sb::append);
45
46         return sb.toString();
47     }
48
49     private String enlistTitleInMD() {
50         String2 s = new String2();
51         s.append("#", level).append(" ").append(name.toMD(0)).append("\n");
52         return s.toString();
53     }
54
55     public ListElement getCurrentHeading(){
56         return currentListElement;
57     }
58
59     public void parse(TokenizerMatch tm){
60
61         if (tm.isGroup(TG_LIST)){
62             parseList(tm);
63             return;
64         }
65
66         currentListElement.parse(tm);
67     }
68
69     private void parseList(TokenizerMatch tm) {
70         String[] listSections = tm.getRegExpGroups();
71         int indent = listSections[0].length();
72         String type = listSections[1];
73         FormattedText title = FormattedText.fromOrg(parseFullListTitle(getPartialTitle(listSections), tm.getTokenizer(), indent));
74
75         if (indent > currentListElement.indent){
76             ListElement newElement = new ListElement(title, indent, currentListElement, type);
77             currentListElement.addContent(newElement);
78             currentListElement = newElement;
79             return;
80         }
81
82         if (indent == currentListElement.indent){
83             ListElement newElement = new ListElement(title, indent, currentListElement.parent, type);
84             currentListElement.parent.addContent(newElement);
85             currentListElement = newElement;
86             return;
87         }
88
89         while (true){
90             if (currentListElement.parent.indent <= indent){
91                 ListElement newElement = new ListElement(title, indent, currentListElement.parent, type);
92                 currentListElement.parent.addContent(newElement);
93                 currentListElement = newElement;
94                 return;
95             }
96             currentListElement = currentListElement.parent;
97         }
98
99     }
100
101     private String getPartialTitle(String[] listSections) {
102         return listSections.length > 2 ? listSections[2] : "";
103     }
104
105     private String parseFullListTitle(String partialTitle, Tokenizer tokenizer, int listIndent){
106         StringBuilder sb = new StringBuilder();
107         sb.append(partialTitle);
108
109         while (tokenizer.hasMoreContent()){
110             final TokenizerMatch tm = tokenizer.getNextToken();
111
112             if (isContentContinuation(tm, listIndent, Helper.TG_NORMAL_TEXT)){
113                 String titleContinuation = tm.token.substring(listIndent).trim();
114                 sb.append("\n").append(titleContinuation);
115                 continue;
116             }
117
118             tokenizer.unreadToken();
119             break;
120         }
121
122         return sb.toString();
123     }
124
125     public static boolean isContentContinuation(TokenizerMatch tm, int requiredIndent, String requiredGroup) {
126         if (tm.token.length() <= requiredIndent) return false;
127
128         return tm.isGroup(requiredGroup) && tm.token.substring(0, requiredIndent +1).trim().length() == 0;
129     }
130
131 }