From d6f00b6a57a19349de058ee17e1f963c0a972cf6 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 4 Nov 2017 11:45:06 +0000
Subject: Use table style mappings during document conversion

---
 .../mammoth/internal/styles/TableMatcher.java      | 30 ++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 src/main/java/org/zwobble/mammoth/internal/styles/TableMatcher.java

(limited to 'src/main/java/org/zwobble/mammoth/internal/styles/TableMatcher.java')

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());
+    }
+}
-- 
cgit v1.2.3