Improvement: Power mod for Linux

bamboohr/calendar.go

@@ -1,7 +1,5 @@
 package bamboohr
 
-import ()
-
 type Calendar struct {
 	Items []Item `xml:"item"`
 }

bamboohr/employee.go

@@ -1,7 +1,5 @@
 package bamboohr
 
-import ()
-
 /*
 * Note: this currently implements the minimum number of fields to fulfill the Away functionality.
 * Undoubtedly there are more fields than this to an employee

jira/issues.go

@@ -1,8 +1,5 @@
 package jira
 
-import (
-)
-
 type Issue struct {
 	Expand string `json:"expand"`
 	ID     string `json:"id"`

jira/search_result.go

@@ -1,7 +1,5 @@
 package jira
 
-import ()
-
 type SearchResult struct {
 	StartAt    int     `json:"startAt"`
 	MaxResults int     `json:"maxResults"`

power/battery.go

@@ -1,3 +1,5 @@
+// +build !linux
+
 package power
 
 import (

power/battery_linux.go

@@ -0,0 +1,118 @@
+// +build linux
+
+package power
+
+import (
+	"fmt"
+	"os/exec"
+	"strconv"
+	"strings"
+
+	"github.com/senorprogrammer/wtf/wtf"
+)
+
+var batteryState string
+
+type Battery struct {
+	args   []string
+	cmd    string
+	result string
+
+	Charge    string
+	Remaining string
+}
+
+func NewBattery() *Battery {
+	var battery *Battery
+	//sh := `upower -i $(upower -e | grep '/battery') | grep --color=never -E "state|to\ full|to\ empty|percentage"`
+	battery = &Battery{
+		//args: []string{"-c", sh},
+		//cmd:  "/bin/sh",
+	}
+	return battery
+}
+
+/* -------------------- Exported Functions -------------------- */
+
+func (battery *Battery) Refresh() {
+	data := battery.execute()
+	battery.result = battery.parse(data)
+}
+
+func (battery *Battery) String() string {
+	return battery.result
+}
+
+/* -------------------- Unexported Functions -------------------- */
+
+func (battery *Battery) execute() string {
+	cmd := exec.Command("upower", "-e")
+	lines := strings.Split(wtf.ExecuteCommand(cmd), "\n")
+	var target string
+	for _, l := range lines {
+		if strings.Contains(l, "/battery") {
+			target = l
+			break
+		}
+	}
+	cmd = exec.Command("upower", "-i", target)
+	return wtf.ExecuteCommand(cmd)
+}
+
+func (battery *Battery) parse(data string) string {
+	lines := strings.Split(data, "\n")
+	if len(lines) < 2 {
+		return "unknown"
+	}
+	table := make(map[string]string)
+	for _, line := range lines {
+		parts := strings.Split(line, ":")
+		if len(parts) < 2 {
+			continue
+		}
+		table[strings.TrimSpace(parts[0])] = strings.TrimSpace(parts[1])
+	}
+	if s := table["time to empty"]; s == "" {
+		table["time to empty"] = "∞"
+	}
+	str := ""
+	str = str + fmt.Sprintf(" %10s: %s\n", "Charge", battery.formatCharge(table["percentage"]))
+	str = str + fmt.Sprintf(" %10s: %s\n", "Remaining", table["time to empty"])
+	str = str + fmt.Sprintf(" %10s: %s\n", "State", battery.formatState(table["state"]))
+	if s := table["time to full"]; s != "" {
+		str = str + fmt.Sprintf(" %10s: %s\n", "TimeToFull", table["time to full"])
+	}
+	batteryState = table["state"]
+	return str
+}
+
+func (battery *Battery) formatCharge(data string) string {
+	percent, _ := strconv.ParseFloat(strings.Replace(data, "%", "", -1), 32)
+	color := ""
+
+	switch {
+	case percent >= 70:
+		color = "[green]"
+	case percent >= 35:
+		color = "[yellow]"
+	default:
+		color = "[red]"
+	}
+
+	return color + data + "[white]"
+}
+
+func (battery *Battery) formatState(data string) string {
+	color := ""
+
+	switch data {
+	case "charging":
+		color = "[green]"
+	case "discharging":
+		color = "[yellow]"
+	default:
+		color = "[white]"
+	}
+
+	return color + data + "[white]"
+}

power/source.go

@@ -1,3 +1,5 @@
+// +build !linux
+
 package power
 
 import (

power/source_linux.go

@@ -0,0 +1,15 @@
+// +build linux
+
+package power
+
+// powerSource returns the name of the current power source, probably one of
+// "AC Power" or "Battery Power"
+func powerSource() string {
+	switch batteryState {
+	case "charging":
+		return "AC Power"
+	case "discharging":
+		return "Battery Power"
+	}
+	return batteryState
+}

todo/item.go

@@ -1,7 +1,5 @@
 package todo
 
-import ()
-
 type Item struct {
 	Checked bool
 	Text    string

wtf/position.go

@@ -1,7 +1,5 @@
 package wtf
 
-import ()
-
 type Position struct {
 	top    int
 	left   int

wtf_tests/utils_test.go

@@ -3,8 +3,8 @@ package wtf_tests
 import (
 	"testing"
 
-	. "github.com/senorprogrammer/wtf/wtf"
 	"github.com/go-test/deep"
+	. "github.com/senorprogrammer/wtf/wtf"
 )
 
 /* -------------------- Exclude() -------------------- */