Clear notifications count optimistically

2 files changed, 5 insertions, 0 deletions

diff --git a/src/state/models/me.ts b/src/state/models/me.ts
index cefd893ca..78c6d2e76 100644
--- a/src/state/models/me.ts
+++ b/src/state/models/me.ts
@@ -49,6 +49,10 @@ export class MeModel {
     }
   }
 
+  clearNotificationCount() {
+    this.notificationCount = 0
+  }
+
   async fetchStateUpdate() {
     const res = await this.rootStore.api.app.bsky.notification.getCount()
     runInAction(() => {
diff --git a/src/state/models/notifications-view.ts b/src/state/models/notifications-view.ts
index c328be62b..b841c7f3a 100644
--- a/src/state/models/notifications-view.ts
+++ b/src/state/models/notifications-view.ts
@@ -332,6 +332,7 @@ export class NotificationsViewModel {
       await this.rootStore.api.app.bsky.notification.updateSeen({
         seenAt: new Date().toISOString(),
       })
+      this.rootStore.me.clearNotificationCount()
     } catch (e) {
       console.log('Failed to update notifications read state', e)
     }