diff --git a/src/context/revoltjs/RevoltClient.tsx b/src/context/revoltjs/RevoltClient.tsx index 26c90bc79847e477632f0cc35eccf76ca96d5707..045b7f9496a9b6c6962de6d8ff45442da8b5eb43 100644 --- a/src/context/revoltjs/RevoltClient.tsx +++ b/src/context/revoltjs/RevoltClient.tsx @@ -51,6 +51,7 @@ function Context({ auth, sync, children, dispatcher }: Props) { (async () => { let db; try { + // Match sw.ts#L23 db = await openDB('state', 3, { upgrade(db) { for (let store of [ "channels", "servers", "users", "members" ]) { diff --git a/src/sw.ts b/src/sw.ts index f0f22383833fc03cd6ff265b9f169cb89e6c0959..b79f0671988eb0d52b7757694057d5bddfd17567 100644 --- a/src/sw.ts +++ b/src/sw.ts @@ -1,11 +1,117 @@ /// <reference lib="webworker" /> +import { Channel, Message, SYSTEM_USER_ID, User } from 'revolt.js' import { precacheAndRoute } from 'workbox-precaching' +import { Server } from 'revolt.js/dist/api/objects' +import { IDBPDatabase, openDB } from 'idb' +import { decodeTime } from 'ulid' declare let self: ServiceWorkerGlobalScope self.addEventListener('message', (event) => { - if (event.data && event.data.type === 'SKIP_WAITING') - self.skipWaiting() + if (event.data && event.data.type === 'SKIP_WAITING') + self.skipWaiting() }) precacheAndRoute(self.__WB_MANIFEST) + +const base_url = `https://autumn.revolt.chat`; +self.addEventListener("push", event => { + async function process() { + if (event.data === null) return; + let data: Message = event.data.json(); + + let db: IDBPDatabase; + try { + // Match RevoltClient.tsx#L55 + db = await openDB('state', 3, { + upgrade(db) { + for (let store of [ "channels", "servers", "users", "members" ]) { + db.createObjectStore(store, { + keyPath: '_id' + }); + } + }, + }); + } catch (err) { + console.error('Failed to open IndexedDB store, continuing without.'); + return; + } + + async function get<T>(store: string, key: string): Promise<T | undefined> { + try { + return await db.get(store, key); + } catch (err) { + return undefined; + } + } + + let image; + if (data.attachments) { + let attachment = data.attachments[0]; + if (attachment.metadata.type === "Image") { + image = `${base_url}/${attachment.tag}/${attachment._id}`; + } + } + + let title = `@${data.author}`; + let channel = await get<Channel>('channels', data.channel); + let user = await get<User>('users', data.author); + let username = user?.username ?? data.author; + + switch (channel?.channel_type) { + case "SavedMessages": break; + case "DirectMessage": title = `@${username}`; break; + case "Group": + if (user?._id === SYSTEM_USER_ID) { + title = channel.name; + } else { + title = `@${user?.username} - ${channel.name}`; + } + break; + case "TextChannel": + { + let server = await get<Server>('servers', channel.server); + title = `@${user?.username} (#${channel.name}, ${server?.name})`; + } + break; + } + + await self.registration.showNotification(title, { + icon: user?.avatar ? `${base_url}/${user.avatar.tag}/${user.avatar._id}` : `https://api.revolt.chat/users/${data.author}/default_avatar`, + image, + body: typeof data.content === "string" ? data.content : JSON.stringify(data.content), + timestamp: decodeTime(data._id), + tag: data.channel, + badge: "https://app.revolt.chat/assets/icons/android-chrome-512x512.png" + }); + } + + event.waitUntil(process()); +}); + +// ? Open the app on notification click. +// https://stackoverflow.com/a/39457287 +self.addEventListener("notificationclick", function(event) { + let url = event.notification.data; + event.notification.close(); + event.waitUntil( + self.clients + .matchAll({ includeUncontrolled: true, type: "window" }) + .then(windowClients => { + // Check if there is already a window/tab open with the target URL + for (var i = 0; i < windowClients.length; i++) { + var client = windowClients[i]; + // If so, just focus it. + if (client.url === url && "focus" in client) { + return client.focus(); + } + } + + // If not, then open the target URL in a new window/tab. + if (self.clients.openWindow) { + return self.clients.openWindow(url); + } + }) + ); +}); + \ No newline at end of file