summaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/mammoth/internal/docx/StylesXml.java
blob: 31d55b7c812f06a235c24f85ec9bb43e61eb7a24 (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
package org.zwobble.mammoth.internal.docx;

import org.zwobble.mammoth.internal.documents.NumberingStyle;
import org.zwobble.mammoth.internal.documents.Style;
import org.zwobble.mammoth.internal.xml.XmlElement;
import org.zwobble.mammoth.internal.xml.XmlElementList;

import java.util.Map;
import java.util.Optional;

import static org.zwobble.mammoth.internal.util.Iterables.lazyFilter;
import static org.zwobble.mammoth.internal.util.Maps.*;

public class StylesXml {
    public static Styles readStylesXmlElement(XmlElement element) {
        XmlElementList styleElements = element.findChildren("w:style");
        return new Styles(
            readStyles(styleElements, "paragraph"),
            readStyles(styleElements, "character"),
            readStyles(styleElements, "table"),
            readNumberingStyles(styleElements)
        );
    }

    private static Map<String, Style> readStyles(XmlElementList styleElements, String styleType) {
        return toMapPreferFirst(
            styleElementsOfType(styleElements, styleType),
            StylesXml::readStyle
        );
    }

    private static Map.Entry<String, Style> readStyle(XmlElement element) {
        String styleId = readStyleId(element);
        Optional<String> styleName = element.findChildOrEmpty("w:name").getAttributeOrNone("w:val");
        return entry(styleId, new Style(styleId, styleName));
    }

    private static Map<String, NumberingStyle> readNumberingStyles(XmlElementList styleElements) {
        return toMapPreferFirst(
            styleElementsOfType(styleElements, "numbering"),
            StylesXml::readNumberingStyle
        );
    }

    private static Map.Entry<String, NumberingStyle> readNumberingStyle(XmlElement element) {
        String styleId = readStyleId(element);
        Optional<String> numId = element
            .findChildOrEmpty("w:pPr")
            .findChildOrEmpty("w:numPr")
            .findChildOrEmpty("w:numId")
            .getAttributeOrNone("w:val");
        return entry(styleId, new NumberingStyle(numId));
    }

    private static String readStyleId(XmlElement element) {
        return element.getAttribute("w:styleId");
    }

    private static Iterable<XmlElement> styleElementsOfType(XmlElementList styleElements, String styleType) {
        return lazyFilter(styleElements, styleElement -> isStyleType(styleElement, styleType));
    }

    private static boolean isStyleType(XmlElement styleElement, String styleType) {
        return styleElement.getAttribute("w:type").equals(styleType);
    }
}