calckey/packages/sw/src/scripts/notification-read.ts

declare var self: ServiceWorkerGlobalScope;

import { get } from "idb-keyval";
import { pushNotificationDataMap } from "@/types";
import { api } from "@/scripts/operations";

type Accounts = {
	[x: string]: {
		queue: string[];
		timeout: number | null;
	};
};

class SwNotificationReadManager {
	private accounts: Accounts = {};

	public async construct() {
		const accounts = await get("accounts");
		if (!accounts) Error("Accounts are not recorded");

		this.accounts = accounts.reduce((acc, e) => {
			acc[e.id] = {
				queue: [],
				timeout: null,
			};
			return acc;
		}, {} as Accounts);

		return this;
	}

	// プッシュ通知の既読をサーバーに送信
	public async read<K extends keyof pushNotificationDataMap>(
		data: pushNotificationDataMap[K],
	) {
		if (data.type !== "notification" || !(data.userId in this.accounts)) return;

		const account = this.accounts[data.userId];

		account.queue.push(data.body.id as string);

		if (account.queue.length >= 20) {
			if (account.timeout) clearTimeout(account.timeout);
			const notificationIds = account.queue;
			account.queue = [];
			await api("notifications/read", data.userId, { notificationIds });
			return;
		}

		// 最後の呼び出しから200ms待ってまとめて処理する
		if (account.timeout) clearTimeout(account.timeout);
		account.timeout = setTimeout(() => {
			account.timeout = null;

			const notificationIds = account.queue;
			account.queue = [];
			api("notifications/read", data.userId, { notificationIds });
		}, 200);
	}
}

export const swNotificationRead = new SwNotificationReadManager().construct();