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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
package org.zwobble.mammoth.docx;
import org.zwobble.mammoth.documents.*;
import org.zwobble.mammoth.results.Result;
import org.zwobble.mammoth.util.MammothOptionals;
import org.zwobble.mammoth.xml.XmlElement;
import org.zwobble.mammoth.xml.XmlElementLike;
import org.zwobble.mammoth.xml.XmlNode;
import java.util.Optional;
import java.util.function.Function;
import static com.google.common.collect.Iterables.filter;
import static com.google.common.collect.Iterables.transform;
import static org.zwobble.mammoth.docx.ReadResult.EMPTY_SUCCESS;
import static org.zwobble.mammoth.docx.ReadResult.success;
import static org.zwobble.mammoth.results.Warning.warning;
import static org.zwobble.mammoth.util.MammothLists.list;
public class BodyXmlReader {
private final Styles styles;
private final Numbering numbering;
private final Relationships relationships;
public BodyXmlReader(Styles styles, Numbering numbering, Relationships relationships) {
this.styles = styles;
this.numbering = numbering;
this.relationships = relationships;
}
public ReadResult readElement(XmlElement element) {
switch (element.getName()) {
case "w:t":
return success(new Text(element.innerText()));
case "w:r":
return readRun(element);
case "w:p":
return readParagraph(element);
case "w:tab":
return success(Tab.TAB);
case "w:tbl":
return readElements(element.children()).map(Table::new);
case "w:tr":
return readElements(element.children()).map(TableRow::new);
case "w:tc":
return readElements(element.children()).map(TableCell::new);
case "w:ins":
case "w:smartTag":
case "w:drawing":
case "w:roundrect":
case "w:shape":
case "w:textbox":
case "w:txbxContent":
return readElements(element.children());
case "w:hyperlink":
return readHyperlink(element);
case "w:pPr":
return EMPTY_SUCCESS;
default:
// TODO: emit warning
return EMPTY_SUCCESS;
}
}
private ReadResult readRun(XmlElement element) {
XmlElementLike properties = element.findChildOrEmpty("w:rPr");
return ReadResult.map(
readRunStyle(properties),
readElements(element.children()),
(style, children) -> new Run(
isBold(properties),
isItalic(properties),
isUnderline(properties),
isStrikethrough(properties),
style,
children));
}
private boolean isBold(XmlElementLike properties) {
return properties.hasChild("w:b");
}
private boolean isItalic(XmlElementLike properties) {
return properties.hasChild("w:i");
}
private boolean isUnderline(XmlElementLike properties) {
return properties.hasChild("w:u");
}
private boolean isStrikethrough(XmlElementLike properties) {
return properties.hasChild("w:strike");
}
private Result<Optional<Style>> readRunStyle(XmlElementLike properties) {
return readStyle(properties, "w:rStyle", "Run", styles::findCharacterStyleById);
}
public ReadResult readElements(Iterable<XmlNode> nodes) {
return ReadResult.concat(
transform(
filter(nodes, XmlElement.class),
this::readElement));
}
private ReadResult readParagraph(XmlElement element) {
XmlElementLike properties = element.findChildOrEmpty("w:pPr");
Optional<NumberingLevel> numbering = readNumbering(properties);
return ReadResult.map(
readParagraphStyle(properties),
readElements(element.children()),
(style, children) -> new Paragraph(style, numbering, children));
}
private Result<Optional<Style>> readParagraphStyle(XmlElementLike properties) {
return readStyle(properties, "w:pStyle", "Paragraph", styles::findParagraphStyleById);
}
private Result<Optional<Style>> readStyle(
XmlElementLike properties,
String styleTagName,
String styleType,
Function<String, Optional<Style>> findStyleById)
{
return readVal(properties, styleTagName)
.map(styleId -> findStyleById(styleType, styleId, findStyleById))
.orElse(Result.empty());
}
private Result<Optional<Style>> findStyleById(
String styleType,
String styleId,
Function<String, Optional<Style>> findStyleById)
{
Optional<Style> style = findStyleById.apply(styleId);
if (style.isPresent()) {
return Result.success(style);
} else {
return new Result<>(
Optional.of(new Style(styleId, Optional.empty())),
list(warning(styleType + " style with ID " + styleId + " was referenced but not defined in the document")));
}
}
private Optional<NumberingLevel> readNumbering(XmlElementLike properties) {
XmlElementLike numberingProperties = properties.findChildOrEmpty("w:numPr");
return MammothOptionals.flatMap(
readVal(numberingProperties, "w:numId"),
readVal(numberingProperties, "w:ilvl"),
numbering::findLevel);
}
private ReadResult readHyperlink(XmlElement element) {
Optional<String> relationshipId = element.getAttributeOrNone("r:id");
Optional<String> anchor = element.getAttributeOrNone("w:anchor");
ReadResult childrenResult = readElements(element.children());
if (relationshipId.isPresent()) {
return childrenResult.map(children -> Hyperlink.href(
relationships.findRelationshipById(relationshipId.get()).getTarget(), children));
} else if (anchor.isPresent()) {
return childrenResult.map(children -> Hyperlink.anchor(
anchor.get(), children));
} else {
return ReadResult.EMPTY_SUCCESS;
}
}
private Optional<String> readVal(XmlElementLike element, String name) {
return element.findChildOrEmpty(name).getAttributeOrNone("w:val");
}
}
|
