66 lines
1.7 KiB
TypeScript
66 lines
1.7 KiB
TypeScript
// Copyright 2022 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import type { AttachmentType } from '../types/Attachment';
|
|
import * as Attachment from '../types/Attachment';
|
|
import { ToastDangerousFileType } from '../components/ToastDangerousFileType';
|
|
import { ToastFileSaved } from '../components/ToastFileSaved';
|
|
import { isFileDangerous } from './isFileDangerous';
|
|
import { showToast } from './showToast';
|
|
import { getMessageById } from '../messages/getMessageById';
|
|
|
|
export async function saveAttachment(
|
|
attachment: AttachmentType,
|
|
timestamp = Date.now(),
|
|
index = 0
|
|
): Promise<void> {
|
|
const { fileName = '' } = attachment;
|
|
|
|
const isDangerous = isFileDangerous(fileName);
|
|
|
|
if (isDangerous) {
|
|
showToast(ToastDangerousFileType);
|
|
return;
|
|
}
|
|
|
|
const { openFileInFolder, readAttachmentData, saveAttachmentToDisk } =
|
|
window.Signal.Migrations;
|
|
|
|
const fullPath = await Attachment.save({
|
|
attachment,
|
|
index: index + 1,
|
|
readAttachmentData,
|
|
saveAttachmentToDisk,
|
|
timestamp,
|
|
});
|
|
|
|
if (fullPath) {
|
|
showToast(ToastFileSaved, {
|
|
onOpenFile: () => {
|
|
openFileInFolder(fullPath);
|
|
},
|
|
});
|
|
}
|
|
}
|
|
|
|
export async function saveAttachmentFromMessage(
|
|
messageId: string,
|
|
providedAttachment?: AttachmentType
|
|
): Promise<void> {
|
|
const message = await getMessageById(messageId);
|
|
if (!message) {
|
|
throw new Error(`saveAttachmentFromMessage: Message ${messageId} missing!`);
|
|
}
|
|
|
|
const { attachments, sent_at: timestamp } = message.attributes;
|
|
if (!attachments || attachments.length < 1) {
|
|
return;
|
|
}
|
|
|
|
const attachment =
|
|
providedAttachment && attachments.includes(providedAttachment)
|
|
? providedAttachment
|
|
: attachments[0];
|
|
|
|
return saveAttachment(attachment, timestamp);
|
|
}
|