termui/box.go

package termui

import tm "github.com/nsf/termbox-go"

const TOP_RIGHT = '┐'
const VERTICAL_LINE = '│'
const HORIZONTAL_LINE = '─'
const TOP_LEFT = '┌'
const BOTTOM_RIGHT = '┘'
const BOTTOM_LEFT = '└'

type Box struct {
	X       int
	Y       int
	Width   int
	Height  int
	FgColor tm.Attribute
	BgColor tm.Attribute
}

type HLine struct {
	X       int
	Y       int
	Length  int
	FgColor tm.Attribute
	BgColor tm.Attribute
}

type VLine struct {
	X       int
	Y       int
	Length  int
	FgColor tm.Attribute
	BgColor tm.Attribute
}

func (l HLine) Buffer() []Point {
	pts := make([]Point, l.Length)
	for i := 0; i < l.Length; i++ {
		pts[i].X = l.X + i
		pts[i].Y = l.Y
		pts[i].Code.Ch = HORIZONTAL_LINE
		pts[i].Code.Bg = l.BgColor
		pts[i].Code.Fg = l.FgColor
	}
	return pts
}

func (l VLine) Buffer() []Point {
	pts := make([]Point, l.Length)
	for i := 0; i < l.Length; i++ {
		pts[i].X = l.X
		pts[i].Y = l.Y + i
		pts[i].Code.Ch = VERTICAL_LINE
		pts[i].Code.Bg = l.BgColor
		pts[i].Code.Fg = l.FgColor
	}
	return pts
}

func (b Box) Buffer() []Point {
	if b.Width < 2 || b.Height < 2 {
		return nil
	}
	pts := make([]Point, 2*b.Width+2*b.Height-4)

	pts[0].X = b.X
	pts[0].Y = b.Y
	pts[0].Code.Fg = b.FgColor
	pts[0].Code.Bg = b.BgColor
	pts[0].Code.Ch = TOP_LEFT

	pts[1].X = b.X + b.Width - 1
	pts[1].Y = b.Y
	pts[1].Code.Fg = b.FgColor
	pts[1].Code.Bg = b.BgColor
	pts[1].Code.Ch = TOP_RIGHT

	pts[2].X = b.X
	pts[2].Y = b.Y + b.Height - 1
	pts[2].Code.Fg = b.FgColor
	pts[2].Code.Bg = b.BgColor
	pts[2].Code.Ch = BOTTOM_LEFT

	pts[3].X = b.X + b.Width - 1
	pts[3].Y = b.Y + b.Height - 1
	pts[3].Code.Fg = b.FgColor
	pts[3].Code.Bg = b.BgColor
	pts[3].Code.Ch = BOTTOM_RIGHT

	copy(pts[4:], (HLine{b.X + 1, b.Y, b.Width - 2, b.FgColor, b.BgColor}).Buffer())
	copy(pts[4+b.Width-2:], (HLine{b.X + 1, b.Y + b.Height - 1, b.Width - 2, b.FgColor, b.BgColor}).Buffer())
	copy(pts[4+2*b.Width-4:], (VLine{b.X, b.Y + 1, b.Height - 2, b.FgColor, b.BgColor}).Buffer())
	copy(pts[4+2*b.Width-4+b.Height-2:], (VLine{b.X + b.Width - 1, b.Y + 1, b.Height - 2, b.FgColor, b.BgColor}).Buffer())

	return pts
}

type LabeledBox struct {
	Box
	Label        string
	LabelFgColor tm.Attribute
	LabelBgColor tm.Attribute
}

func (lb LabeledBox) Buffer() []Point {
	ps := lb.Box.Buffer()
	maxTxtW := lb.Width - 2
	rs := trimStr2Runes(lb.Label, maxTxtW)

	for i := 0; i < len(rs); i++ {
		p := Point{}
		p.X = lb.X + 1 + i
		p.Y = lb.Y
		p.Code.Ch = rs[i]
		p.Code.Fg = lb.LabelFgColor
		p.Code.Bg = lb.LabelBgColor
		ps = append(ps, p)
	}

	return ps
}