diff --git a/src/context/revoltjs/RevoltClient.tsx b/src/context/revoltjs/RevoltClient.tsx
index 26c90bc7..045b7f94 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 f0f22383..b79f0671 100644
--- a/src/sw.ts
+++ b/src/sw.ts
@@ -1,11 +1,117 @@
///
+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(store: string, key: string): Promise {
+ 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('channels', data.channel);
+ let user = await get('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('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