Merge pull request subchen#3 from konsorten/master

Added extended pretty printing
stmath · Mar 1, 2018 · e1029cd · e1029cd
2 parents 63d6b99 + c775891
commit e1029cd
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/document.go b/document.go
@@ -48,3 +48,17 @@ func (d *Document) XMLPretty() string {
 	buf.WriteString(d.Root.XMLPretty())
 	return buf.String()
 }
+
+func (d *Document) XMLPrettyEx(indent string) string {
+	buf := new(bytes.Buffer)
+	if len(d.ProcInst) > 0 {
+		buf.WriteString(d.ProcInst)
+		buf.WriteByte('\n')
+	}
+	for _, directive := range d.Directives {
+		buf.WriteString(directive)
+		buf.WriteByte('\n')
+	}
+	buf.WriteString(d.Root.XMLPrettyEx(indent))
+	return buf.String()
+}
diff --git a/node.go b/node.go
@@ -208,3 +208,9 @@ func (n *Node) XMLPretty() string {
 	printXML(buf, n, 0, "  ")
 	return buf.String()
 }
+
+func (n *Node) XMLPrettyEx(indent string) string {
+	buf := new(bytes.Buffer)
+	printXML(buf, n, 0, indent)
+	return buf.String()
+}