aerc/lib/ui/text.go

68 lines
1.2 KiB
Go

package ui
import (
"github.com/mattn/go-runewidth"
"github.com/gdamore/tcell"
)
const (
TEXT_LEFT = iota
TEXT_CENTER = iota
TEXT_RIGHT = iota
)
type Text struct {
text string
strategy uint
fg tcell.Color
bg tcell.Color
onInvalidate func(d Drawable)
}
func NewText(text string) *Text {
return &Text{text: text}
}
func (t *Text) Text(text string) *Text {
t.text = text
t.Invalidate()
return t
}
func (t *Text) Strategy(strategy uint) *Text {
t.strategy = strategy
t.Invalidate()
return t
}
func (t *Text) Color(fg tcell.Color, bg tcell.Color) *Text {
t.fg = fg
t.bg = bg
t.Invalidate()
return t
}
func (t *Text) Draw(ctx *Context) {
size := runewidth.StringWidth(t.text)
x := 0
if t.strategy == TEXT_CENTER {
x = (ctx.Width() - size) / 2
}
if t.strategy == TEXT_RIGHT {
x = ctx.Width() - size
}
style := tcell.StyleDefault.Background(t.bg).Foreground(t.fg)
ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', style)
ctx.Printf(x, 0, style, t.text)
}
func (t *Text) OnInvalidate(onInvalidate func(d Drawable)) {
t.onInvalidate = onInvalidate
}
func (t *Text) Invalidate() {
if t.onInvalidate != nil {
t.onInvalidate(t)
}
}