@onyx-p/imlib-web 3.0.2 → 3.0.3

package/database/main.cjs

@@ -65,6 +65,7 @@ const READ_METHOD_NAMES = [
     'getMessageById',
     'getMessageByUId',
     'getMessageIdRange',
+    'getMessagesByType',
     'searchMessages',
     'searchTextMessages',
     'getLegacyMigrationState'

package/database/preload.cjs

@@ -12,6 +12,7 @@ const READ_METHOD_NAMES = [
     'getMessageById',
     'getMessageByUId',
     'getMessageIdRange',
+    'getMessagesByType',
     'searchMessages',
     'searchTextMessages',
     'getLegacyMigrationState'

package/database/worker.cjs

@@ -15,6 +15,7 @@ const READ_METHOD_NAMES = [
     'getMessageById',
     'getMessageByUId',
     'getMessageIdRange',
+    'getMessagesByType',
     'searchMessages',
     'searchTextMessages',
     'getLegacyMigrationState'
@@ -133,6 +134,15 @@ 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);
+      `);
+        }
     }
 ];
 SCHEMA_MIGRATIONS[SCHEMA_MIGRATIONS.length - 1]?.version ?? 0;
@@ -268,6 +278,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