mirror of https://github.com/bcicen/ctop.git
130 lines
2.4 KiB
Go
130 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/bcicen/ctop/config"
|
|
"github.com/bcicen/ctop/cwidgets/compact"
|
|
"github.com/bcicen/ctop/logging"
|
|
"github.com/bcicen/ctop/widgets"
|
|
ui "github.com/gizak/termui"
|
|
)
|
|
|
|
var (
|
|
build = "none"
|
|
version = "dev-build"
|
|
|
|
log *logging.CTopLogger
|
|
cursor *GridCursor
|
|
cGrid *compact.CompactGrid
|
|
header *widgets.CTopHeader
|
|
|
|
versionStr = fmt.Sprintf("ctop version %v, build %v", version, build)
|
|
)
|
|
|
|
func main() {
|
|
defer panicExit()
|
|
|
|
// parse command line arguments
|
|
var versionFlag = flag.Bool("v", false, "output version information and exit")
|
|
var helpFlag = flag.Bool("h", false, "display this help dialog")
|
|
var filterFlag = flag.String("f", "", "filter containers")
|
|
var activeOnlyFlag = flag.Bool("a", false, "show active containers only")
|
|
var sortFieldFlag = flag.String("s", "", "select container sort field")
|
|
var reverseSortFlag = flag.Bool("r", false, "reverse container sort order")
|
|
var invertFlag = flag.Bool("i", false, "invert default colors")
|
|
flag.Parse()
|
|
|
|
if *versionFlag {
|
|
fmt.Println(versionStr)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if *helpFlag {
|
|
printHelp()
|
|
os.Exit(0)
|
|
}
|
|
|
|
// init logger
|
|
log = logging.Init()
|
|
|
|
// init global config
|
|
config.Init()
|
|
|
|
// override default config values with command line flags
|
|
if *filterFlag != "" {
|
|
config.Update("filterStr", *filterFlag)
|
|
}
|
|
|
|
if *activeOnlyFlag {
|
|
config.Toggle("allContainers")
|
|
}
|
|
|
|
if *sortFieldFlag != "" {
|
|
validSort(*sortFieldFlag)
|
|
config.Update("sortField", *sortFieldFlag)
|
|
}
|
|
|
|
if *reverseSortFlag {
|
|
config.Toggle("sortReversed")
|
|
}
|
|
|
|
// init ui
|
|
if *invertFlag {
|
|
InvertColorMap()
|
|
}
|
|
ui.ColorMap = ColorMap // override default colormap
|
|
if err := ui.Init(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
defer Shutdown()
|
|
// init grid, cursor, header
|
|
cursor = NewGridCursor()
|
|
cGrid = compact.NewCompactGrid()
|
|
header = widgets.NewCTopHeader()
|
|
|
|
for {
|
|
exit := Display()
|
|
if exit {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func Shutdown() {
|
|
log.Notice("shutting down")
|
|
log.Exit()
|
|
ui.Close()
|
|
}
|
|
|
|
// ensure a given sort field is valid
|
|
func validSort(s string) {
|
|
if _, ok := Sorters[s]; !ok {
|
|
fmt.Printf("invalid sort field: %s\n", s)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func panicExit() {
|
|
if r := recover(); r != nil {
|
|
Shutdown()
|
|
fmt.Printf("panic: %s\n", r)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
var helpMsg = `ctop - container metric viewer
|
|
|
|
usage: ctop [options]
|
|
|
|
options:
|
|
`
|
|
|
|
func printHelp() {
|
|
fmt.Println(helpMsg)
|
|
flag.PrintDefaults()
|
|
}
|