76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// config is ~/.config/dockertop/config: "key = value" lines, rewritten on exit.
|
|
type config struct {
|
|
Sort string // container list
|
|
ProcSort string // process view
|
|
All bool // show stopped containers
|
|
Tree bool // group containers by compose project
|
|
ProcTree bool
|
|
FullCmd bool
|
|
Refresh float64 // seconds
|
|
}
|
|
|
|
var defaults = config{Sort: "CPU", ProcSort: "CPU", All: true, Refresh: 2}
|
|
|
|
func configPath() string {
|
|
dir := os.Getenv("XDG_CONFIG_HOME")
|
|
if dir == "" {
|
|
home, _ := os.UserHomeDir()
|
|
dir = filepath.Join(home, ".config")
|
|
}
|
|
return filepath.Join(dir, "dockertop", "config")
|
|
}
|
|
|
|
func loadConfig() config {
|
|
c := defaults
|
|
b, err := os.ReadFile(configPath())
|
|
if err != nil {
|
|
return c
|
|
}
|
|
for _, line := range strings.Split(string(b), "\n") {
|
|
k, v, ok := strings.Cut(line, "=")
|
|
if !ok || strings.HasPrefix(strings.TrimSpace(line), "#") {
|
|
continue
|
|
}
|
|
k, v = strings.TrimSpace(k), strings.TrimSpace(v)
|
|
switch k {
|
|
case "sort":
|
|
c.Sort = strings.ToUpper(v)
|
|
case "proc_sort":
|
|
c.ProcSort = strings.ToUpper(v)
|
|
case "all":
|
|
c.All = v == "true"
|
|
case "tree":
|
|
c.Tree = v == "true"
|
|
case "proc_tree":
|
|
c.ProcTree = v == "true"
|
|
case "full_cmd":
|
|
c.FullCmd = v == "true"
|
|
case "refresh":
|
|
if f, err := strconv.ParseFloat(v, 64); err == nil && f > 0 {
|
|
c.Refresh = f
|
|
}
|
|
}
|
|
}
|
|
return c
|
|
}
|
|
|
|
func (c config) save() error {
|
|
p := configPath()
|
|
if err := os.MkdirAll(filepath.Dir(p), 0o755); err != nil {
|
|
return err
|
|
}
|
|
return os.WriteFile(p, []byte(fmt.Sprintf(
|
|
"# dockertop configuration; rewritten when dockertop exits. See dockertop(1).\nsort = %s\nproc_sort = %s\nall = %t\ntree = %t\nproc_tree = %t\nfull_cmd = %t\nrefresh = %g\n",
|
|
c.Sort, c.ProcSort, c.All, c.Tree, c.ProcTree, c.FullCmd, c.Refresh)), 0o644)
|
|
}
|