Add badge in spaces button (PSG-944)

Riot/Modules/Home/AllChats/AllChatsViewController.swift

@@ -72,6 +72,10 @@ class AllChatsViewController: HomeViewController {
     private var isOnboardingCoordinatorPreparing: Bool = false
 
     private var allChatsOnboardingCoordinatorBridgePresenter: AllChatsOnboardingCoordinatorBridgePresenter?
+
+    private var theme: Theme {
+        ThemeService.shared().theme
+    }
 
     @IBOutlet private var toolbar: UIToolbar!
     private var isToolbarHidden: Bool = false {
@@ -137,12 +141,13 @@ class AllChatsViewController: HomeViewController {
         searchController.delegate = self
 
         NotificationCenter.default.addObserver(self, selector: #selector(self.setupEditOptions), name: AllChatsLayoutSettingsManager.didUpdateSettings, object: nil)
+        NotificationCenter.default.addObserver(self, selector: #selector(self.updateBadgeButton), name: MXSpaceNotificationCounter.didUpdateNotificationCount, object: nil)
     }
 
     override func viewWillAppear(_ animated: Bool) {
         super.viewWillAppear(animated)
 
-        self.toolbar.tintColor = ThemeService.shared().theme.colors.accent
+        self.toolbar.tintColor = theme.colors.accent
         if self.navigationItem.searchController == nil {
             self.navigationItem.searchController = searchController
         }
@@ -460,7 +465,7 @@ class AllChatsViewController: HomeViewController {
             return
         }
 
-        self.update(with: ThemeService.shared().theme)
+        self.update(with: theme)
     }
 
     private func update(with theme: Theme) {
@@ -500,22 +505,40 @@ class AllChatsViewController: HomeViewController {
             self?.updateToolbar(with: menu)
         }))
         updateEmptyView()
+        updateBadgeButton()
     }
 
     private func updateRightNavigationItem(with menu: UIMenu) {
         self.navigationItem.rightBarButtonItem = UIBarButtonItem(image: UIImage(systemName: "ellipsis.circle"), menu: menu)
     }
 
+    private lazy var spacesButton: BadgedBarButtonItem = {
+        let innerButton = UIButton(type: .system)
+        innerButton.accessibilityLabel = VectorL10n.spaceSelectorTitle
+        innerButton.addTarget(self, action: #selector(self.showSpaceSelectorAction(sender:)), for: .touchUpInside)
+        innerButton.setImage(Asset.Images.allChatsSpacesIcon.image, for: .normal)
+        return BadgedBarButtonItem(withBaseButton: innerButton, theme: theme)
+    }()
+
+    @objc private func updateBadgeButton() {
+        guard isViewLoaded else {
+            return
+        }
+
+        let notificationState = mainSession?.spaceService.notificationCounter.notificationState(forAllSpacesExcept: nil)
+        spacesButton.badgeText = notificationState.map { "\($0.allCount)" }
+        spacesButton.badgeBackgroundColor = notificationState.map {
+            $0.allHighlightCount > 0 ? theme.noticeColor : theme.noticeSecondaryColor
+        } ?? .clear
+    }
+
     private func updateToolbar(with menu: UIMenu) {
         guard isViewLoaded else {
             return
         }
 
         self.isToolbarHidden = false
-        self.update(with: ThemeService.shared().theme)
-
-        let spacesButton = UIBarButtonItem(image: Asset.Images.allChatsSpacesIcon.image, style: .done, target: self, action: #selector(self.showSpaceSelectorAction(sender: )))
-        spacesButton.accessibilityLabel = VectorL10n.spaceSelectorTitle
+        self.update(with: theme)
 
         self.toolbar.items = [
             spacesButton,

changelog.d/pr-7088.change

@@ -0,0 +1 @@
+Add badge for messages in spaces button.