2018-12-28 13:59:09 +00:00
|
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
// Copyright © 2018 WireGuard LLC. All Rights Reserved.
|
|
|
|
|
|
|
|
import Cocoa
|
|
|
|
|
|
|
|
class StatusMenu: NSMenu {
|
|
|
|
|
|
|
|
let tunnelsManager: TunnelsManager
|
2018-12-29 13:14:29 +00:00
|
|
|
var tunnelStatusObservers = [AnyObject]()
|
|
|
|
|
2018-12-29 14:09:14 +00:00
|
|
|
var statusMenuItem: NSMenuItem?
|
|
|
|
var networksMenuItem: NSMenuItem?
|
2019-01-10 09:21:20 +00:00
|
|
|
var firstTunnelMenuItemIndex = 0
|
|
|
|
var numberOfTunnelMenuItems = 0
|
2018-12-28 13:59:09 +00:00
|
|
|
|
2019-01-04 13:03:46 +00:00
|
|
|
var manageTunnelsRootVC: ManageTunnelsRootViewController?
|
2019-01-03 14:13:52 +00:00
|
|
|
lazy var manageTunnelsWindow: NSWindow = {
|
2019-01-04 13:03:46 +00:00
|
|
|
manageTunnelsRootVC = ManageTunnelsRootViewController(tunnelsManager: tunnelsManager)
|
|
|
|
let window = NSWindow(contentViewController: manageTunnelsRootVC!)
|
2019-01-03 17:37:53 +00:00
|
|
|
window.title = tr("macWindowTitleManageTunnels")
|
2019-01-10 10:18:56 +00:00
|
|
|
window.setContentSize(NSSize(width: 800, height: 480))
|
2019-01-03 14:13:52 +00:00
|
|
|
window.setFrameAutosaveName(NSWindow.FrameAutosaveName("ManageTunnelsWindow")) // Auto-save window position and size
|
|
|
|
return window
|
|
|
|
}()
|
|
|
|
|
2018-12-28 13:59:09 +00:00
|
|
|
init(tunnelsManager: TunnelsManager) {
|
|
|
|
self.tunnelsManager = tunnelsManager
|
|
|
|
super.init(title: "WireGuard Status Bar Menu")
|
2018-12-29 14:09:14 +00:00
|
|
|
|
|
|
|
addStatusMenuItems()
|
|
|
|
addItem(NSMenuItem.separator())
|
|
|
|
for index in 0 ..< tunnelsManager.numberOfTunnels() {
|
|
|
|
let isUpdated = updateStatusMenuItems(with: tunnelsManager.tunnel(at: index), ignoreInactive: true)
|
|
|
|
if isUpdated {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-29 10:03:41 +00:00
|
|
|
firstTunnelMenuItemIndex = numberOfItems
|
|
|
|
let isAdded = addTunnelMenuItems()
|
|
|
|
if isAdded {
|
|
|
|
addItem(NSMenuItem.separator())
|
|
|
|
}
|
2018-12-28 19:12:02 +00:00
|
|
|
addTunnelManagementItems()
|
2019-01-08 21:44:59 +00:00
|
|
|
addItem(NSMenuItem.separator())
|
|
|
|
addApplicationItems()
|
2018-12-28 13:59:09 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
required init(coder decoder: NSCoder) {
|
|
|
|
fatalError("init(coder:) has not been implemented")
|
|
|
|
}
|
|
|
|
|
2018-12-29 14:09:14 +00:00
|
|
|
func addStatusMenuItems() {
|
2019-01-03 12:19:32 +00:00
|
|
|
let statusTitle = tr(format: "macStatus (%@)", tr("tunnelStatusInactive"))
|
2018-12-29 14:09:14 +00:00
|
|
|
let statusMenuItem = NSMenuItem(title: statusTitle, action: #selector(manageTunnelsClicked), keyEquivalent: "")
|
|
|
|
statusMenuItem.isEnabled = false
|
|
|
|
addItem(statusMenuItem)
|
|
|
|
let networksMenuItem = NSMenuItem(title: tr("macMenuNetworksInactive"), action: #selector(manageTunnelsClicked), keyEquivalent: "")
|
|
|
|
networksMenuItem.isEnabled = false
|
|
|
|
addItem(networksMenuItem)
|
|
|
|
self.statusMenuItem = statusMenuItem
|
|
|
|
self.networksMenuItem = networksMenuItem
|
|
|
|
}
|
|
|
|
|
|
|
|
@discardableResult
|
2019-01-10 09:21:20 +00:00
|
|
|
//swiftlint:disable:next cyclomatic_complexity
|
2018-12-29 14:09:14 +00:00
|
|
|
func updateStatusMenuItems(with tunnel: TunnelContainer, ignoreInactive: Bool) -> Bool {
|
|
|
|
guard let statusMenuItem = statusMenuItem, let networksMenuItem = networksMenuItem else { return false }
|
|
|
|
var statusText: String
|
|
|
|
|
|
|
|
switch tunnel.status {
|
|
|
|
case .waiting:
|
|
|
|
return false
|
|
|
|
case .inactive:
|
|
|
|
if ignoreInactive {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
statusText = tr("tunnelStatusInactive")
|
|
|
|
case .activating:
|
|
|
|
statusText = tr("tunnelStatusActivating")
|
|
|
|
case .active:
|
|
|
|
statusText = tr("tunnelStatusActive")
|
|
|
|
case .deactivating:
|
|
|
|
statusText = tr("tunnelStatusDeactivating")
|
|
|
|
case .reasserting:
|
|
|
|
statusText = tr("tunnelStatusReasserting")
|
|
|
|
case .restarting:
|
|
|
|
statusText = tr("tunnelStatusRestarting")
|
|
|
|
}
|
|
|
|
|
2019-01-03 12:19:32 +00:00
|
|
|
statusMenuItem.title = tr(format: "macStatus (%@)", statusText)
|
2018-12-29 14:09:14 +00:00
|
|
|
|
2019-01-08 21:19:46 +00:00
|
|
|
if tunnel.status == .inactive {
|
|
|
|
networksMenuItem.title = tr("macMenuNetworksInactive")
|
2018-12-29 14:09:14 +00:00
|
|
|
} else {
|
2019-01-08 21:19:46 +00:00
|
|
|
let addresses = tunnel.tunnelConfiguration?.interface.addresses ?? []
|
|
|
|
let addressesString = addresses.map { $0.stringRepresentation }.joined(separator: ", ")
|
|
|
|
if addressesString.isEmpty {
|
|
|
|
networksMenuItem.title = tr("macMenuNetworksNone")
|
|
|
|
} else {
|
|
|
|
networksMenuItem.title = tr(format: "macMenuNetworks (%@)", addressesString)
|
|
|
|
}
|
2018-12-29 14:09:14 +00:00
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
2018-12-29 10:03:41 +00:00
|
|
|
func addTunnelMenuItems() -> Bool {
|
|
|
|
let numberOfTunnels = tunnelsManager.numberOfTunnels()
|
2018-12-28 13:59:09 +00:00
|
|
|
for index in 0 ..< tunnelsManager.numberOfTunnels() {
|
|
|
|
let tunnel = tunnelsManager.tunnel(at: index)
|
2018-12-29 13:14:29 +00:00
|
|
|
insertTunnelMenuItem(for: tunnel, at: numberOfTunnelMenuItems)
|
2018-12-28 13:59:09 +00:00
|
|
|
}
|
2018-12-29 10:03:41 +00:00
|
|
|
return numberOfTunnels > 0
|
|
|
|
}
|
|
|
|
|
2018-12-28 19:12:02 +00:00
|
|
|
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)
|
|
|
|
}
|
|
|
|
|
2019-01-08 21:44:59 +00:00
|
|
|
func addApplicationItems() {
|
|
|
|
let quitItem = NSMenuItem(title: tr("macMenuQuit"), action: #selector(NSApplication.terminate), keyEquivalent: "")
|
|
|
|
quitItem.target = NSApp
|
|
|
|
addItem(quitItem)
|
|
|
|
}
|
|
|
|
|
2018-12-29 13:14:29 +00:00
|
|
|
@objc func tunnelClicked(sender: AnyObject) {
|
|
|
|
guard let tunnelMenuItem = sender as? NSMenuItem else { return }
|
|
|
|
guard let tunnel = tunnelMenuItem.representedObject as? TunnelContainer else { return }
|
|
|
|
if tunnelMenuItem.state == .off {
|
|
|
|
tunnelsManager.startActivation(of: tunnel)
|
|
|
|
} else {
|
|
|
|
tunnelsManager.startDeactivation(of: tunnel)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-28 19:12:02 +00:00
|
|
|
@objc func manageTunnelsClicked() {
|
2019-01-01 19:37:46 +00:00
|
|
|
NSApp.activate(ignoringOtherApps: true)
|
2019-01-03 14:13:52 +00:00
|
|
|
manageTunnelsWindow.makeKeyAndOrderFront(self)
|
2018-12-28 19:12:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
@objc func importTunnelsClicked() {
|
2019-01-03 14:13:52 +00:00
|
|
|
NSApp.activate(ignoringOtherApps: true)
|
|
|
|
manageTunnelsWindow.makeKeyAndOrderFront(self)
|
2019-01-05 13:46:16 +00:00
|
|
|
ImportPanelPresenter.presentImportPanel(tunnelsManager: tunnelsManager, sourceVC: manageTunnelsRootVC!)
|
2018-12-29 10:03:41 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-29 13:14:29 +00:00
|
|
|
extension StatusMenu {
|
|
|
|
func insertTunnelMenuItem(for tunnel: TunnelContainer, at tunnelIndex: Int) {
|
|
|
|
let menuItem = NSMenuItem(title: tunnel.name, action: #selector(tunnelClicked(sender:)), keyEquivalent: "")
|
|
|
|
menuItem.target = self
|
|
|
|
menuItem.representedObject = tunnel
|
|
|
|
updateTunnelMenuItem(menuItem)
|
2018-12-29 14:09:14 +00:00
|
|
|
let statusObservationToken = tunnel.observe(\.status) { [weak self] tunnel, _ in
|
2018-12-29 13:14:29 +00:00
|
|
|
updateTunnelMenuItem(menuItem)
|
2018-12-29 14:09:14 +00:00
|
|
|
self?.updateStatusMenuItems(with: tunnel, ignoreInactive: false)
|
2018-12-29 13:14:29 +00:00
|
|
|
}
|
|
|
|
tunnelStatusObservers.insert(statusObservationToken, at: tunnelIndex)
|
|
|
|
insertItem(menuItem, at: firstTunnelMenuItemIndex + tunnelIndex)
|
|
|
|
if numberOfTunnelMenuItems == 0 {
|
|
|
|
insertItem(NSMenuItem.separator(), at: firstTunnelMenuItemIndex + tunnelIndex + 1)
|
|
|
|
}
|
|
|
|
numberOfTunnelMenuItems += 1
|
|
|
|
}
|
|
|
|
|
|
|
|
func removeTunnelMenuItem(at tunnelIndex: Int) {
|
|
|
|
removeItem(at: firstTunnelMenuItemIndex + tunnelIndex)
|
|
|
|
tunnelStatusObservers.remove(at: tunnelIndex)
|
|
|
|
numberOfTunnelMenuItems -= 1
|
|
|
|
if numberOfTunnelMenuItems == 0 {
|
|
|
|
if let firstItem = item(at: firstTunnelMenuItemIndex), firstItem.isSeparatorItem {
|
|
|
|
removeItem(at: firstTunnelMenuItemIndex)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func moveTunnelMenuItem(from oldTunnelIndex: Int, to newTunnelIndex: Int) {
|
|
|
|
let oldMenuItem = item(at: firstTunnelMenuItemIndex + oldTunnelIndex)!
|
|
|
|
let oldMenuItemTitle = oldMenuItem.title
|
|
|
|
let oldMenuItemTunnel = oldMenuItem.representedObject
|
|
|
|
removeItem(at: firstTunnelMenuItemIndex + oldTunnelIndex)
|
|
|
|
let menuItem = NSMenuItem(title: oldMenuItemTitle, action: #selector(tunnelClicked(sender:)), keyEquivalent: "")
|
|
|
|
menuItem.target = self
|
|
|
|
menuItem.representedObject = oldMenuItemTunnel
|
|
|
|
insertItem(menuItem, at: firstTunnelMenuItemIndex + newTunnelIndex)
|
|
|
|
let statusObserver = tunnelStatusObservers.remove(at: oldTunnelIndex)
|
|
|
|
tunnelStatusObservers.insert(statusObserver, at: newTunnelIndex)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private func updateTunnelMenuItem(_ tunnelMenuItem: NSMenuItem) {
|
|
|
|
guard let tunnel = tunnelMenuItem.representedObject as? TunnelContainer else { return }
|
|
|
|
tunnelMenuItem.title = tunnel.name
|
|
|
|
let shouldShowCheckmark = (tunnel.status != .inactive && tunnel.status != .deactivating)
|
|
|
|
tunnelMenuItem.state = shouldShowCheckmark ? .on : .off
|
|
|
|
}
|
|
|
|
|
2018-12-29 10:03:41 +00:00
|
|
|
extension StatusMenu: TunnelsManagerListDelegate {
|
|
|
|
func tunnelAdded(at index: Int) {
|
|
|
|
let tunnel = tunnelsManager.tunnel(at: index)
|
2018-12-29 13:14:29 +00:00
|
|
|
insertTunnelMenuItem(for: tunnel, at: index)
|
2019-01-04 13:03:46 +00:00
|
|
|
manageTunnelsRootVC?.tunnelsListVC?.tunnelAdded(at: index)
|
2018-12-29 10:03:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func tunnelModified(at index: Int) {
|
2018-12-29 13:14:29 +00:00
|
|
|
if let tunnelMenuItem = item(at: firstTunnelMenuItemIndex + index) {
|
|
|
|
updateTunnelMenuItem(tunnelMenuItem)
|
2018-12-29 10:03:41 +00:00
|
|
|
}
|
2019-01-04 13:03:46 +00:00
|
|
|
manageTunnelsRootVC?.tunnelsListVC?.tunnelModified(at: index)
|
2018-12-29 10:03:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func tunnelMoved(from oldIndex: Int, to newIndex: Int) {
|
2018-12-29 13:14:29 +00:00
|
|
|
moveTunnelMenuItem(from: oldIndex, to: newIndex)
|
2019-01-04 13:03:46 +00:00
|
|
|
manageTunnelsRootVC?.tunnelsListVC?.tunnelMoved(from: oldIndex, to: newIndex)
|
2018-12-29 10:03:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func tunnelRemoved(at index: Int) {
|
2018-12-29 13:14:29 +00:00
|
|
|
removeTunnelMenuItem(at: index)
|
2019-01-04 13:03:46 +00:00
|
|
|
manageTunnelsRootVC?.tunnelsListVC?.tunnelRemoved(at: index)
|
2018-12-28 19:12:02 +00:00
|
|
|
}
|
2018-12-28 13:59:09 +00:00
|
|
|
}
|
2019-01-05 08:56:20 +00:00
|
|
|
|
|
|
|
extension StatusMenu: TunnelsManagerActivationDelegate {
|
|
|
|
func tunnelActivationAttemptFailed(tunnel: TunnelContainer, error: TunnelsManagerActivationAttemptError) {
|
|
|
|
if let manageTunnelsRootVC = manageTunnelsRootVC, manageTunnelsWindow.isVisible {
|
|
|
|
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, manageTunnelsWindow.isVisible {
|
|
|
|
ErrorPresenter.showErrorAlert(error: error, from: manageTunnelsRootVC)
|
|
|
|
} else {
|
|
|
|
ErrorPresenter.showErrorAlert(error: error, from: nil)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func tunnelActivationSucceeded(tunnel: TunnelContainer) {
|
|
|
|
// Nothing to do
|
|
|
|
}
|
|
|
|
}
|