signal-desktop/ts/messages/saveAndNotify.ts
2025-01-10 08:18:32 +10:00

72 lines
2.2 KiB
TypeScript

// Copyright 2024 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import * as log from '../logging/log';
import { explodePromise } from '../util/explodePromise';
import { saveNewMessageBatcher } from '../util/messageBatcher';
import { handleAttachmentDownloadsForNewMessage } from '../util/queueAttachmentDownloads';
import {
modifyTargetMessage,
ModifyTargetMessageResult,
} from '../util/modifyTargetMessage';
import { shouldReplyNotifyUser } from '../util/shouldReplyNotifyUser';
import { isStory } from './helpers';
import { drop } from '../util/drop';
import type { ConversationModel } from '../models/conversations';
import type { MessageModel } from '../models/messages';
export async function saveAndNotify(
message: MessageModel,
conversation: ConversationModel,
confirm: () => void
): Promise<void> {
const { resolve, promise } = explodePromise<void>();
try {
conversation.addSavePromise(promise);
await saveNewMessageBatcher.add(message.attributes);
log.info('Message saved', message.get('sent_at'));
// Once the message is saved to DB, we queue attachment downloads
await handleAttachmentDownloadsForNewMessage(message, conversation);
// We'd like to check for deletions before scheduling downloads, but if an edit
// comes in, we want to have kicked off attachment downloads for the original
// message.
const result = await modifyTargetMessage(message, conversation, {
isFirstRun: false,
skipEdits: false,
});
if (result === ModifyTargetMessageResult.Deleted) {
confirm();
return;
}
drop(conversation.onNewMessage(message));
if (await shouldReplyNotifyUser(message.attributes, conversation)) {
await conversation.notify(message.attributes);
}
// Increment the sent message count if this is an outgoing message
if (message.get('type') === 'outgoing') {
conversation.incrementSentMessageCount();
}
window.Whisper.events.trigger('incrementProgress');
confirm();
if (!isStory(message.attributes)) {
drop(
conversation.queueJob('updateUnread', () => conversation.updateUnread())
);
}
} finally {
resolve();
conversation.removeSavePromise(promise);
}
}