59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package ansi
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/muesli/reflow/wordwrap"
|
|
)
|
|
|
|
// A ParagraphElement is used to render individual paragraphs.
|
|
type ParagraphElement struct {
|
|
First bool
|
|
}
|
|
|
|
func (e *ParagraphElement) Render(w io.Writer, ctx RenderContext) error {
|
|
bs := ctx.blockStack
|
|
rules := ctx.options.Styles.Paragraph
|
|
|
|
if !e.First {
|
|
_, _ = w.Write([]byte("\n"))
|
|
}
|
|
be := BlockElement{
|
|
Block: &bytes.Buffer{},
|
|
Style: cascadeStyle(bs.Current().Style, rules, false),
|
|
}
|
|
bs.Push(be)
|
|
|
|
renderText(w, ctx.options.ColorProfile, bs.Parent().Style.StylePrimitive, rules.BlockPrefix)
|
|
renderText(bs.Current().Block, ctx.options.ColorProfile, bs.Current().Style.StylePrimitive, rules.Prefix)
|
|
return nil
|
|
}
|
|
|
|
func (e *ParagraphElement) Finish(w io.Writer, ctx RenderContext) error {
|
|
bs := ctx.blockStack
|
|
rules := bs.Current().Style
|
|
|
|
mw := NewMarginWriter(ctx, w, rules)
|
|
if len(strings.TrimSpace(bs.Current().Block.String())) > 0 {
|
|
flow := wordwrap.NewWriter(int(bs.Width(ctx)))
|
|
flow.KeepNewlines = false
|
|
_, _ = flow.Write(bs.Current().Block.Bytes())
|
|
flow.Close()
|
|
|
|
_, err := mw.Write(flow.Bytes())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, _ = mw.Write([]byte("\n"))
|
|
}
|
|
|
|
renderText(w, ctx.options.ColorProfile, bs.Current().Style.StylePrimitive, rules.Suffix)
|
|
renderText(w, ctx.options.ColorProfile, bs.Parent().Style.StylePrimitive, rules.BlockSuffix)
|
|
|
|
bs.Current().Block.Reset()
|
|
bs.Pop()
|
|
return nil
|
|
}
|