Use minimal replacement class for MessageModel

ts/util/doubleCheckMissingQuoteReference.ts

@@ -0,0 +1,94 @@
+// 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);
+  }
+}