@onyx-p/imlib-web 3.0.4 → 3.0.6

package/README.md

@@ -106,7 +106,8 @@ already committed batches are safe to repeat.
 ## 消息搜索
 
 单会话搜索接口保持不变。全局搜索使用 Signal 风格的 FTS5 前缀匹配，
-按消息时间倒序返回，默认最多 500 条：
+按消息时间倒序返回，默认最多 500 条。文本消息按正文搜索，
+文件消息按文件名搜索：
 
 ```js
 const result = await ACIMLib.searchMessages('项目进度')

package/database/main.cjs

@@ -28,6 +28,14 @@ class DatabaseKeyStore {
         this.writeKeys(keys);
         return key;
     }
+    remove(accountId) {
+        const keys = this.readKeys();
+        if (!(accountId in keys)) {
+            return;
+        }
+        delete keys[accountId];
+        this.writeKeys(keys);
+    }
     readKeys() {
         try {
             const parsed = JSON.parse(node_fs.readFileSync(this.filePath, 'utf8'));
@@ -69,6 +77,7 @@ const READ_METHOD_NAMES = [
     'getPinnedMessages',
     'searchMessages',
     'searchTextMessages',
+    'getLegacyMediaReferences',
     'getLegacyMigrationState'
 ];
 const WRITE_METHOD_NAMES = [
@@ -78,6 +87,7 @@ const WRITE_METHOD_NAMES = [
     'removeMessagesByUId',
     'updateMessageReceiptStatus',
     'setMessagePinned',
+    'setMessageReactions',
     'unpinAllMessages',
     'clearBurnAfterReadingExpiredMessages',
     'upsertMessage',
@@ -120,6 +130,19 @@ function parseRequest(value) {
     if (value.type === 'close') {
         return { type: 'close' };
     }
+    if (value.type === 'delete') {
+        if (!isRecord(value.options) ||
+            typeof value.options.appKey !== 'string' ||
+            typeof value.options.userId !== 'string' ||
+            !value.options.appKey ||
+            !value.options.userId) {
+            throw new Error('Invalid database delete options');
+        }
+        return {
+            type: 'delete',
+            options: { appKey: value.options.appKey, userId: value.options.userId }
+        };
+    }
     if (value.type === 'open') {
         if (!isRecord(value.options) ||
             typeof value.options.appKey !== 'string' ||
@@ -158,6 +181,9 @@ function registerDatabaseIpc({ ipcMain, service, isTrustedEvent = isTopLevelSend
             else if (request.type === 'close') {
                 result = await service.close();
             }
+            else if (request.type === 'delete') {
+                result = await service.deleteAccount(request.options);
+            }
             else if (request.type === 'read') {
                 result = await service.read(request.method, request.args);
             }
@@ -250,6 +276,18 @@ class MainDatabaseService {
             }
         }));
     }
+    async deleteAccount(options) {
+        await this.close();
+        const accountId = node_crypto.createHash('sha256')
+            .update(`${options.appKey}:${options.userId}`)
+            .digest('hex');
+        const sqlDirectory = node_path.join(this.options.userDataPath, 'sql');
+        const dbPath = node_path.join(sqlDirectory, `${accountId}.sqlite`);
+        for (const path of [dbPath, `${dbPath}-wal`, `${dbPath}-shm`]) {
+            node_fs.rmSync(path, { force: true });
+        }
+        this.options.keyStore.remove(accountId);
+    }
     assertOpen() {
         if (!this.accountId || this.pool.length === 0) {
             throw new Error('Database service is not open');

package/database/preload.cjs

@@ -16,6 +16,7 @@ const READ_METHOD_NAMES = [
     'getPinnedMessages',
     'searchMessages',
     'searchTextMessages',
+    'getLegacyMediaReferences',
     'getLegacyMigrationState'
 ];
 const WRITE_METHOD_NAMES = [
@@ -25,6 +26,7 @@ const WRITE_METHOD_NAMES = [
     'removeMessagesByUId',
     'updateMessageReceiptStatus',
     'setMessagePinned',
+    'setMessageReactions',
     'unpinAllMessages',
     'clearBurnAfterReadingExpiredMessages',
     'upsertMessage',

package/database/worker.cjs

@@ -19,6 +19,7 @@ const READ_METHOD_NAMES = [
     'getPinnedMessages',
     'searchMessages',
     'searchTextMessages',
+    'getLegacyMediaReferences',
     'getLegacyMigrationState'
 ];
 const WRITE_METHOD_NAMES = [
@@ -28,6 +29,7 @@ const WRITE_METHOD_NAMES = [
     'removeMessagesByUId',
     'updateMessageReceiptStatus',
     'setMessagePinned',
+    'setMessageReactions',
     'unpinAllMessages',
     'clearBurnAfterReadingExpiredMessages',
     'upsertMessage',
@@ -52,6 +54,24 @@ function serializeDatabaseError(error) {
     };
 }
 
+const MessageTypes = {
+    TEXT: 0x00000000,
+    IMAGE: 0x30001000,
+    GIF: 0x3000100B,
+    FILE: 0x3000101A,
+    AUDIO: 0x30001011,
+    VIDEO: 0x30001006,
+    RECALL: 0x6001001e,
+    STORED_RECALL: 0x6001001d,
+    LOCATION: 0x30001005,
+    CHAT_RECORD: 0x30001019,
+    CONTACT: 0x10001002,
+    GROUP_INVITATION: 0x30001003,
+    REDPACKET: 0x10001040,
+    TRANSFER: 0x10001030,
+    LINK: 0x10001004
+};
+
 const SCHEMA_MIGRATIONS = [
     {
         version: 1,
@@ -157,6 +177,20 @@ const SCHEMA_MIGRATIONS = [
           ON messages (dialogId, pinned, sentTime DESC, messageId DESC);
       `);
         }
