Drop redundant DocumentElement

This commit is contained in:
Christian Muehlhaeuser
2019-12-13 03:05:55 +01:00
parent 35342a2061
commit aa7c98e225
2 changed files with 7 additions and 41 deletions
-39
View File
@@ -1,39 +0,0 @@
package gold
import (
"bytes"
"io"
)
type DocumentElement struct {
}
func (e *DocumentElement) Render(w io.Writer, ctx RenderContext) error {
rules := ctx.styles.Document
be := BlockElement{
Block: &bytes.Buffer{},
Style: rules,
}
ctx.blockStack.Push(be)
renderText(ctx.blockStack.Current().Block, rules.StylePrimitive, rules.Prefix)
return nil
}
func (e *DocumentElement) Finish(w io.Writer, ctx RenderContext) error {
bs := ctx.blockStack
rules := ctx.styles.Document
mw := NewMarginWriter(ctx, w, rules)
_, err := mw.Write(bs.Current().Block.Bytes())
if err != nil {
return err
}
renderText(mw, rules.StylePrimitive, rules.Suffix)
bs.Current().Block.Reset()
bs.Pop()
return nil
}
+7 -2
View File
@@ -32,9 +32,14 @@ func (tr *TermRenderer) NewElement(node ast.Node, source []byte) Element {
switch node.Kind() {
// Document
case ast.KindDocument:
e := &BlockElement{
Block: &bytes.Buffer{},
Style: ctx.styles.Document,
Margin: true,
}
return Element{
Renderer: &DocumentElement{},
Finisher: &DocumentElement{},
Renderer: e,
Finisher: e,
}
// Heading