138 lines
2.5 KiB
Go
138 lines
2.5 KiB
Go
package tui
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.tordarus.net/Tordarus/adverr"
|
|
"git.tordarus.net/Tordarus/buf2d"
|
|
"github.com/gdamore/tcell"
|
|
)
|
|
|
|
type Screen struct {
|
|
EventTmpl
|
|
|
|
scr tcell.Screen
|
|
buf *ViewBuffer
|
|
|
|
stopCh chan error
|
|
redrawCh chan struct{}
|
|
|
|
// Root is the root view which is currently shown on screen
|
|
Root View
|
|
}
|
|
|
|
func NewScreen(root View) (*Screen, error) {
|
|
scr, err := tcell.NewScreen()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s := &Screen{
|
|
Root: root,
|
|
scr: scr,
|
|
stopCh: make(chan error, 1),
|
|
redrawCh: make(chan struct{}, 1),
|
|
}
|
|
|
|
s.KeyPressed = CloseOnCtrlC(s)
|
|
|
|
go s.eventloop()
|
|
go s.drawloop()
|
|
|
|
return s, nil
|
|
}
|
|
|
|
func (s *Screen) Start() error {
|
|
err := s.scr.Init()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer s.scr.Fini()
|
|
|
|
s.scr.EnableMouse()
|
|
return <-s.stopCh
|
|
}
|
|
|
|
func (s *Screen) Stop() {
|
|
s.StopWithError(nil)
|
|
}
|
|
|
|
func (s *Screen) StopWithError(err error) {
|
|
s.stopCh <- err
|
|
}
|
|
|
|
func (s *Screen) onKeyPressed(event *KeyEvent) {
|
|
if s.KeyPressed == nil || !s.KeyPressed(event) {
|
|
go s.Root.OnKeyPressed(event)
|
|
}
|
|
s.Redraw()
|
|
}
|
|
|
|
func (s *Screen) onMouseEvent(event *MouseEvent) {
|
|
if s.MouseEvent == nil || !s.MouseEvent(event) {
|
|
go s.Root.OnMouseEvent(event)
|
|
// TODO redraw in same thread
|
|
}
|
|
if event.Button != MouseButtonNone {
|
|
s.Redraw()
|
|
}
|
|
}
|
|
|
|
func convertMouseEvent(original *tcell.EventMouse) *MouseEvent {
|
|
x, y := original.Position()
|
|
return &MouseEvent{
|
|
Position: P(x, y),
|
|
Button: convertMouseButton(original.Buttons()),
|
|
Modifiers: original.Modifiers(),
|
|
}
|
|
}
|
|
|
|
func (s *Screen) Redraw() {
|
|
s.redrawCh <- struct{}{}
|
|
}
|
|
|
|
func (s *Screen) eventloop() {
|
|
defer s.handlePanic("panicked while handling event")
|
|
|
|
for evt := s.scr.PollEvent(); evt != nil; evt = s.scr.PollEvent() {
|
|
switch event := evt.(type) {
|
|
case *tcell.EventResize:
|
|
s.Redraw()
|
|
case *tcell.EventKey:
|
|
s.onKeyPressed(event)
|
|
case *tcell.EventMouse:
|
|
s.onMouseEvent(convertMouseEvent(event))
|
|
default:
|
|
s.StopWithError(fmt.Errorf("%#v", event))
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *Screen) drawloop() {
|
|
defer s.handlePanic("panicked while redrawing")
|
|
|
|
for range s.redrawCh {
|
|
w, h := s.scr.Size()
|
|
|
|
if s.buf == nil || s.buf.Width() != w || s.buf.Height() != h {
|
|
s.buf = buf2d.NewBuffer(w, h, DefaultRune)
|
|
} else {
|
|
s.buf.Fill(DefaultRune)
|
|
}
|
|
|
|
rw, rh := s.Root.Layout()
|
|
s.Root.Draw(truncateBuffer(s.buf, rw, rh))
|
|
drawBuffer(s.scr, s.buf)
|
|
}
|
|
}
|
|
|
|
func (s *Screen) handlePanic(msg string) {
|
|
if err := recover(); err != nil {
|
|
if e, ok := err.(error); ok {
|
|
s.StopWithError(adverr.Wrap(msg, e))
|
|
} else {
|
|
s.StopWithError(adverr.Wrap(msg, fmt.Errorf("%v", err)))
|
|
}
|
|
}
|
|
}
|