Make document indentable

document.go

@@ -0,0 +1,46 @@
+package gold
+
+import (
+	"io"
+
+	bf "gopkg.in/russross/blackfriday.v2"
+)
+
+type DocumentElement struct {
+}
+
+func (e *DocumentElement) Render(w io.Writer, node *bf.Node, tr *TermRenderer) error {
+	rules := tr.style[Document]
+	if rules != nil {
+		if rules.Prefix != "" {
+			renderText(w, rules, rules.Prefix)
+		}
+	}
+	return nil
+}
+
+func (e *DocumentElement) Finish(w io.Writer, node *bf.Node, tr *TermRenderer) error {
+	var indent uint
+	suffix := ""
+	rules := tr.style[Document]
+	if rules != nil {
+		indent = rules.Indent
+		suffix = rules.Suffix
+	}
+
+	iw := &IndentWriter{
+		Indent:  indent,
+		Forward: w,
+	}
+
+	_, err := iw.Write(tr.document.Bytes())
+	if err != nil {
+		return err
+	}
+	tr.document.Reset()
+
+	if suffix != "" {
+		renderText(iw, rules, suffix)
+	}
+	return nil
+}