grindlemire/go-tui code browser

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

package tui

import (
	"io"
	"os"
	"strings"
)

// PrintOption configures single-frame rendering.
type PrintOption func(*printConfig)

type printConfig struct {
	width int // 0 means auto-detect
}

// WithPrintWidth sets an explicit width in characters.
// Default: auto-detect terminal width, falling back to 80 columns.
func WithPrintWidth(w int) PrintOption {
	return func(c *printConfig) {
		c.width = w
	}
}

// Print renders a Viewable to stdout with ANSI styling.
// Width is auto-detected from the terminal unless overridden with WithPrintWidth.
func Print(v Viewable, opts ...PrintOption) {
	Fprint(os.Stdout, v, opts...)
}

// Sprint renders a Viewable to a string with ANSI escape codes.
// Width is auto-detected from the terminal unless overridden with WithPrintWidth.
func Sprint(v Viewable, opts ...PrintOption) string {
	cfg := printConfig{}
	for _, o := range opts {
		o(&cfg)
	}

	width := cfg.width
	if width <= 0 {
		w, _, err := getTerminalSize(int(os.Stdout.Fd()))
		if err != nil || w <= 0 {
			width = 80
		} else {
			width = w
		}
	}

	el := v.GetRoot()
	caps := DetectCapabilities()
	buf, height := renderElementToBuffer(el, width, caps)
	if height == 0 {
		return ""
	}

	var sb strings.Builder
	esc := newEscBuilder(256)
	for row := range height {
		if row > 0 {
			sb.WriteByte('\n')
		}
		sb.WriteString(bufferRowToANSI(buf, row, esc, caps))
	}
	return sb.String()
}

// Fprint renders a Viewable to the given writer with ANSI styling.
// Width is auto-detected from the terminal unless overridden with WithPrintWidth.
func Fprint(w io.Writer, v Viewable, opts ...PrintOption) {
	s := Sprint(v, opts...)
	if s != "" {
		io.WriteString(w, s)
		io.WriteString(w, "\n")
	}
}