diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/lemonbar/input.go | 558 |
1 files changed, 0 insertions, 558 deletions
diff --git a/src/lemonbar/input.go b/src/lemonbar/input.go deleted file mode 100644 index b33441b..0000000 --- a/src/lemonbar/input.go +++ /dev/null @@ -1,558 +0,0 @@ -package main - -import "bufio" -import "bytes" -import "errors" -import "fmt" -import "io" -import "os" -import "os/exec" -import "regexp" -import "strconv" -import "strings" -import "time" - -type BatteryLevel struct { - display string - val float64 -} - -type Status struct { - desktops string - networks string - sound string - battery BatteryLevel - cpu string - ram string - est string - utc string -} - -type CPU_times struct { - user int64 - nice int64 - system int64 - idle int64 - iowait int64 - irq int64 - softirq int64 - steal int64 - guest int64 - guest_nice int64 -} - -func get_dates(est chan string, utc chan string, done chan bool) { - for { - select { - case <-done: - return - default: - est_time := time.Now() - utc_time := est_time.UTC() - - est <- est_time.Format("EST: 2006-01-02 15:04:05") - utc <- utc_time.Format("UTC: 15:04:05") - - time.Sleep(time.Second) - } - } -} - -func battery(bat chan BatteryLevel, done chan bool) { - var cur_level BatteryLevel - - full_fd, err := os.Open("/sys/class/power_supply/BAT0/charge_full") - if err != nil { - panic(err) - } - - var full_buf, cur_buf, status_buf [64]byte - n, err := full_fd.Read(full_buf[:]) - if err != nil { - panic(err) - } - full_fd.Close() - - full_val, err := strconv.ParseInt( - strings.TrimSpace(string(full_buf[:n])), 10, 64) - if err != nil { - panic(err) - } - - cur_fd, err := os.Open("/sys/class/power_supply/BAT0/charge_now") - if err != nil { - panic(err) - } - - status_fd, err := os.Open("/sys/class/power_supply/BAT0/status") - if err != nil { - panic(err) - } - - for { - select { - case <-done: - return - default: - _, err := cur_fd.Seek(0, 0) - if err != nil { - panic(err) - } - - n, err := cur_fd.Read(cur_buf[:]) - if err != nil { - panic(err) - } - - cur_val, err := strconv.ParseInt( - strings.TrimSpace(string(cur_buf[:n])), 10, 64) - if err != nil { - panic(err) - } - - _, err = status_fd.Seek(0, 0) - if err != nil { - panic(err) - } - - n, err = status_fd.Read(status_buf[:]) - if err != nil { - panic(err) - } - - status_val := strings.TrimSpace(string(status_buf[:n])) - if err != nil { - panic(err) - } - - var tmp string - if status_val == "Discharging" { - tmp = "-" - } else if status_val == "Charging" { - tmp = "+" - } else { - tmp = "*" - } - cur_level.val = (float64(cur_val) / - float64(full_val)) * - 100.00 - cur_level.display = fmt.Sprintf( - "BAT: %s%3.2f%%", - tmp, cur_level.val) - - bat <- cur_level - - time.Sleep(5 * time.Second) - } - } -} - -func sound_level(sound chan string, done chan bool) { - - percent_re, err := regexp.CompilePOSIX("[0-9]*%") - if err != nil { - panic(err) - } - - for { - select { - case <-done: - return - default: - sound_out, err := exec.Command("pactl", - "list", "sinks").Output() - if err != nil { - panic(err) - } - - tmp := "VOL: " + string(percent_re.Find(sound_out)) - sound <- tmp - time.Sleep(2 * time.Second) - } - } - -} - -func parse_cpu_times(times [][]byte) CPU_times { - var tmp CPU_times - var err error - - tmp.user, err = strconv.ParseInt( - strings.TrimSpace(string(times[0])), 10, 64) - if err != nil { - panic(err) - } - tmp.nice, err = strconv.ParseInt( - strings.TrimSpace(string(times[1])), 10, 64) - if err != nil { - panic(err) - } - - tmp.system, err = strconv.ParseInt( - strings.TrimSpace(string(times[2])), 10, 64) - if err != nil { - panic(err) - } - - tmp.idle, err = strconv.ParseInt( - strings.TrimSpace(string(times[3])), 10, 64) - if err != nil { - panic(err) - } - - tmp.iowait, err = strconv.ParseInt( - strings.TrimSpace(string(times[4])), 10, 64) - if err != nil { - panic(err) - } - - tmp.irq, err = strconv.ParseInt( - strings.TrimSpace(string(times[5])), 10, 64) - if err != nil { - panic(err) - } - - tmp.softirq, err = strconv.ParseInt( - strings.TrimSpace(string(times[6])), 10, 64) - if err != nil { - panic(err) - } - - tmp.steal, err = strconv.ParseInt( - strings.TrimSpace(string(times[7])), 10, 64) - if err != nil { - panic(err) - } - - tmp.guest, err = strconv.ParseInt( - strings.TrimSpace(string(times[8])), 10, 64) - if err != nil { - panic(err) - } - - tmp.guest_nice, err = strconv.ParseInt( - strings.TrimSpace(string(times[9])), 10, 64) - if err != nil { - panic(err) - } - - return tmp -} - -func cpu_levels(cpu chan string, done chan bool) { - var prev, cur CPU_times - - proc_fd, err := os.Open("/proc/stat") - if err != nil { - panic(err) - } - reader := bufio.NewReader(proc_fd) - - cpu_re, err := regexp.CompilePOSIX( - /* user nice system idle iowait irq softirq steal guest guest_nice*/ - "cpu +([0-9]+) +([0-9]+) +([0-9]+) +([0-9]+) +([0-9]+) +([0-9]+) +([0-9]+) +([0-9]+) +([0-9]+) +([0-9]+).*", - ) - if err != nil { - panic(err) - } - - cpu_line, err := reader.ReadBytes('\n') - if err != nil { - return - panic(err) - } - proc_fd.Seek(0, 0) - reader.Reset(proc_fd) - - vals := cpu_re.FindSubmatch(cpu_line) - if vals == nil || len(vals) < 11 { - panic(errors.New("Could not parse /proc/stat")) - } - - prev = parse_cpu_times(vals[1:]) - - for { - select { - case <-done: - return - default: - cpu_line, err := reader.ReadBytes('\n') - if err != nil { - return - panic(err) - } - - vals := cpu_re.FindSubmatch(cpu_line) - if vals == nil || len(vals) < 11 { - panic(errors.New("Could not parse /proc/stat")) - } - - cur = parse_cpu_times(vals[1:]) - - prev_idle := prev.idle + prev.iowait - idle := cur.idle + cur.iowait - - prev_non_idle := prev.user + prev.nice + prev.system + - prev.irq + prev.softirq + prev.steal - non_idle := cur.user + cur.nice + cur.system + - cur.irq + cur.softirq + cur.steal - - prev_total := prev_idle + prev_non_idle - total := idle + non_idle - - total = total - prev_total - idle = idle - prev_idle - - cpu <- fmt.Sprintf("CPU: %.2f%%", - float64(total-idle)/float64(total)) - - prev = cur - proc_fd.Seek(0, 0) - reader.Reset(proc_fd) - time.Sleep(time.Second) - } - } -} - -func ram_usage(ram chan string, done chan bool) { - var mem_total, mem_free, mem_other int64 - var cnt int - - mem_total = -1 - mem_free = -1 - mem_other = -1 - - proc_fd, err := os.Open("/proc/meminfo") - if err != nil { - panic(err) - } - reader := bufio.NewReader(proc_fd) - - for { - select { - case <-done: - return - default: - cnt = 0 - mem_other = 0 - - for { - ram_bytes, err := reader.ReadBytes('\n') - if err == io.EOF { - break - } else if err != nil { - panic(err) - } - ram_line := string(ram_bytes) - - if strings.HasPrefix(ram_line, "MemTotal:") { - cnt++ - tmp := strings.Fields(ram_line) - - if len(tmp) < 2 { - panic(errors.New("Cannot parse /proc/meminfo")) - } - mem_total, err = strconv.ParseInt( - tmp[1], 10, 64) - if err != nil { - panic(err) - } - - } else if strings.HasPrefix(ram_line, "MemFree:") { - cnt++ - tmp := strings.Fields(ram_line) - if len(tmp) < 2 { - panic(errors.New("Cannot parse /proc/meminfo")) - } - mem_free, err = strconv.ParseInt( - tmp[1], 10, 64) - if err != nil { - panic(err) - } - } else if strings.HasPrefix(ram_line, "Cached:") || - strings.HasPrefix(ram_line, "Buffers:") { - - cnt++ - tmp := strings.Fields(ram_line) - if len(tmp) < 2 { - panic(errors.New("Cannot parse /proc/meminfo")) - } - mem_tmp, err := strconv.ParseInt( - tmp[1], 10, 64) - if err != nil { - panic(err) - } - - mem_other += mem_tmp - } - - if cnt >= 4 { - break - } - } - - ram <- fmt.Sprintf("RAM: %d MiB", (mem_total-mem_free- - mem_other)>>10) - - proc_fd.Seek(0, 0) - reader.Reset(proc_fd) - - time.Sleep(4 * time.Second) - } - } -} - -func networks(net chan string, done chan bool) { - - for { - select { - case <-done: - return - default: - network, err := exec.Command("get_network_info.sh").Output() - if err != nil { - panic(err) - } - - net <- strings.TrimSpace(string(network)) - time.Sleep(10 * time.Second) - } - } -} - -func desktops(desk chan string, done chan bool) { - pipe_fn, err := exec.Command("bspc", "subscribe", "desktop", "-f").Output() - if err != nil { - panic(err) - } - pipe_filename := strings.TrimSpace(string(pipe_fn)) - - pipe, err := os.OpenFile( - pipe_filename, - os.O_RDONLY, 0600) - if err != nil { - panic(err) - } - - cleanup := func() { - pipe.Close() - os.Remove(pipe_filename) - } - defer cleanup() - - reader := bufio.NewReader(pipe) - - desk_names_out, err := exec.Command("bspc", "query", "-D", "--names").Output() - if err != nil { - panic(err) - } - - desk_names := strings.Split(string(desk_names_out), "\n") - - for { - select { - case <-done: - return - default: - output := bytes.Buffer{} - if err != nil { - panic(err) - } - - tmp, err := exec.Command("bspc", "query", "-D", - "-d", "focused", "--names").Output() - if err != nil { - panic(err) - } - focused := strings.TrimSpace(string(tmp)) - - for _, d := range desk_names { - output.WriteString(" | ") - if focused == d { - output.WriteString("%{R}") - output.WriteString(d) - output.WriteString("%{R}") - } else { - output.WriteString(d) - } - } - desk <- string(bytes.Trim(output.Bytes(), " |")) - - _, err = reader.ReadBytes('\n') - } - } -} - -func update_bar(s *Status) { - if s.battery.val < 12.5 { - /*012345 - BAT: +*/ - if s.battery.display[5] == '+' { - /* DESK NET SOUND*/ - fmt.Printf("%%{l}%s%%{r}%s | %s |", - s.desktops, s.networks, s.sound) - /* BAT */ - fmt.Printf("%%{B#FF0000}%%{F#000000}") - fmt.Printf(" %s ", s.battery.display) - fmt.Printf("%%{B#000000}%%{F#FFFFFF}") - /* CPU RAM EST UTC*/ - fmt.Printf("| %s | %s | %s | %s\n", - s.cpu, s.ram, s.est, s.utc) - return - } - fmt.Printf("%%{B#FF0000}%%{F#000000}") - } else { - fmt.Printf("%%{B#000000}%%{F#FFFFFF}") - } - /* DESK NET SOUND BAT CPU RAM EST UTC*/ - fmt.Printf("%%{l}%s%%{r}%s | %s | %s | %s | %s | %s | %s\n", - s.desktops, s.networks, s.sound, - s.battery.display, s.cpu, s.ram, s.est, - s.utc) - -} - -func main() { - desk := make(chan string) - net := make(chan string) - sound := make(chan string) - bat := make(chan BatteryLevel) - cpu := make(chan string) - ram := make(chan string) - est := make(chan string) - utc := make(chan string) - done := make(chan bool) - - go get_dates(est, utc, done) - go battery(bat, done) - go sound_level(sound, done) - go cpu_levels(cpu, done) - go ram_usage(ram, done) - go networks(net, done) - go desktops(desk, done) - - var stats Status - stats.battery.val = 12.6 - update_bar(&stats) - - for { - select { - case stats.desktops = <-desk: - update_bar(&stats) - case stats.networks = <-net: - update_bar(&stats) - case stats.sound = <-sound: - update_bar(&stats) - case stats.battery = <-bat: - update_bar(&stats) - case stats.cpu = <-cpu: - update_bar(&stats) - case stats.ram = <-ram: - update_bar(&stats) - case stats.est = <-est: - update_bar(&stats) - case stats.utc = <-utc: - update_bar(&stats) - } - } -} |