mirror of
https://github.com/bcicen/ctop.git
synced 2025-12-06 15:16:41 +08:00
refactor all container widgets into subpackage
This commit is contained in:
Bradley Cicenas
@@ -1,297 +0,0 @@
|
||||
package widgets
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
|
||||
"github.com/bcicen/ctop/logging"
|
||||
ui "github.com/gizak/termui"
|
||||
)
|
||||
|
||||
var log = logging.Init()
|
||||
|
||||
const (
|
||||
mark = string('\u25C9')
|
||||
vBar = string('\u25AE')
|
||||
colSpacing = 1
|
||||
statusWidth = 3
|
||||
)
|
||||
|
||||
type CompactGrid struct {
|
||||
ui.GridBufferer
|
||||
Rows []ContainerWidgets
|
||||
X, Y int
|
||||
Width int
|
||||
Height int
|
||||
header *CompactHeader
|
||||
}
|
||||
|
||||
func NewCompactGrid() *CompactGrid {
|
||||
return &CompactGrid{
|
||||
header: NewCompactHeader(),
|
||||
}
|
||||
}
|
||||
|
||||
func (c *CompactGrid) Align() {
|
||||
// Update y recursively
|
||||
c.header.SetY(c.Y)
|
||||
y := c.Y + 1
|
||||
for n, r := range c.Rows {
|
||||
r.SetY(y + n)
|
||||
}
|
||||
// Update width recursively
|
||||
c.header.SetWidth(c.Width)
|
||||
for _, r := range c.Rows {
|
||||
r.SetWidth(c.Width)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *CompactGrid) GetHeight() int { return len(c.Rows) }
|
||||
func (c *CompactGrid) SetX(x int) { c.X = x }
|
||||
func (c *CompactGrid) SetY(y int) { c.Y = y }
|
||||
func (c *CompactGrid) SetWidth(w int) { c.Width = w }
|
||||
|
||||
func (c *CompactGrid) Buffer() ui.Buffer {
|
||||
buf := ui.NewBuffer()
|
||||
buf.Merge(c.header.Buffer())
|
||||
for _, r := range c.Rows {
|
||||
buf.Merge(r.Buffer())
|
||||
}
|
||||
return buf
|
||||
}
|
||||
|
||||
type ContainerWidgets interface {
|
||||
Render(int, int)
|
||||
Reset()
|
||||
Buffer() ui.Buffer
|
||||
Highlight()
|
||||
UnHighlight()
|
||||
SetY(int)
|
||||
SetWidth(int)
|
||||
SetStatus(string)
|
||||
SetCPU(int)
|
||||
SetNet(int64, int64)
|
||||
SetMem(int64, int64, int)
|
||||
}
|
||||
|
||||
type CompactHeader struct {
|
||||
pars []*ui.Par
|
||||
}
|
||||
|
||||
func NewCompactHeader() *CompactHeader {
|
||||
fields := []string{"", "NAME", "CID", "CPU", "MEM", "NET RX/TX"}
|
||||
header := &CompactHeader{}
|
||||
for _, f := range fields {
|
||||
header.pars = append(header.pars, slimHeaderPar(f))
|
||||
}
|
||||
return header
|
||||
}
|
||||
|
||||
// Calculate per-column width, given total width and number of items
|
||||
func calcWidth(width, items int) int {
|
||||
spacing := colSpacing * items
|
||||
return (width - statusWidth - spacing) / items
|
||||
}
|
||||
|
||||
func (c *CompactHeader) SetWidth(w int) {
|
||||
x := 1
|
||||
autoWidth := calcWidth(w, 5)
|
||||
for n, col := range c.pars {
|
||||
if n == 0 {
|
||||
col.SetX(x)
|
||||
col.SetWidth(statusWidth)
|
||||
x += statusWidth
|
||||
continue
|
||||
}
|
||||
col.SetX(x)
|
||||
col.SetWidth(autoWidth)
|
||||
x += autoWidth + colSpacing
|
||||
}
|
||||
}
|
||||
|
||||
func (c *CompactHeader) SetY(y int) {
|
||||
for _, p := range c.pars {
|
||||
p.SetY(y)
|
||||
}
|
||||
}
|
||||
|
||||
func (c *CompactHeader) Buffer() ui.Buffer {
|
||||
buf := ui.NewBuffer()
|
||||
for _, p := range c.pars {
|
||||
buf.Merge(p.Buffer())
|
||||
}
|
||||
return buf
|
||||
}
|
||||
|
||||
type Compact struct {
|
||||
Status *ui.Par
|
||||
Cid *ui.Par
|
||||
Net *ui.Par
|
||||
Name *ui.Par
|
||||
Cpu *ui.Gauge
|
||||
Memory *ui.Gauge
|
||||
}
|
||||
|
||||
func NewCompact(id, name, status string) *Compact {
|
||||
w := &Compact{
|
||||
Status: slimPar(mark),
|
||||
Cid: slimPar(id),
|
||||
Name: slimPar(name),
|
||||
}
|
||||
w.Reset()
|
||||
w.SetStatus(status)
|
||||
return w
|
||||
}
|
||||
|
||||
// Set gauges, counters to default unread values
|
||||
func (w *Compact) Reset() {
|
||||
w.Net = slimPar("-")
|
||||
w.Cpu = slimGauge()
|
||||
w.Memory = slimGauge()
|
||||
}
|
||||
|
||||
func (w *Compact) all() []ui.GridBufferer {
|
||||
return []ui.GridBufferer{
|
||||
w.Status,
|
||||
w.Name,
|
||||
w.Cid,
|
||||
w.Cpu,
|
||||
w.Memory,
|
||||
w.Net,
|
||||
}
|
||||
}
|
||||
|
||||
func (w *Compact) SetY(y int) {
|
||||
for _, col := range w.all() {
|
||||
col.SetY(y)
|
||||
}
|
||||
}
|
||||
|
||||
func (w *Compact) SetWidth(width int) {
|
||||
x := 1
|
||||
autoWidth := calcWidth(width, 5)
|
||||
for n, col := range w.all() {
|
||||
if n == 0 {
|
||||
col.SetX(x)
|
||||
col.SetWidth(statusWidth)
|
||||
x += statusWidth
|
||||
continue
|
||||
}
|
||||
col.SetX(x)
|
||||
col.SetWidth(autoWidth)
|
||||
x += autoWidth + colSpacing
|
||||
}
|
||||
}
|
||||
|
||||
func (w *Compact) Render(y, rowWidth int) {}
|
||||
|
||||
func (w *Compact) Buffer() ui.Buffer {
|
||||
buf := ui.NewBuffer()
|
||||
|
||||
buf.Merge(w.Status.Buffer())
|
||||
buf.Merge(w.Name.Buffer())
|
||||
buf.Merge(w.Cid.Buffer())
|
||||
buf.Merge(w.Cpu.Buffer())
|
||||
buf.Merge(w.Memory.Buffer())
|
||||
buf.Merge(w.Net.Buffer())
|
||||
|
||||
return buf
|
||||
}
|
||||
|
||||
func (w *Compact) Highlight() {
|
||||
w.Name.TextFgColor = ui.ColorDefault
|
||||
w.Name.TextBgColor = ui.ColorWhite
|
||||
}
|
||||
|
||||
func (w *Compact) UnHighlight() {
|
||||
w.Name.TextFgColor = ui.ColorWhite
|
||||
w.Name.TextBgColor = ui.ColorDefault
|
||||
}
|
||||
|
||||
func (w *Compact) SetStatus(val string) {
|
||||
switch val {
|
||||
case "running":
|
||||
w.Status.Text = mark
|
||||
w.Status.TextFgColor = ui.ColorGreen
|
||||
case "exited":
|
||||
w.Status.Text = mark
|
||||
w.Status.TextFgColor = ui.ColorRed
|
||||
case "paused":
|
||||
w.Status.Text = fmt.Sprintf("%s%s", vBar, vBar)
|
||||
w.Status.TextFgColor = ui.ColorDefault
|
||||
default:
|
||||
w.Status.Text = mark
|
||||
w.Status.TextFgColor = ui.ColorRed
|
||||
}
|
||||
}
|
||||
|
||||
func (w *Compact) SetCPU(val int) {
|
||||
w.Cpu.BarColor = colorScale(val)
|
||||
w.Cpu.Label = fmt.Sprintf("%s%%", strconv.Itoa(val))
|
||||
if val < 5 {
|
||||
val = 5
|
||||
w.Cpu.BarColor = ui.ColorBlack
|
||||
}
|
||||
w.Cpu.Percent = val
|
||||
}
|
||||
|
||||
func (w *Compact) SetNet(rx int64, tx int64) {
|
||||
w.Net.Text = fmt.Sprintf("%s / %s", byteFormat(rx), byteFormat(tx))
|
||||
}
|
||||
|
||||
func (w *Compact) SetMem(val int64, limit int64, percent int) {
|
||||
w.Memory.Label = fmt.Sprintf("%s / %s", byteFormat(val), byteFormat(limit))
|
||||
if percent < 5 {
|
||||
percent = 5
|
||||
w.Memory.BarColor = ui.ColorBlack
|
||||
} else {
|
||||
w.Memory.BarColor = ui.ColorGreen
|
||||
}
|
||||
w.Memory.Percent = percent
|
||||
}
|
||||
|
||||
func centerParText(p *ui.Par) {
|
||||
var text string
|
||||
var padding string
|
||||
|
||||
// strip existing left-padding
|
||||
for i, ch := range p.Text {
|
||||
if string(ch) != " " {
|
||||
text = p.Text[i:]
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
padlen := (p.InnerWidth() - len(text)) / 2
|
||||
for i := 0; i < padlen; i++ {
|
||||
padding += " "
|
||||
}
|
||||
p.Text = fmt.Sprintf("%s%s", padding, text)
|
||||
}
|
||||
|
||||
func slimHeaderPar(s string) *ui.Par {
|
||||
p := slimPar(s)
|
||||
p.Y = 2
|
||||
p.Height = 2
|
||||
return p
|
||||
}
|
||||
|
||||
func slimPar(s string) *ui.Par {
|
||||
p := ui.NewPar(s)
|
||||
p.Border = false
|
||||
p.Height = 1
|
||||
p.Width = 20
|
||||
p.TextFgColor = ui.ColorWhite
|
||||
return p
|
||||
}
|
||||
|
||||
func slimGauge() *ui.Gauge {
|
||||
g := ui.NewGauge()
|
||||
g.Height = 1
|
||||
g.Border = false
|
||||
g.Percent = 0
|
||||
g.PaddingBottom = 0
|
||||
g.BarColor = ui.ColorGreen
|
||||
g.Label = "-"
|
||||
return g
|
||||
}
|
||||
@@ -1,68 +0,0 @@
|
||||
package widgets
|
||||
|
||||
import (
|
||||
ui "github.com/gizak/termui"
|
||||
)
|
||||
|
||||
type Expanded struct {
|
||||
Info *ui.Table
|
||||
Net *ExpandedNet
|
||||
Cpu *ExpandedCpu
|
||||
Mem *ExpandedMem
|
||||
}
|
||||
|
||||
func NewExpanded(id, name string) *Expanded {
|
||||
return &Expanded{
|
||||
Info: NewInfo(id, name),
|
||||
Net: NewExpandedNet(),
|
||||
Cpu: NewExpandedCpu(),
|
||||
Mem: NewExpandedMem(),
|
||||
}
|
||||
}
|
||||
|
||||
func NewInfo(id, name string) *ui.Table {
|
||||
p := ui.NewTable()
|
||||
p.Rows = [][]string{
|
||||
[]string{"name", name},
|
||||
[]string{"id", id},
|
||||
}
|
||||
p.Height = 4
|
||||
p.Width = 50
|
||||
p.FgColor = ui.ColorWhite
|
||||
p.Seperator = false
|
||||
return p
|
||||
}
|
||||
|
||||
func (w *Expanded) Buffer() ui.Buffer {
|
||||
return ui.NewBuffer()
|
||||
}
|
||||
|
||||
func (w *Expanded) Reset() {}
|
||||
func (w *Expanded) SetY(_ int) {}
|
||||
func (w *Expanded) SetWidth(_ int) {}
|
||||
func (w *Expanded) Highlight() {}
|
||||
func (w *Expanded) UnHighlight() {}
|
||||
func (w *Expanded) SetStatus(val string) {}
|
||||
|
||||
func (w *Expanded) Render(_, _ int) {
|
||||
ui.Render(w.Info, w.Cpu, w.Mem, w.Net)
|
||||
ui.Handle("/timer/1s", func(ui.Event) {
|
||||
ui.Render(w.Info, w.Cpu, w.Mem, w.Net)
|
||||
})
|
||||
ui.Handle("/sys/kbd/", func(ui.Event) {
|
||||
ui.StopLoop()
|
||||
})
|
||||
ui.Loop()
|
||||
}
|
||||
|
||||
func (w *Expanded) SetCPU(val int) {
|
||||
w.Cpu.Update(val)
|
||||
}
|
||||
|
||||
func (w *Expanded) SetNet(rx int64, tx int64) {
|
||||
w.Net.Update(rx, tx)
|
||||
}
|
||||
|
||||
func (w *Expanded) SetMem(val int64, limit int64, percent int) {
|
||||
w.Mem.Update(int(val), int(limit))
|
||||
}
|
||||
@@ -1,28 +0,0 @@
|
||||
package widgets
|
||||
|
||||
import (
|
||||
ui "github.com/gizak/termui"
|
||||
)
|
||||
|
||||
type ExpandedCpu struct {
|
||||
*ui.LineChart
|
||||
hist FloatHist
|
||||
}
|
||||
|
||||
func NewExpandedCpu() *ExpandedCpu {
|
||||
cpu := &ExpandedCpu{ui.NewLineChart(), NewFloatHist(60)}
|
||||
cpu.BorderLabel = "CPU"
|
||||
cpu.Height = 10
|
||||
cpu.Width = 50
|
||||
cpu.X = 0
|
||||
cpu.Y = 4
|
||||
cpu.Data = cpu.hist.data
|
||||
cpu.DataLabels = cpu.hist.labels
|
||||
cpu.AxesColor = ui.ColorDefault
|
||||
cpu.LineColor = ui.ColorGreen
|
||||
return cpu
|
||||
}
|
||||
|
||||
func (w *ExpandedCpu) Update(val int) {
|
||||
w.hist.Append(float64(val))
|
||||
}
|
||||
@@ -1,40 +0,0 @@
|
||||
package widgets
|
||||
|
||||
import (
|
||||
ui "github.com/gizak/termui"
|
||||
)
|
||||
|
||||
type ExpandedMem struct {
|
||||
*ui.BarChart
|
||||
hist IntHist
|
||||
}
|
||||
|
||||
func NewExpandedMem() *ExpandedMem {
|
||||
mem := &ExpandedMem{
|
||||
ui.NewBarChart(),
|
||||
NewIntHist(8),
|
||||
}
|
||||
mem.BorderLabel = "MEM"
|
||||
mem.Height = 10
|
||||
mem.Width = 50
|
||||
mem.BarWidth = 5
|
||||
mem.BarGap = 1
|
||||
mem.X = 0
|
||||
mem.Y = 14
|
||||
mem.TextColor = ui.ColorDefault
|
||||
mem.Data = mem.hist.data
|
||||
mem.BarColor = ui.ColorGreen
|
||||
mem.DataLabels = mem.hist.labels
|
||||
mem.NumFmt = byteFormatInt
|
||||
return mem
|
||||
}
|
||||
|
||||
func (w *ExpandedMem) Update(val int, limit int) {
|
||||
// implement our own scaling for mem graph
|
||||
if val*4 < limit {
|
||||
w.SetMax(val * 4)
|
||||
} else {
|
||||
w.SetMax(limit)
|
||||
}
|
||||
w.hist.Append(val)
|
||||
}
|
||||
@@ -1,52 +0,0 @@
|
||||
package widgets
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
ui "github.com/gizak/termui"
|
||||
)
|
||||
|
||||
type ExpandedNet struct {
|
||||
*ui.Sparklines
|
||||
rxHist DiffHist
|
||||
txHist DiffHist
|
||||
}
|
||||
|
||||
func NewExpandedNet() *ExpandedNet {
|
||||
net := &ExpandedNet{ui.NewSparklines(), NewDiffHist(50), NewDiffHist(50)}
|
||||
net.BorderLabel = "NET"
|
||||
net.Height = 6
|
||||
net.Width = 50
|
||||
net.X = 0
|
||||
net.Y = 24
|
||||
|
||||
rx := ui.NewSparkline()
|
||||
rx.Title = "RX"
|
||||
rx.Height = 1
|
||||
rx.Data = net.rxHist.data
|
||||
rx.TitleColor = ui.ColorDefault
|
||||
rx.LineColor = ui.ColorGreen
|
||||
|
||||
tx := ui.NewSparkline()
|
||||
tx.Title = "TX"
|
||||
tx.Height = 1
|
||||
tx.Data = net.txHist.data
|
||||
tx.TitleColor = ui.ColorDefault
|
||||
tx.LineColor = ui.ColorYellow
|
||||
|
||||
net.Lines = []ui.Sparkline{rx, tx}
|
||||
return net
|
||||
}
|
||||
|
||||
func (w *ExpandedNet) Update(rx int64, tx int64) {
|
||||
var rate string
|
||||
|
||||
w.rxHist.Append(int(rx))
|
||||
rate = strings.ToLower(byteFormatInt(w.rxHist.Last()))
|
||||
w.Lines[0].Title = fmt.Sprintf("RX [%s/s]", rate)
|
||||
|
||||
w.txHist.Append(int(tx))
|
||||
rate = strings.ToLower(byteFormatInt(w.txHist.Last()))
|
||||
w.Lines[1].Title = fmt.Sprintf("TX [%s/s]", rate)
|
||||
}
|
||||
@@ -1,74 +0,0 @@
|
||||
package widgets
|
||||
|
||||
type IntHist struct {
|
||||
data []int
|
||||
labels []string
|
||||
}
|
||||
|
||||
func NewIntHist(max int) IntHist {
|
||||
return IntHist{
|
||||
data: make([]int, max),
|
||||
labels: make([]string, max),
|
||||
}
|
||||
}
|
||||
|
||||
func (h IntHist) Append(val int) {
|
||||
if len(h.data) == cap(h.data) {
|
||||
h.data = append(h.data[:0], h.data[1:]...)
|
||||
}
|
||||
|
||||
h.data = append(h.data, val)
|
||||
}
|
||||
|
||||
type FloatHist struct {
|
||||
data []float64
|
||||
labels []string
|
||||
}
|
||||
|
||||
func NewFloatHist(max int) FloatHist {
|
||||
return FloatHist{
|
||||
data: make([]float64, max),
|
||||
labels: make([]string, max),
|
||||
}
|
||||
}
|
||||
|
||||
func (h FloatHist) Append(val float64) {
|
||||
if len(h.data) == cap(h.data) {
|
||||
h.data = append(h.data[:0], h.data[1:]...)
|
||||
}
|
||||
h.data = append(h.data, val)
|
||||
}
|
||||
|
||||
type DiffHist struct {
|
||||
data []int // data point derivatives
|
||||
srcData []int // principal input data
|
||||
labels []string
|
||||
}
|
||||
|
||||
func NewDiffHist(max int) DiffHist {
|
||||
return DiffHist{
|
||||
data: make([]int, max),
|
||||
srcData: make([]int, max),
|
||||
labels: make([]string, max),
|
||||
}
|
||||
}
|
||||
|
||||
// return most recent value
|
||||
func (h DiffHist) Last() int {
|
||||
return h.data[len(h.data)-1]
|
||||
}
|
||||
|
||||
func (h DiffHist) Append(val int) {
|
||||
if len(h.data) == cap(h.data) {
|
||||
h.data = append(h.data[:0], h.data[1:]...)
|
||||
}
|
||||
if len(h.srcData) == cap(h.srcData) {
|
||||
h.srcData = append(h.srcData[:0], h.srcData[1:]...)
|
||||
}
|
||||
|
||||
diff := val - h.srcData[len(h.srcData)-1]
|
||||
if diff != val { // skip adding to data if this is the initial update
|
||||
h.data = append(h.data, diff)
|
||||
}
|
||||
h.srcData = append(h.srcData, val)
|
||||
}
|
||||
@@ -1,60 +0,0 @@
|
||||
package widgets
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
|
||||
ui "github.com/gizak/termui"
|
||||
)
|
||||
|
||||
const (
|
||||
kb = 1024
|
||||
mb = kb * 1024
|
||||
gb = mb * 1024
|
||||
)
|
||||
|
||||
// convenience method
|
||||
func byteFormatInt(n int) string {
|
||||
return byteFormat(int64(n))
|
||||
}
|
||||
|
||||
func byteFormat(n int64) string {
|
||||
if n < kb {
|
||||
return fmt.Sprintf("%sB", strconv.FormatInt(n, 10))
|
||||
}
|
||||
if n < mb {
|
||||
n = n / kb
|
||||
return fmt.Sprintf("%sK", strconv.FormatInt(n, 10))
|
||||
}
|
||||
if n < gb {
|
||||
n = n / mb
|
||||
return fmt.Sprintf("%sM", strconv.FormatInt(n, 10))
|
||||
}
|
||||
nf := float64(n) / gb
|
||||
return fmt.Sprintf("%sG", unpadFloat(nf))
|
||||
}
|
||||
|
||||
func unpadFloat(f float64) string {
|
||||
return strconv.FormatFloat(f, 'f', getPrecision(f), 64)
|
||||
}
|
||||
|
||||
func getPrecision(f float64) int {
|
||||
frac := int((f - float64(int(f))) * 100)
|
||||
if frac == 0 {
|
||||
return 0
|
||||
}
|
||||
if frac%10 == 0 {
|
||||
return 1
|
||||
}
|
||||
return 2 // default precision
|
||||
}
|
||||
|
||||
func colorScale(n int) ui.Attribute {
|
||||
if n > 70 {
|
||||
return ui.ColorRed
|
||||
}
|
||||
if n > 30 {
|
||||
return ui.ColorYellow
|
||||
}
|
||||
return ui.ColorGreen
|
||||
}
|
||||
Reference in New Issue
Block a user