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
|
// SPDX-License-Identifier: MIT
// Copyright © 2018 WireGuard LLC. All Rights Reserved.
import Cocoa
class StatusMenu: NSMenu {
let tunnelsManager: TunnelsManager
var firstTunnelMenuItemIndex: Int = 0
init(tunnelsManager: TunnelsManager) {
self.tunnelsManager = tunnelsManager
super.init(title: "WireGuard Status Bar Menu")
firstTunnelMenuItemIndex = numberOfItems
let isAdded = addTunnelMenuItems()
if isAdded {
addItem(NSMenuItem.separator())
}
addTunnelManagementItems()
}
required init(coder decoder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
func addTunnelMenuItems() -> Bool {
let numberOfTunnels = tunnelsManager.numberOfTunnels()
for index in 0 ..< tunnelsManager.numberOfTunnels() {
let tunnel = tunnelsManager.tunnel(at: index)
let menuItem = createTunnelMenuItem(for: tunnel)
addItem(menuItem)
}
return numberOfTunnels > 0
}
func createTunnelMenuItem(for tunnel: TunnelContainer) -> NSMenuItem {
let menuItem = NSMenuItem(title: tunnel.name, action: #selector(tunnelClicked(sender:)), keyEquivalent: "")
menuItem.target = self
menuItem.representedObject = tunnel
return menuItem
}
@objc func tunnelClicked(sender: AnyObject) {
guard let tunnel = sender.representedObject as? TunnelContainer else { return }
print("Tunnel \(tunnel.name) clicked")
}
func addTunnelManagementItems() {
let manageItem = NSMenuItem(title: tr("macMenuManageTunnels"), action: #selector(manageTunnelsClicked), keyEquivalent: "")
manageItem.target = self
addItem(manageItem)
let importItem = NSMenuItem(title: tr("macMenuImportTunnels"), action: #selector(importTunnelsClicked), keyEquivalent: "")
importItem.target = self
addItem(importItem)
}
@objc func manageTunnelsClicked() {
print("Unimplemented")
}
@objc func importTunnelsClicked() {
let openPanel = NSOpenPanel()
openPanel.allowedFileTypes = ["conf", "zip"]
openPanel.begin { [weak tunnelsManager] response in
guard let tunnelsManager = tunnelsManager else { return }
guard response == .OK else { return }
guard let url = openPanel.url else { return }
TunnelImporter.importFromFile(url: url, into: tunnelsManager, sourceVC: nil, errorPresenterType: ErrorPresenter.self)
}
}
}
extension StatusMenu: TunnelsManagerListDelegate {
func tunnelAdded(at index: Int) {
let tunnel = tunnelsManager.tunnel(at: index)
let menuItem = createTunnelMenuItem(for: tunnel)
if tunnelsManager.numberOfTunnels() == 1 {
insertItem(NSMenuItem.separator(), at: firstTunnelMenuItemIndex + index)
}
insertItem(menuItem, at: firstTunnelMenuItemIndex + index)
}
func tunnelModified(at index: Int) {
let tunnel = tunnelsManager.tunnel(at: index)
if let menuItem = item(at: firstTunnelMenuItemIndex + index) {
menuItem.title = tunnel.name
}
}
func tunnelMoved(from oldIndex: Int, to newIndex: Int) {
let tunnel = tunnelsManager.tunnel(at: oldIndex)
let menuItem = createTunnelMenuItem(for: tunnel)
removeItem(at: firstTunnelMenuItemIndex + oldIndex)
insertItem(menuItem, at: firstTunnelMenuItemIndex + newIndex)
}
func tunnelRemoved(at index: Int) {
removeItem(at: firstTunnelMenuItemIndex + index)
if tunnelsManager.numberOfTunnels() == 0 {
if let firstItem = item(at: firstTunnelMenuItemIndex), firstItem.isSeparatorItem {
removeItem(at: firstTunnelMenuItemIndex)
}
}
}
}
|
