status-go-monitor/view.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.
	State *State
}

// 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 {
		// IDEA: I can pass a method instead of a function here
		if err := g.SetKeybinding("", b.Key, b.Mod, b.Handler); err != nil {
			return err
		}
	}
	return nil
}

func (vc *ViewController) CursorUp(g *gocui.Gui, v *gocui.View) error {
	// TODO propper error handling?
	vc.State.SetCurrent(vc.State.GetState().Current - 1)
	return nil
}

func (vc *ViewController) CursorDown(g *gocui.Gui, v *gocui.View) error {
	// TODO propper error handling?
	vc.State.SetCurrent(vc.State.GetState().Current + 1)
	return nil
}

func (vc *ViewController) HandleDelete(g *gocui.Gui, v *gocui.View) error {
	currentPeer := vc.State.GetCurrent()
	err := vc.State.Remove(currentPeer)
	if err != nil {
		return err
	}
	return nil
}