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

---
 .../java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java | 8 ++++++++
 1 file changed, 8 insertions(+)

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

diff --git a/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java b/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java
index 4e6e27b..dbd4fa8 100644
--- a/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java
+++ b/src/main/java/org/zwobble/mammoth/internal/styles/StyleMapBuilder.java
@@ -3,6 +3,7 @@ package org.zwobble.mammoth.internal.styles;
 import org.zwobble.mammoth.internal.documents.Break;
 import org.zwobble.mammoth.internal.documents.Paragraph;
 import org.zwobble.mammoth.internal.documents.Run;
+import org.zwobble.mammoth.internal.documents.Table;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -17,6 +18,7 @@ public class StyleMapBuilder {
     private Optional<HtmlPath> commentReference;
     private final List<StyleMapping<Paragraph>> paragraphStyles = new ArrayList<>();
     private final List<StyleMapping<Run>> runStyles = new ArrayList<>();
+    private final List<StyleMapping<Table>> tableStyles = new ArrayList<>();
     private final List<StyleMapping<Break>> breakStyles = new ArrayList<>();
 
     public StyleMapBuilder() {
@@ -68,6 +70,11 @@ public class StyleMapBuilder {
         return this;
     }
 
+    public StyleMapBuilder mapTable(TableMatcher matcher, HtmlPath path) {
+        tableStyles.add(new StyleMapping<>(matcher, path));
+        return this;
+    }
+
     public StyleMapBuilder mapBreak(BreakMatcher matcher, HtmlPath path) {
         breakStyles.add(new StyleMapping<>(matcher, path));
         return this;
@@ -83,6 +90,7 @@ public class StyleMapBuilder {
             commentReference,
             paragraphStyles,
             runStyles,
+            tableStyles,
             breakStyles
         );
     }
-- 
cgit v1.2.3