// // TunnelConfigurationTableViewController.swift // WireGuard // // Created by Jeroen Leenarts on 24-05-18. // Copyright © 2018 Jason A. Donenfeld . All rights reserved. // import UIKit import CoreData import BNRCoreDataStack import PromiseKit protocol TunnelConfigurationTableViewControllerDelegate: class { func didSave(tunnel: Tunnel, tunnelConfigurationTableViewController: TunnelConfigurationTableViewController) } class TunnelConfigurationTableViewController: UITableViewController { @IBOutlet weak var saveButton: UIBarButtonItem! private var viewContext: NSManagedObjectContext! private weak var delegate: TunnelConfigurationTableViewControllerDelegate? private var tunnel: Tunnel! func configure(context: NSManagedObjectContext, delegate: TunnelConfigurationTableViewControllerDelegate? = nil, tunnel: Tunnel? = nil) { viewContext = context self.delegate = delegate self.tunnel = tunnel ?? generateNewTunnelConfig() } private func generateNewTunnelConfig() -> Tunnel { var tunnel: Tunnel! = nil viewContext.performAndWait { tunnel = Tunnel(context: viewContext) let peer = Peer(context: viewContext) peer.allowedIPs = "0.0.0.0/0, ::/0" tunnel.addToPeers(peer) let interface = Interface(context: viewContext) tunnel.interface = interface } return tunnel } @IBAction func addPeer(_ sender: Any) { if let moc = tunnel.managedObjectContext { tableView.beginUpdates() let insertedAt = IndexPath(row: tunnel.peers?.count ?? 0, section: 1) tableView.insertRows(at: [insertedAt], with: .automatic) let peer = Peer(context: moc) peer.allowedIPs = "0.0.0.0/0, ::/0" tunnel.addToPeers(peer) tableView.endUpdates() tableView.scrollToRow(at: insertedAt, at: .middle, animated: true) } } override func numberOfSections(in tableView: UITableView) -> Int { return 3 } override func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int { switch section { case 1: return tunnel?.peers?.count ?? 1 default: return 1 } } override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell { switch indexPath.section { case 0: let cell = tableView.dequeueReusableCell(type: InterfaceTableViewCell.self, for: indexPath) cell.model = tunnel.interface return cell case 1: let cell = tableView.dequeueReusableCell(type: PeerTableViewCell.self, for: indexPath) if let peer = tunnel.peers?.object(at: indexPath.row) as? Peer { cell.peer = peer } else { let peer = Peer(context: tunnel.managedObjectContext!) tunnel.addToPeers(peer) cell.peer = peer } cell.delegate = self return cell default: let cell = tableView.dequeueReusableCell(type: AddPeerTableViewCell.self, for: indexPath) cell.tunnel = tunnel return cell } } @IBAction func saveTunnelConfiguration(_ sender: Any) { Promise(resolver: { (seal) in viewContext.perform({ self.viewContext.saveContext({ (result) in switch result { case .success: seal.fulfill(()) case .failure(let error): seal.reject(error) } }) }) }).then { () -> Promise in self.delegate?.didSave(tunnel: self.tunnel, tunnelConfigurationTableViewController: self) return Promise.value(()) }.catch { error in print("Error saving: \(error)") } } } extension TunnelConfigurationTableViewController: PeerTableViewCellDelegate { func delete(peer: Peer) { if let moc = tunnel.managedObjectContext { tableView.beginUpdates() let deletedAt = IndexPath(row: tunnel.peers?.index(of: peer) ?? 0, section: 1) tableView.deleteRows(at: [deletedAt], with: .automatic) tunnel.removeFromPeers(peer) moc.delete(peer) tableView.endUpdates() } } } class InterfaceTableViewCell: UITableViewCell { var model: Interface! { didSet { nameField.text = model.tunnel?.title privateKeyField.text = model.privateKey listenPortField.text = String(model.listenPort) dnsField.text = model.dns mtuField.text = String(model.mtu) } } @IBOutlet weak var nameField: UITextField! @IBOutlet weak var privateKeyField: UITextField! @IBOutlet weak var publicKeyField: UITextField! @IBOutlet weak var listenPortField: UITextField! @IBOutlet weak var dnsField: UITextField! @IBOutlet weak var mtuField: UITextField! } extension InterfaceTableViewCell: UITextFieldDelegate { @IBAction func textfieldDidChange(_ sender: UITextField) { let string = sender.text if sender == nameField { model.tunnel?.title = string } else if sender == privateKeyField { model.privateKey = string } else if sender == listenPortField { if let string = string, let port = Int16(string) { model.listenPort = port } } else if sender == dnsField { model.dns = string } else if sender == mtuField { if let string = string, let mtu = Int32(string) { model.mtu = mtu } } } } protocol PeerTableViewCellDelegate: class { func delete(peer: Peer) } class PeerTableViewCell: UITableViewCell { var peer: Peer! { didSet { publicKeyField.text = peer.publicKey preSharedKeyField.text = peer.presharedKey allowedIpsField.text = peer.allowedIPs endpointField.text = peer.endpoint persistentKeepaliveField.text = String(peer.persistentKeepalive) } } weak var delegate: PeerTableViewCellDelegate? @IBOutlet weak var publicKeyField: UITextField! @IBOutlet weak var preSharedKeyField: UITextField! @IBOutlet weak var allowedIpsField: UITextField! @IBOutlet weak var endpointField: UITextField! @IBOutlet weak var persistentKeepaliveField: UITextField! @IBAction func deletePeer(_ sender: Any) { delegate?.delete(peer: peer) } } extension PeerTableViewCell: UITextFieldDelegate { @IBAction func textfieldDidChange(_ sender: UITextField) { let string = sender.text if sender == publicKeyField { peer.publicKey = string } else if sender == preSharedKeyField { peer.presharedKey = string } else if sender == allowedIpsField { peer.allowedIPs = string } else if sender == endpointField { peer.endpoint = string } else if sender == persistentKeepaliveField { if let string = string, let persistentKeepalive = Int16(string) { peer.persistentKeepalive = persistentKeepalive } } } } class AddPeerTableViewCell: UITableViewCell { var tunnel: Tunnel! @IBAction func addPeer(_ sender: Any) { if let moc = tunnel.managedObjectContext { tunnel.addToPeers(Peer(context: moc)) } } } extension TunnelConfigurationTableViewController: Identifyable {} extension InterfaceTableViewCell: Identifyable {} extension PeerTableViewCell: Identifyable {} extension AddPeerTableViewCell: Identifyable {}