From 29e293ebdeaeabb2d8a80ebe3d782dfad6bace95 Mon Sep 17 00:00:00 2001
From: Yishi Lin <yishilin14@gmail.com>
Date: Mon, 10 Apr 2017 02:15:49 +0800
Subject: [PATCH] Check armor pgp keys before saving

---
 .../PGPKeyArmorSettingTableViewController.swift    | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/pass/Controllers/PGPKeyArmorSettingTableViewController.swift b/pass/Controllers/PGPKeyArmorSettingTableViewController.swift
index 650e5ae..34f7649 100644
--- a/pass/Controllers/PGPKeyArmorSettingTableViewController.swift
+++ b/pass/Controllers/PGPKeyArmorSettingTableViewController.swift
@@ -37,6 +37,20 @@ class PGPKeyArmorSettingTableViewController: UITableViewController, UITextViewDe
         return savePassphraseAlert
     }
     
+    override func shouldPerformSegue(withIdentifier identifier: String, sender: Any?) -> Bool {
+        if identifier == "savePGPKeySegue" {
+            if armorPublicKeyTextView.text.isEmpty {
+                Utils.alert(title: "Cannot Save", message: "Please set public key first.", controller: self, completion: nil)
+                return false
+            }
+            if armorPrivateKeyTextView.text.isEmpty {
+                Utils.alert(title: "Cannot Save", message: "Please set private key first.", controller: self, completion: nil)
+                return false
+            }
+        }
+        return true
+    }
+    
     @IBAction func save(_ sender: Any) {
         let alert = UIAlertController(title: "Passphrase", message: "Please fill in the passphrase of your PGP secret key.", preferredStyle: UIAlertControllerStyle.alert)
         alert.addAction(UIAlertAction(title: "OK", style: UIAlertActionStyle.default, handler: {_ in