102 lines
2.9 KiB
TypeScript
102 lines
2.9 KiB
TypeScript
// Copyright 2023 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import type { MessageAttributesType } from '../model-types.d';
|
|
import type { MessageModel } from '../models/messages';
|
|
import * as Errors from '../types/errors';
|
|
import * as log from '../logging/log';
|
|
import { drop } from '../util/drop';
|
|
import { filter, size } from '../util/iterables';
|
|
import { getContactId } from '../messages/helpers';
|
|
import { handleEditMessage } from '../util/handleEditMessage';
|
|
import { getMessageSentTimestamp } from '../util/getMessageSentTimestamp';
|
|
|
|
export type EditAttributesType = {
|
|
conversationId: string;
|
|
fromId: string;
|
|
fromDevice: number;
|
|
message: MessageAttributesType;
|
|
targetSentTimestamp: number;
|
|
};
|
|
|
|
const edits = new Set<EditAttributesType>();
|
|
|
|
export function forMessage(message: MessageModel): Array<EditAttributesType> {
|
|
const sentAt = getMessageSentTimestamp(message.attributes, { log });
|
|
const matchingEdits = filter(edits, item => {
|
|
return (
|
|
item.targetSentTimestamp === sentAt &&
|
|
item.fromId === getContactId(message.attributes)
|
|
);
|
|
});
|
|
|
|
if (size(matchingEdits) > 0) {
|
|
const result = Array.from(matchingEdits);
|
|
const editsLogIds = result.map(x => x.message.sent_at);
|
|
log.info(
|
|
`Edits.forMessage(${message.get('sent_at')}): ` +
|
|
`Found early edits for message ${editsLogIds.join(', ')}`
|
|
);
|
|
filter(matchingEdits, item => edits.delete(item));
|
|
return result;
|
|
}
|
|
|
|
return [];
|
|
}
|
|
|
|
export async function onEdit(edit: EditAttributesType): Promise<void> {
|
|
edits.add(edit);
|
|
|
|
const logId = `Edits.onEdit(timestamp=${edit.message.timestamp};target=${edit.targetSentTimestamp})`;
|
|
|
|
try {
|
|
// The conversation the edited message was in; we have to find it in the database
|
|
// to to figure that out.
|
|
const targetConversation =
|
|
await window.ConversationController.getConversationForTargetMessage(
|
|
edit.fromId,
|
|
edit.targetSentTimestamp
|
|
);
|
|
|
|
if (!targetConversation) {
|
|
log.info(`${logId}: No target conversation`);
|
|
|
|
return;
|
|
}
|
|
|
|
// Do not await, since this can deadlock the queue
|
|
drop(
|
|
targetConversation.queueJob('Edits.onEdit', async () => {
|
|
log.info(`${logId}: Handling edit`);
|
|
|
|
const messages = await window.Signal.Data.getMessagesBySentAt(
|
|
edit.targetSentTimestamp
|
|
);
|
|
|
|
// Verify authorship
|
|
const targetMessage = messages.find(
|
|
m =>
|
|
edit.conversationId === m.conversationId &&
|
|
edit.fromId === getContactId(m)
|
|
);
|
|
|
|
if (!targetMessage) {
|
|
log.info(`${logId}: No message`);
|
|
|
|
return;
|
|
}
|
|
|
|
const message = window.MessageController.register(
|
|
targetMessage.id,
|
|
targetMessage
|
|
);
|
|
|
|
await handleEditMessage(message.attributes, edit);
|
|
|
|
edits.delete(edit);
|
|
})
|
|
);
|
|
} catch (error) {
|
|
log.error(`${logId} error:`, Errors.toLogFormat(error));
|
|
}
|
|
}
|