sampler-fork/component/asciibox/asciibox.go

90 lines
1.9 KiB
Go

package asciibox
import (
ui "github.com/gizak/termui/v3"
fl "github.com/mbndr/figlet4go"
"github.com/sqshq/sampler/asset"
"github.com/sqshq/sampler/component"
"github.com/sqshq/sampler/config"
"github.com/sqshq/sampler/console"
"github.com/sqshq/sampler/data"
"image"
"strings"
)
type AsciiBox struct {
*ui.Block
*data.Consumer
alert *data.Alert
text string
ascii string
style ui.Style
render *fl.AsciiRender
options *fl.RenderOptions
palette console.Palette
}
const asciiFontExtension = ".flf"
func NewAsciiBox(c config.AsciiBoxConfig, palette console.Palette) *AsciiBox {
options := fl.NewRenderOptions()
options.FontName = string(*c.Font)
fontStr, err := asset.Asset(options.FontName + asciiFontExtension)
if err != nil {
panic("Can't load the font: " + err.Error())
}
render := fl.NewAsciiRender()
_ = render.LoadBindataFont(fontStr, options.FontName)
color := c.Color
if color == nil {
color = &palette.BaseColor
}
box := AsciiBox{
Block: component.NewBlock(c.Title, true, palette),
Consumer: data.NewConsumer(),
style: ui.NewStyle(*color),
render: render,
options: options,
palette: palette,
}
go func() {
for {
select {
case sample := <-box.SampleChannel:
box.text = strings.TrimSpace(sample.Value)
box.ascii, _ = box.render.RenderOpts(box.text, box.options)
case alert := <-box.AlertChannel:
box.alert = alert
}
}
}()
return &box
}
func (a *AsciiBox) Draw(buffer *ui.Buffer) {
buffer.Fill(ui.NewCell(' ', ui.NewStyle(ui.ColorBlack)), a.GetRect())
a.Block.Draw(buffer)
point := a.Inner.Min
cells := ui.ParseStyles(a.ascii, a.style)
for i := 0; i < len(cells) && point.Y < a.Inner.Max.Y; i++ {
if cells[i].Rune == '\n' {
point = image.Pt(a.Inner.Min.X, point.Y+1)
} else if point.In(a.Inner) {
buffer.SetCell(cells[i], point)
point = point.Add(image.Pt(1, 0))
}
}
component.RenderAlert(a.alert, a.Rectangle, buffer)
}