From 66e802584bd0c5be0c545d87b3aa52c6d182d475 Mon Sep 17 00:00:00 2001
From: Roopesh Chander <roop@roopc.net>
Date: Fri, 2 Nov 2018 13:09:45 +0530
Subject: Tunnel view model: If there's just one peer, populate the scratchpad
 on init

So that exclude private IPs can be computed correctly.

Signed-off-by: Roopesh Chander <roop@roopc.net>
---
 WireGuard/WireGuard/UI/TunnelViewModel.swift | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

(limited to 'WireGuard/WireGuard/UI/TunnelViewModel.swift')

diff --git a/WireGuard/WireGuard/UI/TunnelViewModel.swift b/WireGuard/WireGuard/UI/TunnelViewModel.swift
index 0886848..a9ee6f8 100644
--- a/WireGuard/WireGuard/UI/TunnelViewModel.swift
+++ b/WireGuard/WireGuard/UI/TunnelViewModel.swift
@@ -179,11 +179,9 @@ class TunnelViewModel {
         var validatedConfiguration: PeerConfiguration? = nil
 
         // For exclude private IPs
-        var shouldAllowExcludePrivateIPsControl: Bool = false
-        var excludePrivateIPsValue: Bool = false
-        var numberOfPeers: Int = 0 {
-            didSet { updateExcludePrivateIPsFieldState() }
-        }
+        var shouldAllowExcludePrivateIPsControl: Bool = false /* Read-only from the VC's point of view */
+        var excludePrivateIPsValue: Bool = false /* Read-only from the VC's point of view */
+        fileprivate var numberOfPeers: Int = 0
 
         init(index: Int) {
             self.index = index
@@ -318,6 +316,9 @@ class TunnelViewModel {
                 excludePrivateIPsValue = false
                 return
             }
+            if (scratchpad.isEmpty) {
+                populateScratchpad()
+            }
             let allowedIPStrings = Set<String>(
                 (scratchpad[.allowedIPs] ?? "")
                     .split(separator: ",")
@@ -365,8 +366,8 @@ class TunnelViewModel {
     var peersData: [PeerData]
 
     init(tunnelConfiguration: TunnelConfiguration?) {
-        interfaceData = InterfaceData()
-        peersData = []
+        let interfaceData: InterfaceData = InterfaceData()
+        var peersData: [PeerData] = []
         if let tunnelConfiguration = tunnelConfiguration {
             interfaceData.validatedConfiguration = tunnelConfiguration.interface
             for (i, peerConfiguration) in tunnelConfiguration.peers.enumerated() {
@@ -378,7 +379,10 @@ class TunnelViewModel {
         let numberOfPeers = peersData.count
         for peerData in peersData {
             peerData.numberOfPeers = numberOfPeers
+            peerData.updateExcludePrivateIPsFieldState()
         }
+        self.interfaceData = interfaceData
+        self.peersData = peersData
     }
 
     func appendEmptyPeer() {
@@ -386,6 +390,7 @@ class TunnelViewModel {
         peersData.append(peer)
         for p in peersData {
             p.numberOfPeers = peersData.count
+            p.updateExcludePrivateIPsFieldState()
         }
     }
 
@@ -398,6 +403,7 @@ class TunnelViewModel {
         }
         for p in peersData {
             p.numberOfPeers = peersData.count
+            p.updateExcludePrivateIPsFieldState()
         }
     }
 
-- 
cgit v1.2.3-59-g8ed1b