Create internal package

1 files changed, 62 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java b/src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java
new file mode 100644
index 0000000..164eefc
--- /dev/null
+++ b/src/main/java/org/zwobble/mammoth/internal/html/HtmlWriter.java
@@ -0,0 +1,62 @@
+package org.zwobble.mammoth.internal.html;
+
+import java.util.Map;
+
+import static org.zwobble.mammoth.internal.util.MammothLists.orderedBy;
+
+public class HtmlWriter {
+    public static void write(HtmlNode node, StringBuilder builder) {
+        node.visit(new HtmlNode.Visitor() {
+            @Override
+            public void visit(HtmlElement element) {
+                builder.append("<").append(element.getTagName());
+
+                HtmlWriter.generateAttributes(element.getAttributes(), builder);
+
+                builder.append(">");
+
+                element.getChildren().forEach(node -> write(node, 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(" />");
+            }
+
+            @Override
+            public void visit(HtmlTextNode node) {
+                builder.append(HtmlWriter.escapeText(node.getValue()));
+            }
+
+            @Override
+            public void visit(HtmlForceWrite forceWrite) {
+            }
+        });
+    }
+
+    private static void generateAttributes(Map<String, String> attributes, StringBuilder builder) {
+        for (Map.Entry<String, String> attribute : orderedBy(attributes.entrySet(), Map.Entry::getKey)) {
+            builder
+                .append(" ")
+                .append(attribute.getKey())
+                .append("=\"")
+                .append(HtmlWriter.escapeAttributeValue(attribute.getValue()))
+                .append("\"");
+        }
+    }
+
+    private static String escapeText(String text) {
+        return text.replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;");
+    }
+
+    private static String escapeAttributeValue(String value) {
+        return escapeText(value).replace("\"", "&quot;");
+    }
+}