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
|
// SPDX-License-Identifier: MIT
// Copyright © 2018-2019 WireGuard LLC. All Rights Reserved.
import Cocoa
// Keeps track of tunnels and informs the following objects of changes in tunnels:
// - Status menu
// - Status item controller
// - Tunnels list view controller in the Manage Tunnels window
class TunnelsTracker {
weak var statusMenu: StatusMenu? {
didSet {
statusMenu?.currentTunnel = currentTunnel
}
}
weak var statusItemController: StatusItemController? {
didSet {
statusItemController?.currentTunnel = currentTunnel
}
}
weak var manageTunnelsRootVC: ManageTunnelsRootViewController?
private var tunnelsManager: TunnelsManager
private var tunnelStatusObservers = [AnyObject]()
private(set) var currentTunnel: TunnelContainer? {
didSet {
statusMenu?.currentTunnel = currentTunnel
statusItemController?.currentTunnel = currentTunnel
}
}
init(tunnelsManager: TunnelsManager) {
self.tunnelsManager = tunnelsManager
if let waitingTunnel = tunnelsManager.waitingTunnel() {
currentTunnel = waitingTunnel
} else {
for index in 0 ..< tunnelsManager.numberOfTunnels() {
let tunnel = tunnelsManager.tunnel(at: index)
if tunnel.status != .inactive {
currentTunnel = tunnel
break
}
}
}
for index in 0 ..< tunnelsManager.numberOfTunnels() {
let tunnel = tunnelsManager.tunnel(at: index)
let statusObservationToken = observeStatus(of: tunnel)
tunnelStatusObservers.insert(statusObservationToken, at: index)
}
tunnelsManager.tunnelsListDelegate = self
tunnelsManager.activationDelegate = self
}
func observeStatus(of tunnel: TunnelContainer) -> AnyObject {
return tunnel.observe(\.status) { [weak self] tunnel, _ in
guard let self = self else { return }
if tunnel.status == .deactivating || tunnel.status == .inactive {
if self.currentTunnel == tunnel {
if let waitingTunnel = self.tunnelsManager.waitingTunnel() {
self.currentTunnel = waitingTunnel
} else if tunnel.status == .inactive {
self.currentTunnel = nil
}
}
} else {
self.currentTunnel = tunnel
}
}
}
}
extension TunnelsTracker: TunnelsManagerListDelegate {
func tunnelAdded(at index: Int) {
let tunnel = tunnelsManager.tunnel(at: index)
if tunnel.status != .deactivating && tunnel.status != .inactive {
self.currentTunnel = tunnel
}
let statusObservationToken = observeStatus(of: tunnel)
tunnelStatusObservers.insert(statusObservationToken, at: index)
statusMenu?.insertTunnelMenuItem(for: tunnel, at: index)
manageTunnelsRootVC?.tunnelsListVC?.tunnelAdded(at: index)
}
func tunnelModified(at index: Int) {
manageTunnelsRootVC?.tunnelsListVC?.tunnelModified(at: index)
}
func tunnelMoved(from oldIndex: Int, to newIndex: Int) {
let statusObserver = tunnelStatusObservers.remove(at: oldIndex)
tunnelStatusObservers.insert(statusObserver, at: newIndex)
statusMenu?.moveTunnelMenuItem(from: oldIndex, to: newIndex)
manageTunnelsRootVC?.tunnelsListVC?.tunnelMoved(from: oldIndex, to: newIndex)
}
func tunnelRemoved(at index: Int, tunnel: TunnelContainer) {
tunnelStatusObservers.remove(at: index)
statusMenu?.removeTunnelMenuItem(at: index)
manageTunnelsRootVC?.tunnelsListVC?.tunnelRemoved(at: index)
}
}
extension TunnelsTracker: TunnelsManagerActivationDelegate {
func tunnelActivationAttemptFailed(tunnel: TunnelContainer, error: TunnelsManagerActivationAttemptError) {
if let manageTunnelsRootVC = manageTunnelsRootVC, manageTunnelsRootVC.view.window?.isVisible ?? false {
ErrorPresenter.showErrorAlert(error: error, from: manageTunnelsRootVC)
} else {
ErrorPresenter.showErrorAlert(error: error, from: nil)
}
}
func tunnelActivationAttemptSucceeded(tunnel: TunnelContainer) {
// Nothing to do
}
func tunnelActivationFailed(tunnel: TunnelContainer, error: TunnelsManagerActivationError) {
if let manageTunnelsRootVC = manageTunnelsRootVC, manageTunnelsRootVC.view.window?.isVisible ?? false {
ErrorPresenter.showErrorAlert(error: error, from: manageTunnelsRootVC)
} else {
ErrorPresenter.showErrorAlert(error: error, from: nil)
}
}
func tunnelActivationSucceeded(tunnel: TunnelContainer) {
// Nothing to do
}
}
|