+    },
+    {
+        version: 7,
+        up(database) {
+            database
+                .prepare(`UPDATE messages
+           SET searchText = CASE
+             WHEN json_type(json, '$.content.title') = 'text'
+               THEN json_extract(json, '$.content.title')
+             ELSE ''
+           END
+           WHERE messageType = $fileMessageType;`)
+                .run({ fileMessageType: MessageTypes.FILE });
+        }
     }
 ];
 SCHEMA_MIGRATIONS[SCHEMA_MIGRATIONS.length - 1]?.version ?? 0;
@@ -187,6 +221,167 @@ function runMigrations(database, migrations = SCHEMA_MIGRATIONS) {
     }
 }
 
+var ConnectionStatus;
+(function (ConnectionStatus) {
+    ConnectionStatus[ConnectionStatus["CONNECTED"] = 0] = "CONNECTED";
+    ConnectionStatus[ConnectionStatus["CONNECTING"] = 1] = "CONNECTING";
+    ConnectionStatus[ConnectionStatus["DISCONNECTED"] = 2] = "DISCONNECTED";
+    ConnectionStatus[ConnectionStatus["NETWORK_UNAVAILABLE"] = 3] = "NETWORK_UNAVAILABLE";
+    ConnectionStatus[ConnectionStatus["CONNECTION_CLOSED"] = 4] = "CONNECTION_CLOSED";
+    ConnectionStatus[ConnectionStatus["KICKED_OFFLINE_BY_OTHER_CLIENT"] = 6] = "KICKED_OFFLINE_BY_OTHER_CLIENT";
+    ConnectionStatus[ConnectionStatus["WEBSOCKET_UNAVAILABLE"] = 7] = "WEBSOCKET_UNAVAILABLE";
+    ConnectionStatus[ConnectionStatus["WEBSOCKET_ERROR"] = 8] = "WEBSOCKET_ERROR";
+    ConnectionStatus[ConnectionStatus["BLOCKED"] = 9] = "BLOCKED";
+    ConnectionStatus[ConnectionStatus["DISCONNECT_BY_SERVER"] = 13] = "DISCONNECT_BY_SERVER";
+})(ConnectionStatus || (ConnectionStatus = {}));
+var ErrorCode;
+(function (ErrorCode) {
+    ErrorCode[ErrorCode["TIMEOUT"] = -1] = "TIMEOUT";
+    ErrorCode[ErrorCode["UNKNOWN"] = -2] = "UNKNOWN";
+    ErrorCode[ErrorCode["PARAMETER_ERROR"] = -3] = "PARAMETER_ERROR";
+    ErrorCode[ErrorCode["SUCCESS"] = 0] = "SUCCESS";
+    ErrorCode[ErrorCode["NETWORK_ERROR"] = 7] = "NETWORK_ERROR";
+    ErrorCode[ErrorCode["MSG_ENCRYPT_ERROR"] = 6001] = "MSG_ENCRYPT_ERROR";
+    ErrorCode[ErrorCode["RC_DISCUSSION_GROUP_ID_INVALID"] = 20407] = "RC_DISCUSSION_GROUP_ID_INVALID";
+    ErrorCode[ErrorCode["SEND_FREQUENCY_TOO_FAST"] = 20604] = "SEND_FREQUENCY_TOO_FAST";
+    ErrorCode[ErrorCode["FORBIDDEN_IN_GROUP"] = 22408] = "FORBIDDEN_IN_GROUP";
+    ErrorCode[ErrorCode["RECALL_MESSAGE"] = 25101] = "RECALL_MESSAGE";
+    ErrorCode[ErrorCode["NOT_IN_GROUP"] = 22406] = "NOT_IN_GROUP";
+    ErrorCode[ErrorCode["SENSITIVE_SHIELD"] = 21501] = "SENSITIVE_SHIELD";
+    ErrorCode[ErrorCode["GET_USERINFO_ERROR"] = 23407] = "GET_USERINFO_ERROR";
+    ErrorCode[ErrorCode["REJECTED_BY_BLACKLIST"] = 405] = "REJECTED_BY_BLACKLIST";
+    ErrorCode[ErrorCode["RC_NET_CHANNEL_INVALID"] = 30001] = "RC_NET_CHANNEL_INVALID";
+    ErrorCode[ErrorCode["RC_NET_UNAVAILABLE"] = 30002] = "RC_NET_UNAVAILABLE";
+    ErrorCode[ErrorCode["RC_MSG_RESP_TIMEOUT"] = 30003] = "RC_MSG_RESP_TIMEOUT";
+    ErrorCode[ErrorCode["RC_SOCKET_NOT_CREATED"] = 30010] = "RC_SOCKET_NOT_CREATED";
+    ErrorCode[ErrorCode["RC_SOCKET_DISCONNECTED"] = 30011] = "RC_SOCKET_DISCONNECTED";
+    ErrorCode[ErrorCode["RC_PONG_RECV_FAIL"] = 30013] = "RC_PONG_RECV_FAIL";
+    ErrorCode[ErrorCode["RC_MSG_SEND_FAIL"] = 30014] = "RC_MSG_SEND_FAIL";
+    ErrorCode[ErrorCode["RC_MSG_CONTENT_EXCEED_LIMIT"] = 30016] = "RC_MSG_CONTENT_EXCEED_LIMIT";
+    ErrorCode[ErrorCode["RC_CONN_ACK_TIMEOUT"] = 31000] = "RC_CONN_ACK_TIMEOUT";
+    ErrorCode[ErrorCode["RC_CONN_ID_REJECT"] = 31002] = "RC_CONN_ID_REJECT";
+    ErrorCode[ErrorCode["RC_CONN_SERVER_UNAVAILABLE"] = 31003] = "RC_CONN_SERVER_UNAVAILABLE";
+    ErrorCode[ErrorCode["RC_CONN_USER_OR_PASSWD_ERROR"] = 31004] = "RC_CONN_USER_OR_PASSWD_ERROR";
+    ErrorCode[ErrorCode["RC_CONN_NOT_AUTHRORIZED"] = 31005] = "RC_CONN_NOT_AUTHRORIZED";
+    ErrorCode[ErrorCode["RC_DISCONN_KICK"] = 31010] = "RC_DISCONN_KICK";
+    ErrorCode[ErrorCode["RC_DISCONN_SAME_CLIENT_ON_LINE"] = 31023] = "RC_DISCONN_SAME_CLIENT_ON_LINE";
+    ErrorCode[ErrorCode["BIZ_ERROR_CLIENT_NOT_INIT"] = 33001] = "BIZ_ERROR_CLIENT_NOT_INIT";
+    ErrorCode[ErrorCode["BIZ_ERROR_INVALID_PARAMETER"] = 33003] = "BIZ_ERROR_INVALID_PARAMETER";
+    ErrorCode[ErrorCode["BIZ_ERROR_RECONNECT_SUCCESS"] = 33005] = "BIZ_ERROR_RECONNECT_SUCCESS";
+    ErrorCode[ErrorCode["BIZ_ERROR_CONNECTING"] = 33006] = "BIZ_ERROR_CONNECTING";
+    ErrorCode[ErrorCode["MSG_INSERT_ERROR"] = 33008] = "MSG_INSERT_ERROR";
+    ErrorCode[ErrorCode["MSG_DEL_ERROR"] = 33009] = "MSG_DEL_ERROR";
+    ErrorCode[ErrorCode["CONVER_REMOVE_ERROR"] = 34001] = "CONVER_REMOVE_ERROR";
+    ErrorCode[ErrorCode["CONVER_GETLIST_ERROR"] = 34002] = "CONVER_GETLIST_ERROR";
+    ErrorCode[ErrorCode["CONVER_SETOP_ERROR"] = 34003] = "CONVER_SETOP_ERROR";
+    ErrorCode[ErrorCode["CONVER_TOTAL_UNREAD_ERROR"] = 34004] = "CONVER_TOTAL_UNREAD_ERROR";
+    ErrorCode[ErrorCode["CONVER_TYPE_UNREAD_ERROR"] = 34005] = "CONVER_TYPE_UNREAD_ERROR";
+    ErrorCode[ErrorCode["CONVER_ID_TYPE_UNREAD_ERROR"] = 34006] = "CONVER_ID_TYPE_UNREAD_ERROR";
+    ErrorCode[ErrorCode["CONVER_OUT_LIMIT_ERROR"] = 34013] = "CONVER_OUT_LIMIT_ERROR";
+    ErrorCode[ErrorCode["MEDIA_EXCEPTION"] = 34018] = "MEDIA_EXCEPTION";
+    ErrorCode[ErrorCode["CONVER_GET_ERROR"] = 35021] = "CONVER_GET_ERROR";
+    ErrorCode[ErrorCode["GROUP_SYNC_ERROR"] = 35001] = "GROUP_SYNC_ERROR";
+    ErrorCode[ErrorCode["CAN_NOT_RECONNECT"] = 35007] = "CAN_NOT_RECONNECT";
+    ErrorCode[ErrorCode["HAS_OHTER_SAME_CLIENT_ON_LINE"] = 35010] = "HAS_OHTER_SAME_CLIENT_ON_LINE";
+    ErrorCode[ErrorCode["METHOD_NOT_AVAILABLE"] = 35011] = "METHOD_NOT_AVAILABLE";
+    ErrorCode[ErrorCode["METHOD_NOT_SUPPORT"] = 35012] = "METHOD_NOT_SUPPORT";
+    ErrorCode[ErrorCode["MSG_LIMIT_ERROR"] = 35013] = "MSG_LIMIT_ERROR";
+    ErrorCode[ErrorCode["METHOD_ONLY_SUPPORT_ULTRA_GROUP"] = 35014] = "METHOD_ONLY_SUPPORT_ULTRA_GROUP";
+    ErrorCode[ErrorCode["UPLOAD_FILE_FAILED"] = 35020] = "UPLOAD_FILE_FAILED";
+    ErrorCode[ErrorCode["DRAF_GET_ERROR"] = 38001] = "DRAF_GET_ERROR";
+    ErrorCode[ErrorCode["DRAF_SAVE_ERROR"] = 38002] = "DRAF_SAVE_ERROR";
+    ErrorCode[ErrorCode["DRAF_REMOVE_ERROR"] = 38003] = "DRAF_REMOVE_ERROR";
+    ErrorCode[ErrorCode["NOT_SUPPORT"] = 39002] = "NOT_SUPPORT";
+})(ErrorCode || (ErrorCode = {}));
+var ConversationType;
+(function (ConversationType) {
+    ConversationType[ConversationType["PRIVATE"] = 1] = "PRIVATE";
+    ConversationType[ConversationType["GROUP"] = 3] = "GROUP";
+})(ConversationType || (ConversationType = {}));
+var MentionedType;
+(function (MentionedType) {
+    MentionedType[MentionedType["ALL"] = 1] = "ALL";
+    MentionedType[MentionedType["SINGAL"] = 2] = "SINGAL";
+})(MentionedType || (MentionedType = {}));
+var NotificationStatus;
+(function (NotificationStatus) {
+    NotificationStatus[NotificationStatus["OPEN"] = 1] = "OPEN";
+    NotificationStatus[NotificationStatus["CLOSE"] = 2] = "CLOSE";
+})(NotificationStatus || (NotificationStatus = {}));
+var MessageDirection;
+(function (MessageDirection) {
+    MessageDirection[MessageDirection["SEND"] = 1] = "SEND";
+    MessageDirection[MessageDirection["RECEIVE"] = 2] = "RECEIVE";
+})(MessageDirection || (MessageDirection = {}));
+var SentStatus;
+(function (SentStatus) {
+    SentStatus[SentStatus["SENDING"] = 10] = "SENDING";
+    SentStatus[SentStatus["FAILED"] = 20] = "FAILED";
+    SentStatus[SentStatus["SENT"] = 30] = "SENT";
+})(SentStatus || (SentStatus = {}));
+var ReceivedStatus;
+(function (ReceivedStatus) {
+    ReceivedStatus[ReceivedStatus["IDLE"] = 0] = "IDLE";
+    ReceivedStatus[ReceivedStatus["RECEIVED"] = 1] = "RECEIVED";
+    ReceivedStatus[ReceivedStatus["READ"] = 2] = "READ";
+    ReceivedStatus[ReceivedStatus["LISTENED"] = 3] = "LISTENED";
+})(ReceivedStatus || (ReceivedStatus = {}));
+var NotificationLevel;
+(function (NotificationLevel) {
+    NotificationLevel[NotificationLevel["ALL_MESSAGE"] = -1] = "ALL_MESSAGE";
+    NotificationLevel[NotificationLevel["NOT_SET"] = 0] = "NOT_SET";
+    NotificationLevel[NotificationLevel["AT_MESSAGE_NOTIFICATION"] = 1] = "AT_MESSAGE_NOTIFICATION";
+    NotificationLevel[NotificationLevel["AT_USER_NOTIFICATION"] = 2] = "AT_USER_NOTIFICATION";
+    NotificationLevel[NotificationLevel["AT_GROUP_ALL_USER_NOTIFICATION"] = 4] = "AT_GROUP_ALL_USER_NOTIFICATION";
+    NotificationLevel[NotificationLevel["NOT_MESSAGE_NOTIFICATION"] = 5] = "NOT_MESSAGE_NOTIFICATION";
+})(NotificationLevel || (NotificationLevel = {}));
+
+function extractLegacyMediaReferences(message) {
+    const { messageId, conversationType, targetId } = message;
+    if (!Number.isSafeInteger(messageId) ||
+        typeof conversationType !== 'number' ||
+        !Number.isSafeInteger(conversationType) ||
+        typeof targetId !== 'string' ||
+        !targetId) {
+        return [];
+    }
+    const result = [];
+    const content = message.content ?? {};
+    const add = (kind, value) => {
+        const objectKey = typeof value === 'string' ? value.trim() : '';
+        if (!objectKey) {
+            return;
+        }
+        result.push({
+            messageId,
+            conversationType,
+            targetId,
+            objectKey,
+            kind
+        });
+    };
+    switch (message.messageType) {
+        case MessageTypes.IMAGE:
+            add('image', content.originalObjectKey);
+            add('image-thumbnail', content.thumbnailObjectKey);
+            break;
+        case MessageTypes.GIF:
+            add('gif', content.originalObjectKey);
+            break;
+        case MessageTypes.VIDEO:
+            add('video', content.videoObjectKey);
+            add('video-thumbnail', content.thumbnailObjectKey);
+            break;
+        case MessageTypes.FILE:
+            add('file', content.fileKey);
+            break;
+        case MessageTypes.AUDIO:
+            add('audio', content.audioObjectKey);
+            break;
+    }
+    return result;
+}
+
 const LEGACY_MIGRATION_KEY = 'legacy_indexeddb_migration';
 function parseMessage(row) {
     if (!row) {
@@ -196,8 +391,38 @@ function parseMessage(row) {
     message.pinned = message.pinned ?? false;
     return message;
 }
+function mergeStoredReceiptStatus(incoming, existing) {
+    if (!existing) {
+        return incoming;
+    }
+    const existingStatus = existing.receivedStatus ?? -1;
+    const incomingStatus = incoming.receivedStatus ?? -1;
+    const merged = existingStatus <= incomingStatus
+        ? {
+            ...incoming,
+            messageId: existing.messageId
+        }
+        : {
+            ...incoming,
+            messageId: existing.messageId,
+            receivedStatus: existing.receivedStatus
+        };
+    if (incoming.reactions == null && existing.reactions != null) {
+        merged.reactions = existing.reactions;
+    }
+    if (incoming.burnAfterReadingStartedAt == null &&
+        existing.burnAfterReadingStartedAt != null) {
+        merged.burnAfterReadingStartedAt = existing.burnAfterReadingStartedAt;
+    }
+    return merged;
+}
 function extractSearchText(message) {
     const content = message.content;
+    if (message.messageType === MessageTypes.FILE &&
+        content &&
+        typeof content.title === 'string') {
+        return content.title;
+    }
     if (typeof content === 'string') {
         return content;
     }
@@ -213,6 +438,32 @@ function normalizeInteger(value, name) {
     }
     return number;
 }
+function isBurnAfterReadingMessageVisible(message, now) {
+    if (!message.burnAfterReadingFlag) {
+        return true;
+    }
+    const burnAfterReadingTime = message.burnAfterReadingTime ?? 0;
+    const sentTime = Number(message.sentTime);
+    if (message.messageDirection === MessageDirection.RECEIVE) {
+        if ((message.receivedStatus ?? ReceivedStatus.IDLE) < ReceivedStatus.READ) {
+            return true;
+        }
+        if (burnAfterReadingTime <= 0) {
+            return false;
+        }
+        const startedAt = message.burnAfterReadingStartedAt ?? sentTime;
+        return !Number.isFinite(startedAt) || startedAt + burnAfterReadingTime > now;
+    }
+    if (message.messageDirection === MessageDirection.SEND) {
+        if (message.sentStatus !== SentStatus.SENT) {
+            return true;
+        }
+        return (burnAfterReadingTime > 0 &&
+            (!Number.isFinite(sentTime) || sentTime + burnAfterReadingTime > now));
+    }
+    return (burnAfterReadingTime > 0 &&
+        (!Number.isFinite(sentTime) || sentTime + burnAfterReadingTime > now));
+}
 class MessageStore {
     database;
     constructor(database) {
@@ -244,20 +495,24 @@ class MessageStore {
                 if (message.isPersited === false) {
                     continue;
                 }
-                const messageId = normalizeInteger(message.messageId, 'messageId');
-                const sentTime = normalizeInteger(message.sentTime, 'sentTime');
+                const existing = message.messageUId
+                    ? this.getMessageByUId(message.messageUId)
+                    : this.getMessageById(normalizeInteger(message.messageId, 'messageId'));
+                const merged = mergeStoredReceiptStatus(message, existing);
+                const messageId = normalizeInteger(merged.messageId, 'messageId');
+                const sentTime = normalizeInteger(merged.sentTime, 'sentTime');
                 upsert.run({
                     messageId,
-                    messageUId: message.messageUId ?? null,
+                    messageUId: merged.messageUId ?? null,
                     dialogId,
                     sentTime,
-                    receivedStatus: message.receivedStatus ?? null,
-                    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)
+                    receivedStatus: merged.receivedStatus ?? null,
+                    burnAfterReadingFlag: merged.burnAfterReadingFlag ? 1 : 0,
+                    burnAfterReadingTime: merged.burnAfterReadingTime ?? null,
+                    messageType: merged.messageType ?? null,
+                    pinned: merged.pinned ? 1 : 0,
+                    json: JSON.stringify(merged),
+                    searchText: extractSearchText(merged)
                 });
             }
             if (isEnd !== undefined) {
@@ -270,33 +525,54 @@ class MessageStore {
     }
     getMessages(dialogId, timestamp = '0', count = 20, isForward = true) {
         const safeCount = Math.max(0, Math.floor(count));
+        const batchSize = Math.max(safeCount + 1, 50);
         const params = {
             dialogId,
-            limit: safeCount + 1
+            limit: batchSize
         };
+        let offset = 0;
         let timeClause = '';
         if (timestamp !== '0') {
             params.timestamp = normalizeInteger(timestamp, 'timestamp');
-            timeClause = isForward ? 'AND sentTime < $timestamp' : 'AND sentTime > $timestamp';
+            timeClause = isForward
+                ? 'AND sentTime < $timestamp'
+                : 'AND sentTime > $timestamp';
         }
         const direction = isForward ? 'DESC' : 'ASC';
-        const rows = this.database
-            .prepare(`SELECT json FROM messages
-         WHERE dialogId = $dialogId ${timeClause}
-         ORDER BY sentTime ${direction}, messageId ${direction}
-         LIMIT $limit;`)
-            .all(params);
-        const hasAdditionalRow = rows.length > safeCount;
-        const messages = rows
+        const select = this.database.prepare(`SELECT json FROM messages
+       WHERE dialogId = $dialogId ${timeClause}
+       ORDER BY sentTime ${direction}, messageId ${direction}
+       LIMIT $limit OFFSET $offset;`);
+        const visibleMessages = [];
+        const now = Date.now();
+        while (visibleMessages.length <= safeCount) {
+            const rows = select.all({ ...params, offset });
+            for (const row of rows) {
+                const message = parseMessage(row);
+                if (message && isBurnAfterReadingMessageVisible(message, now)) {
+                    visibleMessages.push(message);
+                    if (visibleMessages.length > safeCount) {
+                        break;
+                    }
+                }
+            }
+            if (rows.length < batchSize || visibleMessages.length > safeCount) {
+                break;
+            }
+            offset += rows.length;
+        }
+        const hasAdditionalMessage = visibleMessages.length > safeCount;
+        const messages = visibleMessages
             .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 || !this.getDialogLoadedState(dialogId)
+            hasMore: isForward
+                ? hasAdditionalMessage || !this.getDialogLoadedState(dialogId)
+                : hasAdditionalMessage
         };
     }
     getMessagesByType(dialogId, messageTypes, timestamp = '0', count = 50, isForward = true) {
@@ -431,7 +707,7 @@ class MessageStore {
             }
         })();
     }
-    updateMessageReceiptStatus(messageUIds, receivedStatus) {
+    updateMessageReceiptStatus(messageUIds, receivedStatus, burnAfterReadingStartedAt) {
         const select = this.database.prepare('SELECT json, receivedStatus FROM messages WHERE messageUId = $messageUId;');
         const update = this.database.prepare(`
       UPDATE messages SET receivedStatus = $receivedStatus, json = $json
@@ -440,12 +716,31 @@ class MessageStore {
         this.database.transaction(() => {
             for (const messageUId of messageUIds) {
                 const row = select.get({ messageUId });
-                if (!row || (row.receivedStatus ?? -1) >= receivedStatus) {
+                if (!row) {
                     continue;
                 }
                 const message = JSON.parse(row.json);
-                message.receivedStatus = receivedStatus;
-                update.run({ messageUId, receivedStatus, json: JSON.stringify(message) });
+                const shouldAdvanceStatus = (row.receivedStatus ?? -1) < receivedStatus;
+                const shouldStartBurn = burnAfterReadingStartedAt != null &&
+                    receivedStatus === ReceivedStatus.READ &&
+                    message.messageDirection === MessageDirection.RECEIVE &&
+                    message.burnAfterReadingFlag === true &&
+                    message.burnAfterReadingStartedAt == null;
+                if (!shouldAdvanceStatus && !shouldStartBurn) {
+                    continue;
+                }
+                const nextReceivedStatus = shouldAdvanceStatus
+                    ? receivedStatus
+                    : row.receivedStatus ?? receivedStatus;
+                message.receivedStatus = nextReceivedStatus;
+                if (shouldStartBurn) {
+                    message.burnAfterReadingStartedAt = burnAfterReadingStartedAt;
+                }
+                update.run({
+                    messageUId,
+                    receivedStatus: nextReceivedStatus,
+                    json: JSON.stringify(message)
+                });
             }
         })();
     }
@@ -469,6 +764,31 @@ class MessageStore {
             json: JSON.stringify(message)
         });
     }
+    setMessageReactions(dialogId, messageUId, reactions) {
+        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 false;
+        }
+        if (reactions.length) {
+            message.reactions = reactions;
+        }
+        else {
+            delete message.reactions;
+        }
+        this.database
+            .prepare(`UPDATE messages SET json = $json
+         WHERE dialogId = $dialogId AND messageUId = $messageUId;`)
+            .run({
+            dialogId,
+            messageUId,
+            json: JSON.stringify(message)
+        });
+        return true;
+    }
     unpinAllMessages(dialogId) {
         const rows = this.database
             .prepare(`SELECT messageUId, json FROM messages
@@ -493,20 +813,23 @@ class MessageStore {
     }
     clearBurnAfterReadingExpiredMessages(dialogId, now = Date.now()) {
         const rows = this.database
-            .prepare(`SELECT messageUId FROM messages
-         WHERE dialogId = $dialogId
-           AND burnAfterReadingFlag = 1
-           AND (burnAfterReadingTime IS NULL OR burnAfterReadingTime = 0
-             OR sentTime + burnAfterReadingTime <= $now);`, { pluck: true })
-            .all({ dialogId, now });
-        this.database
-            .prepare(`DELETE FROM messages
+            .prepare(`SELECT messageId, messageUId, json FROM messages
          WHERE dialogId = $dialogId
-           AND burnAfterReadingFlag = 1
-           AND (burnAfterReadingTime IS NULL OR burnAfterReadingTime = 0
-             OR sentTime + burnAfterReadingTime <= $now);`)
-            .run({ dialogId, now });
-        return rows;
+           AND burnAfterReadingFlag = 1;`)
+            .all({
+            dialogId
+        });
+        const expiredRows = rows.filter(row => {
+            const message = parseMessage({ json: row.json });
+            return message != null && !isBurnAfterReadingMessageVisible(message, now);
+        });
+        const remove = this.database.prepare('DELETE FROM messages WHERE messageId = $messageId;');
+        this.database.transaction(() => {
+            for (const row of expiredRows) {
+                remove.run({ messageId: row.messageId });
+            }
+        })();
+        return expiredRows.flatMap(row => row.messageUId == null ? [] : [row.messageUId]);
     }
     convertToRecallMessages(messageUIds, recallMessageType) {
         const select = this.database.prepare('SELECT json FROM messages WHERE messageUId = $messageUId;');
@@ -576,6 +899,29 @@ class MessageStore {
         })
             .map(row => parseMessage(row));
     }
+    getLegacyMediaReferences(afterMessageId = 0, limit = 500) {
+        const safeCursor = Number.isSafeInteger(afterMessageId)
+            ? afterMessageId
+            : 0;
+        const safeLimit = Math.min(1000, Math.max(1, Math.floor(limit)));
+        const rows = this.database
+            .prepare(`SELECT messageId, json
+         FROM messages
+         WHERE messageId > $afterMessageId
+         ORDER BY messageId ASC
+         LIMIT $limit;`)
+            .all({
+            afterMessageId: safeCursor,
+            limit: safeLimit
+        });
+        const nextCursor = rows.length === safeLimit
+            ? rows[rows.length - 1]?.messageId
+            : undefined;
+        return {
+            items: rows.flatMap(row => extractLegacyMediaReferences(JSON.parse(row.json))),
+            ...(nextCursor === undefined ? {} : { nextCursor })
+        };
+    }
     getLegacyMigrationState() {
         const value = this.database
             .prepare('SELECT value FROM metadata WHERE key = $key;', { pluck: true })