131 lines
2.9 KiB
Go
131 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/jroimartin/gocui"
|
|
)
|
|
|
|
// Struct for more succint definitions of key bindings
|
|
type Binding struct {
|
|
Key interface{} // so both gocui.Key and rune work
|
|
Mod gocui.Modifier
|
|
Handler func(g *gocui.Gui, v *gocui.View) error
|
|
}
|
|
|
|
// Default Gocui views arent granular enough
|
|
// so I'm adding a custom one to have more control.
|
|
type ViewController struct {
|
|
Name string
|
|
Title string
|
|
Placeholder string
|
|
Enabled bool
|
|
Wrap bool
|
|
Cursor bool
|
|
Current bool
|
|
Highlight bool
|
|
OnTop bool
|
|
TopLeft func(int, int) (int, int)
|
|
BotRight func(int, int) (int, int)
|
|
SelBgColor gocui.Attribute
|
|
SelFgColor gocui.Attribute
|
|
Keybindings []Binding
|
|
// Extra field for view state. Might need different name.
|
|
StateCtrl *StateController
|
|
}
|
|
|
|
// To combine all existing views into one
|
|
type ViewManager struct {
|
|
g *gocui.Gui
|
|
views []*ViewController
|
|
}
|
|
|
|
func (m *ViewManager) Layout(g *gocui.Gui) error {
|
|
mx, my := g.Size()
|
|
|
|
for _, cfg := range m.views {
|
|
if !cfg.Enabled {
|
|
continue
|
|
}
|
|
x0, y0 := cfg.TopLeft(mx, my)
|
|
x1, y1 := cfg.BotRight(mx, my)
|
|
|
|
v, err := g.SetView(cfg.Name, x0, y0, x1, y1)
|
|
|
|
// Some settings can be set only once
|
|
if err == gocui.ErrUnknownView {
|
|
cfg.SetKeybindings(g)
|
|
if cfg.Cursor {
|
|
v.SetCursor(0, 0)
|
|
}
|
|
if cfg.Placeholder != "" {
|
|
fmt.Fprintln(v, cfg.Placeholder)
|
|
}
|
|
} else if err != nil {
|
|
log.Panicln(err)
|
|
}
|
|
|
|
v.Title = cfg.Title
|
|
v.Wrap = cfg.Wrap
|
|
v.SelFgColor = cfg.SelFgColor
|
|
v.SelBgColor = cfg.SelBgColor
|
|
v.Highlight = cfg.Highlight
|
|
|
|
if cfg.Current {
|
|
g.SetCurrentView(cfg.Name)
|
|
}
|
|
if cfg.OnTop {
|
|
g.SetViewOnTop(cfg.Name)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (v *ViewController) SetKeybindings(g *gocui.Gui) error {
|
|
for _, b := range v.Keybindings {
|
|
if err := g.SetKeybinding("", b.Key, b.Mod, b.Handler); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (vc *ViewController) Refresh(g *gocui.Gui, v *gocui.View) error {
|
|
// TODO proper error handling?
|
|
vc.StateCtrl.Fetch()
|
|
return nil
|
|
}
|
|
|
|
func (vc *ViewController) CursorUp(g *gocui.Gui, v *gocui.View) error {
|
|
// TODO proper error handling?
|
|
current := vc.StateCtrl.State.GetData().Current
|
|
vc.StateCtrl.State.SetCurrentPeer(current - 1)
|
|
return nil
|
|
}
|
|
|
|
func (vc *ViewController) CursorDown(g *gocui.Gui, v *gocui.View) error {
|
|
// TODO proper error handling?
|
|
current := vc.StateCtrl.State.GetData().Current
|
|
vc.StateCtrl.State.SetCurrentPeer(current + 1)
|
|
return nil
|
|
}
|
|
|
|
func (vc *ViewController) HandleDelete(g *gocui.Gui, v *gocui.View) error {
|
|
currentPeer := vc.StateCtrl.State.GetCurrent()
|
|
err := vc.StateCtrl.RemovePeer(currentPeer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (vc *ViewController) HandleTrust(g *gocui.Gui, v *gocui.View) error {
|
|
currentPeer := vc.StateCtrl.State.GetCurrent()
|
|
err := vc.StateCtrl.TrustPeer(currentPeer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|