Populate credential store if user has enabled before launching app

duckduckgo · Dec 18, 2024 · c59c6cb · c59c6cb
1 parent 1e4ecf0
commit c59c6cb
Showing 1 changed file with 19 additions and 4 deletions.
diff --git a/DuckDuckGo/AutofillUsageMonitor.swift b/DuckDuckGo/AutofillUsageMonitor.swift
@@ -19,19 +19,34 @@
 
 import Core
 import AuthenticationServices
+import BrowserServicesKit
 
 final class AutofillUsageMonitor {
 
+    private lazy var credentialIdentityStoreManager: AutofillCredentialIdentityStoreManager? = {
+        guard let vault = try? AutofillSecureVaultFactory.makeVault(reporter: SecureVaultReporter()) else {
+            return nil
+        }
+
+        return AutofillCredentialIdentityStoreManager(vault: vault,
+                                                      tld: AppDependencyProvider.shared.storageCache.tld)
+    }()
+
     init() {
         NotificationCenter.default.addObserver(self, selector: #selector(didReceiveSaveEvent), name: .autofillSaveEvent, object: nil)
 
-        ASCredentialIdentityStore.shared.getState({ state in
+        ASCredentialIdentityStore.shared.getState({ [weak self] state in
             if state.isEnabled {
-                self.autofillExtensionEnabled = true
+                if self?.autofillExtensionEnabled == nil {
+                    Task {
+                        await self?.credentialIdentityStoreManager?.populateCredentialStore()
+                    }
+                }
+                self?.autofillExtensionEnabled = true
             } else {
-                if self.autofillExtensionEnabled != nil {
+                if self?.autofillExtensionEnabled != nil {
                     Pixel.fire(pixel: .autofillExtensionDisabled)
-                    self.autofillExtensionEnabled = false
+                    self?.autofillExtensionEnabled = false
                 }
             }
         })