Implement sign-out flow for DBP

duckduckgo · Dec 27, 2023 · c3af223 · c3af223
1 parent 3e020ba
commit c3af223
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 3 deletions.
diff --git a/DuckDuckGo/DBP/DataBrokerProtectionFeatureDisabler.swift b/DuckDuckGo/DBP/DataBrokerProtectionFeatureDisabler.swift
@@ -19,6 +19,10 @@
 import Foundation
 import DataBrokerProtection
 
+public extension Notification.Name {
+    static let dbpWasDisabled = Notification.Name("com.duckduckgo.DBP.DBPWasDisabled")
+}
+
 protocol DataBrokerProtectionFeatureDisabling {
     func disableAndDelete()
 }
@@ -40,6 +44,8 @@ struct DataBrokerProtectionFeatureDisabler: DataBrokerProtectionFeatureDisabling
             scheduler.disableLoginItem()
 
             dataManager.removeAllData()
+
+            NotificationCenter.default.post(name: .dbpWasDisabled, object: nil)
         }
     }
 }
diff --git a/DuckDuckGo/DBP/DataBrokerProtectionSubscriptionEventHandler.swift b/DuckDuckGo/DBP/DataBrokerProtectionSubscriptionEventHandler.swift
@@ -25,11 +25,14 @@ final class DataBrokerProtectionSubscriptionEventHandler {
 
     private let accountManager: Subscription.AccountManaging
     private let authRepository: AuthenticationRepository
+    private let featureDisabler: DataBrokerProtectionFeatureDisabling
 
-    init(accountManager: Subscription.AccountManaging = Subscription.AccountManager(),
-         authRepository: AuthenticationRepository = KeychainAuthenticationData()) {
+    init(accountManager: Subscription.AccountManaging = AccountManager(),
+         authRepository: AuthenticationRepository = KeychainAuthenticationData(),
+         featureDisabler: DataBrokerProtectionFeatureDisabling = DataBrokerProtectionFeatureDisabler()) {
         self.accountManager = accountManager
         self.authRepository = authRepository
+        self.featureDisabler = featureDisabler
     }
 
     func registerForSubscriptionAccountManagerEvents() {
@@ -48,7 +51,7 @@ final class DataBrokerProtectionSubscriptionEventHandler {
     }
 
     @objc private func handleAccountDidSignOut() {
-        // Going to be defined here: https://app.asana.com/0/1204006570077678/1206206961074916/f
+        featureDisabler.disableAndDelete()
     }
 }
 

diff --git a/DuckDuckGo/Tab/View/BrowserTabViewController.swift b/DuckDuckGo/Tab/View/BrowserTabViewController.swift
@@ -114,6 +114,10 @@ final class BrowserTabViewController: NSViewController {
                                                object: nil)
 
 #if DBP
+        NotificationCenter.default.addObserver(self,
+                                               selector: #selector(onDBPFeatureDisabled),
+                                               name: .dbpWasDisabled,
+                                               object: nil)
         NotificationCenter.default.addObserver(self,
                                                selector: #selector(onCloseDataBrokerProtection),
                                                name: .dbpDidClose,
@@ -165,6 +169,11 @@ final class BrowserTabViewController: NSViewController {
     }
 
 #if DBP
+    @objc
+    private func onDBPFeatureDisabled(_ notification: Notification) {
+        tabCollectionViewModel.removeAll(with: .dataBrokerProtection)
+    }
+
     @objc
     private func onCloseDataBrokerProtection(_ notification: Notification) {
         guard let activeTab = tabCollectionViewModel.selectedTabViewModel?.tab,

diff --git a/DuckDuckGo/TabBar/ViewModel/TabCollectionViewModel.swift b/DuckDuckGo/TabBar/ViewModel/TabCollectionViewModel.swift
@@ -361,6 +361,16 @@ final class TabCollectionViewModel: NSObject {
 
     // MARK: - Removal
 
+    func removeAll(with content: Tab.TabContent) {
+        let tabs = tabCollection.tabs.filter { $0.content == content }
+
+        for tab in tabs {
+            if let index = indexInAllTabs(of: tab) {
+                remove(at: index)
+            }
+        }
+    }
+
     func remove(at index: TabIndex, published: Bool = true, forceChange: Bool = false) {
         switch index {
         case .unpinned(let i):