update

internal/tui/options.go

@@ -0,0 +1,71 @@
+package tui
+
+import (
+	"bytes"
+
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+var ErrorWriter = &bytes.Buffer{}
+
+type options struct {
+	// public
+	altScreen tea.ProgramOption
+	styles    *Styles
+	// private (not capable of being set)
+	mouseCellMotion tea.ProgramOption
+	errorWriter     *bytes.Buffer
+}
+
+type Options interface {
+	apply(*options)
+}
+
+type funcOption struct {
+	f func(*options)
+}
+
+func (fo *funcOption) apply(opt *options) {
+	fo.f(opt)
+}
+
+func newFuncOption(f func(*options)) *funcOption {
+	return &funcOption{f: f}
+}
+
+func WithAltScreen(b bool) Options {
+	return newFuncOption(func(opt *options) {
+		if !b {
+			opt.altScreen = noOpt
+			return
+		}
+		opt.altScreen = tea.WithAltScreen()
+	})
+}
+
+func WithStyles(styles *Styles) Options {
+	return newFuncOption(func(opt *options) {
+		if styles != nil {
+			opt.styles = styles
+		}
+	})
+}
+
+func WithErrWriter(b *bytes.Buffer) Options {
+	return newFuncOption(func(opt *options) {
+		if b != nil {
+			opt.errorWriter = b
+		}
+	})
+}
+
+func defaultOptions() *options {
+	return &options{
+		altScreen:       tea.WithAltScreen(),
+		styles:          DefaultStyles(),
+		mouseCellMotion: tea.WithMouseCellMotion(),
+		errorWriter:     ErrorWriter,
+	}
+}
+
+func noOpt(*tea.Program) {}