@onyx-p/imlib-web 3.0.3 → 3.0.4

package/database/main.cjs

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

package/database/preload.cjs

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

package/database/worker.cjs

@@ -16,6 +16,7 @@ const READ_METHOD_NAMES = [
     'getMessageByUId',
     'getMessageIdRange',
     'getMessagesByType',
+    'getPinnedMessages',
     'searchMessages',
     'searchTextMessages',
     'getLegacyMigrationState'
@@ -26,6 +27,8 @@ const WRITE_METHOD_NAMES = [
     'clearAllCache',
     'removeMessagesByUId',
     'updateMessageReceiptStatus',
+    'setMessagePinned',
+    'unpinAllMessages',
     'clearBurnAfterReadingExpiredMessages',
     'upsertMessage',
     'convertToRecallMessages',
@@ -143,6 +146,17 @@ const SCHEMA_MIGRATIONS = [
           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;
@@ -175,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;
@@ -203,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,
@@ -216,6 +235,7 @@ class MessageStore {
         burnAfterReadingFlag = excluded.burnAfterReadingFlag,
         burnAfterReadingTime = excluded.burnAfterReadingTime,
         messageType = excluded.messageType,
+        pinned = excluded.pinned,
         json = excluded.json,
         searchText = excluded.searchText;
     `);
@@ -235,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)
                 });
@@ -358,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;')
@@ -411,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
@@ -431,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(() => {
@@ -447,6 +527,7 @@ class MessageStore {
                 message.isCounted = false;
                 message.isMentioned = false;
                 message.disableNotification = true;
+                message.pinned = false;
                 update.run({
                     messageUId,
                     messageType: recallMessageType,