Fixing retry and loading cases

brave · May 26, 2023 · bd15c97 · bd15c97
1 parent 8fcb8c8
commit bd15c97
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/Sources/BraveVPN/InstallVPNViewController.swift b/Sources/BraveVPN/InstallVPNViewController.swift
@@ -59,9 +59,14 @@ class InstallVPNViewController: VPNSetupLoadingController {
         self.dismiss(animated: true) {
           self.showSuccessAlert()
         }
-      } else if self.installVPNProfileRetryCount < 2 {
-        // Retry installing profile if it fails first time
-        presentErrorVPNInstallProfile()
+      } else {
+        // Retry installing profile twice if it fails
+        // Error generated is WireGuard capabilities are not yet enabled on this node
+        if self.installVPNProfileRetryCount < 2 {
+          installVPNAction()
+        } else {
+          presentErrorVPNInstallProfile()
+        }
       }
     }
 
@@ -83,7 +88,7 @@ class InstallVPNViewController: VPNSetupLoadingController {
 
     BraveVPN.connectToVPN() { [weak self] status in
       DispatchQueue.main.async {
-        self?.isLoading = true
+        self?.isLoading = false
       }
 
       completion(status)