import db from "./db"; import { topicUrl } from "./utils"; class SubscriptionManager { /** All subscriptions, including "new count"; this is a JOIN, see https://dexie.org/docs/API-Reference#joining */ async all() { const subscriptions = await db.subscriptions.toArray(); return Promise.all( subscriptions.map(async (s) => ({ ...s, new: await db.notifications.where({ subscriptionId: s.id, new: 1 }).count(), })) ); } async get(subscriptionId) { return db.subscriptions.get(subscriptionId); } async add(baseUrl, topic, internal) { const id = topicUrl(baseUrl, topic); const existingSubscription = await this.get(id); if (existingSubscription) { return existingSubscription; } const subscription = { id: topicUrl(baseUrl, topic), baseUrl, topic, mutedUntil: 0, last: null, internal: internal || false, }; await db.subscriptions.put(subscription); return subscription; } async syncFromRemote(remoteSubscriptions, remoteReservations) { console.log(`[SubscriptionManager] Syncing subscriptions from remote`, remoteSubscriptions); // Add remote subscriptions const remoteIds = await Promise.all( remoteSubscriptions.map(async (remote) => { const local = await this.add(remote.base_url, remote.topic, false); const reservation = remoteReservations?.find((r) => remote.base_url === config.base_url && remote.topic === r.topic) || null; await this.update(local.id, { displayName: remote.display_name, // May be undefined reservation, // May be null! }); return local.id; }) ); // Remove local subscriptions that do not exist remotely const localSubscriptions = await db.subscriptions.toArray(); await Promise.all( localSubscriptions.map(async (local) => { const remoteExists = remoteIds.includes(local.id); if (!local.internal && !remoteExists) { await this.remove(local.id); } }) ); } async updateState(subscriptionId, state) { db.subscriptions.update(subscriptionId, { state }); } async remove(subscriptionId) { await db.subscriptions.delete(subscriptionId); await db.notifications.where({ subscriptionId }).delete(); } async first() { return db.subscriptions.toCollection().first(); // May be undefined } async getNotifications(subscriptionId) { // This is quite awkward, but it is the recommended approach as per the Dexie docs. // It's actually fine, because the reading and filtering is quite fast. The rendering is what's // killing performance. See https://dexie.org/docs/Collection/Collection.offset()#a-better-paging-approach return db.notifications .orderBy("time") // Sort by time first .filter((n) => n.subscriptionId === subscriptionId) .reverse() .toArray(); } async getAllNotifications() { return db.notifications .orderBy("time") // Efficient, see docs .reverse() .toArray(); } /** Adds notification, or returns false if it already exists */ async addNotification(subscriptionId, notification) { const exists = await db.notifications.get(notification.id); if (exists) { return false; } try { await db.notifications.add({ ...notification, subscriptionId, // New marker (used for bubble indicator); cannot be boolean; Dexie index limitation new: 1, }); // FIXME consider put() for double tab await db.subscriptions.update(subscriptionId, { last: notification.id, }); } catch (e) { console.error(`[SubscriptionManager] Error adding notification`, e); } return true; } /** Adds/replaces notifications, will not throw if they exist */ async addNotifications(subscriptionId, notifications) { const notificationsWithSubscriptionId = notifications.map((notification) => ({ ...notification, subscriptionId })); const lastNotificationId = notifications.at(-1).id; await db.notifications.bulkPut(notificationsWithSubscriptionId); await db.subscriptions.update(subscriptionId, { last: lastNotificationId, }); } async updateNotification(notification) { const exists = await db.notifications.get(notification.id); if (!exists) { return false; } try { await db.notifications.put({ ...notification }); } catch (e) { console.error(`[SubscriptionManager] Error updating notification`, e); } return true; } async deleteNotification(notificationId) { await db.notifications.delete(notificationId); } async deleteNotifications(subscriptionId) { await db.notifications.where({ subscriptionId }).delete(); } async markNotificationRead(notificationId) { await db.notifications.where({ id: notificationId }).modify({ new: 0 }); } async markNotificationsRead(subscriptionId) { await db.notifications.where({ subscriptionId, new: 1 }).modify({ new: 0 }); } async setMutedUntil(subscriptionId, mutedUntil) { await db.subscriptions.update(subscriptionId, { mutedUntil, }); } async setDisplayName(subscriptionId, displayName) { await db.subscriptions.update(subscriptionId, { displayName, }); } async setReservation(subscriptionId, reservation) { await db.subscriptions.update(subscriptionId, { reservation, }); } async update(subscriptionId, params) { await db.subscriptions.update(subscriptionId, params); } async pruneNotifications(thresholdTimestamp) { await db.notifications.where("time").below(thresholdTimestamp).delete(); } } const subscriptionManager = new SubscriptionManager(); export default subscriptionManager;