Observe site publisher to update site title and URL

Author: itsmeichigo

WooCommerce/Classes/ViewRelated/Hub Menu/HubMenu.swift

@@ -34,7 +34,7 @@ struct HubMenu: View {
         VStack {
             TopBar(avatarURL: viewModel.avatarURL,
                    storeTitle: viewModel.storeTitle,
-                   storeURL: viewModel.storeURL.absoluteString) {
+                   storeURL: viewModel.storeURL?.absoluteString) {
                 viewModel.presentSwitchStore()
             }
                    .padding([.leading, .trailing], Constants.padding)

WooCommerce/Classes/ViewRelated/Hub Menu/HubMenuViewModel.swift

@@ -29,23 +29,11 @@ final class HubMenuViewModel: ObservableObject {
         return url
     }
 
-    var storeTitle: String {
-        stores.sessionManager.defaultSite?.name ?? Localization.myStore
-    }
+    @Published private(set) var storeTitle = Localization.myStore
 
-    var storeURL: URL {
-        guard let urlString = stores.sessionManager.defaultSite?.url, let url = URL(string: urlString) else {
-            return WooConstants.URLs.blog.asURL()
-        }
-        return url
-    }
-    var woocommerceAdminURL: URL {
-        guard let urlString = stores.sessionManager.defaultSite?.adminURL, let url = URL(string: urlString) else {
-            return stores.sessionManager.defaultSite?.adminURLWithFallback() ??
-            WooConstants.URLs.blog.asURL()
-        }
-        return url
-    }
+    @Published private(set) var storeURL: URL?
+
+    @Published private(set) var woocommerceAdminURL: URL?
 
     /// Child items
     ///
@@ -61,8 +49,6 @@ final class HubMenuViewModel: ObservableObject {
 
     private var storePickerCoordinator: StorePickerCoordinator?
 
-    private var cancellables = Set<AnyCancellable>()
-
     init(siteID: Int64,
          navigationController: UINavigationController? = nil,
          featureFlagService: FeatureFlagService = ServiceLocator.featureFlagService,
@@ -139,9 +125,27 @@ final class HubMenuViewModel: ObservableObject {
     }
 
     private func observeSiteForUIUpdates() {
-        stores.site.sink { site in
-            // This will be useful in the future for updating some info of the screen depending on the store site info
-        }.store(in: &cancellables)
+        stores.site
+            .compactMap { site -> URL? in
+                guard let urlString = site?.url, let url = URL(string: urlString) else {
+                    return nil
+                }
+                return url
+            }
+            .assign(to: &$storeURL)
+
+        stores.site
+            .compactMap { $0?.name }
+            .assign(to: &$storeTitle)
+
+        stores.site
+            .compactMap { site -> URL? in
+                guard let urlString = site?.adminURL, let url = URL(string: urlString) else {
+                    return site?.adminURLWithFallback()
+                }
+                return url
+            }
+            .assign(to: &$woocommerceAdminURL)
     }
 }