summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/ui/borders.go73
-rw-r--r--lib/ui/context.go109
-rw-r--r--lib/ui/drawable.go10
-rw-r--r--lib/ui/grid.go191
-rw-r--r--lib/ui/interactive.go15
-rw-r--r--lib/ui/tab.go115
-rw-r--r--lib/ui/text.go71
-rw-r--r--lib/ui/ui.go79
8 files changed, 663 insertions, 0 deletions
diff --git a/lib/ui/borders.go b/lib/ui/borders.go
new file mode 100644
index 0000000..08071ad
--- /dev/null
+++ b/lib/ui/borders.go
@@ -0,0 +1,73 @@
+package ui
+
+import (
+ tb "github.com/nsf/termbox-go"
+)
+
+const (
+ BORDER_LEFT = 1 << iota
+ BORDER_TOP = 1 << iota
+ BORDER_RIGHT = 1 << iota
+ BORDER_BOTTOM = 1 << iota
+)
+
+type Bordered struct {
+ borders uint
+ content Drawable
+ onInvalidate func(d Drawable)
+}
+
+func NewBordered(content Drawable, borders uint) *Bordered {
+ b := &Bordered{
+ borders: borders,
+ content: content,
+ }
+ content.OnInvalidate(b.contentInvalidated)
+ return b
+}
+
+func (bordered *Bordered) contentInvalidated(d Drawable) {
+ bordered.Invalidate()
+}
+
+func (bordered *Bordered) Invalidate() {
+ if bordered.onInvalidate != nil {
+ bordered.onInvalidate(bordered)
+ }
+}
+
+func (bordered *Bordered) OnInvalidate(onInvalidate func(d Drawable)) {
+ bordered.onInvalidate = onInvalidate
+}
+
+func (bordered *Bordered) Draw(ctx *Context) {
+ x := 0
+ y := 0
+ width := ctx.Width()
+ height := ctx.Height()
+ cell := tb.Cell{
+ Ch: ' ',
+ Fg: tb.ColorBlack,
+ Bg: tb.ColorWhite,
+ }
+ if bordered.borders&BORDER_LEFT != 0 {
+ ctx.Fill(0, 0, 1, ctx.Height(), cell)
+ x += 1
+ width -= 1
+ }
+ if bordered.borders&BORDER_TOP != 0 {
+ ctx.Fill(0, 0, ctx.Width(), 1, cell)
+ y += 1
+ height -= 1
+ }
+ if bordered.borders&BORDER_RIGHT != 0 {
+ ctx.Fill(ctx.Width()-1, 0, 1, ctx.Height(), cell)
+ width -= 1
+ }
+ if bordered.borders&BORDER_BOTTOM != 0 {
+ ctx.Fill(0, ctx.Height()-1, ctx.Width(), 1, cell)
+ height -= 1
+ }
+ subctx := ctx.Subcontext(x, y, width, height)
+ bordered.content.Draw(subctx)
+}
diff --git a/lib/ui/context.go b/lib/ui/context.go
new file mode 100644
index 0000000..ca3f452
--- /dev/null
+++ b/lib/ui/context.go
@@ -0,0 +1,109 @@
+package ui
+
+import (
+ "fmt"
+
+ "github.com/mattn/go-runewidth"
+ tb "github.com/nsf/termbox-go"
+)
+
+// A context allows you to draw in a sub-region of the terminal
+type Context struct {
+ x int
+ y int
+ width int
+ height int
+}
+
+func (ctx *Context) X() int {
+ return ctx.x
+}
+
+func (ctx *Context) Y() int {
+ return ctx.y
+}
+
+func (ctx *Context) Width() int {
+ return ctx.width
+}
+
+func (ctx *Context) Height() int {
+ return ctx.height
+}
+
+func NewContext(width, height int) *Context {
+ return &Context{0, 0, width, height}
+}
+
+func (ctx *Context) Subcontext(x, y, width, height int) *Context {
+ if x+width > ctx.width || y+height > ctx.height {
+ panic(fmt.Errorf("Attempted to create context larger than parent"))
+ }
+ return &Context{
+ x: ctx.x + x,
+ y: ctx.y + y,
+ width: width,
+ height: height,
+ }
+}
+
+func (ctx *Context) SetCell(x, y int, ch rune, fg, bg tb.Attribute) {
+ if x >= ctx.width || y >= ctx.height {
+ panic(fmt.Errorf("Attempted to draw outside of context"))
+ }
+ tb.SetCell(ctx.x+x, ctx.y+y, ch, fg, bg)
+}
+
+func (ctx *Context) Printf(x, y int, ref tb.Cell,
+ format string, a ...interface{}) int {
+
+ if x >= ctx.width || y >= ctx.height {
+ panic(fmt.Errorf("Attempted to draw outside of context"))
+ }
+
+ str := fmt.Sprintf(format, a...)
+
+ x += ctx.x
+ y += ctx.y
+ old_x := x
+
+ newline := func() bool {
+ x = old_x
+ y++
+ return y < ctx.height
+ }
+ for _, ch := range str {
+ if str == " こんにちは " {
+ fmt.Printf("%c\n", ch)
+ }
+ switch ch {
+ case '\n':
+ if !newline() {
+ return runewidth.StringWidth(str)
+ }
+ case '\r':
+ x = old_x
+ default:
+ tb.SetCell(x, y, ch, ref.Fg, ref.Bg)
+ x += runewidth.RuneWidth(ch)
+ if x == old_x+ctx.width {
+ if !newline() {
+ return runewidth.StringWidth(str)
+ }
+ }
+ }
+ }
+
+ return runewidth.StringWidth(str)
+}
+
+func (ctx *Context) Fill(x, y, width, height int, ref tb.Cell) {
+ _x := x
+ _y := y
+ for ; y < _y+height && y < ctx.height; y++ {
+ for ; x < _x+width && x < ctx.width; x++ {
+ ctx.SetCell(x, y, ref.Ch, ref.Fg, ref.Bg)
+ }
+ x = _x
+ }
+}
diff --git a/lib/ui/drawable.go b/lib/ui/drawable.go
new file mode 100644
index 0000000..ef09451
--- /dev/null
+++ b/lib/ui/drawable.go
@@ -0,0 +1,10 @@
+package ui
+
+type Drawable interface {
+ // Called when this renderable should draw itself
+ Draw(ctx *Context)
+ // Specifies a function to call when this cell needs to be redrawn
+ OnInvalidate(callback func(d Drawable))
+ // Invalidates the drawable
+ Invalidate()
+}
diff --git a/lib/ui/grid.go b/lib/ui/grid.go
new file mode 100644
index 0000000..ede7d0c
--- /dev/null
+++ b/lib/ui/grid.go
@@ -0,0 +1,191 @@
+package ui
+
+import (
+ "fmt"
+ "math"
+)
+
+type Grid struct {
+ rows []GridSpec
+ rowLayout []gridLayout
+ columns []GridSpec
+ columnLayout []gridLayout
+ Cells []*GridCell
+ onInvalidate func(d Drawable)
+ invalid bool
+}
+
+const (
+ SIZE_EXACT = iota
+ SIZE_WEIGHT = iota
+)
+
+// Specifies the layout of a single row or column
+type GridSpec struct {
+ // One of SIZE_EXACT or SIZE_WEIGHT
+ Strategy int
+ // If Strategy = SIZE_EXACT, this is the number of cells this row/col shall
+ // occupy. If SIZE_WEIGHT, the space left after all exact rows/cols are
+ // measured is distributed amonst the remainder weighted by this value.
+ Size int
+}
+
+// Used to cache layout of each row/column
+type gridLayout struct {
+ Offset int
+ Size int
+}
+
+type GridCell struct {
+ Row int
+ Column int
+ RowSpan int
+ ColSpan int
+ Content Drawable
+ invalid bool
+}
+
+func NewGrid() *Grid {
+ return &Grid{invalid: true}
+}
+
+func (cell *GridCell) At(row, col int) *GridCell {
+ cell.Row = row
+ cell.Column = col
+ return cell
+}
+
+func (cell *GridCell) Span(rows, cols int) *GridCell {
+ cell.RowSpan = rows
+ cell.ColSpan = cols
+ return cell
+}
+
+func (grid *Grid) Rows(spec []GridSpec) *Grid {
+ grid.rows = spec
+ return grid
+}
+
+func (grid *Grid) Columns(spec []GridSpec) *Grid {
+ grid.columns = spec
+ return grid
+}
+
+func (grid *Grid) Draw(ctx *Context) {
+ invalid := grid.invalid
+ if invalid {
+ grid.reflow(ctx)
+ }
+ for _, cell := range grid.Cells {
+ if !cell.invalid && !invalid {
+ continue
+ }
+ rows := grid.rowLayout[cell.Row : cell.Row+cell.RowSpan]
+ cols := grid.columnLayout[cell.Column : cell.Column+cell.ColSpan]
+ x := cols[0].Offset
+ y := rows[0].Offset
+ width := 0
+ height := 0
+ for _, col := range cols {
+ width += col.Size
+ }
+ for _, row := range rows {
+ height += row.Size
+ }
+ subctx := ctx.Subcontext(x, y, width, height)
+ cell.Content.Draw(subctx)
+ }
+}
+
+func (grid *Grid) reflow(ctx *Context) {
+ grid.rowLayout = nil
+ grid.columnLayout = nil
+ flow := func(specs *[]GridSpec, layouts *[]gridLayout, extent int) {
+ exact := 0
+ weight := 0
+ nweights := 0
+ for _, spec := range *specs {
+ if spec.Strategy == SIZE_EXACT {
+ exact += spec.Size
+ } else if spec.Strategy == SIZE_WEIGHT {
+ nweights += 1
+ weight += spec.Size
+ }
+ }
+ offset := 0
+ for _, spec := range *specs {
+ layout := gridLayout{Offset: offset}
+ if spec.Strategy == SIZE_EXACT {
+ layout.Size = spec.Size
+ } else if spec.Strategy == SIZE_WEIGHT {
+ size := float64(spec.Size) / float64(weight)
+ size *= float64(extent - exact)
+ layout.Size = int(math.Floor(size))
+ }
+ offset += layout.Size
+ *layouts = append(*layouts, layout)
+ }
+ }
+ flow(&grid.rows, &grid.rowLayout, ctx.Height())
+ flow(&grid.columns, &grid.columnLayout, ctx.Width())
+ grid.invalid = false
+}
+
+func (grid *Grid) invalidateLayout() {
+ grid.invalid = true
+ if grid.onInvalidate != nil {
+ grid.onInvalidate(grid)
+ }
+}
+
+func (grid *Grid) Invalidate() {
+ grid.invalidateLayout()
+ for _, cell := range grid.Cells {
+ cell.Content.Invalidate()
+ }
+}
+
+func (grid *Grid) OnInvalidate(onInvalidate func(d Drawable)) {
+ grid.onInvalidate = onInvalidate
+}
+
+func (grid *Grid) AddChild(content Drawable) *GridCell {
+ cell := &GridCell{
+ RowSpan: 1,
+ ColSpan: 1,
+ Content: content,
+ invalid: true,
+ }
+ grid.Cells = append(grid.Cells, cell)
+ cell.Content.OnInvalidate(grid.cellInvalidated)
+ cell.invalid = true
+ grid.invalidateLayout()
+ return cell
+}
+
+func (grid *Grid) RemoveChild(cell *GridCell) {
+ for i, _cell := range grid.Cells {
+ if _cell == cell {
+ grid.Cells = append(grid.Cells[:i], grid.Cells[i+1:]...)
+ break
+ }
+ }
+ grid.invalidateLayout()
+}
+
+func (grid *Grid) cellInvalidated(drawable Drawable) {
+ var cell *GridCell
+ for _, cell = range grid.Cells {
+ if cell.Content == drawable {
+ break
+ }
+ cell = nil
+ }
+ if cell == nil {
+ panic(fmt.Errorf("Attempted to invalidate unknown cell"))
+ }
+ cell.invalid = true
+ if grid.onInvalidate != nil {
+ grid.onInvalidate(grid)
+ }
+}
diff --git a/lib/ui/interactive.go b/lib/ui/interactive.go
new file mode 100644
index 0000000..8bdf592
--- /dev/null
+++ b/lib/ui/interactive.go
@@ -0,0 +1,15 @@
+package ui
+
+import (
+ tb "github.com/nsf/termbox-go"
+)
+
+type Interactive interface {
+ // Returns true if the event was handled by this component
+ Event(event tb.Event) bool
+}
+
+type Simulator interface {
+ // Queues up the given input events for simulation
+ Simulate(events []tb.Event)
+}
diff --git a/lib/ui/tab.go b/lib/ui/tab.go
new file mode 100644
index 0000000..e6a8aa5
--- /dev/null
+++ b/lib/ui/tab.go
@@ -0,0 +1,115 @@
+package ui
+
+import (
+ tb "github.com/nsf/termbox-go"
+)
+
+type Tabs struct {
+ Tabs []*Tab
+ TabStrip *TabStrip
+ TabContent *TabContent
+ Selected int
+
+ onInvalidateStrip func(d Drawable)
+ onInvalidateContent func(d Drawable)
+}
+
+type Tab struct {
+ Content Drawable
+ Name string
+ invalid bool
+}
+
+type TabStrip Tabs
+type TabContent Tabs
+
+func NewTabs() *Tabs {
+ tabs := &Tabs{}
+ tabs.TabStrip = (*TabStrip)(tabs)
+ tabs.TabContent = (*TabContent)(tabs)
+ return tabs
+}
+
+func (tabs *Tabs) Add(content Drawable, name string) {
+ tabs.Tabs = append(tabs.Tabs, &Tab{
+ Content: content,
+ Name: name,
+ })
+ tabs.TabStrip.Invalidate()
+ content.OnInvalidate(tabs.invalidateChild)
+}
+
+func (tabs *Tabs) invalidateChild(d Drawable) {
+ for i, tab := range tabs.Tabs {
+ if tab.Content == d {
+ if i == tabs.Selected {
+ tabs.TabContent.Invalidate()
+ }
+ return
+ }
+ }
+}
+
+func (tabs *Tabs) Remove(content Drawable) {
+ for i, tab := range tabs.Tabs {
+ if tab.Content == content {
+ tabs.Tabs = append(tabs.Tabs[:i], tabs.Tabs[i+1:]...)
+ break
+ }
+ }
+ tabs.TabStrip.Invalidate()
+}
+
+func (tabs *Tabs) Select(index int) {
+ if tabs.Selected != index {
+ tabs.Selected = index
+ tabs.TabStrip.Invalidate()
+ tabs.TabContent.Invalidate()
+ }
+}
+
+// TODO: Color repository
+func (strip *TabStrip) Draw(ctx *Context) {
+ x := 0
+ for i, tab := range strip.Tabs {
+ cell := tb.Cell{
+ Fg: tb.ColorBlack,
+ Bg: tb.ColorWhite,
+ }
+ if strip.Selected == i {
+ cell.Fg = tb.ColorDefault
+ cell.Bg = tb.ColorDefault
+ }
+ x += ctx.Printf(x, 0, cell, " %s ", tab.Name)
+ }
+ cell := tb.Cell{
+ Fg: tb.ColorBlack,
+ Bg: tb.ColorWhite,
+ }
+ ctx.Fill(x, 0, ctx.Width()-x, 1, cell)
+}
+
+func (strip *TabStrip) Invalidate() {
+ if strip.onInvalidateStrip != nil {
+ strip.onInvalidateStrip(strip)
+ }
+}
+
+func (strip *TabStrip) OnInvalidate(onInvalidate func(d Drawable)) {
+ strip.onInvalidateStrip = onInvalidate
+}
+
+func (content *TabContent) Draw(ctx *Context) {
+ tab := content.Tabs[content.Selected]
+ tab.Content.Draw(ctx)
+}
+
+func (content *TabContent) Invalidate() {
+ if content.onInvalidateContent != nil {
+ content.onInvalidateContent(content)
+ }
+}
+
+func (content *TabContent) OnInvalidate(onInvalidate func(d Drawable)) {
+ content.onInvalidateContent = onInvalidate
+}
diff --git a/lib/ui/text.go b/lib/ui/text.go
new file mode 100644
index 0000000..6164837
--- /dev/null
+++ b/lib/ui/text.go
@@ -0,0 +1,71 @@
+package ui
+
+import (
+ "github.com/mattn/go-runewidth"
+ tb "github.com/nsf/termbox-go"
+)
+
+const (
+ TEXT_LEFT = iota
+ TEXT_CENTER = iota
+ TEXT_RIGHT = iota
+)
+
+type Text struct {
+ text string
+ strategy uint
+ fg tb.Attribute
+ bg tb.Attribute
+ 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 tb.Attribute, bg tb.Attribute) *Text {
+ t.fg = fg
+ t.bg = bg
+ t.Invalidate()
+ return t
+}
+
+func (t *Text) Draw(ctx *Context) {
+ size := runewidth.StringWidth(t.text)
+ cell := tb.Cell{
+ Ch: ' ',
+ Fg: t.fg,
+ Bg: t.bg,
+ }
+ x := 0
+ if t.strategy == TEXT_CENTER {
+ x = (ctx.Width() - size) / 2
+ }
+ if t.strategy == TEXT_RIGHT {
+ x = ctx.Width() - size
+ }
+ ctx.Fill(0, 0, ctx.Width(), ctx.Height(), cell)
+ ctx.Printf(x, 0, cell, "%s", t.text)
+}
+
+func (t *Text) OnInvalidate(onInvalidate func(d Drawable)) {
+ t.onInvalidate = onInvalidate
+}
+
+func (t *Text) Invalidate() {
+ if t.onInvalidate != nil {
+ t.onInvalidate(t)
+ }
+}
diff --git a/lib/ui/ui.go b/lib/ui/ui.go
new file mode 100644
index 0000000..9ea037c
--- /dev/null
+++ b/lib/ui/ui.go
@@ -0,0 +1,79 @@
+package ui
+
+import (
+ tb "github.com/nsf/termbox-go"
+
+ "git.sr.ht/~sircmpwn/aerc2/config"
+)
+
+type UI struct {
+ Exit bool
+ Content Drawable
+ ctx *Context
+
+ interactive []Interactive
+
+ tbEvents chan tb.Event
+ invalidations chan interface{}
+}
+
+func Initialize(conf *config.AercConfig, content Drawable) (*UI, error) {
+ if err := tb.Init(); err != nil {
+ return nil, err
+ }
+ width, height := tb.Size()
+ state := UI{
+ Content: content,
+ ctx: NewContext(width, height),
+
+ tbEvents: make(chan tb.Event, 10),
+ invalidations: make(chan interface{}),
+ }
+ tb.SetInputMode(tb.InputEsc | tb.InputMouse)
+ tb.SetOutputMode(tb.Output256)
+ go (func() {
+ for !state.Exit {
+ state.tbEvents <- tb.PollEvent()
+ }
+ })()
+ go (func() { state.invalidations <- nil })()
+ content.OnInvalidate(func(_ Drawable) {
+ go (func() { state.invalidations <- nil })()
+ })
+ return &state, nil
+}
+
+func (state *UI) Close() {
+ tb.Close()
+}
+
+func (state *UI) Tick() bool {
+ select {
+ case event := <-state.tbEvents:
+ switch event.Type {
+ case tb.EventKey:
+ if event.Key == tb.KeyEsc {
+ state.Exit = true
+ }
+ case tb.EventResize:
+ tb.Clear(tb.ColorDefault, tb.ColorDefault)
+ state.ctx = NewContext(event.Width, event.Height)
+ state.Content.Invalidate()
+ }
+ if state.interactive != nil {
+ for _, i := range state.interactive {
+ i.Event(event)
+ }
+ }
+ case <-state.invalidations:
+ state.Content.Draw(state.ctx)
+ tb.Flush()
+ default:
+ return false
+ }
+ return true
+}
+
+func (state *UI) AddInteractive(i Interactive) {
+ state.interactive = append(state.interactive, i)
+}