ctop/grid.go

package main

import (
	"fmt"

	"github.com/bcicen/ctop/config"
	"github.com/bcicen/ctop/widgets"
	ui "github.com/gizak/termui"
)

var cGrid = widgets.NewCompactGrid()

func maxRows() int {
	return ui.TermHeight() - 2 - cGrid.Y
}

type Grid struct {
	cursorID   string // id of currently selected container
	cSource    ContainerSource
	containers Containers // sorted slice of containers
	header     *widgets.CTopHeader
}

func NewGrid() *Grid {
	cs := NewDockerContainerSource()
	g := &Grid{
		cSource:    cs,
		containers: cs.All(),
		header:     widgets.NewCTopHeader(),
	}
	return g
}

// Set an initial cursor position, if possible
func (g *Grid) cursorReset() {
	if len(g.containers) > 0 {
		g.cursorID = g.containers[0].id
		g.containers[0].widgets.Highlight()
	}
}

// Return current cursor index
func (g *Grid) cursorIdx() int {
	for n, c := range g.containers {
		if c.id == g.cursorID {
			return n
		}
	}
	return 0
}

func (g *Grid) cursorUp() {
	idx := g.cursorIdx()
	// decrement if possible
	if idx <= 0 {
		return
	}
	active := g.containers[idx]
	next := g.containers[idx-1]

	active.widgets.UnHighlight()
	g.cursorID = next.id
	next.widgets.Highlight()

	ui.Render(cGrid)
}

func (g *Grid) cursorDown() {
	idx := g.cursorIdx()
	// increment if possible
	if idx >= (len(g.containers) - 1) {
		return
	}
	if idx >= maxRows()-1 {
		return
	}
	active := g.containers[idx]
	next := g.containers[idx+1]

	active.widgets.UnHighlight()
	g.cursorID = next.id
	next.widgets.Highlight()
	ui.Render(cGrid)
}

func (g *Grid) redrawRows() {
	// reinit body rows
	cGrid.Rows = []widgets.ContainerWidgets{}

	// build layout
	y := 1
	if config.GetSwitchVal("enableHeader") {
		g.header.SetCount(len(g.containers))
		g.header.SetFilter(config.GetVal("filterStr"))
		g.header.Render()
		y += g.header.Height()
	}
	cGrid.SetY(y)

	var cursorVisible bool
	max := maxRows()
	for n, c := range g.containers.Filter() {
		if n >= max {
			break
		}
		cGrid.Rows = append(cGrid.Rows, c.widgets)
		if c.id == g.cursorID {
			cursorVisible = true
		}
	}

	if !cursorVisible {
		g.cursorReset()
	}

	ui.Clear()
	if config.GetSwitchVal("enableHeader") {
		g.header.Render()
	}
	cGrid.Align()
	ui.Render(cGrid)
}

func (g *Grid) ExpandView() {
	ui.Clear()
	ui.DefaultEvtStream.ResetHandlers()
	defer ui.DefaultEvtStream.ResetHandlers()
	container, _ := g.cSource.Get(g.cursorID)
	container.Expand()
}

func logEvent(e ui.Event) {
	var s string
	s += fmt.Sprintf("Type: %s\n", e.Type)
	s += fmt.Sprintf("Path: %s\n", e.Path)
	s += fmt.Sprintf("From: %s\n", e.From)
	s += fmt.Sprintf("To: %s", e.To)
	log.Debugf("new event:\n%s", s)
}

func Display(g *Grid) bool {
	var menu func()
	var expand bool

	cGrid.SetWidth(ui.TermWidth())
	ui.DefaultEvtStream.Hook(logEvent)

	// initial draw
	g.redrawRows()

	ui.Handle("/sys/kbd/<up>", func(ui.Event) {
		g.cursorUp()
	})
	ui.Handle("/sys/kbd/<down>", func(ui.Event) {
		g.cursorDown()
	})
	ui.Handle("/sys/kbd/<enter>", func(ui.Event) {
		expand = true
		ui.StopLoop()
	})

	ui.Handle("/sys/kbd/a", func(ui.Event) {
		config.Toggle("allContainers")
		g.redrawRows()
	})
	ui.Handle("/sys/kbd/f", func(ui.Event) {
		menu = FilterMenu
		ui.StopLoop()
	})
	ui.Handle("/sys/kbd/h", func(ui.Event) {
		menu = HelpMenu
		ui.StopLoop()
	})
	ui.Handle("/sys/kbd/H", func(ui.Event) {
		config.Toggle("enableHeader")
		g.redrawRows()
	})
	ui.Handle("/sys/kbd/q", func(ui.Event) {
		ui.StopLoop()
	})
	ui.Handle("/sys/kbd/r", func(e ui.Event) {
		config.Toggle("sortReversed")
	})
	ui.Handle("/sys/kbd/s", func(ui.Event) {
		menu = SortMenu
		ui.StopLoop()
	})

	ui.Handle("/timer/1s", func(e ui.Event) {
		g.containers = g.cSource.All() // refresh containers for current sort order
		g.redrawRows()
	})

	ui.Handle("/sys/wnd/resize", func(e ui.Event) {
		g.header.Align()
		cGrid.SetWidth(ui.TermWidth())
		log.Infof("resize: width=%v max-rows=%v", cGrid.Width, maxRows())
		g.redrawRows()
	})

	ui.Loop()
	if menu != nil {
		menu()
		return false
	}
	if expand {
		g.ExpandView()
		return false
	}
	return true
}