Added property to control auto dismissing web auth

octopus-energy · Jan 31, 2023 · 37b9a52 · 37b9a52
1 parent df79c36
commit 37b9a52
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Sources/TeslaSwift/TeslaWebLoginViewContoller.swift b/Sources/TeslaSwift/TeslaWebLoginViewContoller.swift
@@ -12,6 +12,8 @@ import WebKit
 public class TeslaWebLoginViewController: UIViewController {
     var webView = WKWebView()
     var result: ((Result<URL, Error>) -> Void)?
+
+    public var shouldDismiss: Bool = true
 
     required init?(coder: NSCoder) {
         fatalError("not supported")
@@ -32,7 +34,9 @@ extension TeslaWebLoginViewController: WKNavigationDelegate {
     public func webView(_ webView: WKWebView, decidePolicyFor navigationAction: WKNavigationAction, decisionHandler: @escaping (WKNavigationActionPolicy) -> Void) {
         if let url = navigationAction.request.url, url.absoluteString.starts(with: "https://auth.tesla.com/void/callback")  {
             decisionHandler(.cancel)
-            self.dismiss(animated: true, completion: nil)
+            if shouldDismiss {
+                self.dismiss(animated: true, completion: nil)
+            }
             self.result?(Result.success(url))
         } else {
             decisionHandler(.allow)
@@ -41,7 +45,9 @@ extension TeslaWebLoginViewController: WKNavigationDelegate {
 
     public func webView(_ webView: WKWebView, didFail navigation: WKNavigation!, withError error: Error) {
         self.result?(Result.failure(TeslaError.authenticationFailed))
-        self.dismiss(animated: true, completion: nil)
+        if shouldDismiss {
+            self.dismiss(animated: true, completion: nil)
+        }
     }
 }