diff --git a/Riot/Modules/Spaces/SpaceList/SpaceListViewModel.swift b/Riot/Modules/Spaces/SpaceList/SpaceListViewModel.swift index 9a7c030c8..ed11d2df9 100644 --- a/Riot/Modules/Spaces/SpaceList/SpaceListViewModel.swift +++ b/Riot/Modules/Spaces/SpaceList/SpaceListViewModel.swift @@ -53,6 +53,8 @@ final class SpaceListViewModel: SpaceListViewModelType { self.userSessionsService = userSessionsService NotificationCenter.default.addObserver(self, selector: #selector(self.sessionDidSync(notification:)), name: MXSpaceService.didBuildSpaceGraph, object: nil) + + NotificationCenter.default.addObserver(self, selector: #selector(self.counterDidUpdateNotificationCount(notification:)), name: MXSpaceNotificationCounter.didUpdateNotificationCount, object: nil) } @@ -122,6 +124,10 @@ final class SpaceListViewModel: SpaceListViewModelType { loadData() } + @objc private func counterDidUpdateNotificationCount(notification: Notification) { + loadData() + } + private func loadData() { guard let session = self.userSessionsService.mainUserSession?.matrixSession else { return