signal-desktop/ts/util/doubleCheckMissingQuoteReference.ts
2025-01-10 08:18:32 +10:00

94 lines
2.5 KiB
TypeScript

// Copyright 2019 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import { getMessageById } from '../messages/getMessageById';
import type { MessageModel } from '../models/messages';
import { hydrateStoryContext } from './hydrateStoryContext';
import { getMessageIdForLogging } from './idForLogging';
import * as log from '../logging/log';
import {
isQuoteAMatch,
shouldTryToCopyFromQuotedMessage,
} from '../messages/helpers';
import { copyQuoteContentFromOriginal } from '../messages/copyQuote';
import { queueUpdateMessage } from './messageBatcher';
export async function doubleCheckMissingQuoteReference(
message: MessageModel
): Promise<void> {
const logId = getMessageIdForLogging(message.attributes);
const storyId = message.get('storyId');
if (storyId) {
log.warn(
`doubleCheckMissingQuoteReference/${logId}: missing story reference`
);
const storyMessage = await getMessageById(storyId);
if (!storyMessage) {
return;
}
if (message.get('storyReplyContext')) {
message.set({ storyReplyContext: undefined });
}
await hydrateStoryContext(message.id, storyMessage.attributes, {
shouldSave: true,
});
return;
}
const quote = message.get('quote');
if (!quote) {
log.warn(`doubleCheckMissingQuoteReference/${logId}: Missing quote!`);
return;
}
const { authorAci, author, id: sentAt, referencedMessageNotFound } = quote;
const contact = window.ConversationController.get(authorAci || author);
// Is the quote really without a reference? Check with our in memory store
// first to make sure it's not there.
if (
contact &&
shouldTryToCopyFromQuotedMessage({
referencedMessageNotFound,
quoteAttachment: quote.attachments.at(0),
})
) {
const matchingMessage = await window.MessageCache.findBySentAt(
Number(sentAt),
model =>
isQuoteAMatch(model.attributes, message.get('conversationId'), quote)
);
if (!matchingMessage) {
log.info(
`doubleCheckMissingQuoteReference/${logId}: No match for ${sentAt}.`
);
return;
}
message.set({
quote: {
...quote,
referencedMessageNotFound: false,
},
});
log.info(
`doubleCheckMissingQuoteReference/${logId}: Found match for ${sentAt}, updating.`
);
await copyQuoteContentFromOriginal(matchingMessage, quote);
message.set({
quote: {
...quote,
referencedMessageNotFound: false,
},
});
queueUpdateMessage(message.attributes);
}
}