lntop/ui/views/menu.go

package views

import (
	"fmt"

	"github.com/edouardparis/lntop/ui/color"
	"github.com/jroimartin/gocui"
)

const (
	MENU        = "menu"
	MENU_HEADER = "menu_header"
	MENU_FOOTER = "menu_footer"
)

var menu = []string{
	"CHANNEL",
	"TRANSAC",
	"ROUTING",
}

type Menu struct {
	view *gocui.View

	cy, oy int
}

func (h Menu) Name() string {
	return MENU
}

func (h *Menu) Wrap(v *gocui.View) View {
	h.view = v
	return h
}

func (h Menu) Origin() (int, int) {
	return 0, h.oy
}

func (h Menu) Cursor() (int, int) {
	return 0, h.cy
}

func (h Menu) Speed() (int, int, int, int) {
	down := 0
	if h.cy+h.oy < len(menu)-1 {
		down = 1
	}
	return 0, 0, down, 1
}

func (h *Menu) SetCursor(x, y int) error {
	err := h.view.SetCursor(x, y)
	if err != nil {
		return err
	}
	h.cy = y
	return nil
}

func (h *Menu) SetOrigin(x, y int) error {
	err := h.view.SetOrigin(x, y)
	if err != nil {
		return err
	}

	h.oy = y
	return nil
}

func (h Menu) Current() string {
	_, y := h.view.Cursor()
	if y < len(menu) {
		switch menu[y] {
		case "CHANNEL":
			return CHANNELS
		case "TRANSAC":
			return TRANSACTIONS
		case "ROUTING":
			return ROUTING
		}
	}
	return ""
}

func (c Menu) Delete(g *gocui.Gui) error {
	err := g.DeleteView(MENU_HEADER)
	if err != nil {
		return err
	}

	err = g.DeleteView(MENU_FOOTER)
	if err != nil {
		return err
	}

	return g.DeleteView(MENU)
}

func (h Menu) Set(g *gocui.Gui, x0, y0, x1, y1 int) error {
	setCursor := false
	header, err := g.SetView(MENU_HEADER, x0-1, y0, x1, y0+2)
	if err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		setCursor = true
	}
	header.Frame = false
	header.BgColor = gocui.ColorGreen
	header.FgColor = gocui.ColorBlack

	header.Clear()
	fmt.Fprintln(header, " MENU")

	h.view, err = g.SetView(MENU, x0-1, y0+1, x1, y1-2)
	if err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		setCursor = true
	}

	h.view.Frame = false
	h.view.Highlight = true
	h.view.SelBgColor = gocui.ColorCyan
	h.view.SelFgColor = gocui.ColorBlack
	if setCursor {
		ox, oy := h.Origin()
		err := h.SetOrigin(ox, oy)
		if err != nil {
			return err
		}

		cx, cy := h.Cursor()
		err = h.SetCursor(cx, cy)
		if err != nil {
			return err
		}
	}

	h.view.Clear()
	for i := range menu {
		fmt.Fprintln(h.view, fmt.Sprintf(" %-9s", menu[i]))
	}
	_, err = g.SetCurrentView(MENU)
	if err != nil {
		return err
	}

	footer, err := g.SetView(MENU_FOOTER, x0-1, y1-2, x1, y1)
	if err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
	}
	footer.Frame = false
	footer.BgColor = gocui.ColorCyan
	footer.FgColor = gocui.ColorBlack
	footer.Clear()
	blackBg := color.Black(color.Background)
	fmt.Fprintln(footer, fmt.Sprintf("%s%s",
		blackBg("F2"), "Close",
	))
	return nil
}

func NewMenu() *Menu { return &Menu{} }