// 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);
  }
}