Use table style mappings during document conversion

1 files changed, 30 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/styles/TableMatcher.java b/src/main/java/org/zwobble/mammoth/internal/styles/TableMatcher.java
new file mode 100644
index 0000000..7dbba82
--- /dev/null
+++ b/src/main/java/org/zwobble/mammoth/internal/styles/TableMatcher.java
@@ -0,0 +1,30 @@
+package org.zwobble.mammoth.internal.styles;
+
+import org.zwobble.mammoth.internal.documents.Table;
+
+import java.util.Optional;
+
+public class TableMatcher implements DocumentElementMatcher<Table> {
+    public static final TableMatcher ANY = new TableMatcher(Optional.empty(), Optional.empty());
+
+    public static TableMatcher styleId(String styleId) {
+        return new TableMatcher(Optional.of(styleId), Optional.empty());
+    }
+
+    public static TableMatcher styleName(String styleName) {
+        return new TableMatcher(Optional.empty(), Optional.of(new EqualToStringMatcher(styleName)));
+    }
+
+    private final Optional<String> styleId;
+    private final Optional<StringMatcher> styleName;
+
+    public TableMatcher(Optional<String> styleId, Optional<StringMatcher> styleName) {
+        this.styleId = styleId;
+        this.styleName = styleName;
+    }
+
+    @Override
+    public boolean matches(Table table) {
+        return DocumentElementMatching.matchesStyle(styleId, styleName, table.getStyle());
+    }
+}