service: inform UIs it is time to quit so they can kill tray

1 files changed, 19 insertions, 0 deletions

diff --git a/service/ipc_client.go b/service/ipc_client.go
index a6241af2..41e71f22 100644
--- a/service/ipc_client.go
+++ b/service/ipc_client.go
@@ -32,6 +32,7 @@ type NotificationType int
 const (
 	TunnelChangeNotificationType NotificationType = iota
 	TunnelsChangeNotificationType
+	ManagerStoppingNotificationType
 )
 
 var rpcClient *rpc.Client
@@ -48,6 +49,12 @@ type TunnelsChangeCallback struct {
 
 var tunnelsChangeCallbacks = make(map[*TunnelsChangeCallback]bool)
 
+type ManagerStoppingCallback struct {
+	cb func()
+}
+
+var managerStoppingCallbacks = make(map[*ManagerStoppingCallback]bool)
+
 func InitializeIPCClient(reader *os.File, writer *os.File, events *os.File) {
 	rpcClient = rpc.NewClient(&pipeRWC{reader, writer})
 	go func() {
@@ -95,6 +102,10 @@ func InitializeIPCClient(reader *os.File, writer *os.File, events *os.File) {
 				for cb := range tunnelsChangeCallbacks {
 					cb.cb()
 				}
+			case ManagerStoppingNotificationType:
+				for cb := range managerStoppingCallbacks {
+					cb.cb()
+				}
 			}
 		}
 	}()
@@ -181,3 +192,11 @@ func IPCClientRegisterTunnelsChange(cb func()) *TunnelsChangeCallback {
 func (cb *TunnelsChangeCallback) Unregister() {
 	delete(tunnelsChangeCallbacks, cb)
 }
+func IPCClientRegisterManagerStopping(cb func()) *ManagerStoppingCallback {
+	s := &ManagerStoppingCallback{cb}
+	managerStoppingCallbacks[s] = true
+	return s
+}
+func (cb *ManagerStoppingCallback) Unregister() {
+	delete(managerStoppingCallbacks, cb)
+}