Importing: Simplify TunnelImporter

Signed-off-by: Roopesh Chander <roop@roopc.net>

WireGuard/WireGuard/UI/TunnelImporter.swift

@@ -9,86 +9,77 @@ class TunnelImporter {
             completionHandler?()
             return
         }
-        if urls.count > 1 {
         let dispatchGroup = DispatchGroup()
         var configs = [TunnelConfiguration?]()
+        var lastFileImportErrorText: (title: String, message: String)?
         for url in urls {
             if url.pathExtension.lowercased() == "zip" {
                 dispatchGroup.enter()
                 ZipImporter.importConfigFiles(from: url) { result in
+                    if let error = result.error {
+                        lastFileImportErrorText = error.alertText
+                    }
                     if let configsInZip = result.value {
                         configs.append(contentsOf: configsInZip)
                     }
                     dispatchGroup.leave()
                 }
-                } else {
+            } else { /* if it is not a zip, we assume it is a conf */
-                    let fileBaseName = url.deletingPathExtension().lastPathComponent.trimmingCharacters(in: .whitespacesAndNewlines)
-                    let fileContents = try? String(contentsOf: url)
-                    let tunnelConfiguration = try? TunnelConfiguration(fromWgQuickConfig: fileContents ?? "", called: fileBaseName)
-                    configs.append(tunnelConfiguration)
-                }
-            }
-            dispatchGroup.notify(queue: .main) {
-                tunnelsManager.addMultiple(tunnelConfigurations: configs.compactMap { $0 }) { numberSuccessful in
-                    if numberSuccessful == configs.count {
-                        completionHandler?()
-                        return
-                    }
-                    let title = tr(format: "alertImportedFromMultipleFilesTitle (%d)", numberSuccessful)
-                    let message = tr(format: "alertImportedFromMultipleFilesMessage (%1$d of %2$d)", numberSuccessful, configs.count)
-                    errorPresenterType.showErrorAlert(title: title, message: message, from: sourceVC, onPresented: completionHandler)
-                }
-            }
-            return
-        }
-        assert(urls.count == 1)
-        let url = urls.first!
-        if url.pathExtension.lowercased() == "zip" {
-            ZipImporter.importConfigFiles(from: url) { result in
-                if let error = result.error {
-                    errorPresenterType.showErrorAlert(error: error, from: sourceVC)
-                    return
-                }
-                let configs = result.value!
-                tunnelsManager.addMultiple(tunnelConfigurations: configs.compactMap { $0 }) { numberSuccessful in
-                    if numberSuccessful == configs.count {
-                        completionHandler?()
-                        return
-                    }
-                    let title = tr(format: "alertImportedFromZipTitle (%d)", numberSuccessful)
-                    let message = tr(format: "alertImportedFromZipMessage (%1$d of %2$d)", numberSuccessful, configs.count)
-                    errorPresenterType.showErrorAlert(title: title, message: message, from: sourceVC, onPresented: completionHandler)
-                }
-            }
-        } else /* if (url.pathExtension == "conf") -- we assume everything else is a conf */ {
                 let fileName = url.lastPathComponent
                 let fileBaseName = url.deletingPathExtension().lastPathComponent.trimmingCharacters(in: .whitespacesAndNewlines)
+                dispatchGroup.enter()
+                DispatchQueue.global(qos: .userInitiated).async {
                     let fileContents: String
                     do {
                         fileContents = try String(contentsOf: url)
                     } catch let error {
-                let message: String
                         if let cocoaError = error as? CocoaError, cocoaError.isFileError {
-                    message = error.localizedDescription
+                            lastFileImportErrorText = (title: tr("alertCantOpenInputConfFileTitle"), message: error.localizedDescription)
                         } else {
-                    message = tr(format: "alertCantOpenInputConfFileMessage (%@)", fileName)
+                            lastFileImportErrorText = (title: tr("alertCantOpenInputConfFileTitle"), message: tr(format: "alertCantOpenInputConfFileMessage (%@)", fileName))
+                        }
+                        DispatchQueue.main.async {
+                            configs.append(nil)
+                            dispatchGroup.leave()
                         }
-                errorPresenterType.showErrorAlert(title: tr("alertCantOpenInputConfFileTitle"), message: message, from: sourceVC, onPresented: completionHandler)
                         return
                     }
-            if let tunnelConfiguration = try? TunnelConfiguration(fromWgQuickConfig: fileContents, called: fileBaseName) {
+                    let tunnelConfiguration = try? TunnelConfiguration(fromWgQuickConfig: fileContents, called: fileBaseName)
-                tunnelsManager.add(tunnelConfiguration: tunnelConfiguration) { result in
+                    if tunnelConfiguration == nil {
-                    if let error = result.error {
+                        lastFileImportErrorText = (title: tr("alertBadConfigImportTitle"), message: tr(format: "alertBadConfigImportMessage (%@)", fileName))
-                        errorPresenterType.showErrorAlert(error: error, from: sourceVC, onPresented: completionHandler)
+                    }
+                    DispatchQueue.main.async {
+                        configs.append(tunnelConfiguration)
+                        dispatchGroup.leave()
+                    }
+                }
+            }
+        }
+        dispatchGroup.notify(queue: .main) {
+            tunnelsManager.addMultiple(tunnelConfigurations: configs.compactMap { $0 }) { numberSuccessful in
+                if !configs.isEmpty && numberSuccessful == configs.count {
+                    completionHandler?()
+                    return
+                }
+                let title: String
+                let message: String
+                if urls.count == 1 {
+                    if urls.first!.pathExtension.lowercased() == "zip" && !configs.isEmpty {
+                        title = tr(format: "alertImportedFromZipTitle (%d)", numberSuccessful)
+                        message = tr(format: "alertImportedFromZipMessage (%1$d of %2$d)", numberSuccessful, configs.count)
+                    } else if let lastFileImportErrorText = lastFileImportErrorText {
+                        title = lastFileImportErrorText.title
+                        message = lastFileImportErrorText.message
                     } else {
                         completionHandler?()
-                    }
+                        return
                     }
                 } else {
-                errorPresenterType.showErrorAlert(title: tr("alertBadConfigImportTitle"), message: tr(format: "alertBadConfigImportMessage (%@)", fileName),
+                    title = tr(format: "alertImportedFromMultipleFilesTitle (%d)", numberSuccessful)
-                                              from: sourceVC, onPresented: completionHandler)
+                    message = tr(format: "alertImportedFromMultipleFilesMessage (%1$d of %2$d)", numberSuccessful, configs.count)
+                }
+                errorPresenterType.showErrorAlert(title: title, message: message, from: sourceVC, onPresented: completionHandler)
             }
         }
     }
-
 }