From f95ca164d63ea372e812b61e3c622a5cd4fe7400 Mon Sep 17 00:00:00 2001
From: tamaina <tamaina@hotmail.co.jp>
Date: Wed, 10 Feb 2021 01:54:07 +0900
Subject: [PATCH] wip

---
 src/client/components/notification.vue       |  13 +-
 src/client/components/notifications.vue      |  15 ++
 src/client/scripts/mark-notification-read.ts |   3 +
 src/client/sw/compose-notification.ts        | 107 ---------------
 src/client/sw/lang.ts                        |  44 ++++++
 src/client/sw/notification.ts                | 137 +++++++++++++++++++
 src/client/sw/sw.ts                          |  97 +++++--------
 src/server/api/common/read-notification.ts   |  11 +-
 src/services/create-notification.ts          |   4 +-
 src/services/messages/create.ts              |   2 +-
 src/services/push-notification.ts            |   2 +-
 11 files changed, 259 insertions(+), 176 deletions(-)
 create mode 100644 src/client/scripts/mark-notification-read.ts
 delete mode 100644 src/client/sw/compose-notification.ts
 create mode 100644 src/client/sw/lang.ts
 create mode 100644 src/client/sw/notification.ts

diff --git a/src/client/components/notification.vue b/src/client/components/notification.vue
index 2fe8500891..4ae8547162 100644
--- a/src/client/components/notification.vue
+++ b/src/client/components/notification.vue
@@ -68,6 +68,7 @@ import notePage from '../filters/note';
 import { userPage } from '../filters/user';
 import { i18n } from '@/i18n';
 import * as os from '@/os';
