From d28861448e3db59d421c9dc8db1b397a72915c62 Mon Sep 17 00:00:00 2001
From: Mingshen Sun <bob@mssun.me>
Date: Tue, 31 Aug 2021 18:25:40 -0700
Subject: [PATCH] Fix issue of displying extension

---
 .../Controllers/PasscodeExtensionDisplay.swift              | 6 +++++-
 passExtension/Controllers/ExtensionViewController.swift     | 4 ----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/passAutoFillExtension/Controllers/PasscodeExtensionDisplay.swift b/passAutoFillExtension/Controllers/PasscodeExtensionDisplay.swift
index 72c0863..55c8499 100644
--- a/passAutoFillExtension/Controllers/PasscodeExtensionDisplay.swift
+++ b/passAutoFillExtension/Controllers/PasscodeExtensionDisplay.swift
@@ -24,7 +24,11 @@ class PasscodeExtensionDisplay {
             before?()
             passcodeLockVC.successCallback = after
             passcodeLockVC.modalPresentationStyle = .fullScreen
-            sender.parent?.present(passcodeLockVC, animated: false)
+            sender.parent?.present(passcodeLockVC, animated: false) {
+                after?()
+            }
+        } else {
+            after?()
         }
     }
 }
diff --git a/passExtension/Controllers/ExtensionViewController.swift b/passExtension/Controllers/ExtensionViewController.swift
index 4869e90..7eb190d 100644
--- a/passExtension/Controllers/ExtensionViewController.swift
+++ b/passExtension/Controllers/ExtensionViewController.swift
@@ -47,10 +47,6 @@ class ExtensionViewController: UIViewController {
     override func viewWillAppear(_ animated: Bool) {
         super.viewWillAppear(animated)
         prepareCredentialList()
-    }
-
-    override func viewDidAppear(_ animated: Bool) {
-        super.viewDidAppear(animated)
         passcodelock.presentPasscodeLockIfNeeded(self, after: { [unowned self] in
             self.view.isHidden = false
         })