From cc122d7463f98ab3f26d46ab849173ffd58d7951 Mon Sep 17 00:00:00 2001
From: Roopesh Chander <roop@roopc.net>
Date: Mon, 12 Nov 2018 14:02:09 +0530
Subject: Model, Tunnels manager: Rewrite the model for VPN-on-demand

The VPN-on-demand settings should not be part of the tunnel
configuration. Rather, the onDemandRules stored in the
tunnel provider configuration serve as the one place
where the VPN-on-demand settings are stored.

Signed-off-by: Roopesh Chander <roop@roopc.net>
---
 WireGuard/WireGuard/VPN/TunnelsManager.swift | 47 +++++++---------------------
 1 file changed, 11 insertions(+), 36 deletions(-)

(limited to 'WireGuard/WireGuard/VPN/TunnelsManager.swift')

diff --git a/WireGuard/WireGuard/VPN/TunnelsManager.swift b/WireGuard/WireGuard/VPN/TunnelsManager.swift
index 8eb2112..d75e6c0 100644
--- a/WireGuard/WireGuard/VPN/TunnelsManager.swift
+++ b/WireGuard/WireGuard/VPN/TunnelsManager.swift
@@ -54,7 +54,9 @@ class TunnelsManager {
         #endif
     }
 
-    func add(tunnelConfiguration: TunnelConfiguration, completionHandler: @escaping (TunnelContainer?, TunnelManagementError?) -> Void) {
+    func add(tunnelConfiguration: TunnelConfiguration,
+             activateOnDemandSetting: ActivateOnDemandSetting = ActivateOnDemandSetting.defaultSetting,
+             completionHandler: @escaping (TunnelContainer?, TunnelManagementError?) -> Void) {
         let tunnelName = tunnelConfiguration.interface.name
         if tunnelName.isEmpty {
             completionHandler(nil, TunnelManagementError.tunnelAlreadyExistsWithThatName)
@@ -72,13 +74,7 @@ class TunnelsManager {
         tunnelProviderManager.localizedDescription = tunnelName
         tunnelProviderManager.isEnabled = true
 
-        if (tunnelConfiguration.activationType == .activateManually) {
-            tunnelProviderManager.onDemandRules = []
-            tunnelProviderManager.isOnDemandEnabled = false
-        } else {
-            tunnelProviderManager.onDemandRules = onDemandRules(for: tunnelConfiguration.activationType)
-            tunnelProviderManager.isOnDemandEnabled = true
-        }
+        activateOnDemandSetting.apply(on: tunnelProviderManager)
 
         tunnelProviderManager.saveToPreferences { [weak self] (error) in
             defer { self?.isAddingTunnel = false }
@@ -114,7 +110,8 @@ class TunnelsManager {
         }
     }
 
-    func modify(tunnel: TunnelContainer, with tunnelConfiguration: TunnelConfiguration, completionHandler: @escaping (TunnelManagementError?) -> Void) {
+    func modify(tunnel: TunnelContainer, tunnelConfiguration: TunnelConfiguration,
+                activateOnDemandSetting: ActivateOnDemandSetting, completionHandler: @escaping (TunnelManagementError?) -> Void) {
         let tunnelName = tunnelConfiguration.interface.name
         if tunnelName.isEmpty {
             completionHandler(TunnelManagementError.tunnelAlreadyExistsWithThatName)
@@ -138,13 +135,7 @@ class TunnelsManager {
         tunnelProviderManager.localizedDescription = tunnelName
         tunnelProviderManager.isEnabled = true
 
-        if (tunnelConfiguration.activationType == .activateManually) {
-            tunnelProviderManager.onDemandRules = []
-            tunnelProviderManager.isOnDemandEnabled = false
-        } else {
-            tunnelProviderManager.onDemandRules = onDemandRules(for: tunnelConfiguration.activationType)
-            tunnelProviderManager.isOnDemandEnabled = true
-        }
+        activateOnDemandSetting.apply(on: tunnelProviderManager)
 
         tunnelProviderManager.saveToPreferences { [weak self] (error) in
             defer { self?.isModifyingTunnel = false }
@@ -229,26 +220,6 @@ class TunnelsManager {
             t.refreshConnectionStatus()
         }
     }
-
-    func onDemandRules(for activationType: ActivationType) -> [NEOnDemandRule] {
-        switch (activationType) {
-        case .activateManually: return []
-        case .useOnDemandOverWifiAndCellular:
-            return [NEOnDemandRuleConnect()]
-        case .useOnDemandOverWifiOnly:
-            let connectOnWifiRule = NEOnDemandRuleConnect()
-            connectOnWifiRule.interfaceTypeMatch = .wiFi
-            let disconnectOnCellularRule = NEOnDemandRuleDisconnect()
-            disconnectOnCellularRule.interfaceTypeMatch = .cellular
-            return [connectOnWifiRule, disconnectOnCellularRule]
-        case .useOnDemandOverCellularOnly:
-            let connectOnCellularRule = NEOnDemandRuleConnect()
-            connectOnCellularRule.interfaceTypeMatch = .cellular
-            let disconnectOnWifiRule = NEOnDemandRuleDisconnect()
-            disconnectOnWifiRule.interfaceTypeMatch = .wiFi
-            return [connectOnCellularRule, disconnectOnWifiRule]
-        }
-    }
 }
 
 class TunnelContainer: NSObject {
@@ -275,6 +246,10 @@ class TunnelContainer: NSObject {
         return (tunnelProvider.protocolConfiguration as! NETunnelProviderProtocol).tunnelConfiguration()
     }
 
+    func activateOnDemandSetting() -> ActivateOnDemandSetting {
+        return ActivateOnDemandSetting(from: tunnelProvider)
+    }
+
     func refreshConnectionStatus() {
         let status = TunnelStatus(from: self.tunnelProvider.connection.status)
         self.status = status
-- 
cgit v1.2.3-59-g8ed1b