Bug 1536840 - Deleting FxA account on-device leaves app in state that…

… appears connected (#4630)
mozilla-mobile · Mar 20, 2019 · baf102c · baf102c
1 parent c874695
commit baf102c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Client/Frontend/Settings/FxAContentViewController.swift b/Client/Frontend/Settings/FxAContentViewController.swift
@@ -29,6 +29,7 @@ class FxAContentViewController: SettingsContentViewController, WKScriptMessageHa
         case changePassword = "change_password"
         case sessionStatus = "session_status"
         case signOut = "sign_out"
+        case deleteAccount = "delete_account"
     }
 
     weak var delegate: FxAContentViewControllerDelegate?
@@ -122,6 +123,13 @@ class FxAContentViewController: SettingsContentViewController, WKScriptMessageHa
         injectData("message", content: ["status": "error"])
     }
 
+    // The user has deleted their Firefox Account. Disconnect them!
+    fileprivate func onDeleteAccount(_ data: JSON) {
+        FxALoginHelper.sharedInstance.applicationDidDisconnect(UIApplication.shared)
+        LeanPlumClient.shared.set(attributes: [LPAttributeKey.signedInSync: profile.hasAccount()])
+        dismiss(animated: true)
+    }
+
     // The user has signed in to a Firefox Account.  We're done!
     fileprivate func onLogin(_ data: JSON) {
         injectData("message", content: ["status": "login"])
@@ -180,6 +188,8 @@ class FxAContentViewController: SettingsContentViewController, WKScriptMessageHa
                 onSessionStatus(data)
             case .signOut:
                 onSignOut(data)
+            case .deleteAccount:
+                onDeleteAccount(data)
             }
         }
     }