diff options
4 files changed, 144 insertions, 0 deletions
diff --git a/src/main/java/org/zwobble/mammoth/docx/Style.java b/src/main/java/org/zwobble/mammoth/docx/Style.java new file mode 100644 index 0000000..551a93f --- /dev/null +++ b/src/main/java/org/zwobble/mammoth/docx/Style.java @@ -0,0 +1,15 @@ +package org.zwobble.mammoth.docx; + +import java.util.Optional; + +public class Style { + private final Optional<String> name; + + public Style(Optional<String> name) { + this.name = name; + } + + public Optional<String> getName() { + return name; + } +} diff --git a/src/main/java/org/zwobble/mammoth/docx/Styles.java b/src/main/java/org/zwobble/mammoth/docx/Styles.java new file mode 100644 index 0000000..89098de --- /dev/null +++ b/src/main/java/org/zwobble/mammoth/docx/Styles.java @@ -0,0 +1,22 @@ +package org.zwobble.mammoth.docx; + +import java.util.Map; +import java.util.Optional; + +public class Styles { + private final Map<String, Style> paragraphStyles; + private final Map<String, Style> characterStyles; + + public Styles(Map<String, Style> paragraphStyles, Map<String, Style> characterStyles) { + this.paragraphStyles = paragraphStyles; + this.characterStyles = characterStyles; + } + + public Optional<Style> findParagraphStyleById(String id) { + return Optional.ofNullable(paragraphStyles.get(id)); + } + + public Optional<Style> findCharacterStyleById(String id) { + return Optional.ofNullable(characterStyles.get(id)); + } +} diff --git a/src/main/java/org/zwobble/mammoth/docx/StylesXml.java b/src/main/java/org/zwobble/mammoth/docx/StylesXml.java new file mode 100644 index 0000000..18061d1 --- /dev/null +++ b/src/main/java/org/zwobble/mammoth/docx/StylesXml.java @@ -0,0 +1,31 @@ +package org.zwobble.mammoth.docx; + +import org.zwobble.mammoth.xml.XmlElement; +import org.zwobble.mammoth.xml.XmlElementList; + +import java.util.Map; + +import static com.google.common.collect.Iterables.filter; +import static com.google.common.collect.Maps.immutableEntry; +import static org.zwobble.mammoth.util.MammothMaps.toMap; + +public class StylesXml { + public static Styles readStylesXmlElement(XmlElement element) { + XmlElementList styleElements = element.findChildren("w:style"); + return new Styles( + readStyles(styleElements, "paragraph"), + readStyles(styleElements, "character")); + } + + private static Map<String, Style> readStyles(XmlElementList styleElements, String styleType) { + return toMap( + filter(styleElements, styleElement -> styleElement.getAttribute("w:type").equals(styleType)), + StylesXml::readStyle); + } + + private static Map.Entry<String, Style> readStyle(XmlElement element) { + return immutableEntry( + element.getAttribute("w:styleId"), + new Style(element.findChildOrEmpty("w:name").getAttributeOrNone("w:val"))); + } +} diff --git a/src/test/java/org/zwobble/mammoth/tests/docx/StylesXmlTests.java b/src/test/java/org/zwobble/mammoth/tests/docx/StylesXmlTests.java new file mode 100644 index 0000000..eb28af9 --- /dev/null +++ b/src/test/java/org/zwobble/mammoth/tests/docx/StylesXmlTests.java @@ -0,0 +1,76 @@ +package org.zwobble.mammoth.tests.docx; + +import com.google.common.collect.ImmutableList; +import com.google.common.collect.ImmutableMap; +import org.junit.Test; +import org.zwobble.mammoth.docx.Styles; +import org.zwobble.mammoth.xml.XmlElement; + +import java.util.Optional; + +import static org.junit.Assert.assertEquals; +import static org.zwobble.mammoth.docx.StylesXml.readStylesXmlElement; +import static org.zwobble.mammoth.xml.XmlNodes.element; + +public class StylesXmlTests { + @Test + public void paragraphStyleIsNoneIfNoStyleWithThatIdExists() { + XmlElement element = element("w:styles"); + + Styles styles = readStylesXmlElement(element); + + assertEquals(Optional.empty(), styles.findParagraphStyleById("Heading1")); + } + + @Test + public void paragraphStyleCanBeFoundById() { + XmlElement element = element("w:styles", ImmutableList.of( + element("w:style", ImmutableMap.of("w:type", "paragraph", "w:styleId", "Heading1"), ImmutableList.of( + nameElement("Heading 1"))))); + + Styles styles = readStylesXmlElement(element); + + assertEquals(Optional.of("Heading 1"), styles.findParagraphStyleById("Heading1").get().getName()); + } + + @Test + public void characterStyleCanBeFoundById() { + XmlElement element = element("w:styles", ImmutableList.of( + element("w:style", ImmutableMap.of("w:type", "character", "w:styleId", "Heading1Char"), ImmutableList.of( + nameElement("Heading 1 Char"))))); + + Styles styles = readStylesXmlElement(element); + + assertEquals(Optional.of("Heading 1 Char"), styles.findCharacterStyleById("Heading1Char").get().getName()); + } + + @Test + public void paragraphAndCharacterStylesAreDistinct() { + XmlElement element = element("w:styles", ImmutableList.of( + element("w:style", ImmutableMap.of("w:type", "paragraph", "w:styleId", "Heading1"), ImmutableList.of( + nameElement("Heading 1"))), + element("w:style", ImmutableMap.of("w:type", "character", "w:styleId", "Heading1Char"), ImmutableList.of( + nameElement("Heading 1 Char"))))); + + Styles styles = readStylesXmlElement(element); + + assertEquals(Optional.empty(), styles.findCharacterStyleById("Heading1")); + assertEquals(Optional.empty(), styles.findParagraphStyleById("Heading1Char")); + } + + @Test + public void styleNameIsNoneIfNameElementDoesNotExist() { + XmlElement element = element("w:styles", ImmutableList.of( + element("w:style", ImmutableMap.of("w:type", "paragraph", "w:styleId", "Heading1")), + element("w:style", ImmutableMap.of("w:type", "character", "w:styleId", "Heading1Char")))); + + Styles styles = readStylesXmlElement(element); + + assertEquals(Optional.empty(), styles.findParagraphStyleById("Heading1").get().getName()); + assertEquals(Optional.empty(), styles.findCharacterStyleById("Heading1Char").get().getName()); + } + + private XmlElement nameElement(String name) { + return element("w:name", ImmutableMap.of("w:val", name)); + } +} |