summaryrefslogtreecommitdiff
path: root/main.go
blob: 7d7b0371f975ee91dbfcceacfb604642e23fed63 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package main // import "hhvn.uk/hbspbar"

import (
	"os"
	"os/signal"
	"strings"
	"syscall"

	"hhvn.uk/hbspbar/bspc"
	"hhvn.uk/hbspbar/common"
	"hhvn.uk/hbspbar/drw"
	"hhvn.uk/hbspbar/status"
)

func main() {
	signals := make(chan os.Signal, 1)
	signal.Notify(signals,
		syscall.SIGINT,
		syscall.SIGHUP,
		syscall.SIGTERM)

	if drw.InitErr != nil {
		common.Error("Couldn't initialize X: %s\n", drw.InitErr)
		return
	}

	if bspc.InitErr != nil {
		common.Error("Couldn't subscribe to bspwm: %s\n", bspc.InitErr)
		return
	}
	defer bspc.Cleanup()

	state := <- bspc.NewState

	for _, m := range state.Monitors {
		if err := Create(state, m.ID); err != nil {
			common.Error("%s\n", err)
		}
	}

	var blocks []*status.Block = nil

	mainloop: for {
		select {
		case s := <-signals:
			common.Error("%s\n", s)
			break mainloop
		case <- drw.XClosed:
			break mainloop
		case err := <-bspc.EvErr:
			common.Error("Couldn't read event: %s\n", err)
			break mainloop
		case event := <-bspc.Event:
			if strings.HasPrefix(event.Name, "monitor_") {
				id, err := common.Intify(event.Tokens[0])
				if err == nil {
					switch event.Name {
					case "monitor_add":
						fallthrough
					case "monitor_geometry":
						err = Create(state, id)
						if err != nil {
							common.Error("%s\n", err)
						}
					case "monitor_remove":
						err = Destroy(state, id)
						if err != nil {
							common.Error("%s\n", err)
						}
					}
				} else {
					common.Perror("Intify", err)
				}
			}
		case err := <-bspc.StErr:
			common.Error("Couldn't load bspwm state: %s\n", err)
			break mainloop
		case blocks = <- status.NewBlocks:
			select {
			case <- drw.XClosed:
				break mainloop
			default:
				// Will hang if X is closed
				Redraw(state, blocks)
			}
		case state = <- bspc.NewState:
			// Would've used a fallthrough,
			// but go doesn't allow it
			select {
			case <- drw.XClosed:
				break mainloop
			default:
				Redraw(state, blocks)
			}
		}
	}

	Cleanup(state)
}