passepartout-apple/Passepartout-iOS/Scenes/Organizer/OrganizerViewController.swift

400 lines
13 KiB
Swift
Raw Normal View History

2018-10-11 07:13:19 +00:00
//
// OrganizerViewController.swift
// Passepartout-iOS
//
// Created by Davide De Rosa on 9/2/18.
// Copyright (c) 2018 Davide De Rosa. All rights reserved.
//
// https://github.com/keeshux
//
// 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 UIKit
// XXX: convoluted due to the separation of provider/host profiles
class OrganizerViewController: UITableViewController, TableModelHost {
private let service = TransientStore.shared.service
private var providerProfiles: [ProviderConnectionProfile] = []
private var hostProfiles: [HostConnectionProfile] = []
private var availableProviderNames: [Infrastructure.Name]?
// MARK: TableModelHost
let model: TableModel<SectionType, RowType> = {
let model: TableModel<SectionType, RowType> = TableModel()
model.add(.providers)
model.add(.hosts)
model.add(.about)
model.add(.destruction)
model.setHeader(L10n.Organizer.Sections.Providers.header, for: .providers)
model.setHeader(L10n.Organizer.Sections.Hosts.header, for: .hosts)
model.setFooter(L10n.Organizer.Sections.Providers.footer, for: .providers)
model.setFooter(L10n.Organizer.Sections.Hosts.footer, for: .hosts)
model.set([.openAbout], in: .about)
model.set([.uninstall], in: .destruction)
return model
}()
func reloadModel() {
providerProfiles.removeAll()
hostProfiles.removeAll()
service.profileIds().forEach {
let profile = service.profile(withId: $0)
if let p = profile as? ProviderConnectionProfile {
providerProfiles.append(p)
} else if let p = profile as? HostConnectionProfile {
hostProfiles.append(p)
} else {
fatalError("Unexpected profile type \(type(of: profile))")
}
}
providerProfiles.sort { $0.name.rawValue < $1.name.rawValue }
hostProfiles.sort { $0.title < $1.title }
var providers = [RowType](repeating: .profile, count: providerProfiles.count)
var hosts = [RowType](repeating: .profile, count: hostProfiles.count)
providers.append(.addProvider)
hosts.append(.addHost)
model.set(providers, in: .providers)
model.set(hosts, in: .hosts)
}
// MARK: UIViewController
override func awakeFromNib() {
super.awakeFromNib()
applyMasterTitle(Theme.current)
}
override func viewDidLoad() {
super.viewDidLoad()
title = GroupConstants.App.title
navigationItem.rightBarButtonItem = editButtonItem
Cells.destructive.register(with: tableView)
reloadModel()
tableView.reloadData()
if let ip = selectedIndexPath {
tableView.scrollToRow(at: ip, at: .middle, animated: false)
}
service.delegate = self
}
override func shouldPerformSegue(withIdentifier identifier: String, sender: Any?) -> Bool {
if let cell = sender as? UITableViewCell, let indexPath = tableView.indexPath(for: cell) {
return model.row(at: indexPath) == .profile
}
// fall back to active profile if no selection
return service.hasActiveProfile()
}
override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
let destination = (segue.destination as? UINavigationController)?.topViewController
if let vc = destination as? ServiceViewController {
var selectedProfile: ConnectionProfile?
// XXX: sender can be a cell or a profile
selectedProfile = sender as? ConnectionProfile
if selectedProfile == nil, let cell = sender as? UITableViewCell, let indexPath = tableView.indexPath(for: cell) {
selectedProfile = profile(at: indexPath)
}
assert(selectedProfile != nil, "No selected profile")
vc.profile = selectedProfile
} else if let vc = destination as? Wizard {
if let providerVC = vc as? WizardProviderViewController {
providerVC.availableNames = availableProviderNames ?? []
}
vc.delegate = self
}
}
// MARK: Actions
@IBAction private func about() {
perform(segue: StoryboardSegue.Organizer.aboutSegueIdentifier, sender: nil)
}
private func addNewProvider() {
var names = Set(InfrastructureFactory.shared.allNames)
let createdNames = providerProfiles.map { $0.name }
names.formSymmetricDifference(createdNames)
guard !names.isEmpty else {
let alert = Macros.alert(
L10n.Organizer.Sections.Providers.header,
L10n.Organizer.Alerts.ExhaustedProviders.message
)
alert.addCancelAction(L10n.Global.ok)
present(alert, animated: true, completion: nil)
return
}
availableProviderNames = names.sorted { $0.rawValue < $1.rawValue }
perform(segue: StoryboardSegue.Organizer.addProviderSegueIdentifier)
}
private func addNewHost() {
let alert = Macros.alert(
L10n.Organizer.Sections.Hosts.header,
L10n.Organizer.Alerts.AddHost.message
)
alert.addCancelAction(L10n.Global.ok)
present(alert, animated: true, completion: nil)
}
private func removeProfile(at indexPath: IndexPath) {
let sectionObject = model.section(for: indexPath.section)
let rowProfile = profile(at: indexPath)
switch sectionObject {
case .providers:
providerProfiles.remove(at: indexPath.row)
case .hosts:
hostProfiles.remove(at: indexPath.row)
default:
return
}
// var fallbackSection: SectionType?
let total = providerProfiles.count + hostProfiles.count
// removed all profiles
if total == 0 {
VPN.shared.disconnect(completionHandler: nil)
}
// removed active profile
else if service.isActiveProfile(rowProfile) {
// let anyProvider = providerProfiles.first
// let anyHost = hostProfiles.first
// guard let anyProfile: ConnectionProfile = firstProvider ?? firstHost else {
// fatalError("There must be one profile somewhere")
// }
// fallbackSection = (anyProvider != nil) ? .providers : .hosts
// store.service.activateProfile(only)
VPN.shared.disconnect(completionHandler: nil)
}
tableView.beginUpdates()
model.deleteRow(in: sectionObject, at: indexPath.row)
tableView.deleteRows(at: [indexPath], with: .automatic)
// if let fallbackSection = fallbackSection {
// let section = model.index(ofSection: fallbackSection)
// tableView.reloadRows(at: [IndexPath(row: 0, section: section)], with: .none)
// }
tableView.endUpdates()
let _ = service.removeProfile(rowProfile)
splitViewController?.serviceViewController?.hideProfileIfDeleted()
TransientStore.shared.serialize() // delete
}
private func confirmVpnProfileDeletion() {
let alert = Macros.alert(
L10n.Organizer.Cells.Uninstall.caption,
L10n.Organizer.Alerts.DeleteVpnProfile.message
)
alert.addDefaultAction(L10n.Global.ok) {
VPN.shared.uninstall(completionHandler: nil)
}
alert.addCancelAction(L10n.Global.cancel)
present(alert, animated: true, completion: nil)
}
}
// MARK: -
extension OrganizerViewController {
enum SectionType: Int {
case providers
case hosts
case about
case destruction
}
enum RowType: Int {
case profile
case addProvider
case addHost
case openAbout
case uninstall
}
private var selectedIndexPath: IndexPath? {
guard let active = service.activeProfile?.id else {
return nil
}
if let row = providerProfiles.index(where: { $0.id == active }) {
return IndexPath(row: row, section: 0)
}
if let row = hostProfiles.index(where: { $0.id == active }) {
return IndexPath(row: row, section: 1)
}
return nil
}
override func numberOfSections(in tableView: UITableView) -> Int {
return model.count
}
override func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? {
return model.header(for: section)
}
override func tableView(_ tableView: UITableView, titleForFooterInSection section: Int) -> String? {
return model.footer(for: section)
}
override func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
return model.count(for: section)
}
override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
switch model.row(at: indexPath) {
case .profile:
let cell = Cells.setting.dequeue(from: tableView, for: indexPath)
let rowProfile = profile(at: indexPath)
cell.leftText = rowProfile.title
cell.applyChecked(service.isActiveProfile(rowProfile), Theme.current)
2018-10-11 07:13:19 +00:00
return cell
case .addProvider:
let cell = Cells.setting.dequeue(from: tableView, for: indexPath)
cell.applyAction(Theme.current)
cell.leftText = L10n.Organizer.Cells.AddProvider.caption
return cell
case .addHost:
let cell = Cells.setting.dequeue(from: tableView, for: indexPath)
cell.applyAction(Theme.current)
cell.leftText = L10n.Organizer.Cells.AddHost.caption
return cell
case .openAbout:
let cell = Cells.setting.dequeue(from: tableView, for: indexPath)
cell.leftText = L10n.Organizer.Cells.About.caption(GroupConstants.App.name)
return cell
case .uninstall:
let cell = Cells.destructive.dequeue(from: tableView, for: indexPath)
cell.caption = L10n.Organizer.Cells.Uninstall.caption
return cell
}
}
override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
switch model.row(at: indexPath) {
case .profile:
// selectedProfileId = profile(at: indexPath).id
break
case .addProvider:
addNewProvider()
case .addHost:
addNewHost()
case .openAbout:
about()
case .uninstall:
confirmVpnProfileDeletion()
}
tableView.deselectRow(at: indexPath, animated: true)
}
override func tableView(_ tableView: UITableView, canEditRowAt indexPath: IndexPath) -> Bool {
guard tableView.isEditing else {
return false
}
return model.row(at: indexPath) == .profile
}
override func tableView(_ tableView: UITableView, commit editingStyle: UITableViewCell.EditingStyle, forRowAt indexPath: IndexPath) {
removeProfile(at: indexPath)
}
// MARK: Helpers
private func sectionProfiles(at indexPath: IndexPath) -> [ConnectionProfile] {
let sectionProfiles: [ConnectionProfile]
let sectionObject = model.section(for: indexPath.section)
switch sectionObject {
case .providers:
sectionProfiles = providerProfiles
case .hosts:
sectionProfiles = hostProfiles
default:
fatalError("Unexpected section: \(sectionObject)")
}
guard indexPath.row < sectionProfiles.count else {
fatalError("No profile found at \(indexPath), is it an add cell?")
}
return sectionProfiles
}
private func profile(at indexPath: IndexPath) -> ConnectionProfile {
return sectionProfiles(at: indexPath)[indexPath.row]
}
}
// MARK: -
extension OrganizerViewController: ConnectionServiceDelegate {
func connectionService(didDeactivate profile: ConnectionProfile) {
tableView.reloadData()
}
func connectionService(didActivate profile: ConnectionProfile) {
tableView.reloadData()
}
}
extension OrganizerViewController: WizardDelegate {
func wizard(didCreate profile: ConnectionProfile, withCredentials credentials: Credentials) {
service.addOrReplaceProfile(profile, credentials: credentials)
TransientStore.shared.serialize() // add
reloadModel()
tableView.reloadData()
perform(segue: StoryboardSegue.Organizer.selectProfileSegueIdentifier, sender: profile)
}
}