Scan QR Code to import SSH private key

pass/Controllers/GitCredential.swift

@@ -57,6 +57,7 @@ struct GitCredential {
                 attempts += 1
                 lastPassword = newPassword
                 credential = try? GTCredential(userName: userName, publicKeyURL: nil, privateKeyURL: privateKeyFile, passphrase: newPassword!)
+                print(privateKeyFile)
             }
             return credential
         }

pass/Controllers/GitSSHKeyArmorSettingTableViewController.swift

@@ -9,17 +9,77 @@
 import UIKit
 import SwiftyUserDefaults
 
-class GitSSHKeyArmorSettingTableViewController: UITableViewController, UITextViewDelegate {
+class GitSSHKeyArmorSettingTableViewController: UITableViewController, UITextViewDelegate, QRScannerControllerDelegate {
     @IBOutlet weak var armorPrivateKeyTextView: UITextView!
+    @IBOutlet weak var scanPrivateKeyCell: UITableViewCell!
+    
     var gitSSHPrivateKeyPassphrase: String?
     let passwordStore = PasswordStore.shared
     
     private var recentPastedText = ""
     
+    class ScannedSSHKey {
+        static let maxNumberOfGif = 100
+        var numberOfSegments = 0
+        var previousSegment = ""
+        var key = ""
+        var message = ""
+        var hasStarted = false
+        var isDone = false
+        
+        func reset() {
+            numberOfSegments = 0
+            previousSegment = ""
+            key = ""
+            message = "Looking for the starting frame."
+            hasStarted = false
+            isDone = false
+        }
+        
+        func addSegment(segment: String) {
+            // skip duplicated segments
+            guard segment != previousSegment else {
+                return
+            }
+            previousSegment = segment
+            
+            // check whether we have found the first block
+            if hasStarted == false {
+                hasStarted = segment.contains("-----BEGIN")
+            }
+            guard hasStarted == true else {
+                return
+            }
+            
+            // check the number of segments
+            numberOfSegments = numberOfSegments + 1
+            guard numberOfSegments <= ScannedSSHKey.maxNumberOfGif else {
+                key = "Too many QR codes"
+                return
+            }
+            
+            // update full text and check whether we are done
+            key.append(segment)
+            if let index1 = key.range(of: "-----END")?.lowerBound,
+                let _ = key.substring(from: index1).range(of: "KEY-----")?.lowerBound {
+                isDone = true
+            }
+            
+            // update message
+            message = "\(numberOfSegments) scanned QR codes."
+        }
+    }
+    var scanned = ScannedSSHKey()
+    
     override func viewDidLoad() {
         super.viewDidLoad()
         armorPrivateKeyTextView.text = Defaults[.gitSSHPrivateKeyArmor]
         armorPrivateKeyTextView.delegate = self
+        
+        scanPrivateKeyCell?.textLabel?.text = "Scan Private Key QR Codes"
+        scanPrivateKeyCell?.textLabel?.textColor = Globals.blue
+        scanPrivateKeyCell?.selectionStyle = .default
+        scanPrivateKeyCell?.accessoryType = .disclosureIndicator
     }
     
     @IBAction func doneButtonTapped(_ sender: Any) {
@@ -46,4 +106,46 @@ class GitSSHKeyArmorSettingTableViewController: UITableViewController, UITextVie
         }
         return true
     }
+    
+    override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
+        let selectedCell = tableView.cellForRow(at: indexPath)
+        if selectedCell == scanPrivateKeyCell {
+            scanned.reset()
+            self.performSegue(withIdentifier: "showSSHScannerSegue", sender: self)
+        }
+        tableView.deselectRow(at: indexPath, animated: true)
+    }
+    
+    
+    // MARK: - QRScannerControllerDelegate Methods
+    func checkScannedOutput(line: String) -> (accept: Bool, message: String) {
+        scanned.addSegment(segment: line)
+        if scanned.isDone {
+            return (accept: true, message: "Done")
+        } else {
+            return (accept: false, message: scanned.message)
+        }
+    }
+    
+    // MARK: - QRScannerControllerDelegate Methods
+    func handleScannedOutput(line: String) {
+        armorPrivateKeyTextView.text = scanned.key
+    }
+    
+    override func prepare(for segue: UIStoryboardSegue, sender: Any?) {
+        if segue.identifier == "showSSHScannerSegue" {
+            if let navController = segue.destination as? UINavigationController {
+                if let viewController = navController.topViewController as? QRScannerController {
+                    viewController.delegate = self
+                }
+            } else if let viewController = segue.destination as? QRScannerController {
+                viewController.delegate = self
+            }
+        }
+    }
+    
+    @IBAction private func cancelSSHScanner(segue: UIStoryboardSegue) {
+        
+    }
+
 }

pass/Controllers/GitServerSettingTableViewController.swift

@@ -120,8 +120,7 @@ class GitServerSettingTableViewController: UITableViewController {
     }
     
     private func gitSSHKeyExists() -> Bool {
-        return FileManager.default.fileExists(atPath: Globals.gitSSHPublicKeyPath) &&
-            FileManager.default.fileExists(atPath: Globals.gitSSHPrivateKeyPath)
+        return FileManager.default.fileExists(atPath: Globals.gitSSHPrivateKeyPath)
     }
     
     func showSSHKeyActionSheet() {