okami.mom/glamour

git clone https://git.nkp.pet/okami.mom/glamour

Kieran Klukaschore: fix pre-existing lint failuresc293649

main
branchesmainall refs →
1.7 KiB66 linesraw
1package ansi
2
3import (
4	"bytes"
5	"fmt"
6	"io"
7
8	"charm.land/lipgloss/v2"
9)
10
11// BlockElement provides a render buffer for children of a block element.
12// After all children have been rendered into it, it applies indentation and
13// margins around them and writes everything to the parent rendering buffer.
14type BlockElement struct {
15	Block   *bytes.Buffer
16	Style   StyleBlock
17	Margin  bool
18	Newline bool
19}
20
21// Render renders a BlockElement.
22func (e *BlockElement) Render(w io.Writer, ctx RenderContext) error {
23	bs := ctx.blockStack
24	bs.Push(*e)
25
26	_, _ = renderText(w, bs.Parent().Style.StylePrimitive, e.Style.BlockPrefix)
27	_, _ = renderText(bs.Current().Block, bs.Current().Style.StylePrimitive, e.Style.Prefix)
28	return nil
29}
30
31// Finish finishes rendering a BlockElement.
32func (e *BlockElement) Finish(w io.Writer, ctx RenderContext) error {
33	bs := ctx.blockStack
34
35	if e.Margin { //nolint: nestif
36		s := lipgloss.Wrap(
37			bs.Current().Block.String(),
38			int(bs.Width(ctx)),
39			" ,.;-+|",
40		)
41
42		mw := NewMarginWriter(ctx, w, bs.Current().Style)
43		defer mw.Close() //nolint:errcheck
44		if _, err := io.WriteString(mw, s); err != nil {
45			return fmt.Errorf("glamour: error writing to writer: %w", err)
46		}
47
48		if e.Newline {
49			if _, err := io.WriteString(mw, "\n"); err != nil {
50				return fmt.Errorf("glamour: error writing to writer: %w", err)
51			}
52		}
53	} else {
54		_, err := bs.Parent().Block.Write(bs.Current().Block.Bytes())
55		if err != nil {
56			return fmt.Errorf("glamour: error writing to writer: %w", err)
57		}
58	}
59
60	_, _ = renderText(w, bs.Current().Style.StylePrimitive, e.Style.Suffix)
61	_, _ = renderText(w, bs.Parent().Style.StylePrimitive, e.Style.BlockSuffix)
62
63	bs.Current().Block.Reset()
64	bs.Pop()
65	return nil
66}