Extending Alert Utils

Trust/Extensions/UIAlertController.swift

@@ -7,14 +7,14 @@ import Result
 extension UIAlertController {
 
     static func askPassword(
-        title: String = "",
-        message: String = "",
-        completion: @escaping (Result<String, ConfirmationError>) -> Void
+            title: String = "",
+            message: String = "",
+            completion: @escaping (Result<String, ConfirmationError>) -> Void
     ) -> UIAlertController {
         let alertController = UIAlertController(
-            title: title,
-            message: message,
-            preferredStyle: .alert
+                title: title,
+                message: message,
+                preferredStyle: .alert
         )
         alertController.addAction(UIAlertAction(title: NSLocalizedString("OK", value: "OK", comment: ""), style: .default, handler: { _ -> Void in
             let textField = alertController.textFields![0] as UITextField
@@ -24,10 +24,35 @@ extension UIAlertController {
         alertController.addAction(UIAlertAction(title: NSLocalizedString("Cancel", value: "Cancel", comment: ""), style: .cancel, handler: { _ in
             completion(.failure(ConfirmationError.cancel))
         }))
-        alertController.addTextField(configurationHandler: {(textField: UITextField!) -> Void in
+        alertController.addTextField(configurationHandler: { (textField: UITextField!) -> Void in
             textField.placeholder = NSLocalizedString("Password", value: "Password", comment: "")
             textField.isSecureTextEntry = true
         })
         return alertController
     }
+
+    static func alert(_ title: String,
+                      message: String,
+                      alertButtonTitles: [String],
+                      alertButtonStyles: [UIAlertActionStyle],
+                      viewController: UIViewController,
+                      completion: ((Int) -> Void)?) {
+
+        let alertController = UIAlertController(
+                title: title,
+                message: message,
+                preferredStyle: UIAlertControllerStyle.alert)
+
+        alertButtonTitles.forEach { title in
+            let alertStyle: UIAlertActionStyle = alertButtonStyles[alertButtonTitles.index(of: title)!]
+            let action = UIAlertAction(title: title, style: alertStyle, handler: { action in
+                if completion != nil {
+                    completion!(alertButtonTitles.index(of: action.title!)!)
+                }
+            })
+            alertController.addAction(action)
+        }
+        viewController.present(alertController, animated: true, completion: nil)
+    }
+
 }