summaryrefslogtreecommitdiff
path: root/main.go
blob: b720f98f7c846205145fbd4cd477e61cc431b3a7 (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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package main

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

	"hhvn.uk/hbspbar/common"
	"hhvn.uk/hbspbar/bspc"
	"hhvn.uk/hbspbar/bar"
)

var reloadEvents = []string{
	"monitor_rename",
	"monitor_swap",
	"monitor_focus",
	"desktop_add",
	"desktop_rename",
	"desktop_remove",
	"desktop_swap",
	"desktop_transfer",
	"desktop_focus",
	"desktop_activate",
	"desktop_layout",
	"node_add",
	"node_remove",
	"node_swap",
	"node_transfer" }

func reloadState(current *bspc.State) (*bspc.State, error) {
	state, err := bspc.LoadState()

	if err != nil {
		return current, err
	} else {
		return state, nil
	}
}

func handleEvent(events *bspc.Subscriber, state *bspc.State, bar *bar.Handle) (bool) {
	var event *bspc.Event

	select {
	case event = <-events.Event:
		break
	default:
		return false
	}

	if event.Err != nil {
		common.Error("Couldn't read event: %s\n", event.Err)
		return false
	}

	if strings.HasPrefix(event.Name, "monitor_") {
		id, err := common.Intify(event.Tokens[0])
		if err == nil {
			switch (event.Name) {
			case "monitor_add": bar.Create <- id
			case "monitor_remove": bar.Destroy <- id
			}
		}
	}

	for i := 0; i < len(reloadEvents); i++ {
		if event.Name == reloadEvents[i] {
			return true
		}
	}

	return false
}

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

	state, err := bspc.LoadState()
	if err != nil {
		common.Error("Couldn't load bspwm state: %s\n", err)
		return
	}

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

	barhandle, err := bar.Init(state)
	if err != nil {
		common.Error("Couldn't initialize X: %s\n", err)
		return
	}

	for _, m := range state.Monitors {
		barhandle.Create <- m.ID
	}
	defer bar.Cleanup()

	for {
		if reload := handleEvent(events, state, barhandle); reload {
			state, err = reloadState(state)
			if err != nil {
				common.Error("Couldn't reload bspwm state: %s\n", err)
			}

			barhandle.NewState <- state
		}

		select {
		case s := <- signals:
			common.Error("%s\n", s)
			return
		case err := <- barhandle.Err:
			common.Error("%s\n", err)
			return
		default:
		}
	}
}