File: monitor.go

package info (click to toggle)
go-dlib 5.6.0.9%2Bdfsg-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 3,212 kB
  • sloc: ansic: 4,664; xml: 1,456; makefile: 20; sh: 15
file content (156 lines) | stat: -rw-r--r-- 3,054 bytes parent folder | download | duplicates (3)
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
package gsettings

import (
	"path"
	"strings"
	"sync"

	"github.com/godbus/dbus"
)

func addMatch(bus *dbus.Conn, rule string) error {
	err := bus.BusObject().Call("org.freedesktop.DBus.AddMatch", 0, rule).Err
	return err
}

func initBus() (*dbus.Conn, error) {
	bus, err := dbus.SessionBus()
	if err != nil {
		return nil, err
	}

	err = addMatch(bus, "type='signal',sender='ca.desrt.dconf',path='/ca/desrt/dconf/Writer/user',interface='ca.desrt.dconf.Writer',member='Notify'")
	if err != nil {
		return nil, err
	}
	return bus, nil
}

type changedCallback struct {
	all bool
	fns []ChangedCallbackFunc
}

type ChangedCallbackFunc func(key string)

func toPath(schemaOrPath string) string {
	if schemaOrPath[0] == '/' {
		// is path
		if schemaOrPath[len(schemaOrPath)-1] == '/' {
			schemaOrPath = schemaOrPath[:len(schemaOrPath)-1]
		}
		return schemaOrPath
	}
	// is schema id
	return "/" + strings.Replace(schemaOrPath, ".", "/", -1)
}

var changedCallbackMap map[string]*changedCallback
var changedCallbackMapMu sync.RWMutex

func ConnectChanged(schemaOrPath, key string, fn ChangedCallbackFunc) {
	if schemaOrPath == "" || key == "" {
		return
	}
	var all bool
	if key == "*" {
		all = true
	}

	keyPath := toPath(schemaOrPath)
	if !all {
		keyPath += "/" + key
	}

	//log.Println("ConnectChanged", keyPath)

	changedCallbackMapMu.Lock()
	if changedCallbackMap == nil {
		changedCallbackMap = make(map[string]*changedCallback)
	}

	callback, ok := changedCallbackMap[keyPath]
	if ok {
		if callback.all == all {
			callback.fns = append(callback.fns, fn)
		}
	} else {
		changedCallbackMap[keyPath] = &changedCallback{
			all: all,
			fns: []ChangedCallbackFunc{fn},
		}
	}
	changedCallbackMapMu.Unlock()
}

var started bool
var startedMu sync.Mutex

func StartMonitor() error {
	startedMu.Lock()
	defer startedMu.Unlock()
	if started {
		return nil
	}

	bus, err := initBus()
	if err != nil {
		return err
	}
	signalCh := make(chan *dbus.Signal, 10)
	bus.Signal(signalCh)

	go func() {
		for signal := range signalCh {
			if signal.Name == "ca.desrt.dconf.Writer.Notify" &&
				signal.Path == "/ca/desrt/dconf/Writer/user" {
				if len(signal.Body) == 3 {
					keyPath, ok := signal.Body[0].(string)
					if !ok {
						continue
					}
					subPathList, ok := signal.Body[1].([]string)
					if !ok || len(subPathList) == 0 {
						handleSignal(keyPath)
					} else {
						for _, subPath := range subPathList {
							handleSignal(keyPath + subPath)
						}
					}
				}
			}
		}

	}()

	started = true
	return nil
}

func handleSignal(keyPath string) {
	parent, key := path.Split(keyPath)
	if parent == "" || key == "" {
		return
	}
	if parent[len(parent)-1] == '/' {
		parent = parent[:len(parent)-1]
	}

	changedCallbackMapMu.RLock()

	callback := changedCallbackMap[parent]
	if callback != nil && callback.all {
		for _, fn := range callback.fns {
			go fn(key)
		}
	}

	callback = changedCallbackMap[keyPath]
	if callback != nil && !callback.all {
		for _, fn := range callback.fns {
			go fn(key)
		}
	}

	changedCallbackMapMu.RUnlock()
}