Update: reload MySites view on sign in

Author: hassaanelgarem

WordPress/Classes/ViewRelated/System/Coordinators/MySitesCoordinator.swift

@@ -1,4 +1,5 @@
 import UIKit
+import WordPressAuthenticator
 
 @objc
 class MySitesCoordinator: NSObject {
@@ -18,8 +19,9 @@ class MySitesCoordinator: NSObject {
     init(meScenePresenter: ScenePresenter, onBecomeActiveTab becomeActiveTab: @escaping () -> Void) {
         self.meScenePresenter = meScenePresenter
         self.becomeActiveTab = becomeActiveTab
-
         super.init()
+
+        addSignInObserver()
     }
 
     // MARK: - Root View Controller
@@ -73,9 +75,13 @@ class MySitesCoordinator: NSObject {
     }()
 
     private lazy var mySiteViewController: MySiteViewController = {
-        MySiteViewController(meScenePresenter: self.meScenePresenter)
+        makeMySiteViewController()
     }()
 
+    private func makeMySiteViewController() -> MySiteViewController {
+        MySiteViewController(meScenePresenter: self.meScenePresenter)
+    }
+
     // MARK: - Navigation
 
     func showRootViewController() {
@@ -219,4 +225,19 @@ class MySitesCoordinator: NSObject {
 
         navigationController.pushViewController(listViewController, animated: false)
     }
+
+    // MARK: Notifications Handling
+
+    private func addSignInObserver() {
+        let notificationName = NSNotification.Name(WordPressAuthenticator.WPSigninDidFinishNotification)
+        NotificationCenter.default.addObserver(self,
+                                               selector: #selector(signinDidFinish),
+                                               name: notificationName,
+                                               object: nil)
+    }
+
+    @objc func signinDidFinish() {
+        mySiteViewController = makeMySiteViewController()
+        navigationController.viewControllers = [rootContentViewController]
+    }
 }

WordPress/Classes/ViewRelated/System/WPTabBarController.m

@@ -252,7 +252,6 @@ - (void)reloadSplitViewControllers
     _readerNavigationController = nil;
     _notificationsNavigationController = nil;
     _notificationsSplitViewController = nil;
-    _mySitesCoordinator = nil;
 
     [self setViewControllers:[self tabViewControllers]];