From 585b9e022974cddb559bcbeb967fd344ff36581a Mon Sep 17 00:00:00 2001
From: Daniel Roe <daniel@roe.dev>
Date: Fri, 2 Dec 2022 02:21:10 +0000
Subject: [PATCH] feat: show and stream new notifications (#282)

Co-authored-by: Anthony Fu <anthonyfu117@hotmail.com>
---
 components/common/CommonPaginator.vue         |  5 ++-
 components/nav/NavSide.vue                    | 15 ++++++-
 .../notification/NotificationPaginator.vue    | 14 +++++--
 composables/paginator.ts                      |  4 +-
 composables/users.ts                          | 39 ++++++++++++++++++-
 pages/notifications.vue                       |  7 +++-
 6 files changed, 74 insertions(+), 10 deletions(-)

diff --git a/components/common/CommonPaginator.vue b/components/common/CommonPaginator.vue
index e035b916..ad106717 100644
--- a/components/common/CommonPaginator.vue
+++ b/components/common/CommonPaginator.vue
@@ -4,11 +4,12 @@ import { DynamicScroller } from 'vue-virtual-scroller'
 import 'vue-virtual-scroller/dist/vue-virtual-scroller.css'
 import type { Paginator, WsEvents } from 'masto'
 
-const { paginator, stream, keyProp = 'id', virtualScroller = false } = defineProps<{
+const { paginator, stream, keyProp = 'id', virtualScroller = false, eventType = 'update' } = defineProps<{
   paginator: Paginator<any, any[]>
   keyProp?: string
   virtualScroller?: boolean
   stream?: WsEvents
+  eventType?: 'notification' | 'update'
 }>()
 
 defineSlots<{
@@ -23,7 +24,7 @@ defineSlots<{
   loading: {}
 }>()
 
-const { items, prevItems, update, state, endAnchor, error } = usePaginator(paginator, stream)
+const { items, prevItems, update, state, endAnchor, error } = usePaginator(paginator, stream, eventType)
 </script>
 
 <template>
diff --git a/components/nav/NavSide.vue b/components/nav/NavSide.vue
index da59c6de..72c753ed 100644
--- a/components/nav/NavSide.vue
+++ b/components/nav/NavSide.vue
@@ -1,8 +1,21 @@
+<script setup>
+const { notifications } = useNotifications()
+</script>
+
 <template>
   <nav px3 py4 flex="~ col gap2" text-lg>
     <template v-if="currentUser">
       <NavSideItem :text="$t('nav_side.home')" to="/home" icon="i-ri:home-5-line" />
-      <NavSideItem :text="$t('nav_side.notifications')" to="/notifications" icon="i-ri:notification-4-line" />
+      <NavSideItem :text="$t('nav_side.notifications')" to="/notifications" icon="i-ri:notification-4-line">
+        <template #icon>
+          <div flex relative>
+            <div class="i-ri:notification-4-line" />
+            <div v-if="notifications" class="top-[-0.3rem] right-[-0.3rem]" absolute font-bold rounded-full h-4 w-4 text-xs bg-primary text-inverted flex items-center justify-center>
+              {{ notifications < 10 ? notifications : '•' }}
+            </div>
+          </div>
+        </template>
+      </NavSideItem>
     </template>
     <NavSideItem :text="$t('nav_side.explore')" to="/explore" icon="i-ri:hashtag" />
     <NavSideItem :text="$t('nav_side.local')" to="/public/local" icon="i-ri:group-2-line " />
diff --git a/components/notification/NotificationPaginator.vue b/components/notification/NotificationPaginator.vue
index 65f34440..7792b485 100644
--- a/components/notification/NotificationPaginator.vue
+++ b/components/notification/NotificationPaginator.vue
@@ -1,9 +1,10 @@
 <script setup lang="ts">
-import type { Notification, Paginator } from 'masto'
+import type { Notification, Paginator, WsEvents } from 'masto'
 import type { GroupedNotifications } from '~/types'
 
-const { paginator } = defineProps<{
+const { paginator, stream } = defineProps<{
   paginator: Paginator<any, Notification[]>
+  stream?: WsEvents
 }>()
 
 function groupItems(items: Notification[]): (Notification | GroupedNotifications)[] {
@@ -41,10 +42,17 @@ function groupItems(items: Notification[]): (Notification | GroupedNotifications
 
   return results
 }
+
+const { clearNotifications } = useNotifications()
 </script>
 
 <template>
-  <CommonPaginator :paginator="paginator">
+  <CommonPaginator :paginator="paginator" :stream="stream" event-type="notification">
+    <template #updater="{ number, update }">
+      <button py-4 border="b base" flex="~ col" p-3 w-full text-primary font-bold @click="() => { update(); clearNotifications() }">
+        {{ $t('timeline.show_new_items', [number]) }}
+      </button>
+    </template>
     <template #items="{ items }">
       <template v-for="item of groupItems(items)" :key="item.id">
         <NotificationGroupedFollow
diff --git a/composables/paginator.ts b/composables/paginator.ts
index 45ae70af..e6b09f17 100644
--- a/composables/paginator.ts
+++ b/composables/paginator.ts
@@ -2,7 +2,7 @@ import type { Paginator, WsEvents } from 'masto'
 import { useDeactivated } from './lifecycle'
 import type { PaginatorState } from '~/types'
 
-export function usePaginator<T>(paginator: Paginator<any, T[]>, stream?: WsEvents) {
+export function usePaginator<T>(paginator: Paginator<any, T[]>, stream?: WsEvents, eventType: 'notification' | 'update' = 'update') {
   const state = ref<PaginatorState>('idle')
   const items = ref<T[]>([])
   const nextItems = ref<T[]>([])
@@ -19,7 +19,7 @@ export function usePaginator<T>(paginator: Paginator<any, T[]>, stream?: WsEvent
     prevItems.value = []
   }
 
-  stream?.on('update', (status) => {
+  stream?.on(eventType, (status) => {
     prevItems.value.unshift(status as any)
   })
 
diff --git a/composables/users.ts b/composables/users.ts
index ff7d32df..019f67d8 100644
--- a/composables/users.ts
+++ b/composables/users.ts
@@ -1,5 +1,5 @@
 import { login as loginMasto } from 'masto'
-import type { AccountCredentials, Instance } from 'masto'
+import type { AccountCredentials, Instance, WsEvents } from 'masto'
 import { clearUserDrafts } from './statusDrafts'
 import type { UserLogin } from '~/types'
 import { DEFAULT_POST_CHARS_LIMIT, DEFAULT_SERVER, STORAGE_KEY_CURRENT_USER, STORAGE_KEY_SERVERS, STORAGE_KEY_USERS } from '~/constants'
@@ -97,6 +97,43 @@ export async function signout() {
   await loginTo(currentUser.value)
 }
 
+const notifications = reactive<Record<string, undefined | [Promise<WsEvents>, number]>>({})
+
+export const useNotifications = () => {
+  const id = currentUser.value?.account.id
+
+  const clearNotifications = () => {
+    if (!id || !notifications[id])
+      return
+    notifications[id]![1] = 0
+  }
+
+  async function connect(): Promise<void> {
+    if (!id || notifications[id])
+      return
+
+    const masto = useMasto()
+    const stream = masto.stream.streamUser()
+    notifications[id] = [stream, 0]
+    ;(await stream).on('notification', () => {
+      if (notifications[id])
+        notifications[id]![1]++
+    })
+  }
+
+  function disconnect(): void {
+    if (!id || !notifications[id])
+      return
+    notifications[id]![0].then(stream => stream.disconnect())
+    notifications[id] = undefined
+  }
+
+  watch(currentUser, disconnect)
+  connect()
+
+  return { notifications: computed(() => id ? notifications[id]?.[1] ?? 0 : 0), disconnect, clearNotifications }
+}
+
 export function checkLogin() {
   if (!currentUser.value) {
     openSigninDialog()
diff --git a/pages/notifications.vue b/pages/notifications.vue
index b50f6b19..41aab53e 100644
--- a/pages/notifications.vue
+++ b/pages/notifications.vue
@@ -8,6 +8,11 @@ const { t } = useI18n()
 const paginatorAll = useMasto().notifications.getIterator()
 const paginatorMention = useMasto().notifications.getIterator({ types: ['mention'] })
 
+const { clearNotifications } = useNotifications()
+onActivated(clearNotifications)
+
+const stream = await useMasto().stream.streamUser()
+
 const tabs = $computed(() => [
   {
     name: 'all',
@@ -43,7 +48,7 @@ useHeadFixed({
       <CommonTabs v-model="tab" :options="tabs" />
     </template>
     <slot>
-      <NotificationPaginator :key="tab" :paginator="paginator" />
+      <NotificationPaginator :key="tab" v-bind="{ paginator, stream }" />
     </slot>
   </MainContent>
 </template>