diff options
| -rw-r--r-- | go.mod | 5 | ||||
| -rw-r--r-- | go.sum | 2 | ||||
| -rw-r--r-- | statusbar.conf | 15 | ||||
| -rw-r--r-- | statusbar.go | 193 |
4 files changed, 215 insertions, 0 deletions
@@ -0,0 +1,5 @@ +module statusbar + +go 1.14 + +require github.com/BurntSushi/toml v0.3.1 // indirect @@ -0,0 +1,2 @@ +github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ= +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= diff --git a/statusbar.conf b/statusbar.conf new file mode 100644 index 0000000..dc46741 --- /dev/null +++ b/statusbar.conf @@ -0,0 +1,15 @@ +separator = " | " + +[ip] +label = "IP: " +interfaces = [ "eno1" ] +ipv4only = true + +[mounts] +label = "mounts: " +mountpoints = [ "/", "/mnt/storage", "/mnt/storage1" ] + +[music] +labelplaying = "▶ " +labelpaused = "⏸ " +labelstopped = "-" diff --git a/statusbar.go b/statusbar.go new file mode 100644 index 0000000..607b545 --- /dev/null +++ b/statusbar.go @@ -0,0 +1,193 @@ +package main + +import ( + "fmt" + "net" + "syscall" + "strings" + "strconv" + "os" + "os/exec" + "io/ioutil" + "time" + + "github.com/BurntSushi/toml" +) + +type Config struct { + Separator string + IP IPConfig + Mounts MountConfig + Music MusicConfig +} + +type IPConfig struct { + Label string + Interfaces []string + IPv4Only bool `toml:"ipv4only"` +} + +type MountConfig struct { + Label string + MountPoints []string `toml:"mountpoints"` +} + +type MusicConfig struct { + LabelPlaying string `toml:"labelplaying"` + LabelPaused string `toml:"labelpaused"` + LabelStopped string `toml:"labelstopped"` +} + +func diskSizes(mounts []string) string { + var b strings.Builder + var statfs syscall.Statfs_t + const div = float64(1024 * 1024 * 1024) + const unit = "G" + sep := "" + for _, mount := range mounts { + b.WriteString(sep) + err := syscall.Statfs(mount, &statfs) + if err != nil { + panic(err) + } + availBytes := float64(statfs.Bavail * uint64(statfs.Bsize)) + avail := strconv.FormatFloat(availBytes / div, 'f', 2, 64) + b.WriteString(fmt.Sprintf("%v : %v %v", mount, avail, unit)) + sep = " " + } + return b.String() +} + +func ips(interfaces []string, IPv4Only bool) string { + var b strings.Builder + sep := "" + for _, interfaceName := range interfaces { + b.WriteString(sep) + b.WriteString(interfaceName) + b.WriteString(": ") + if len(interfaces) > 1 { + b.WriteString(" [") + } + b.WriteString(ip(interfaceName, IPv4Only)) + if len(interfaces) > 1 { + b.WriteString(" ]") + } + sep = " " + } + return b.String() +} + +func ip(interfaceName string, IPv4Only bool) string { + var b strings.Builder + sep := "" + in, err := net.InterfaceByName(interfaceName) + if err != nil { + panic(err) + } + addrs, err := in.Addrs() + if err != nil { + panic(err) + } + for _, addr := range addrs { + var ip net.IP + switch v := addr.(type) { + case *net.IPAddr: + ip = v.IP + case *net.IPNet: + ip = v.IP + default: + } + if IPv4Only { + if ip.To4() != nil { + b.WriteString(ip.String()) + b.WriteString(sep) + sep = " " + } + } else { + b.WriteString(sep) + b.WriteString(ip.String()) + sep = " " + } + } + return b.String() +} + +func clock() string { + t := time.Now() + return t.Format("Monday 2006-01-02 15:04:05") +} + +func nowPlayingMPD() string { + cmd := exec.Command("mpc", "current") + out, err := cmd.Output() + if err != nil { + panic(err) + } + sOut := string(out) + sOut = strings.ReplaceAll(sOut, "\n", "") + return sOut +} + +// Returns 0 if playing, 1 if paused and 2 if stopped +func statusMPD() int { + cmd := exec.Command("mpc", "status") + out, err := cmd.Output() + if err != nil { + panic(err) + } + sOut := string(out) + if strings.Contains(sOut, "playing") { + return 0 + } else if strings.Contains(sOut, "paused") { + return 1 + } else { + return 2 + } +} + +func main() { + var configPath = "/etc/statusbar.conf" + var content []byte + var err error + if _, err := os.Stat(configPath); err != nil { + content, err = ioutil.ReadAll(os.Stdin) + } else { + content, err = ioutil.ReadFile(configPath) + } + if err != nil { + panic(err) + } + var config Config + _, err = toml.Decode(string(content), &config) + if err != nil { + panic(err) + } + for true { + var b strings.Builder + switch statusMPD() { + case 0: + b.WriteString(config.Music.LabelPlaying) + case 1: + b.WriteString(config.Music.LabelPaused) + case 2: + b.WriteString(config.Music.LabelStopped) + default: + b.WriteString("Unknown MPD status") + } + b.WriteString(nowPlayingMPD()) + b.WriteString(config.Separator) + b.WriteString(config.IP.Label) + b.WriteString("[ ") + b.WriteString(ips(config.IP.Interfaces, config.IP.IPv4Only)) + b.WriteString(" ]") + b.WriteString(config.Separator) + b.WriteString(config.Mounts.Label) + b.WriteString("[ ") + b.WriteString(diskSizes(config.Mounts.MountPoints)) + b.WriteString(" ]") + b.WriteString(config.Separator) + b.WriteString(clock()) + fmt.Println(b.String()) + time.Sleep(10 * time.Second) + } +} |