+import { markNotificationRead } from '@/scripts/mark-notification-read';
 
 export default defineComponent({
 	components: {
@@ -113,7 +114,17 @@ export default defineComponent({
 			this.readObserver.observe(this.$el);
 
 			this.connection = os.stream.useSharedConnection('main');
-			this.connection.on('readAllNotifications', () => this.readObserver.unobserve(this.$el));
+
+			this.connection.on('readAllNotifications', () => {
+				this.readObserver.unobserve(this.$el);
+				this.notification = markNotificationRead(this.notification);
+			});
+			this.connection.on('readNotifications', notificationIds => {
+				if (notificationIds.includes(this.notification.id)) {
+					this.readObserver.unobserve(this.$el);
+					this.notification = markNotificationRead(this.notification);
+				}
+			})
 		}
 	},
 
diff --git a/src/client/components/notifications.vue b/src/client/components/notifications.vue
index 9759cc2395..b19b7ff7a3 100644
--- a/src/client/components/notifications.vue
+++ b/src/client/components/notifications.vue
@@ -19,6 +19,7 @@
 <script lang="ts">
 import { defineComponent, PropType } from 'vue';
 import paging from '@/scripts/paging';
+import { markNotificationRead } from '@/scripts/mark-notification-read';
 import XNotification from './notification.vue';
 import XList from './date-separated-list.vue';
 import XNote from './note.vue';
@@ -85,6 +86,20 @@ export default defineComponent({
 	mounted() {
 		this.connection = os.stream.useSharedConnection('main');
 		this.connection.on('notification', this.onNotification);
+
+		// queueに対してのみ既読処理を行う
+		this.connection.on('readAllNotifications', () => {
+			this.queue = this.queue.map(markNotificationRead);
+		});
+		this.connection.on('readNotifications', notificationIds => {
+			if (this.queue.length === 0) return;
+
+			for (let i = 0; i < this.queue.length; i++) {
+				if (notificationIds.includes(this.queue[i].id)) {
+					this.queue[i] = markNotificationRead(this.queue[i]);
+				}
+			}
+		});
 	},
 
 	beforeUnmount() {
diff --git a/src/client/scripts/mark-notification-read.ts b/src/client/scripts/mark-notification-read.ts
new file mode 100644
index 0000000000..36aaec365f
--- /dev/null
+++ b/src/client/scripts/mark-notification-read.ts
@@ -0,0 +1,3 @@
+export function markNotificationRead(notification: any) {
+    return { ...notification, isRead: true };
+}
diff --git a/src/client/sw/compose-notification.ts b/src/client/sw/compose-notification.ts
deleted file mode 100644
index 72ad4a8f8a..0000000000
--- a/src/client/sw/compose-notification.ts
+++ /dev/null
@@ -1,107 +0,0 @@
-/**
- * Notification composer of Service Worker
- */
-declare var self: ServiceWorkerGlobalScope;
-
-import { getNoteSummary } from '../../misc/get-note-summary';
-import getUserName from '../../misc/get-user-name';
-
-export default async function(data, i18n): Promise<[string, NotificationOptions] | null | undefined> {
-	if (!i18n) {
-		console.log('no i18n');
-		return;
-	}
-
-	switch (data.type) {
-		case 'driveFileCreated': // TODO (Server Side)
-			return [i18n.t('_notification.fileUploaded'), {
-				body: data.body.name,
-				icon: data.body.url,
-				data
-			}];
-		case 'notification':
-			switch (data.body.type) {
-				case 'mention':
-					return [i18n.t('_notification.youGotMention', { name: getUserName(data.body.user) }), {
-						body: getNoteSummary(data.body.note, i18n.locale),
-						icon: data.body.user.avatarUrl,
-						data,
-						actions: [
-							{
-								action: 'showUser',
-								title: 'showUser'
-							}
-						]
-					}];
-
-				case 'reply':
-					return [i18n.t('_notification.youGotReply', { name: getUserName(data.body.user) }), {
-						body: getNoteSummary(data.body.note, i18n.locale),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'renote':
-					return [i18n.t('_notification.youRenoted', { name: getUserName(data.body.user) }), {
-						body: getNoteSummary(data.body.note, i18n.locale),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'quote':
-					return [i18n.t('_notification.youGotQuote', { name: getUserName(data.body.user) }), {
-						body: getNoteSummary(data.body.note, i18n.locale),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'reaction':
-					return [`${data.body.reaction} ${getUserName(data.body.user)}`, {
-						body: getNoteSummary(data.body.note, i18n.locale),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'pollVote':
-					return [i18n.t('_notification.youGotPoll', { name: getUserName(data.body.user) }), {
-						body: getNoteSummary(data.body.note, i18n.locale),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'follow':
-					return [i18n.t('_notification.youWereFollowed'), {
-						body: getUserName(data.body.user),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'receiveFollowRequest':
-					return [i18n.t('_notification.youReceivedFollowRequest'), {
-						body: getUserName(data.body.user),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'followRequestAccepted':
-					return [i18n.t('_notification.yourFollowRequestAccepted'), {
-						body: getUserName(data.body.user),
-						icon: data.body.user.avatarUrl
-					}];
-
-				case 'groupInvited':
-					return [i18n.t('_notification.youWereInvitedToGroup'), {
-						body: data.body.group.name
-					}];
-
-				default:
-					return null;
-			}
-		case 'unreadMessagingMessage':
-			if (data.body.groupId === null) {
-				return [i18n.t('_notification.youGotMessagingMessageFromUser', { name: getUserName(data.body.user) }), {
-					icon: data.body.user.avatarUrl,
-					tag: `messaging:user:${data.body.user.id}`
-				}];
-			}
-			return [i18n.t('_notification.youGotMessagingMessageFromGroup', { name: data.body.group.name }), {
-				icon: data.body.user.avatarUrl,
-				tag: `messaging:group:${data.body.group.id}`
-			}];
-		default:
-			return null;
-	}
-}
diff --git a/src/client/sw/lang.ts b/src/client/sw/lang.ts
new file mode 100644
index 0000000000..06a16729d9
--- /dev/null
+++ b/src/client/sw/lang.ts
@@ -0,0 +1,44 @@
+/*
+ * Language manager for SW
+ */
+
+declare var self: ServiceWorkerGlobalScope;
+
+import { get, set } from 'idb-keyval';
+import { I18n } from '@/scripts/i18n';
+
+class SwLang {
+    public cacheName = `mk-cache-${_VERSION_}`;
+
+    public lang: Promise<string> = get('lang').then(async prelang => {
+        if (!prelang) return 'en-US';
+        return prelang;
+    });
+
+    public i18n: I18n<any> | null = null;
+
+    public setLang(newLang: string) {
+        this.i18n = null;
+        this.lang = Promise.resolve(newLang);
+		set('lang', newLang);
+		return this.fetchLocale();
+    } 
+
+    public async fetchLocale() {
+        // Service Workerは何度も起動しそのたびにlocaleを読み込むので、CacheStorageを使う
+        const localeUrl = `/assets/locales/${await this.lang}.${_VERSION_}.json`;
+        let localeRes = await caches.match(localeUrl);
+    
+        if (!localeRes) {
+            localeRes = await fetch(localeUrl);
+            const clone = localeRes?.clone();
+            if (!clone?.clone().ok) return;
+    
+            caches.open(this.cacheName).then(cache => cache.put(localeUrl, clone));
+        }
+
+        return this.i18n = new I18n(await localeRes.json());
+    }
+}
+
+export const swLang = new SwLang();
diff --git a/src/client/sw/notification.ts b/src/client/sw/notification.ts
new file mode 100644
index 0000000000..3680554efe
--- /dev/null
+++ b/src/client/sw/notification.ts
@@ -0,0 +1,137 @@
+/*
+ * Notification manager for SW
+ */
+
+declare var self: ServiceWorkerGlobalScope;
+
+import { getNoteSummary } from '../../misc/get-note-summary';
+import getUserName from '../../misc/get-user-name';
+import { swLang } from '@/sw/lang'
+
+class SwNotification {
+	private queue: any[] = [];
+
+	private fetching = false;
+
+	public async append(data) {
+		if (swLang.i18n) {
+			const n = await this.composeNotification(data);
+			if (n) return self.registration.showNotification(...n);
+		} else {
+			this.queue.push(data);
+			if (this.fetching == false) {
+				this.fetching = true;
+				await swLang.fetchLocale();
+				this.fetching = false;
+				const promises = this.queue.map(this.composeNotification).map(n => {
+					if (!n) return;
+					return self.registration.showNotification(...n);
+				})
+				this.queue = [];
+				return Promise.all(promises);
+			}
+		}
+	}
+
+	private composeNotification(data): [string, NotificationOptions] | null | undefined {
+		const { i18n } = swLang;
+		if (!i18n) return;
+		const { t } = i18n;
+
+		switch (data.type) {
+			case 'driveFileCreated': // TODO (Server Side)
+				return [t('_notification.fileUploaded'), {
+					body: data.body.name,
+					icon: data.body.url,
+					data
+				}];
+			case 'notification':
+				switch (data.body.type) {
+					case 'mention':
+						return [t('_notification.youGotMention', { name: getUserName(data.body.user) }), {
+							body: getNoteSummary(data.body.note, i18n.locale),
+							icon: data.body.user.avatarUrl,
+							data,
+							actions: [
+								{
+									action: 'showUser',
+									title: 'showUser'
+								}
+							]
+						}];
+	
+					case 'reply':
+						return [t('_notification.youGotReply', { name: getUserName(data.body.user) }), {
+							body: getNoteSummary(data.body.note, i18n.locale),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'renote':
+						return [t('_notification.youRenoted', { name: getUserName(data.body.user) }), {
+							body: getNoteSummary(data.body.note, i18n.locale),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'quote':
+						return [t('_notification.youGotQuote', { name: getUserName(data.body.user) }), {
+							body: getNoteSummary(data.body.note, i18n.locale),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'reaction':
+						return [`${data.body.reaction} ${getUserName(data.body.user)}`, {
+							body: getNoteSummary(data.body.note, i18n.locale),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'pollVote':
+						return [t('_notification.youGotPoll', { name: getUserName(data.body.user) }), {
+							body: getNoteSummary(data.body.note, i18n.locale),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'follow':
+						return [t('_notification.youWereFollowed'), {
+							body: getUserName(data.body.user),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'receiveFollowRequest':
+						return [t('_notification.youReceivedFollowRequest'), {
+							body: getUserName(data.body.user),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'followRequestAccepted':
+						return [t('_notification.yourFollowRequestAccepted'), {
+							body: getUserName(data.body.user),
+							icon: data.body.user.avatarUrl
+						}];
+	
+					case 'groupInvited':
+						return [t('_notification.youWereInvitedToGroup'), {
+							body: data.body.group.name
+						}];
+	
+					default:
+						return null;
+				}
+			case 'unreadMessagingMessage':
+				if (data.body.groupId === null) {
+					return [t('_notification.youGotMessagingMessageFromUser', { name: getUserName(data.body.user) }), {
+						icon: data.body.user.avatarUrl,
+						tag: `messaging:user:${data.body.user.id}`
+					}];
+				}
+				return [t('_notification.youGotMessagingMessageFromGroup', { name: data.body.group.name }), {
+					icon: data.body.user.avatarUrl,
+					tag: `messaging:group:${data.body.group.id}`
+				}];
+			default:
+				return null;
+		}
+	}
+	
+}
+
+export const swNotification = new SwNotification();
diff --git a/src/client/sw/sw.ts b/src/client/sw/sw.ts
index 385f1be60d..0569b87ce6 100644
--- a/src/client/sw/sw.ts
+++ b/src/client/sw/sw.ts
@@ -3,49 +3,13 @@
  */
 declare var self: ServiceWorkerGlobalScope;
 
-import { get, set } from 'idb-keyval';
-import composeNotification from '@/sw/compose-notification';
-import { I18n } from '@/scripts/i18n';
+import { get } from 'idb-keyval';
+import { swNotification } from '@/sw/notification';
+import { swLang } from '@/sw/lang';
 
 //#region Variables
-const version = _VERSION_;
-const cacheName = `mk-cache-${version}`;
+// const cacheName = `mk-cache-${_VERSION_}`;
 const apiUrl = `${location.origin}/api/`;
-
-let lang: Promise<string> = get('lang').then(async prelang => {
-	if (!prelang) return 'en-US';
-	return prelang;
-});
-let i18n: I18n<any>;
-let pushesPool: any[] = [];
-//#endregion
-
-//#region Function: (Re)Load i18n instance
-async function fetchLocale() {
-	//#region localeファイルの読み込み
-	// Service Workerは何度も起動しそのたびにlocaleを読み込むので、CacheStorageを使う
-	const localeUrl = `/assets/locales/${await lang}.${version}.json`;
-	let localeRes = await caches.match(localeUrl);
-
-	if (!localeRes) {
-		localeRes = await fetch(localeUrl);
-		const clone = localeRes?.clone();
-		if (!clone?.clone().ok) return;
-
-		caches.open(cacheName).then(cache => cache.put(localeUrl, clone));
-	}
-
-	i18n = new I18n(await localeRes.json());
-	//#endregion
-
-	//#region i18nをきちんと読み込んだ後にやりたい処理
-	for (const data of pushesPool) {
-		const n = await composeNotification(data, i18n);
-		if (n) self.registration.showNotification(...n);
-	}
-	pushesPool = [];
-	//#endregion
-}
 //#endregion
 
 //#region Lifecycle: Install
@@ -60,7 +24,7 @@ self.addEventListener('activate', ev => {
 		caches.keys()
 			.then(cacheNames => Promise.all(
 				cacheNames
-					.filter((v) => v !== cacheName)
+					.filter((v) => v !== swLang.cacheName)
 					.map(name => caches.delete(name))
 			))
 			.then(() => self.clients.claim())
@@ -93,11 +57,22 @@ self.addEventListener('push', ev => {
 
 		const data = ev.data?.json();
 
-		// localeを読み込めておらずi18nがundefinedだった場合はpushesPoolにためておく
-		if (!i18n) return pushesPool.push(data);
-
-		const n = await composeNotification(data, i18n);
-		if (n) return self.registration.showNotification(...n);
+		switch (data.type) {
+			case 'notification':
+			case 'driveFileCreated':
+			case 'unreadMessagingMessage':
+				return swNotification.append(data);
+			case 'readAllNotifications':
+				for (const n of await self.registration.getNotifications()) {
+					n.close();
+				}
+				break;
+			case 'readNotifications':
+				for (const n of await self.registration.getNotifications()) {
+					if (data.notificationIds.includes(n.data.body.id)) n.close();
+				}
+				break;
+		}
 	}));
 });
 //#endregion
@@ -108,16 +83,18 @@ self.addEventListener('notificationclick', ev => {
 	const { data } = notification;
 	const { origin } = location;
 
+	const suffix = `?loginId=${data.userId}`;
+
 	switch (action) {
 		case 'showUser':
 			switch (data.body.type) {
 				case 'reaction':
-					self.clients.openWindow(`${origin}/users/${data.body.user.id}`);
+					self.clients.openWindow(`${origin}/users/${data.body.user.id}${suffix}`);
 					break;
 
 				default:
 					if ('note' in data.body) {
-						self.clients.openWindow(`${origin}/notes/${data.body.note.id}`);
+						self.clients.openWindow(`${origin}/users/${data.body.note.user.id}${suffix}`);
 					}
 			}
 			break;
@@ -128,8 +105,10 @@ self.addEventListener('notificationclick', ev => {
 });
 
 self.addEventListener('notificationclose', async ev => {
-	self.registration.showNotification('notificationclose');
 	const { notification } = ev;
+	if (notification.title !== 'notificationclose') {
+		self.registration.showNotification('notificationclose', { body: `${notification.data.id}` });
+	}
 	const { data } = notification;
 
 	if (data.isNotification) {
@@ -140,13 +119,15 @@ self.addEventListener('notificationclose', async ev => {
 
 		if (!account) return;
 
-		fetch(`${origin}/api/notifications/read`, {
-			method: 'POST',
-			body: JSON.stringify({
-				i: account.token,
-				notificationIds: [data.data.id]
-			})
-		});
+		if (data.type === 'notification') {
+			fetch(`${origin}/api/notifications/read`, {
+				method: 'POST',
+				body: JSON.stringify({
+					i: account.token,
+					notificationIds: [data.body.id]
+				})
+			});
+		}
 	}
 });
 //#endregion
@@ -166,9 +147,7 @@ self.addEventListener('message', ev => {
 
 		if (otype === 'object') {
 			if (ev.data.msg === 'initialize') {
-				lang = Promise.resolve(ev.data.lang);
-				set('lang', ev.data.lang);
-				fetchLocale();
+				swLang.setLang(ev.data.lang);
 			}
 		}
 	}
diff --git a/src/server/api/common/read-notification.ts b/src/server/api/common/read-notification.ts
index 4cc50468e0..3811d2db0e 100644
--- a/src/server/api/common/read-notification.ts
+++ b/src/server/api/common/read-notification.ts
@@ -1,4 +1,5 @@
 import { publishMainStream } from '../../../services/stream';
+import { pushNotification } from '../../../services/push-notification';
 import { User } from '../../../models/entities/user';
 import { Notification } from '../../../models/entities/notification';
 import { Notifications, Users } from '../../../models';
@@ -12,7 +13,7 @@ export async function readNotification(
 	notificationIds: Notification['id'][]
 ) {
 	// Update documents
-	const updatedNotificatons = await Notifications.update({
+	await Notifications.update({
 		id: In(notificationIds),
 		isRead: false
 	}, {
@@ -23,10 +24,10 @@ export async function readNotification(
 		// ユーザーのすべての通知が既読だったら、
 		// 全ての(いままで未読だった)通知を(これで)読みましたよというイベントを発行
 		publishMainStream(userId, 'readAllNotifications');
+		pushNotification(userId, 'readAllNotifications', undefined);
 	} else {
-		// まだすべて既読になっていなければ、
-		// アップデートしたという
-		const updatedNotificationsIds = updatedNotificatons.generatedMaps.map(e => e.id);
-		publishMainStream(userId, 'readNotifications', updatedNotificationsIds);
+		// まだすべて既読になっていなければ、各クライアントにnotificationIdsを伝達
+		publishMainStream(userId, 'readNotifications', notificationIds);
+		pushNotification(userId, 'readNotifications', { notificationIds });
 	}
 }
diff --git a/src/services/create-notification.ts b/src/services/create-notification.ts
index 5dddaa5727..4593f33f00 100644
--- a/src/services/create-notification.ts
+++ b/src/services/create-notification.ts
@@ -1,5 +1,5 @@
 import { publishMainStream } from './stream';
-import pushSw from './push-notification';
+import { pushNotification } from './push-notification';
 import { Notifications, Mutings, UserProfiles } from '../models';
 import { genId } from '../misc/gen-id';
 import { User } from '../models/entities/user';
@@ -50,7 +50,7 @@ export async function createNotification(
 
 			publishMainStream(notifieeId, 'unreadNotification', packed);
 
-			pushSw(notifieeId, 'notification', packed);
+			pushNotification(notifieeId, 'notification', packed);
 		}
 	}, 2000);
 
diff --git a/src/services/messages/create.ts b/src/services/messages/create.ts
index 8646ce37fc..d8fb749af9 100644
--- a/src/services/messages/create.ts
+++ b/src/services/messages/create.ts
@@ -5,7 +5,7 @@ import { MessagingMessages, UserGroupJoinings, Mutings, Users } from '../../mode
 import { genId } from '../../misc/gen-id';
 import { MessagingMessage } from '../../models/entities/messaging-message';
 import { publishMessagingStream, publishMessagingIndexStream, publishMainStream, publishGroupMessagingStream } from '../stream';
-import pushNotification from '../push-notification';
+import { pushNotification } from '../push-notification';
 import { Not } from 'typeorm';
 import { Note } from '../../models/entities/note';
 import renderNote from '../../remote/activitypub/renderer/note';
diff --git a/src/services/push-notification.ts b/src/services/push-notification.ts
index 939a8b635c..f879dc094e 100644
--- a/src/services/push-notification.ts
+++ b/src/services/push-notification.ts
@@ -12,7 +12,7 @@ type pushNotificationsTypes = {
 	'readAllNotifications': undefined;
 };
 
-export default async function<T extends keyof pushNotificationsTypes>(userId: string, type: T, body: pushNotificationsTypes[T]) {
+export async function pushNotification<T extends keyof pushNotificationsTypes>(userId: string, type: T, body: pushNotificationsTypes[T]) {
 	const meta = await fetchMeta();
 
 	if (!meta.enableServiceWorker || meta.swPublicKey == null || meta.swPrivateKey == null) return;