aerc/ui/context.go

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) 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
	}
}