Add style mappings for runs

1 files changed, 28 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/styles/RunMatcher.java b/src/main/java/org/zwobble/mammoth/styles/RunMatcher.java
new file mode 100644
index 0000000..75e883c
--- /dev/null
+++ b/src/main/java/org/zwobble/mammoth/styles/RunMatcher.java
@@ -0,0 +1,28 @@
+package org.zwobble.mammoth.styles;
+
+import org.zwobble.mammoth.documents.Run;
+
+import java.util.Optional;
+
+public class RunMatcher implements DocumentElementMatcher<Run> {
+    public static RunMatcher styleId(String styleId) {
+        return new RunMatcher(Optional.of(styleId), Optional.empty());
+    }
+
+    public static RunMatcher styleName(String styleName) {
+        return new RunMatcher(Optional.empty(), Optional.of(styleName));
+    }
+
+    private final Optional<String> styleId;
+    private final Optional<String> styleName;
+
+    public RunMatcher(Optional<String> styleId, Optional<String> styleName) {
+        this.styleId = styleId;
+        this.styleName = styleName;
+    }
+
+    @Override
+    public boolean matches(Run run) {
+        return DocumentElementMatching.matchesStyle(styleId, styleName, run.getStyle());
+    }
+}