ctop/cwidgets/compact/status.go

package compact

import (
	ui "github.com/gizak/termui"
)

const (
	mark       = string('\u25C9')
	healthMark = string('\u207A')
	vBar       = string('\u25AE') + string('\u25AE')
)

// Status indicator
type Status struct {
	*ui.Block
	status []ui.Cell
	health []ui.Cell
}

func NewStatus() *Status {
	s := &Status{Block: ui.NewBlock()}
	s.Height = 1
	s.Border = false
	s.Set("")
	return s
}

func (s *Status) Buffer() ui.Buffer {
	buf := s.Block.Buffer()
	x := 0
	for _, c := range s.status {
		buf.Set(s.InnerX()+x, s.InnerY(), c)
		x += c.Width()
	}
	for _, c := range s.health {
		buf.Set(s.InnerX()+x, s.InnerY(), c)
		x += c.Width()
	}
	return buf
}

func (s *Status) Set(val string) {
	// defaults
	text := mark
	color := ui.ColorDefault

	switch val {
	case "running":
		color = ui.ThemeAttr("status.ok")
	case "exited":
		color = ui.ThemeAttr("status.danger")
	case "paused":
		text = vBar
	}

	var cells []ui.Cell
	for _, ch := range text {
		cells = append(cells, ui.Cell{Ch: ch, Fg: color})
	}
	s.status = cells
}

func (s *Status) SetHealth(val string) {
	if val == "" {
		return
	}
	color := ui.ColorDefault

	switch val {
	case "healthy":
		color = ui.ThemeAttr("status.ok")
	case "unhealthy":
		color = ui.ThemeAttr("status.danger")
	case "starting":
		color = ui.ThemeAttr("status.warn")
	}

	var cells []ui.Cell
	for _, ch := range healthMark {
		cells = append(cells, ui.Cell{Ch: ch, Fg: color})
	}
	s.health = cells
}