passepartout-apple/Passepartout/App/macOS/Scenes/Service/Customization/ProfileCustomizationViewController.swift
2021-01-03 22:28:08 +01:00

259 lines
11 KiB
Swift

//
// ProfileCustomizationViewController.swift
// Passepartout
//
// Created by Davide De Rosa on 6/19/19.
// Copyright (c) 2021 Davide De Rosa. All rights reserved.
//
// https://github.com/passepartoutvpn
//
// This file is part of Passepartout.
//
// Passepartout is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Passepartout is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Passepartout. If not, see <http://www.gnu.org/licenses/>.
//
import Cocoa
import PassepartoutCore
import TunnelKit
protocol ProfileCustomization: class {
var profile: ConnectionProfile? { get set }
var delegate: ProfileCustomizationDelegate? { get set }
}
protocol ProfileCustomizationDelegate: class {
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateEndpointWithAddress newAddress: String?)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateEndpointWithProtocol newEndpointProtocol: EndpointProtocol?)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdatePreset newPreset: InfrastructurePreset)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateConfiguration newConfiguration: OpenVPN.ConfigurationBuilder)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateTrustedNetworks newTrustedNetworks: TrustedNetworks)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateGateway choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateDNS choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateProxy choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings)
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateMTU choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings)
}
class ProfileCustomizationContainerViewController: NSViewController {
@IBOutlet private weak var buttonOK: NSButton!
@IBOutlet private weak var buttonCancel: NSButton!
fileprivate weak var dnsController: DNSViewController?
fileprivate weak var proxyController: ProxyViewController?
var profile: ConnectionProfile?
// MARK: Pending (provider)
private var pendingAddress: String?
private var pendingProtocol: EndpointProtocol?
private var pendingPreset: InfrastructurePreset?
// MARK: Pending (host)
private var pendingParameters: OpenVPN.ConfigurationBuilder?
// MARK: Pending
private var pendingTrustedNetworks: TrustedNetworks?
private var pendingChoices: ProfileNetworkChoices?
private let pendingManualNetworkSettings = ProfileNetworkSettings()
override func viewDidLoad() {
super.viewDidLoad()
buttonOK.title = L10n.Core.Global.ok
buttonCancel.title = L10n.Core.Global.cancel
pendingAddress = (profile as? ProviderConnectionProfile)?.customAddress
pendingProtocol = (profile as? ProviderConnectionProfile)?.customProtocol
pendingPreset = (profile as? ProviderConnectionProfile)?.preset
pendingTrustedNetworks = profile?.trustedNetworks
pendingParameters = (profile as? HostConnectionProfile)?.parameters.sessionConfiguration.builder()
pendingChoices = ProfileNetworkChoices.with(profile: profile)
}
override func prepare(for segue: NSStoryboardSegue, sender: Any?) {
guard let customVC = segue.destinationController as? ProfileCustomizationViewController else {
return
}
customVC.containerController = self
customVC.profile = profile
}
// MARK: Actions
@IBAction private func commitChanges(_ sender: Any?) {
dnsController?.commitManualSettings()
proxyController?.commitManualSettings()
if let providerProfile = profile as? ProviderConnectionProfile {
if let pending = pendingPreset {
providerProfile.presetId = pending.id
}
} else if let hostProfile = profile as? HostConnectionProfile, let pendingParameters = pendingParameters {
var builder = hostProfile.parameters.builder()
builder.sessionConfiguration = pendingParameters.build()
hostProfile.parameters = builder.build()
}
profile?.customAddress = pendingAddress
profile?.customProtocol = pendingProtocol
profile?.trustedNetworks = pendingTrustedNetworks
if let choices = pendingChoices {
let settings = profile?.manualNetworkSettings ?? ProfileNetworkSettings()
if choices.gateway == .manual {
settings.copyGateway(from: pendingManualNetworkSettings)
}
if choices.dns == .manual {
settings.copyDNS(from: pendingManualNetworkSettings)
}
if choices.proxy == .manual {
settings.copyProxy(from: pendingManualNetworkSettings)
}
if choices.mtu == .manual {
settings.copyMTU(from: pendingManualNetworkSettings)
}
profile?.networkChoices = choices
profile?.manualNetworkSettings = settings
}
TransientStore.shared.serialize(withProfiles: true) // customize
if let profile = profile, TransientStore.shared.service.isActiveProfile(ProfileKey(profile)) {
let vpn = GracefulVPN(service: TransientStore.shared.service)
if vpn.isEnabled {
switch vpn.status {
case .connected, .connecting:
let alert = Macros.warning(
L10n.App.Configuration.title,
L10n.App.Configuration.Alerts.Commit.message
)
if alert.presentModally(
withOK: L10n.App.Configuration.Alerts.Commit.Buttons.reconnect,
cancel: L10n.App.Configuration.Alerts.Commit.Buttons.skip) {
vpn.reconnect(completionHandler: nil)
} else {
vpn.reinstall(completionHandler: nil)
}
default:
vpn.reinstall(completionHandler: nil)
}
}
}
dismiss(self)
}
}
extension ProfileCustomizationContainerViewController: ProfileCustomizationDelegate {
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateEndpointWithAddress newAddress: String?) {
pendingAddress = newAddress
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateEndpointWithProtocol newEndpointProtocol: EndpointProtocol?) {
pendingProtocol = newEndpointProtocol
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdatePreset newPreset: InfrastructurePreset) {
pendingPreset = newPreset
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateConfiguration newConfiguration: OpenVPN.ConfigurationBuilder) {
pendingParameters = newConfiguration
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateTrustedNetworks newTrustedNetworks: TrustedNetworks) {
pendingTrustedNetworks = newTrustedNetworks
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateGateway choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings) {
pendingChoices?.gateway = choice
pendingManualNetworkSettings.gatewayPolicies = newSettings.gatewayPolicies
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateDNS choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings) {
pendingChoices?.dns = choice
pendingManualNetworkSettings.dnsSearchDomains = newSettings.dnsSearchDomains
pendingManualNetworkSettings.dnsServers = newSettings.dnsServers
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateProxy choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings) {
pendingChoices?.proxy = choice
pendingManualNetworkSettings.proxyAddress = newSettings.proxyAddress
pendingManualNetworkSettings.proxyPort = newSettings.proxyPort
pendingManualNetworkSettings.proxyBypassDomains = newSettings.proxyBypassDomains
}
func profileCustomization(_ profileCustomization: ProfileCustomization, didUpdateMTU choice: NetworkChoice, withManualSettings newSettings: ProfileNetworkSettings) {
pendingChoices?.mtu = choice
pendingManualNetworkSettings.mtuBytes = newSettings.mtuBytes
}
}
//
class ProfileCustomizationViewController: NSTabViewController {
fileprivate weak var containerController: ProfileCustomizationContainerViewController?
fileprivate var profile: ConnectionProfile? {
didSet {
for item in tabViewItems {
guard let custom = item.viewController as? ProfileCustomization else {
continue
}
custom.profile = profile
custom.delegate = containerController
if let vc = custom as? DNSViewController {
containerController?.dnsController = vc
} else if let vc = custom as? ProxyViewController {
containerController?.proxyController = vc
}
}
}
}
override func viewDidLoad() {
super.viewDidLoad()
let expectedTabs = 7
assert(tabViewItems.count == expectedTabs, "Customization tabs misconfigured (expected \(expectedTabs))")
tabViewItems[0].label = L10n.Core.Endpoint.title
tabViewItems[1].label = L10n.App.Configuration.title
tabViewItems[2].label = L10n.Core.Service.Sections.Trusted.header
tabViewItems[3].label = L10n.Core.NetworkSettings.Gateway.title
tabViewItems[4].label = L10n.Core.NetworkSettings.Dns.title
tabViewItems[5].label = L10n.Core.NetworkSettings.Proxy.title
tabViewItems[6].label = L10n.Core.NetworkSettings.Mtu.title
}
}