utility-screen/screen_main.go

79 lines
2.0 KiB
Go

package main
import (
"fmt"
"os/exec"
"strconv"
"strings"
"time"
"gogs.bullercodeworks.com/brian/termbox-util"
"github.com/nsf/termbox-go"
)
type mainScreen struct {
initialized bool
tabIdx int
battStat *termboxUtil.ProgressBar
}
func (screen *mainScreen) handleKeyEvent(event termbox.Event) int {
if event.Type == termbox.EventKey {
if event.Key == termbox.KeyEsc {
return exitScreenIndex
} else if event.Ch == 'r' {
// Refresh data
screen.battStat.SetProgress(getBatteryPct())
} else if event.Ch == 'q' {
return exitScreenIndex
}
}
return mainScreenIndex
}
func (screen *mainScreen) performLayout(style style) {
if !screen.initialized {
w, _ := termbox.Size()
screen.battStat = termboxUtil.CreateProgressBar(100, w-14, 1, style.defaultFg, style.defaultBg)
screen.battStat.SetProgress(getBatteryPct())
}
}
func (screen *mainScreen) drawScreen(style style) {
w, h := termbox.Size()
termboxUtil.DrawStringAtPoint(time.Now().Format("15:04 MST"), 0, 0, style.defaultFg, style.defaultBg)
termboxUtil.DrawStringAtPoint(time.Now().Format("02 Jan 2006"), 0, 1, style.defaultFg, style.defaultBg)
screen.battStat.Draw()
termboxUtil.DrawStringAtPoint(fmt.Sprintf("% 3d%s", screen.battStat.GetPercent(), "%"),
screen.battStat.GetX()+screen.battStat.GetWidth(),
screen.battStat.GetY(), style.defaultFg, style.defaultBg)
quitTxt := "(q)uit"
termboxUtil.DrawStringAtPoint(quitTxt, (w - len(quitTxt)), h-1, style.defaultFg, style.defaultBg)
}
func getBatteryPct() int {
powerApp := "upower"
powerArg1 := "-i"
powerArg2 := "/org/freedesktop/UPower/devices/battery_BAT0"
cmd := exec.Command(powerApp, powerArg1, powerArg2)
stdout, err := cmd.Output()
if err != nil {
return -1
}
tst := string(stdout)
for _, k := range strings.Split(tst, "\n") {
pair := strings.Split(k, ":")
if len(pair) > 1 {
pair[0] = strings.Trim(pair[0], " ")
pair[1] = strings.Trim(pair[1], " %")
if pair[0] == "percentage" {
if pct, err := strconv.Atoi(pair[1]); err == nil {
return pct
}
}
}
}
return -1
}