@onyx-p/imlib-web 3.0.2 → 3.0.4

package/database/main.cjs

@@ -65,6 +65,8 @@ const READ_METHOD_NAMES = [
     'getMessageById',
     'getMessageByUId',
     'getMessageIdRange',
+    'getMessagesByType',
+    'getPinnedMessages',
     'searchMessages',
     'searchTextMessages',
     'getLegacyMigrationState'
@@ -75,6 +77,8 @@ const WRITE_METHOD_NAMES = [
     'clearAllCache',
     'removeMessagesByUId',
     'updateMessageReceiptStatus',
+    'setMessagePinned',
+    'unpinAllMessages',
     'clearBurnAfterReadingExpiredMessages',
     'upsertMessage',
     'convertToRecallMessages',

package/database/preload.cjs

@@ -12,6 +12,8 @@ const READ_METHOD_NAMES = [
     'getMessageById',
     'getMessageByUId',
     'getMessageIdRange',
+    'getMessagesByType',
+    'getPinnedMessages',
     'searchMessages',
     'searchTextMessages',
     'getLegacyMigrationState'
@@ -22,6 +24,8 @@ const WRITE_METHOD_NAMES = [
     'clearAllCache',
     'removeMessagesByUId',
     'updateMessageReceiptStatus',
+    'setMessagePinned',
+    'unpinAllMessages',
     'clearBurnAfterReadingExpiredMessages',
     'upsertMessage',
     'convertToRecallMessages',

package/database/worker.cjs

@@ -15,6 +15,8 @@ const READ_METHOD_NAMES = [
     'getMessageById',
     'getMessageByUId',
     'getMessageIdRange',
+    'getMessagesByType',
+    'getPinnedMessages',
     'searchMessages',
     'searchTextMessages',
     'getLegacyMigrationState'
@@ -25,6 +27,8 @@ const WRITE_METHOD_NAMES = [
     'clearAllCache',
     'removeMessagesByUId',
     'updateMessageReceiptStatus',
+    'setMessagePinned',
+    'unpinAllMessages',
     'clearBurnAfterReadingExpiredMessages',
     'upsertMessage',
     'convertToRecallMessages',
@@ -133,6 +137,26 @@ const SCHEMA_MIGRATIONS = [
           ON messages (sentTime DESC, messageId DESC);
       `);
         }
+    },
+    {
+        version: 5,
+        up(database) {
+            database.exec(`
+        CREATE INDEX messages_by_dialog_type_time
+          ON messages (dialogId, messageType, sentTime DESC, messageId DESC);
+      `);
+        }
+    },
+    {
+        version: 6,
+        up(database) {
+            database.exec(`
+        ALTER TABLE messages ADD COLUMN pinned INTEGER NOT NULL DEFAULT 0;
+
+        CREATE INDEX messages_by_dialog_pinned_time
+          ON messages (dialogId, pinned, sentTime DESC, messageId DESC);
+      `);
+        }
     }
 ];
 SCHEMA_MIGRATIONS[SCHEMA_MIGRATIONS.length - 1]?.version ?? 0;
@@ -165,7 +189,12 @@ function runMigrations(database, migrations = SCHEMA_MIGRATIONS) {
 
 const LEGACY_MIGRATION_KEY = 'legacy_indexeddb_migration';
 function parseMessage(row) {
-    return row ? JSON.parse(row.json) : null;
+    if (!row) {
+        return null;
+    }
+    const message = JSON.parse(row.json);
+    message.pinned = message.pinned ?? false;
+    return message;
 }
 function extractSearchText(message) {
     const content = message.content;
@@ -193,10 +222,10 @@ class MessageStore {
         const upsert = this.database.prepare(`
       INSERT INTO messages (
         messageId, messageUId, dialogId, sentTime, receivedStatus,
-        burnAfterReadingFlag, burnAfterReadingTime, messageType, json, searchText
+        burnAfterReadingFlag, burnAfterReadingTime, messageType, pinned, json, searchText
       ) VALUES (
         $messageId, $messageUId, $dialogId, $sentTime, $receivedStatus,
-        $burnAfterReadingFlag, $burnAfterReadingTime, $messageType, $json, $searchText
+        $burnAfterReadingFlag, $burnAfterReadingTime, $messageType, $pinned, $json, $searchText
       )
       ON CONFLICT(messageId) DO UPDATE SET
         messageUId = excluded.messageUId,
@@ -206,6 +235,7 @@ class MessageStore {
         burnAfterReadingFlag = excluded.burnAfterReadingFlag,
         burnAfterReadingTime = excluded.burnAfterReadingTime,
         messageType = excluded.messageType,
+        pinned = excluded.pinned,
         json = excluded.json,
         searchText = excluded.searchText;
     `);
@@ -225,6 +255,7 @@ class MessageStore {
                     burnAfterReadingFlag: message.burnAfterReadingFlag ? 1 : 0,
                     burnAfterReadingTime: message.burnAfterReadingTime ?? null,
                     messageType: message.messageType ?? null,
+                    pinned: message.pinned ? 1 : 0,
                     json: JSON.stringify(message),
                     searchText: extractSearchText(message)
                 });
@@ -268,6 +299,44 @@ class MessageStore {
             hasMore: hasAdditionalRow || !this.getDialogLoadedState(dialogId)
         };
     }
+    getMessagesByType(dialogId, messageTypes, timestamp = '0', count = 50, isForward = true) {
+        const types = [
+            ...new Set(messageTypes.map(type => normalizeInteger(type, 'messageType')))
+        ];
+        const safeCount = Math.max(0, Math.floor(count));
+        if (types.length === 0 || safeCount === 0) {
+            return { messages: [], hasMore: false };
+        }
+        const params = [dialogId, ...types];
+        const typePlaceholders = types.map(() => '?').join(', ');
+        let timeClause = '';
+        if (timestamp !== '0') {
+            params.push(normalizeInteger(timestamp, 'timestamp'));
+            timeClause = isForward ? 'AND sentTime < ?' : 'AND sentTime > ?';
+        }
+        params.push(safeCount + 1);
+        const direction = isForward ? 'DESC' : 'ASC';
+        const rows = this.database
+            .prepare(`SELECT json FROM messages
+         WHERE dialogId = ?
+           AND messageType IN (${typePlaceholders})
+           ${timeClause}
+         ORDER BY sentTime ${direction}, messageId ${direction}
+         LIMIT ?;`)
+            .all(params);
+        const hasAdditionalRow = rows.length > safeCount;
+        const messages = rows
+            .slice(0, safeCount)
+            .map(row => parseMessage(row))
+            .sort((left, right) => {
+            const byTime = Number(left.sentTime) - Number(right.sentTime);
+            return byTime || left.messageId - right.messageId;
+        });
+        return {
+            messages,
+            hasMore: hasAdditionalRow
+        };
+    }
     getLatestMessage(dialogId) {
         return parseMessage(this.database
             .prepare(`SELECT json FROM messages
@@ -310,6 +379,23 @@ class MessageStore {
             .prepare('SELECT json FROM messages WHERE messageUId = $messageUId;')
             .get({ messageUId }));
     }
+    getPinnedMessages(dialogId, count = 50) {
+        const safeCount = Math.max(0, Math.floor(count));
+        if (safeCount === 0) {
+            return [];
+        }
+        return this.database
+            .prepare(`SELECT json FROM messages
+         WHERE dialogId = $dialogId AND pinned = 1
+         ORDER BY sentTime DESC, messageId DESC
+         LIMIT $limit;`)
+            .all({ dialogId, limit: safeCount })
+            .map(row => parseMessage(row))
+            .sort((left, right) => {
+            const byTime = Number(left.sentTime) - Number(right.sentTime);
+            return byTime || left.messageId - right.messageId;
+        });
+    }
     getMessageIdRange() {
         const row = this.database
             .prepare('SELECT MIN(messageId) AS oldest, MAX(messageId) AS newest FROM messages;')
@@ -363,6 +449,48 @@ class MessageStore {
             }
         })();
     }
+    setMessagePinned(dialogId, messageUId, pinned) {
+        const row = this.database
+            .prepare(`SELECT json FROM messages
+         WHERE dialogId = $dialogId AND messageUId = $messageUId;`)
+            .get({ dialogId, messageUId });
+        const message = parseMessage(row);
+        if (!message) {
+            return;
+        }
+        message.pinned = pinned;
+        this.database
+            .prepare(`UPDATE messages SET pinned = $pinned, json = $json
+         WHERE dialogId = $dialogId AND messageUId = $messageUId;`)
+            .run({
+            dialogId,
+            messageUId,
+            pinned: pinned ? 1 : 0,
+            json: JSON.stringify(message)
+        });
+    }
+    unpinAllMessages(dialogId) {
+        const rows = this.database
+            .prepare(`SELECT messageUId, json FROM messages
+         WHERE dialogId = $dialogId AND pinned = 1;`)
+            .all({ dialogId });
+        const update = this.database.prepare(`UPDATE messages SET pinned = 0, json = $json
+       WHERE dialogId = $dialogId AND messageUId = $messageUId;`);
+        this.database.transaction(() => {
+            for (const row of rows) {
+                const message = parseMessage(row);
+                if (!message) {
+                    continue;
+                }
+                message.pinned = false;
+                update.run({
+                    dialogId,
+                    messageUId: row.messageUId,
+                    json: JSON.stringify(message)
+                });
+            }
+        })();
+    }
     clearBurnAfterReadingExpiredMessages(dialogId, now = Date.now()) {
         const rows = this.database
             .prepare(`SELECT messageUId FROM messages
@@ -383,7 +511,7 @@ class MessageStore {
     convertToRecallMessages(messageUIds, recallMessageType) {
         const select = this.database.prepare('SELECT json FROM messages WHERE messageUId = $messageUId;');
         const update = this.database.prepare(`
-      UPDATE messages SET messageType = $messageType, json = $json, searchText = ''
+      UPDATE messages SET messageType = $messageType, pinned = 0, json = $json, searchText = ''
       WHERE messageUId = $messageUId;
     `);
         this.database.transaction(() => {
@@ -399,6 +527,7 @@ class MessageStore {
                 message.isCounted = false;
                 message.isMentioned = false;
                 message.disableNotification = true;
+                message.pinned = false;
                 update.run({
                     messageUId,
                     messageType: recallMessageType,