From 559101b2464a9e5299c05adfac4a1fac8af8ef23 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 28 Nov 2016 16:18:23 +0000
Subject: Use HtmlElement for void elements

---
 .../zwobble/mammoth/internal/html/HtmlWriter.java  | 24 ++++++++++------------
 1 file changed, 11 insertions(+), 13 deletions(-)

(limited to 'src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java')

diff --git a/src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java b/src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java
index 59bb90a..910fd81 100644
--- a/src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java
+++ b/src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java
@@ -2,6 +2,7 @@ package org.zwobble.mammoth.internal.html;
 
 import java.util.Map;
 
+import static org.zwobble.mammoth.internal.html.Html.isVoidElement;
 import static org.zwobble.mammoth.internal.util.Lists.orderedBy;
 
 public class HtmlWriter {
@@ -13,21 +14,18 @@ public class HtmlWriter {
 
                 HtmlWriter.generateAttributes(element.getAttributes(), builder);
 
-                builder.append(">");
+                if (isVoidElement(element)) {
+                    builder.append(" />");
+                } else {
+                    builder.append(">");
 
-                element.getChildren().forEach(child -> write(child, builder));
+                    element.getChildren().forEach(child -> write(child, builder));
 
-                builder
-                    .append("</")
-                    .append(element.getTagName())
-                    .append(">");
-            }
-
-            @Override
-            public void visit(HtmlSelfClosingElement element) {
-                builder.append("<").append(element.getTagName());
-                HtmlWriter.generateAttributes(element.getAttributes(), builder);
-                builder.append(" />");
+                    builder
+                        .append("</")
+                        .append(element.getTagName())
+                        .append(">");
+                }
             }
 
             @Override
-- 
cgit v1.2.3