upstream-rocketchat/apps/meteor/client/lib/chats/flows/uploadFiles.ts

import { t } from '../../../../app/utils/lib/i18n';
import { MAX_MULTIPLE_UPLOADED_FILES } from '../../../../lib/constants';
import { e2e } from '../../e2ee';
import { settings } from '../../settings';
import { dispatchToastMessage } from '../../toast';
import type { ChatAPI } from '../ChatAPI';

export const uploadFiles = async (
	chat: ChatAPI,
	{ files, resetFileInput }: { files: readonly File[]; resetFileInput?: () => void },
): Promise<void> => {
	const uploadsStore = chat.composer?.uploads;
	if (!uploadsStore) {
		throw new Error('No uploads store found in composer');
	}

	const mergedFilesLength = files.length + uploadsStore.get().length;
	if (mergedFilesLength > MAX_MULTIPLE_UPLOADED_FILES) {
		return dispatchToastMessage({
			type: 'error',
			message: t('You_cant_upload_more_than__count__files', { count: MAX_MULTIPLE_UPLOADED_FILES }),
		});
	}

	const room = await chat.data.getRoom();

	if (room.encrypted && !settings.peek('E2E_Allow_Unencrypted_Messages') && !settings.peek('E2E_Enable_Encrypt_Files')) {
		return dispatchToastMessage({
			type: 'error',
			message: t('You_cant_send_unencrypted_files_in_an_encrypted_room'),
		});
	}

	const uploadFile = async (file: File) => {
		Object.defineProperty(file, 'name', {
			writable: true,
			value: file.name,
		});

		const e2eRoom = await e2e.getInstanceByRoomId(room._id);

		if (!e2eRoom || !settings.peek('E2E_Enable_Encrypt_Files')) {
			await uploadsStore.send(file);
			return;
		}

		const encryptedFile = await e2eRoom.encryptFile(file);

		if (!e2eRoom.isReady() || !encryptedFile) {
			dispatchToastMessage({
				type: 'error',
				message: t('Error_encrypting_file'),
			});
			return;
		}

		const fileContentData = {
			type: file.type,
			typeGroup: file.type.split('/')[0],
			name: file.name,
			encryption: {
				key: encryptedFile.key,
				iv: encryptedFile.iv,
			},
			hashes: {
				sha256: encryptedFile.hash,
			},
		};

		const fileContent = {
			raw: fileContentData,
			encrypted: await e2eRoom.encryptMessageContent(fileContentData),
		};

		await uploadsStore.send(encryptedFile.file, { rawFile: file, fileContent, encryptedFile });
	};

	resetFileInput?.();
	chat?.action.performContinuously('uploading');

	try {
		await Promise.allSettled(files.map((file) => uploadFile(file)));
	} finally {
		chat.composer?.focus();
	}
};